Daniel Amendoeira commited on
Commit
baf59da
·
verified ·
1 Parent(s): 1c2e3e1

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +7 -8
agent.py CHANGED
@@ -117,14 +117,13 @@ 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
- if isinstance(content, list):
122
- # multimodal list: pull only the text parts
123
- texts = [obj["text"] for obj in content if isinstance(obj, dict) and obj.get("type") == "text"]
124
- final_response = "\n".join(texts).strip()
125
- else:
126
- # plain string fallback
127
- final_response = str(content).strip()
128
 
129
  try:
130
  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_message = result["messages"][-1].content # extract content from the last message
121
+ texts = [
122
+ obj["text"]
123
+ for obj in content_message
124
+ if isinstance(obj, dict) and obj.get("type") == "text"
125
+ ]
126
+ final_response = "\n".join(texts).strip() # join and clean
 
127
 
128
  try:
129
  return final_response.split("FINAL ANSWER:")[-1].strip() # parse out only what's after "FINAL ANSWER:"