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

Ajuste modelos e temperatura

Browse files
Files changed (2) hide show
  1. agent.py +3 -3
  2. 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(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,7 +27,7 @@ class 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,7 +35,7 @@ class 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,
 
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
- 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")
 
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")