Update agent.py
Browse files
agent.py
CHANGED
|
@@ -2,13 +2,13 @@ import os
|
|
| 2 |
import logging
|
| 3 |
from typing import List, Dict, Any, Optional
|
| 4 |
|
| 5 |
-
from
|
| 6 |
-
from
|
| 7 |
-
from
|
| 8 |
-
from
|
| 9 |
-
from
|
| 10 |
-
from
|
| 11 |
-
from
|
| 12 |
|
| 13 |
# Configurazione del logging
|
| 14 |
logging.basicConfig(
|
|
@@ -20,7 +20,7 @@ logger = logging.getLogger("GaiaAgent")
|
|
| 20 |
class GaiaAgent:
|
| 21 |
"""
|
| 22 |
Agente AI per superare il test GAIA livello 1.
|
| 23 |
-
Utilizza
|
| 24 |
"""
|
| 25 |
|
| 26 |
def __init__(self):
|
|
|
|
| 2 |
import logging
|
| 3 |
from typing import List, Dict, Any, Optional
|
| 4 |
|
| 5 |
+
from smolagents.agent import Agent
|
| 6 |
+
from smolagents.llms import OpenAILLM
|
| 7 |
+
from smolagents.tools import Tool
|
| 8 |
+
from smolagents.tools.web_search import DuckDuckGoSearchTool
|
| 9 |
+
from smolagents.tools.wikipedia import WikipediaSearchTool
|
| 10 |
+
from smolagents.tools.arxiv import ArxivSearchTool
|
| 11 |
+
from smolagents.tools.code_interpreter import CodeInterpreterTool
|
| 12 |
|
| 13 |
# Configurazione del logging
|
| 14 |
logging.basicConfig(
|
|
|
|
| 20 |
class GaiaAgent:
|
| 21 |
"""
|
| 22 |
Agente AI per superare il test GAIA livello 1.
|
| 23 |
+
Utilizza smolagents con capacità di ricerca Wikipedia, DuckDuckGo, arXiv e code execution.
|
| 24 |
"""
|
| 25 |
|
| 26 |
def __init__(self):
|