1 files changed, 29 insertions, 0 deletions
diff --git a/server/continuedev/plugins/steps/custom_command.py b/server/continuedev/plugins/steps/custom_command.py
new file mode 100644
index 00000000..4128415b
--- /dev/null
+++ b/server/continuedev/plugins/steps/custom_command.py
@@ -0,0 +1,29 @@
+from ...core.main import Step
+from ...core.sdk import ContinueSDK, Models
+from ...libs.util.templating import render_templated_string
+from ..steps.chat import SimpleChatStep
+
+
+class CustomCommandStep(Step):
+    name: str
+    prompt: str
+    user_input: str
+    slash_command: str
+    hide: bool = True
+
+    async def describe(self, models: Models):
+        return self.prompt
+
+    async def run(self, sdk: ContinueSDK):
+        task = render_templated_string(self.prompt)
+
+        prompt_user_input = f"Task: {task}. Additional info: {self.user_input}"
+        messages = await sdk.get_chat_context()
+        # Find the last chat message with this slash command and replace it with the user input
+        for i in range(len(messages) - 1, -1, -1):
+            if messages[i].role == "user" and messages[i].content.startswith(
+                self.slash_command
+            ):
+                messages[i] = messages[i].copy(update={"content": prompt_user_input})
+                break
+        await sdk.run_step(SimpleChatStep(messages=messages))