Update app.py

app.py

@@ -141,6 +141,7 @@ class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
     proposed_action: str
     information: str
+    raw_output: str
     output: str
     confidence: float
     judge_explanation: str
@@ -547,6 +548,8 @@ Information:
     raw_output = reasoning_generate(prompt)
     # raw_output = generate(prompt)
 
+    state["raw_output"] = raw_output
+
     logger.info(f"Raw Output: {raw_output}")
 
     # output = raw_output.split("Response:")[-1].strip()