Daniel Amendoeira commited on
Commit
cff0002
·
verified ·
1 Parent(s): d8a0003

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -117,9 +117,9 @@ class LangGraphAgent:
117
  print(f"Running LangGraphAgent with input: {question[:200]}...")
118
  result = gaia_agent.invoke(input_state, {"recursion_limit": 30}) # prevents infinite looping when the LLM keeps calling tools over and over
119
 
120
- content_list = result["messages"][-1]["content"] # extract content list from the last message
121
  texts = [obj["text"] for obj in content_list if obj.get("type") == "text"] # pull out all text fragments (required for gpt-4o-mini)
122
- final_response = "\n".join(texts).strip()
123
 
124
  try:
125
  return final_response.split("FINAL ANSWER:")[-1].strip() # parse out only what's after "FINAL ANSWER:"
 
117
  print(f"Running LangGraphAgent with input: {question[:200]}...")
118
  result = gaia_agent.invoke(input_state, {"recursion_limit": 30}) # prevents infinite looping when the LLM keeps calling tools over and over
119
 
120
+ content_list = result["messages"][-1].content # extract content list from the last message
121
  texts = [obj["text"] for obj in content_list if obj.get("type") == "text"] # pull out all text fragments (required for gpt-4o-mini)
122
+ final_response = "\n".join(texts).strip() # join and clean
123
 
124
  try:
125
  return final_response.split("FINAL ANSWER:")[-1].strip() # parse out only what's after "FINAL ANSWER:"