Spaces:
Sleeping
Sleeping
Ajuste temperatura e debug
Browse files
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",
|
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",
|
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 |
|