summaryrefslogtreecommitdiff
path: root/continuedev
diff options
context:
space:
mode:
authorNate Sesti <sestinj@gmail.com>2023-06-16 23:42:10 -0700
committerNate Sesti <sestinj@gmail.com>2023-06-16 23:42:10 -0700
commit5a6fa9ae784c5942f1e37c7a0ae3acdd6337c459 (patch)
tree29faca1d57e17bde00d48709dd31b78335be100b /continuedev
parent472984ab63461b90cf41b645e49bcc95bdf2ef1e (diff)
downloadsncontinue-5a6fa9ae784c5942f1e37c7a0ae3acdd6337c459.tar.gz
sncontinue-5a6fa9ae784c5942f1e37c7a0ae3acdd6337c459.tar.bz2
sncontinue-5a6fa9ae784c5942f1e37c7a0ae3acdd6337c459.zip
polishing
Diffstat (limited to 'continuedev')
-rw-r--r--continuedev/src/continuedev/core/config.py15
-rw-r--r--continuedev/src/continuedev/core/sdk.py3
-rw-r--r--continuedev/src/continuedev/libs/llm/proxy_server.py2
-rw-r--r--continuedev/src/continuedev/libs/util/step_name_to_steps.py2
-rw-r--r--continuedev/src/continuedev/steps/clear_history.py10
-rw-r--r--continuedev/src/continuedev/steps/core/core.py2
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])