Spaces:
Running
Running
Ajuste modelos e temperatura
Browse files- agent.py +3 -3
- constantes.py +3 -2
agent.py
CHANGED
|
@@ -19,7 +19,7 @@ class Agent:
|
|
| 19 |
|
| 20 |
print("--> Audio Agent")
|
| 21 |
self.audio_agent = create_react_agent(
|
| 22 |
-
model=init_chat_model(
|
| 23 |
tools=[extract_text_from_url_tool, extract_text_from_file_tool, clean_ingredient_measure_tool],
|
| 24 |
prompt= AUDIO_AGENT_PROMPT,
|
| 25 |
name="audio_agent",
|
|
@@ -27,7 +27,7 @@ class Agent:
|
|
| 27 |
|
| 28 |
print("--> Web Search Agent")
|
| 29 |
self.web_search_agent = create_react_agent(
|
| 30 |
-
model=init_chat_model(
|
| 31 |
tools=[search_web_tool],
|
| 32 |
prompt= WEB_SEARCH_AGENT_PROMPT,
|
| 33 |
name="web_research_agent",
|
|
@@ -35,7 +35,7 @@ class Agent:
|
|
| 35 |
|
| 36 |
print("--> Supervisor")
|
| 37 |
self.supervisor = create_supervisor(
|
| 38 |
-
model=init_chat_model(
|
| 39 |
agents=[self.web_search_agent, self.audio_agent],
|
| 40 |
tools=[bird_video_count_tool,chess_image_to_fen_tool,chess_fen_get_best_next_move_tool,
|
| 41 |
get_excel_columns_tool, calculate_excel_sum_by_columns_tool,execute_python_code_tool,
|
|
|
|
| 19 |
|
| 20 |
print("--> Audio Agent")
|
| 21 |
self.audio_agent = create_react_agent(
|
| 22 |
+
model=init_chat_model(GPT_4_0_MINI_MODEL),
|
| 23 |
tools=[extract_text_from_url_tool, extract_text_from_file_tool, clean_ingredient_measure_tool],
|
| 24 |
prompt= AUDIO_AGENT_PROMPT,
|
| 25 |
name="audio_agent",
|
|
|
|
| 27 |
|
| 28 |
print("--> Web Search Agent")
|
| 29 |
self.web_search_agent = create_react_agent(
|
| 30 |
+
model=init_chat_model(GPT_4_0_MINI_MODEL),
|
| 31 |
tools=[search_web_tool],
|
| 32 |
prompt= WEB_SEARCH_AGENT_PROMPT,
|
| 33 |
name="web_research_agent",
|
|
|
|
| 35 |
|
| 36 |
print("--> Supervisor")
|
| 37 |
self.supervisor = create_supervisor(
|
| 38 |
+
model=init_chat_model(GPT_4_0_MINI_MODEL, temperature=0.4),
|
| 39 |
agents=[self.web_search_agent, self.audio_agent],
|
| 40 |
tools=[bird_video_count_tool,chess_image_to_fen_tool,chess_fen_get_best_next_move_tool,
|
| 41 |
get_excel_columns_tool, calculate_excel_sum_by_columns_tool,execute_python_code_tool,
|
constantes.py
CHANGED
|
@@ -1,8 +1,9 @@
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
|
| 4 |
-
|
| 5 |
-
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
|
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
|
| 4 |
+
GPT_4_0_MINI_MODEL="openai:gpt-4o-mini"
|
| 5 |
+
GPT_4_1_MINI_MODEL="openai:gpt-4.1-mini"
|
| 6 |
+
GPT_4o_MODEL="openai:gpt-4o"
|
| 7 |
|
| 8 |
|
| 9 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|