Spaces:
Build error
Build error
rename final node
Browse files- 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
|
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",
|
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(
|