Umer797 commited on
Commit
9d73d9a
·
verified ·
1 Parent(s): 0d42337

Update graph_builder.py

Browse files
Files changed (1) hide show
  1. 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