Spaces:
Sleeping
Sleeping
Update graph_builder.py
Browse files- graph_builder.py +1 -1
graph_builder.py
CHANGED
|
@@ -38,7 +38,7 @@ def build_graph():
|
|
| 38 |
|
| 39 |
graph.add_node("llm", llm_step)
|
| 40 |
graph.add_node("formatter", formatter_step)
|
| 41 |
-
graph.set_entry_point("llm")
|
| 42 |
graph.add_edge("llm", "formatter")
|
| 43 |
graph.set_finish_point("formatter")
|
| 44 |
|
|
|
|
| 38 |
|
| 39 |
graph.add_node("llm", llm_step)
|
| 40 |
graph.add_node("formatter", formatter_step)
|
| 41 |
+
graph.set_entry_point("llm", input_key="question")
|
| 42 |
graph.add_edge("llm", "formatter")
|
| 43 |
graph.set_finish_point("formatter")
|
| 44 |
|