Update agent.py

agent.py

@@ -26,12 +26,12 @@ if not TAVILY_API_KEY:
 
 
 MAX_WIKIPEDIA_RESULTS = 3
-MAX_TAVILY_RESULTS = 5 # Tavily può restituire risultati più ricchi
-MAX_ARXIV_RESULTS = 3
-MAX_TOOL_ITERATIONS = 7
+MAX_TAVILY_RESULTS = 3 # Tavily può restituire risultati più ricchi
+MAX_ARXIV_RESULTS = 2
+MAX_TOOL_ITERATIONS = 5
 
 class GaiaAgent:
-    def __init__(self, model_name="gpt-4o"):
+    def __init__(self, model_name="gpt-4.1"):
         self.model_name = model_name
         self.openai_client = openai.OpenAI() # Inizializza il client OpenAI
         if TAVILY_API_KEY:
@@ -180,20 +180,8 @@ class GaiaAgent:
         ]
 
         system_prompt = (
-            "Sei GaiaAgent, un assistente AI progettato per rispondere a domande in modo accurato e completo, specialmente per task complessi come quelli di GAIA. "
-            "Hai accesso ai seguenti strumenti: ricerca Wikipedia, ricerca web Tavily, ricerca Arxiv e un interprete di codice Python. " # Aggiornato DuckDuckGo a Tavily
-            "Segui questi passaggi per rispondere alla domanda dell'utente:\n"
-            "1. **Analizza la Domanda**: Comprendi a fondo cosa viene chiesto. Se la domanda è complessa, scomponila in sotto-problemi.\n"
-            "2. **Pianifica l'Uso degli Strumenti**: Decidi quali strumenti sono più appropriati e in quale ordine. Puoi usare più strumenti in sequenza.\n"
-            "   - `search_wikipedia`: per conoscenza generale consolidata.\n"
-            "   - `search_tavily`: per informazioni recenti, argomenti di nicchia, o per confermare/trovare URL specifici. Può fornire risposte dirette.\n" # Aggiornato
-            "   - `search_arxiv`: per ricerca scientifica e tecnica.\n"
-            "   - `execute_python_code`: per calcoli, elaborazione di dati testuali/numerici, o simulazioni. Il codice deve stampare esplicitamente i risultati.\n"
-            "3. **Esegui e Itera**: Usa gli strumenti. Se uno strumento non fornisce l'informazione necessaria, valuta se riformulare la query, provare un altro strumento, o concludere che l'informazione non è reperibile con gli strumenti attuali.\n"
-            "4. **Sintetizza e Rispondi**: Combina le informazioni raccolte e la tua conoscenza interna per formulare una risposta finale chiara, concisa e che risponda direttamente alla domanda originale. Se non riesci a trovare una risposta definitiva, indicalo chiaramente.\n"
-            "Limita il numero di chiamate agli strumenti a un massimo di {} iterazioni per domanda. ".format(MAX_TOOL_ITERATIONS) +
-            "Fornisci solo la risposta finale senza commentare il processo di ragionamento o l'uso degli strumenti, a meno che non sia esplicitamente richiesto o necessario per chiarire la provenienza di un'informazione cruciale."
-        )
+            "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."
+            )
 
         messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": question}]