summaryrefslogtreecommitdiff
path: root/continuedev/src/continuedev/recipes/CreatePipelineRecipe
diff options
context:
space:
mode:
Diffstat (limited to 'continuedev/src/continuedev/recipes/CreatePipelineRecipe')
-rw-r--r--continuedev/src/continuedev/recipes/CreatePipelineRecipe/main.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/continuedev/src/continuedev/recipes/CreatePipelineRecipe/main.py b/continuedev/src/continuedev/recipes/CreatePipelineRecipe/main.py
index 4a4604d6..55c25da4 100644
--- a/continuedev/src/continuedev/recipes/CreatePipelineRecipe/main.py
+++ b/continuedev/src/continuedev/recipes/CreatePipelineRecipe/main.py
@@ -11,7 +11,7 @@ class CreatePipelineRecipe(Step):
hide: bool = True
async def run(self, sdk: ContinueSDK):
- await sdk.run_step(
+ text_observation = await sdk.run_step(
MessageStep(message=dedent("""\
This recipe will walk you through the process of creating a dlt pipeline for your chosen data source. With the help of Continue, you will:
- Create a Python virtual environment with dlt installed
@@ -21,7 +21,10 @@ class CreatePipelineRecipe(Step):
- Test that the API call works
- Load the data into a local DuckDB instance
- Write a query to view the data""")) >>
- WaitForUserInputStep(prompt="What API do you want to load data from?") >>
- SetupPipelineStep(api_description="WeatherAPI.com API") >>
+ WaitForUserInputStep(
+ prompt="What API do you want to load data from?")
+ )
+ await sdk.run_step(
+ SetupPipelineStep(api_description=text_observation.text) >>
ValidatePipelineStep()
)