Spaces:
Sleeping
Sleeping
testando gpt 4o para busca
Browse files- agent.py +3 -3
- constantes.py +4 -0
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_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(GTP_4o_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_MINI_MODEL, temperature=0),
|
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,6 +1,10 @@
|
|
1 |
import os
|
2 |
|
3 |
|
|
|
|
|
|
|
|
|
4 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
5 |
TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
|
6 |
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|
|
|
1 |
import os
|
2 |
|
3 |
|
4 |
+
GPT_MINI_MODEL="openai:gpt-4o-mini"
|
5 |
+
GTP_4o_MODEL="openai:gpt-4o"
|
6 |
+
|
7 |
+
|
8 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
9 |
TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
|
10 |
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|