Frazer2810 commited on
Commit
d1ea6fd
·
verified ·
1 Parent(s): 1e866d2

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +5 -5
agent.py CHANGED
@@ -25,10 +25,10 @@ if not TAVILY_API_KEY:
25
  print("La ricerca Tavily non funzionerà senza una chiave API valida.")
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 = 6
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 that uses the tools available. I will ask you a question. You must output only the exact answer to the question with no comments, 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 use blank spaces. If you are asked for a code number, give the code number nothing else. Do not add a dot at the end of the answer. If you are asked for 'the number of', answer with the count number."
184
  )
185
 
186
  messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": question}]
 
25
  print("La ricerca Tavily non funzionerà senza una chiave API valida.")
26
 
27
 
28
+ MAX_WIKIPEDIA_RESULTS = 4
29
+ MAX_TAVILY_RESULTS = 4 # Tavily può restituire risultati più ricchi
30
+ MAX_ARXIV_RESULTS = 3
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 that uses the tools available. I will ask you a question. You must output only the exact answer to the question with no comments, 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 use blank spaces. If you are asked for a code number, give the code number nothing else. Do not add a dot at the end of the answer. If you are asked for 'the number of', answer with the count number. Pay attention at first and last names in the questions."
184
  )
185
 
186
  messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": question}]