Spaces:
Build error
Build error
fix: simplify return structure in generate_summary function
Browse files
planning_ai/nodes/map_node.py
CHANGED
@@ -11,10 +11,7 @@ from planning_ai.states import DocumentState, OverallState
|
|
11 |
|
12 |
def generate_summary(state: DocumentState):
|
13 |
response = map_chain.invoke({"context": state["document"]})
|
14 |
-
return {
|
15 |
-
"summaries": [{"summary": response, "document": state["document"]}],
|
16 |
-
"document": state["document"],
|
17 |
-
}
|
18 |
|
19 |
|
20 |
def map_summaries(state: OverallState):
|
|
|
11 |
|
12 |
def generate_summary(state: DocumentState):
|
13 |
response = map_chain.invoke({"context": state["document"]})
|
14 |
+
return {"summaries": [{"summary": response, "document": state["document"]}]}
|
|
|
|
|
|
|
15 |
|
16 |
|
17 |
def map_summaries(state: OverallState):
|