from smolagents import CodeAgent, HfApiModel
from tools.gaia_tool import GaiaTool
from tools.final_answer import FinalAnswerTool
from tools.python_interpreter import PythonInterpreterTool
from Gradio_UI import GradioUI

# Настройка модели
model = HfApiModel(
    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",  # Можешь заменить на другую при желании
    max_tokens=2048,
    temperature=0.3,
)

# Настройка агента
agent = CodeAgent(
    model=model,
    tools=[
        GaiaTool(),  # Это обязательно — чтобы агент получал задачи GAIA
        PythonInterpreterTool(),  # Чтобы агент мог выполнять вычисления
        FinalAnswerTool(),       # Чтобы агент понимал, где завершить мысль
    ],
    max_steps=6,  # Не даём агенту уйти в бесконечное планирование
    verbosity_level=2,  # Показываем шаги агента для отладки
    name="MyGAIAgent",
    description="Agent that solves GAIA level 1 tasks with tools and code execution.",
)

# Интерфейс запуска
GradioUI(agent).launch()