diff options
author | Nate Sesti <sestinj@gmail.com> | 2023-06-16 23:42:10 -0700 |
---|---|---|
committer | Nate Sesti <sestinj@gmail.com> | 2023-06-16 23:42:10 -0700 |
commit | 5a6fa9ae784c5942f1e37c7a0ae3acdd6337c459 (patch) | |
tree | 29faca1d57e17bde00d48709dd31b78335be100b /continuedev | |
parent | 472984ab63461b90cf41b645e49bcc95bdf2ef1e (diff) | |
download | sncontinue-5a6fa9ae784c5942f1e37c7a0ae3acdd6337c459.tar.gz sncontinue-5a6fa9ae784c5942f1e37c7a0ae3acdd6337c459.tar.bz2 sncontinue-5a6fa9ae784c5942f1e37c7a0ae3acdd6337c459.zip |
polishing
Diffstat (limited to 'continuedev')
-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 8f703758..1ee3a7f8 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 7639d010..4ab2f027 100644 --- a/continuedev/src/continuedev/core/sdk.py +++ b/continuedev/src/continuedev/core/sdk.py @@ -192,3 +192,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 59af5f38..330f60ed 100644 --- a/continuedev/src/continuedev/steps/core/core.py +++ b/continuedev/src/continuedev/steps/core/core.py @@ -186,6 +186,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]) |