Update agent.py
Browse files
agent.py
CHANGED
|
@@ -25,13 +25,13 @@ 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="
|
| 35 |
self.model_name = model_name
|
| 36 |
self.openai_client = openai.OpenAI() # Inizializza il client OpenAI
|
| 37 |
if 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="o4-mini"):
|
| 35 |
self.model_name = model_name
|
| 36 |
self.openai_client = openai.OpenAI() # Inizializza il client OpenAI
|
| 37 |
if TAVILY_API_KEY:
|