1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
from typing import Dict
from ...core.main import Step
from ...core.steps import UserInputStep
from ...libs.util.logging import logger
from ...plugins.recipes.AddTransformRecipe.main import AddTransformRecipe
from ...plugins.recipes.CreatePipelineRecipe.main import CreatePipelineRecipe
from ...plugins.recipes.DDtoBQRecipe.main import DDtoBQRecipe
from ...plugins.recipes.DeployPipelineAirflowRecipe.main import (
DeployPipelineAirflowRecipe,
)
from ...plugins.steps.chat import SimpleChatStep
from ...plugins.steps.clear_history import ClearHistoryStep
from ...plugins.steps.comment_code import CommentCodeStep
from ...plugins.steps.feedback import FeedbackStep
from ...plugins.steps.help import HelpStep
from ...plugins.steps.main import EditHighlightedCodeStep
from ...plugins.steps.on_traceback import DefaultOnTracebackStep
from ...plugins.steps.open_config import OpenConfigStep
# 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,
"DefaultOnTracebackStep": DefaultOnTracebackStep,
"ClearHistoryStep": ClearHistoryStep,
"OpenConfigStep": OpenConfigStep,
"HelpStep": HelpStep,
}
def get_step_from_name(step_name: str, params: Dict) -> Step:
try:
return step_name_to_step_class[step_name](**params)
except:
logger.error(
f"Incorrect parameters for step {step_name}. Parameters provided were: {params}"
)
raise
|
