tweaks to config and userInputStepContainer

2 files changed, 58 insertions, 40 deletions

diff --git a/continuedev/src/continuedev/core/config.py b/continuedev/src/continuedev/core/config.py
index 3208e63d..9e8541dc 100644
--- a/continuedev/src/continuedev/core/config.py
+++ b/continuedev/src/continuedev/core/config.py
@@ -22,6 +22,45 @@ class OnTracebackSteps(BaseModel):
     params: Optional[Dict] = {}
 
 
+DEFAULT_SLASH_COMMANDS = [
+    # SlashCommand(
+    #     name="pytest",
+    #     description="Write pytest unit tests for the current file",
+    #     step_name="WritePytestsRecipe",
+    #     params=??)
+    SlashCommand(
+        name="edit",
+        description="Edit code in the current file or the highlighted code",
+        step_name="EditHighlightedCodeStep",
+    ),
+    SlashCommand(
+        name="explain",
+        description="Reply to instructions or a question with previous steps and the highlighted code or current file as context",
+        step_name="SimpleChatStep",
+    ),
+    SlashCommand(
+        name="config",
+        description="Open the config file to create new and edit existing slash commands",
+        step_name="OpenConfigStep",
+    ),
+    SlashCommand(
+        name="comment",
+        description="Write comments for the current file or highlighted code",
+        step_name="CommentCodeStep",
+    ),
+    SlashCommand(
+        name="feedback",
+        description="Send feedback to improve Continue",
+        step_name="FeedbackStep",
+    ),
+    SlashCommand(
+        name="clear",
+        description="Clear step history",
+        step_name="ClearHistoryStep",
+    )
+]
+
+
 class ContinueConfig(BaseModel):
     """
     A pydantic class for the continue config file.
@@ -33,46 +72,15 @@ class ContinueConfig(BaseModel):
     default_model: Literal["gpt-3.5-turbo", "gpt-3.5-turbo-16k",
                            "gpt-4"] = 'gpt-4'
     custom_commands: Optional[List[CustomCommand]] = []
-    slash_commands: Optional[List[SlashCommand]] = [
-        # SlashCommand(
-        #     name="pytest",
-        #     description="Write pytest unit tests for the current file",
-        #     step_name="WritePytestsRecipe",
-        #     params=??)
-        SlashCommand(
-            name="edit",
-            description="Edit code in the current file or the highlighted code",
-            step_name="EditHighlightedCodeStep",
-        ),
-        SlashCommand(
-            name="explain",
-            description="Reply to instructions or a question with previous steps and the highlighted code or current file as context",
-            step_name="SimpleChatStep",
-        ),
-        SlashCommand(
-            name="config",
-            description="Open the config file to create new and edit existing slash commands",
-            step_name="OpenConfigStep",
-        ),
-        SlashCommand(
-            name="comment",
-            description="Write comments for the current file or highlighted code",
-            step_name="CommentCodeStep",
-        ),
-        SlashCommand(
-            name="feedback",
-            description="Send feedback to improve Continue",
-            step_name="FeedbackStep",
-        ),
-        SlashCommand(
-            name="clear",
-            description="Clear step history",
-            step_name="ClearHistoryStep",
-        )
-    ]
+    slash_commands: Optional[List[SlashCommand]] = DEFAULT_SLASH_COMMANDS
     on_traceback: Optional[List[OnTracebackSteps]] = [
         OnTracebackSteps(step_name="DefaultOnTracebackStep")]
 
+    # Want to force these to be the slash commands for now
+    @validator('slash_commands', pre=True)
+    def default_slash_commands_validator(cls, v):
+        return DEFAULT_SLASH_COMMANDS
+
 
 def load_config(config_file: str) -> ContinueConfig:
     """
@@ -142,4 +150,4 @@ def update_global_config(config: ContinueConfig):
     else:
         config_path = os.path.join(global_dir, 'config.json')
         with open(config_path, 'w') as f:
-            json.dump(config.dict(), f)
+            json.dump(config.dict(exclude_unset=False), f, indent=4)
diff --git a/continuedev/src/continuedev/steps/open_config.py b/continuedev/src/continuedev/steps/open_config.py
index 9cffdb3a..91459113 100644
--- a/continuedev/src/continuedev/steps/open_config.py
+++ b/continuedev/src/continuedev/steps/open_config.py
@@ -1,3 +1,4 @@
+from textwrap import dedent
 from ..core.main import Step
 from ..core.sdk import ContinueSDK
 import os
@@ -7,9 +8,18 @@ class OpenConfigStep(Step):
     name: str = "Open config"
 
     async def describe(self, models):
-        return "Config.json is now open. Create a new or edit an existing slash command here. Here is an example: { custom_commands : [ { 'name;: 'test', 'prompt': 'write me a unit test' } ] }"
+        return dedent("""\
+            Config.json is now open. You can add a custom slash command in the `\"custom_commands\"` section, like in this example:
+            ```json
+            "custom_commands": [
+                {
+                    "name": "test",
+                    "prompt": "write me a comprehensive unit test for this function, that covers all edge cases. Use pytest."
+                }
+            ],
+            ```""")
 
     async def run(self, sdk: ContinueSDK):
         global_dir = os.path.expanduser('~/.continue')
         config_path = os.path.join(global_dir, 'config.json')
-        await sdk.ide.setFileOpen(config_path)
\ No newline at end of file
+        await sdk.ide.setFileOpen(config_path)