diff options
Diffstat (limited to 'continuedev/src')
| -rw-r--r-- | continuedev/src/continuedev/libs/util/step_name_to_steps.py | 12 | 
1 files changed, 11 insertions, 1 deletions
| diff --git a/continuedev/src/continuedev/libs/util/step_name_to_steps.py b/continuedev/src/continuedev/libs/util/step_name_to_steps.py index 4dd9c430..b2bb838a 100644 --- a/continuedev/src/continuedev/libs/util/step_name_to_steps.py +++ b/continuedev/src/continuedev/libs/util/step_name_to_steps.py @@ -6,13 +6,23 @@ from ...steps.main import EditHighlightedCodeStep  from ...steps.chat import SimpleChatStep  from ...steps.comment_code import CommentCodeStep  from ...steps.feedback import FeedbackStep +from ...recipes.AddTransformRecipe.main import AddTransformRecipe +from ...recipes.CreatePipelineRecipe.main import CreatePipelineRecipe +from ...recipes.DDtoBQRecipe.main import DDtoBQRecipe +from ...recipes.DeployPipelineAirflowRecipe.main import DeployPipelineAirflowRecipe +# This mapping is used to convert from string in ContinueConfig json to corresponding Step class. +# Used for example in slash_commands and steps_on_startup  step_name_to_step_class = {      "UserInputStep": UserInputStep,      "EditHighlightedCodeStep": EditHighlightedCodeStep,      "SimpleChatStep": SimpleChatStep,      "CommentCodeStep": CommentCodeStep,      "FeedbackStep": FeedbackStep, +    "AddTransformRecipe": AddTransformRecipe, +    "CreatePipelineRecipe": CreatePipelineRecipe, +    "DDtoBQRecipe": DDtoBQRecipe, +    "DeployPipelineAirflowRecipe": DeployPipelineAirflowRecipe,  } @@ -22,4 +32,4 @@ def get_step_from_name(step_name: str, params: Dict) -> Step:      except:          print(              f"Incorrect parameters for step {step_name}. Parameters provided were: {params}") -        raise
\ No newline at end of file +        raise | 
