File size: 1,228 Bytes
b1d82fd 6aae614 b1d82fd 9b5b26a b1d82fd e121372 b1d82fd 13d500a 8c01ffb b1d82fd 8c01ffb 8fe992b b1d82fd 8fe992b b1d82fd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | 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()
|