cjber commited on
Commit
83e127b
·
1 Parent(s): 5d56d37

fix: simplify return structure in generate_summary function

Browse files
Files changed (1) hide show
  1. planning_ai/nodes/map_node.py +1 -4
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):