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

tentanto extrair final_content

Browse files
Files changed (1) hide show
  1. agent.py +9 -9
agent.py CHANGED
@@ -5,6 +5,7 @@ from prompts import *
5
  from tools import *
6
  from langgraph_supervisor import create_supervisor
7
  from langchain.chat_models import init_chat_model
 
8
 
9
  import glob
10
 
@@ -65,17 +66,16 @@ class Agent:
65
  )
66
 
67
  print(f"Resposta LLM: {response}")
68
- messages = response.get("messages", [])
69
-
70
  # Extrair o conteúdo das mensagens do tipo AIMessage
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}")
 
5
  from tools import *
6
  from langgraph_supervisor import create_supervisor
7
  from langchain.chat_models import init_chat_model
8
+ from langchain_core.messages import AIMessage
9
 
10
  import glob
11
 
 
66
  )
67
 
68
  print(f"Resposta LLM: {response}")
 
 
69
  # Extrair o conteúdo das mensagens do tipo AIMessage
70
  final_content = ""
71
+ for m in reversed(response['messages']):
72
+ print(f"buscando resposta em {m.content}")
73
+ if isinstance(m, AIMessage):
74
+ print('AI Message')
75
+ if "FINAL ANSWER:" in m.content.upper():
76
+ print("Tem Final Answer")
77
+ final_content = m.content
78
+ break
79
 
80
  # Extrair o valor final
81
  print(f"Final Content: {final_content}")