cjber commited on
Commit
228495e
·
1 Parent(s): d5b7cf9

fix: ensure model can see divide between reports

Browse files
Files changed (1) hide show
  1. planning_ai/nodes/reduce_node.py +3 -1
planning_ai/nodes/reduce_node.py CHANGED
@@ -136,5 +136,7 @@ def final_output(final_docs):
136
  policies = generate_policy_output(policy_groups)
137
 
138
  batch_executive = batch_generate_executive_summaries(docs)
139
- executive = reduce_chain_final.invoke({"context": "\n\n".join(batch_executive)})
 
 
140
  return {"executive": executive, "documents": docs, "policies": policies}
 
136
  policies = generate_policy_output(policy_groups)
137
 
138
  batch_executive = batch_generate_executive_summaries(docs)
139
+ executive = reduce_chain_final.invoke(
140
+ {"context": "Executive Report:\n\n".join(batch_executive)}
141
+ )
142
  return {"executive": executive, "documents": docs, "policies": policies}