cjber commited on
Commit
7e9928e
·
1 Parent(s): 79e766b

add filename to hallucinations

Browse files
planning_ai/nodes/hallucination_node.py CHANGED
@@ -24,6 +24,7 @@ def check_hallucination(state: DocumentState):
24
  {
25
  "hallucination": response,
26
  "document": state["document"],
 
27
  "summary": state["summary"],
28
  "iteration": state["iteration"] + 1,
29
  }
 
24
  {
25
  "hallucination": response,
26
  "document": state["document"],
27
+ "filename": state["filename"],
28
  "summary": state["summary"],
29
  "iteration": state["iteration"] + 1,
30
  }