Alexis-alexis's picture
Update app.py
b1d82fd verified
raw
history blame
1.23 kB
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()