Putting back step_name_to_step_class for dlt

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