diff options
| author | Ty Dunn <ty@tydunn.com> | 2023-07-06 12:22:15 -0700 |
|---|---|---|
| committer | Ty Dunn <ty@tydunn.com> | 2023-07-06 12:22:15 -0700 |
| commit | 9f32c906023c596a7610ee4d1d6cff9f1201a5dc (patch) | |
| tree | 676005406bb2840f29605c6dbde2df606ed1ce17 /continuedev/src/continuedev/core | |
| parent | 1f28a09fa8af14e09e85f0407e219c9cade5b4de (diff) | |
| download | sncontinue-9f32c906023c596a7610ee4d1d6cff9f1201a5dc.tar.gz sncontinue-9f32c906023c596a7610ee4d1d6cff9f1201a5dc.tar.bz2 sncontinue-9f32c906023c596a7610ee4d1d6cff9f1201a5dc.zip | |
description for custom commands
Diffstat (limited to 'continuedev/src/continuedev/core')
| -rw-r--r-- | continuedev/src/continuedev/core/autopilot.py | 2 | ||||
| -rw-r--r-- | continuedev/src/continuedev/core/config.py | 2 | ||||
| -rw-r--r-- | continuedev/src/continuedev/core/policy.py | 2 |
3 files changed, 4 insertions, 2 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 |