gdms commited on
Commit
9edff99
·
1 Parent(s): e2d2cd5

mudança na extração das mensagens

Browse files
Files changed (1) hide show
  1. agent.py +6 -6
agent.py CHANGED
@@ -71,14 +71,14 @@ class Agent:
71
  final_content = ""
72
  for message in messages:
73
  print(f"buscando resposta em {message}")
74
- if isinstance(message, dict) and message.get("content"):
75
- print("TEM CONTENT")
76
- content = message["content"]
77
- if "FINAL ANSWER:" in content.upper():
78
- final_content = content
79
- break
80
 
81
  # Extrair o valor final
 
82
  final_answer = re.sub(r"(?i)^.*FINAL ANSWER:\s*", "", final_content).strip()
83
  print("Final Answer:", final_answer)
84
 
 
71
  final_content = ""
72
  for message in messages:
73
  print(f"buscando resposta em {message}")
74
+ content = getattr(message, "content", "")
75
+ print(f"Content: {content}")
76
+ if isinstance(content, str) and "FINAL ANSWER:" in content.upper():
77
+ final_content = content
78
+ break
 
79
 
80
  # Extrair o valor final
81
+ print(f"Final Content: {final_content}")
82
  final_answer = re.sub(r"(?i)^.*FINAL ANSWER:\s*", "", final_content).strip()
83
  print("Final Answer:", final_answer)
84