Frazer2810 commited on
Commit
ec841ad
·
verified ·
1 Parent(s): 1367953

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -26,9 +26,9 @@ if not TAVILY_API_KEY:
26
 
27
 
28
  MAX_WIKIPEDIA_RESULTS = 3
29
- MAX_TAVILY_RESULTS = 3 # Tavily può restituire risultati più ricchi
30
  MAX_ARXIV_RESULTS = 2
31
- MAX_TOOL_ITERATIONS = 5
32
 
33
  class GaiaAgent:
34
  def __init__(self, model_name="gpt-4.1"):
@@ -180,7 +180,7 @@ class GaiaAgent:
180
  ]
181
 
182
  system_prompt = (
183
- "You are a general AI assistant. I will ask you a question. Your output needs to be checked versus an exact match test, so your final answer must be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
184
  )
185
 
186
  messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": question}]
 
26
 
27
 
28
  MAX_WIKIPEDIA_RESULTS = 3
29
+ MAX_TAVILY_RESULTS = 4 # Tavily può restituire risultati più ricchi
30
  MAX_ARXIV_RESULTS = 2
31
+ MAX_TOOL_ITERATIONS = 7
32
 
33
  class GaiaAgent:
34
  def __init__(self, model_name="gpt-4.1"):
 
180
  ]
181
 
182
  system_prompt = (
183
+ "You are a general AI assistant. I will ask you a question. Your output needs to be checked versus an exact match test, so your final answer must be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string. In comma separated lists, do not add blank spaces. If you are asked for a code, give the code in a string and nothing else."
184
  )
185
 
186
  messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": question}]