gdms commited on
Commit
5879225
·
1 Parent(s): 283ad74

Ajuste temperatura e debug

Browse files
Files changed (1) hide show
  1. agent.py +5 -3
agent.py CHANGED
@@ -14,7 +14,7 @@ class Agent:
14
 
15
  print("--> Audio Agent")
16
  self.audio_agent = create_react_agent(
17
- model="openai:gpt-4o-mini", # gpt-4o-mini-2024-07-18
18
  tools=[extract_text_from_url_tool, extract_text_from_file_tool],
19
  prompt= AUDIO_AGENT_PROMPT,
20
  name="audio_agent",
@@ -22,7 +22,7 @@ class Agent:
22
 
23
  print("--> Web Search Agent")
24
  self.web_search_agent = create_react_agent(
25
- model="openai:gpt-4o-mini", # gpt-4o-mini-2024-07-18
26
  tools=[search_web_tool],
27
  prompt= WEB_SEARCH_AGENT_PROMPT,
28
  name="web_research_agent",
@@ -30,7 +30,7 @@ class Agent:
30
 
31
  print("--> Supervisor")
32
  self.supervisor = create_supervisor(
33
- model=init_chat_model("openai:gpt-4o-mini"),
34
  agents=[self.web_search_agent, self.audio_agent],
35
  tools=[bird_video_count_tool,chess_image_to_fen_tool,chess_fen_get_best_next_move_tool,
36
  get_excel_columns_tool, calculate_excel_sum_by_columns_tool,execute_python_code_tool,
@@ -103,7 +103,9 @@ class Agent:
103
  print(f"Nenhuma mensagem retornada para task {task_id}.")
104
  return "Desculpe, não houve resposta."
105
 
 
106
  last_msg = messages[-1]["content"]
 
107
  final_answer = re.sub(r"^FINAL ANSWER:\s*", "", last_msg.strip(), flags=re.IGNORECASE)
108
  print(f"Agent returning answer for task {task_id}: {final_answer}")
109
 
 
14
 
15
  print("--> Audio Agent")
16
  self.audio_agent = create_react_agent(
17
+ model=init_chat_model("openai:gpt-4o-mini", temperature=0),
18
  tools=[extract_text_from_url_tool, extract_text_from_file_tool],
19
  prompt= AUDIO_AGENT_PROMPT,
20
  name="audio_agent",
 
22
 
23
  print("--> Web Search Agent")
24
  self.web_search_agent = create_react_agent(
25
+ model=init_chat_model("openai:gpt-4o-mini", temperature=0),
26
  tools=[search_web_tool],
27
  prompt= WEB_SEARCH_AGENT_PROMPT,
28
  name="web_research_agent",
 
30
 
31
  print("--> Supervisor")
32
  self.supervisor = create_supervisor(
33
+ model=init_chat_model("openai:gpt-4o-mini", temperature=0),
34
  agents=[self.web_search_agent, self.audio_agent],
35
  tools=[bird_video_count_tool,chess_image_to_fen_tool,chess_fen_get_best_next_move_tool,
36
  get_excel_columns_tool, calculate_excel_sum_by_columns_tool,execute_python_code_tool,
 
103
  print(f"Nenhuma mensagem retornada para task {task_id}.")
104
  return "Desculpe, não houve resposta."
105
 
106
+ print("Extraindo last_message")
107
  last_msg = messages[-1]["content"]
108
+ print(f"Last Message: {last_msg}")
109
  final_answer = re.sub(r"^FINAL ANSWER:\s*", "", last_msg.strip(), flags=re.IGNORECASE)
110
  print(f"Agent returning answer for task {task_id}: {final_answer}")
111