minor

2 files changed, 12 insertions, 1 deletions

diff --git a/continuedev/src/continuedev/libs/steps/draft/dlt.py b/continuedev/src/continuedev/libs/steps/draft/dlt.py
index 73762327..f3417c9d 100644
--- a/continuedev/src/continuedev/libs/steps/draft/dlt.py
+++ b/continuedev/src/continuedev/libs/steps/draft/dlt.py
@@ -50,6 +50,15 @@ class SetupPipelineStep(Step):
 
 
 class ValidatePipelineStep(Step):
+
+    async def describe(self, models: Models):
+        return dedent("""\
+        This step will validate that your dlt pipeline is working as expected:
+        - Test that the API call works
+        - Load the data into a local DuckDB instance
+        - Write a query to view the data
+        """)
+
     async def run(self, sdk: ContinueSDK):
         source_name = sdk.history.last_observation().values["source_name"]
         filename = f'{source_name}.py'
diff --git a/continuedev/src/continuedev/libs/steps/main.py b/continuedev/src/continuedev/libs/steps/main.py
index c70d5c2c..aefbe084 100644
--- a/continuedev/src/continuedev/libs/steps/main.py
+++ b/continuedev/src/continuedev/libs/steps/main.py
@@ -218,8 +218,10 @@ class StarCoderEditHighlightedCodeStep(Step):
     hide = False
     _prompt: str = "<commit_before>{code}<commit_msg>{user_request}<commit_after>"
 
+    _prompt_and_completion: str = ""
+
     async def describe(self, models: Models) -> Coroutine[str, None, None]:
-        return "Editing highlighted code"
+        return (await models.gpt35()).complete(f"{self._prompt_and_completion}\n\nPlease give brief a description of the changes made above using markdown bullet points:")
 
     async def run(self, sdk: ContinueSDK) -> Coroutine[Observation, None, None]:
         range_in_files = await sdk.ide.getHighlightedCode()