diff options
Diffstat (limited to 'continuedev/src')
| -rw-r--r-- | continuedev/src/continuedev/core/config.py | 15 | ||||
| -rw-r--r-- | continuedev/src/continuedev/core/sdk.py | 3 | ||||
| -rw-r--r-- | continuedev/src/continuedev/libs/llm/proxy_server.py | 2 | ||||
| -rw-r--r-- | continuedev/src/continuedev/libs/util/step_name_to_steps.py | 2 | ||||
| -rw-r--r-- | continuedev/src/continuedev/steps/clear_history.py | 10 | ||||
| -rw-r--r-- | continuedev/src/continuedev/steps/core/core.py | 2 | 
6 files changed, 31 insertions, 3 deletions
| diff --git a/continuedev/src/continuedev/core/config.py b/continuedev/src/continuedev/core/config.py index 859c6188..652320fb 100644 --- a/continuedev/src/continuedev/core/config.py +++ b/continuedev/src/continuedev/core/config.py @@ -52,6 +52,11 @@ class ContinueConfig(BaseModel):              name="feedback",              description="Send feedback to improve Continue",              step_name="FeedbackStep", +        ), +        SlashCommand( +            name="clear", +            description="Clear step history", +            step_name="ClearHistoryStep",          )      ]      on_traceback: Optional[List[OnTracebackSteps]] = [ @@ -68,10 +73,16 @@ def load_config(config_file: str) -> ContinueConfig:      _, ext = os.path.splitext(config_file)      if ext == '.yaml':          with open(config_file, 'r') as f: -            config_dict = yaml.safe_load(f) +            try: +                config_dict = yaml.safe_load(f) +            except: +                return ContinueConfig()      elif ext == '.json':          with open(config_file, 'r') as f: -            config_dict = json.load(f) +            try: +                config_dict = json.load(f) +            except: +                return ContinueConfig()      else:          raise ValueError(f'Unknown config file extension: {ext}')      return ContinueConfig(**config_dict) diff --git a/continuedev/src/continuedev/core/sdk.py b/continuedev/src/continuedev/core/sdk.py index 7159beaa..d6acc404 100644 --- a/continuedev/src/continuedev/core/sdk.py +++ b/continuedev/src/continuedev/core/sdk.py @@ -204,3 +204,6 @@ class ContinueSDK(AbstractContinueSDK):      async def update_ui(self):          await self.__autopilot.update_subscribers() + +    async def clear_history(self): +        await self.__autopilot.clear_history() diff --git a/continuedev/src/continuedev/libs/llm/proxy_server.py b/continuedev/src/continuedev/libs/llm/proxy_server.py index 93f2d48a..dd0d3aca 100644 --- a/continuedev/src/continuedev/libs/llm/proxy_server.py +++ b/continuedev/src/continuedev/libs/llm/proxy_server.py @@ -18,7 +18,7 @@ CHAT_MODELS = {  }  # SERVER_URL = "http://127.0.0.1:8080" -SERVER_URL = "https://proxy-server-l6vsfbzhba-uc.a.run.app" +SERVER_URL = "https://proxy-server-l6vsfbzhba-uw.a.run.app"  class ProxyServer(LLM): diff --git a/continuedev/src/continuedev/libs/util/step_name_to_steps.py b/continuedev/src/continuedev/libs/util/step_name_to_steps.py index 2c4474af..f431f317 100644 --- a/continuedev/src/continuedev/libs/util/step_name_to_steps.py +++ b/continuedev/src/continuedev/libs/util/step_name_to_steps.py @@ -11,6 +11,7 @@ from ...recipes.CreatePipelineRecipe.main import CreatePipelineRecipe  from ...recipes.DDtoBQRecipe.main import DDtoBQRecipe  from ...recipes.DeployPipelineAirflowRecipe.main import DeployPipelineAirflowRecipe  from ...steps.on_traceback import DefaultOnTracebackStep +from ...steps.clear_history import ClearHistoryStep  # This mapping is used to convert from string in ContinueConfig json to corresponding Step class.  # Used for example in slash_commands and steps_on_startup @@ -25,6 +26,7 @@ step_name_to_step_class = {      "DDtoBQRecipe": DDtoBQRecipe,      "DeployPipelineAirflowRecipe": DeployPipelineAirflowRecipe,      "DefaultOnTracebackStep": DefaultOnTracebackStep, +    "ClearHistoryStep": ClearHistoryStep,  } diff --git a/continuedev/src/continuedev/steps/clear_history.py b/continuedev/src/continuedev/steps/clear_history.py new file mode 100644 index 00000000..a875c6d3 --- /dev/null +++ b/continuedev/src/continuedev/steps/clear_history.py @@ -0,0 +1,10 @@ +from ..core.main import Step +from ..core.sdk import ContinueSDK + + +class ClearHistoryStep(Step): +    name: str = "Clear History" +    hide: bool = True + +    async def run(self, sdk: ContinueSDK): +        await sdk.clear_history() diff --git a/continuedev/src/continuedev/steps/core/core.py b/continuedev/src/continuedev/steps/core/core.py index d580e2d2..417398b7 100644 --- a/continuedev/src/continuedev/steps/core/core.py +++ b/continuedev/src/continuedev/steps/core/core.py @@ -233,6 +233,8 @@ class DefaultModelEditCodeStep(Step):              completion = completion.removesuffix(eot_token)              # Remove tags and If it accidentally includes prefix or suffix, remove it +            if completion.strip().startswith("```"): +                completion = completion.strip().removeprefix("```").removesuffix("```")              completion = completion.replace("<file_prefix>", "").replace("<file_suffix>", "").replace(                  "<commit_before>", "").replace("<commit_msg>", "").replace("<commit_after>", "")              completion = completion.removeprefix(segs[0]) | 
