gdms commited on
Commit
9829c50
·
1 Parent(s): 417f798

testando gpt 4o para busca

Browse files
Files changed (2) hide show
  1. agent.py +3 -3
  2. 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("openai:gpt-4o-mini"),
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("openai:gpt-4o-mini"),
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("openai:gpt-4o-mini", 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,
 
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")