From 0b19ccaea4ea36a3bdda97ce93b1a6a9758da038 Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Mon, 12 Jun 2023 14:54:51 -0700
Subject: fixing feedback on dlt recipes
---
continuedev/src/continuedev/core/main.py | 2 +-
.../src/continuedev/recipes/AddTransformRecipe/main.py | 2 +-
.../src/continuedev/recipes/AddTransformRecipe/steps.py | 15 +++++----------
continuedev/src/continuedev/steps/core/core.py | 2 +-
4 files changed, 8 insertions(+), 13 deletions(-)
(limited to 'continuedev')
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)
--
cgit v1.2.3-70-g09d2