Update agent.py
Browse files
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 =
|
| 29 |
-
MAX_TAVILY_RESULTS =
|
| 30 |
-
MAX_ARXIV_RESULTS =
|
| 31 |
-
MAX_TOOL_ITERATIONS =
|
| 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}]
|