From 5e1216968b4bb1d67438e9d1b329932c5d55daab Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Fri, 2 Jun 2023 12:40:06 -0400
Subject: minor

---
 continuedev/src/continuedev/libs/steps/draft/dlt.py | 9 +++++++++
 continuedev/src/continuedev/libs/steps/main.py      | 4 +++-
 2 files changed, 12 insertions(+), 1 deletion(-)

(limited to 'continuedev/src')

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()
-- 
cgit v1.2.3-70-g09d2