Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
|
|
|
| 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()
|