WeByT3 commited on
Commit
62fc861
·
verified ·
1 Parent(s): bfea96a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -6,6 +6,7 @@ import pandas as pd
6
  from agent import build_agent
7
  from langchain_core.messages import AnyMessage, HumanMessage, AIMessage
8
  import traceback
 
9
 
10
  # (Keep Constants as is)
11
  # --- Constants ---
@@ -21,7 +22,8 @@ class BasicAgent:
21
  print(f"Agent received question (first 50 chars): {question[:50]}...")
22
  question = [HumanMessage(content=question)]
23
  final_answer = self.agent.invoke({"messages": question})
24
- print(f"Agent returning fixed answer: {final_answer}")
 
25
  text = final_answer["messages"][-1].content
26
  if "FINAL ANSWER:" in text:
27
  return text.partition("FINAL ANSWER:")[2].strip()
 
6
  from agent import build_agent
7
  from langchain_core.messages import AnyMessage, HumanMessage, AIMessage
8
  import traceback
9
+ from pprint import pprint
10
 
11
  # (Keep Constants as is)
12
  # --- Constants ---
 
22
  print(f"Agent received question (first 50 chars): {question[:50]}...")
23
  question = [HumanMessage(content=question)]
24
  final_answer = self.agent.invoke({"messages": question})
25
+ print("Agent returning fixed answer:")
26
+ pprint(final_answer)
27
  text = final_answer["messages"][-1].content
28
  if "FINAL ANSWER:" in text:
29
  return text.partition("FINAL ANSWER:")[2].strip()