fixing feedback on dlt recipes

4 files changed, 8 insertions, 13 deletions

diff --git a/continuedev/src/continuedev/core/main.py b/continuedev/src/continuedev/core/main.py
index 3053e5a1..81aaaf2e 100644
--- a/continuedev/src/continuedev/core/main.py
+++ b/continuedev/src/continuedev/core/main.py
@@ -147,7 +147,7 @@ class Step(ContinueBaseModel):
         if self.description is not None:
             d["description"] = self.description
         else:
-            d["description"] = "`Description loading...`"
+            d["description"] = "`Step in progress...`"
         return d
 
     @validator("name", pre=True, always=True)
diff --git a/continuedev/src/continuedev/recipes/AddTransformRecipe/main.py b/continuedev/src/continuedev/recipes/AddTransformRecipe/main.py
index e9a998e3..fdd343f5 100644
--- a/continuedev/src/continuedev/recipes/AddTransformRecipe/main.py
+++ b/continuedev/src/continuedev/recipes/AddTransformRecipe/main.py
@@ -19,7 +19,7 @@ class AddTransformRecipe(Step):
                 - Run the pipeline and view the transformed data in a Streamlit app"""), name="Add transformation to a dlt pipeline") >>
             SetUpChessPipelineStep() >>
             WaitForUserInputStep(
-                prompt="How do you want to transform the Chess.com API data before loading it? For example, you could use the `python-chess` library to decode the moves or filter out certain games")
+                prompt="How do you want to transform the Chess.com API data before loading it? For example, you could filter out games that ended in a draw.")
         )
         await sdk.run_step(
             AddTransformStep(
diff --git a/continuedev/src/continuedev/recipes/AddTransformRecipe/steps.py b/continuedev/src/continuedev/recipes/AddTransformRecipe/steps.py
index 7bb0fc23..f042424c 100644
--- a/continuedev/src/continuedev/recipes/AddTransformRecipe/steps.py
+++ b/continuedev/src/continuedev/recipes/AddTransformRecipe/steps.py
@@ -31,12 +31,11 @@ class SetUpChessPipelineStep(Step):
             'pip install -r requirements.txt',
             'pip install pandas streamlit'  # Needed for the pipeline show step later
         ], name="Set up Python environment", description=dedent(f"""\
-            Running the following commands:
-            - `python3 -m venv env`: Create a Python virtual environment
-            - `source env/bin/activate`: Activate the virtual environment
-            - `pip install dlt`: Install dlt
-            - `dlt init chess duckdb`: Create a new dlt pipeline called "chess" that loads data into a local DuckDB instance
-            - `pip install -r requirements.txt`: Install the Python dependencies for the pipeline"""))
+            - Create a Python virtual environment: `python3 -m venv env`
+            - Activate the virtual environment: `source env/bin/activate`
+            - Install dlt: `pip install dlt`
+            - Create a new dlt pipeline called "chess" that loads data into a local DuckDB instance: `dlt init chess duckdb`
+            - Install the Python dependencies for the pipeline: `pip install -r requirements.txt`"""))
 
 
 class AddTransformStep(Step):
@@ -58,10 +57,6 @@ class AddTransformStep(Step):
 
         # Open the file and highlight the function to be edited
         await sdk.ide.setFileOpen(abs_filepath)
-        await sdk.ide.highlightCode(range_in_file=RangeInFile(
-            filepath=abs_filepath,
-            range=Range.from_shorthand(47, 0, 51, 0)
-        ))
 
         with open(os.path.join(os.path.dirname(__file__), 'dlt_transform_docs.md')) as f:
             dlt_transform_docs = f.read()
diff --git a/continuedev/src/continuedev/steps/core/core.py b/continuedev/src/continuedev/steps/core/core.py
index 53df65cc..392339c6 100644
--- a/continuedev/src/continuedev/steps/core/core.py
+++ b/continuedev/src/continuedev/steps/core/core.py
@@ -88,7 +88,7 @@ class ShellCommandsStep(Step):
                     This is a brief summary of the error followed by a suggestion on how it can be fixed:"""), with_history=sdk.chat_context)
 
                 sdk.raise_exception(
-                    title="Error while running query", message=output, with_step=MessageStep(name=f"Suggestion to solve error {AI_ASSISTED_STRING}", message=suggestion)
+                    title="Error while running query", message=output, with_step=MessageStep(name=f"Suggestion to solve error {AI_ASSISTED_STRING}", message=f"{suggestion}\n\nYou can click the retry button on the failed step to try again.")
                 )
 
         return TextObservation(text=output)