description for custom commands

4 files changed, 7 insertions, 4 deletions

diff --git a/continuedev/src/continuedev/core/autopilot.py b/continuedev/src/continuedev/core/autopilot.py
index b1c4f471..49fbacc5 100644
--- a/continuedev/src/continuedev/core/autopilot.py
+++ b/continuedev/src/continuedev/core/autopilot.py
@@ -75,7 +75,7 @@ class Autopilot(ContinueBaseModel):
 
     def get_available_slash_commands(self) -> List[Dict]:
         custom_commands = list(map(lambda x: {
-                               "name": x.name, "description": x.prompt}, self.continue_sdk.config.custom_commands)) or []
+                               "name": x.name, "description": x.description}, self.continue_sdk.config.custom_commands)) or []
         slash_commands = list(map(lambda x: {
                               "name": x.name, "description": x.description}, self.continue_sdk.config.slash_commands)) or []
         return custom_commands + slash_commands
diff --git a/continuedev/src/continuedev/core/config.py b/continuedev/src/continuedev/core/config.py
index ff7b8cb0..55f5bc60 100644
--- a/continuedev/src/continuedev/core/config.py
+++ b/continuedev/src/continuedev/core/config.py
@@ -15,6 +15,7 @@ class SlashCommand(BaseModel):
 class CustomCommand(BaseModel):
     name: str
     prompt: str
+    description: str
 
 
 class OnTracebackSteps(BaseModel):
@@ -73,6 +74,7 @@ class ContinueConfig(BaseModel):
                            "gpt-4"] = 'gpt-4'
     custom_commands: Optional[List[CustomCommand]] = [CustomCommand(
         name="test",
+        description="This is an example custom command. Use /config to edit it and create more",
         prompt="Write a comprehensive set of unit tests for the selected code. It should setup, run tests that check for correctness including important edge cases, and teardown. Ensure that the tests are complete and sophisticated. Give the tests just as chat output, don't edit any file.",
     )]
     slash_commands: Optional[List[SlashCommand]] = DEFAULT_SLASH_COMMANDS
diff --git a/continuedev/src/continuedev/core/policy.py b/continuedev/src/continuedev/core/policy.py
index fc9266ab..6ee2d03f 100644
--- a/continuedev/src/continuedev/core/policy.py
+++ b/continuedev/src/continuedev/core/policy.py
@@ -46,7 +46,7 @@ def parse_custom_command(inp: str, config: ContinueConfig) -> Union[None, Step]:
             slash_command = parse_slash_command(custom_cmd.prompt, config)
             if slash_command is not None:
                 return slash_command
-            return CustomCommandStep(name=custom_cmd.name, prompt=custom_cmd.prompt, user_input=after_command)
+            return CustomCommandStep(name=custom_cmd.name, description=custom_cmd.description, prompt=custom_cmd.prompt, user_input=after_command)
     return None
 
 
diff --git a/continuedev/src/continuedev/steps/open_config.py b/continuedev/src/continuedev/steps/open_config.py
index 91459113..441cb0e7 100644
--- a/continuedev/src/continuedev/steps/open_config.py
+++ b/continuedev/src/continuedev/steps/open_config.py
@@ -9,12 +9,13 @@ class OpenConfigStep(Step):
 
     async def describe(self, models):
         return dedent("""\
-            Config.json is now open. You can add a custom slash command in the `\"custom_commands\"` section, like in this example:
+            `\"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."
+                    "description": "Write unit tests like I do for the highlighted code"
+                    "prompt": "Write a comprehensive set of unit tests for the selected code. It should setup, run tests that check for correctness including important edge cases, and teardown. Ensure that the tests are complete and sophisticated."
                 }
             ],
             ```""")