cjber commited on
Commit
648f926
·
1 Parent(s): 3f35591

rename final node

Browse files
Files changed (1) hide show
  1. planning_ai/graph.py +2 -2
planning_ai/graph.py CHANGED
@@ -14,7 +14,7 @@ from planning_ai.nodes.map_node import (
14
  map_summaries,
15
  retrieve_themes,
16
  )
17
- from planning_ai.nodes.reduce_node import generate_final_summary
18
  from planning_ai.states import OverallState
19
 
20
 
@@ -36,7 +36,7 @@ def create_graph():
36
  graph.add_node("generate_summary", generate_summary)
37
  graph.add_node("check_hallucination", check_hallucination)
38
  graph.add_node("fix_hallucination", fix_hallucination)
39
- graph.add_node("generate_final_summary", generate_final_summary)
40
 
41
  graph.add_edge(START, "add_entities")
42
  graph.add_conditional_edges(
 
14
  map_summaries,
15
  retrieve_themes,
16
  )
17
+ from planning_ai.nodes.reduce_node import generate_final_report
18
  from planning_ai.states import OverallState
19
 
20
 
 
36
  graph.add_node("generate_summary", generate_summary)
37
  graph.add_node("check_hallucination", check_hallucination)
38
  graph.add_node("fix_hallucination", fix_hallucination)
39
+ graph.add_node("generate_final_summary", generate_final_report)
40
 
41
  graph.add_edge(START, "add_entities")
42
  graph.add_conditional_edges(