File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -40,11 +40,11 @@ class GraphBuilder:
40
40
ValueError: If 'api_key' is not included in llm_config.
41
41
"""
42
42
43
- def __init__ (self , user_prompt : str , config : dict ):
43
+ def __init__ (self , prompt : str , config : dict ):
44
44
"""
45
45
Initializes the GraphBuilder with a user prompt and language model configuration.
46
46
"""
47
- self .user_prompt = user_prompt
47
+ self .prompt = prompt
48
48
self .config = config
49
49
self .llm = self ._create_llm (config ["llm" ])
50
50
self .nodes_description = self ._generate_nodes_description ()
@@ -122,7 +122,7 @@ def build_graph(self):
122
122
Returns:
123
123
dict: A JSON representation of the graph configuration.
124
124
"""
125
- return self .chain .invoke (self .user_prompt )
125
+ return self .chain .invoke (self .prompt )
126
126
127
127
@staticmethod
128
128
def convert_json_to_graphviz (json_data , format : str = 'pdf' ):
You can’t perform that action at this time.
0 commit comments