app.py

from smolagents import CodeAgent, HfApiModel, load_tool
import yaml
from smolagents import tool
from duckduckgo_search import DDGS

from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI

@tool
def DuckDuckGoSearchTool(query: str) -> str:
    """
    Инструмент для поиска информации в интернете с помощью DuckDuckGo.
    Args:
        query: Поисковый запрос.
    """
    with DDGS() as ddgs:
        results = [r for r in ddgs.text(query, max_results=5)]  # Ограничиваем до 5 результатов
        if not results:
            return "По вашему запросу ничего не найдено."
        formatted_results = "\n\n".join(
            f"**Заголовок:** {r['title']}\n**Ссылка:** {r['href']}\n**Краткое содержание:** {r['body']}"
            for r in results
        )
    return formatted_results

final_answer = FinalAnswerTool()

model = HfApiModel(
    max_tokens=10000,
    temperature=1,
    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
    custom_role_conversions=None,
)

# Import tool from Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)

# Загрузка prompt_templates из prompts.yaml
with open("prompts.yaml", 'r') as stream:
    try:
        prompt_templates = yaml.safe_load(stream)
    except yaml.YAMLError as exc:
        print(exc)

# Проверка, что prompt_templates имеет правильный формат
if isinstance(prompt_templates, dict) and 'system_prompt' in prompt_templates:
    # Если все ок, используем загруженный шаблон
    system_prompt = prompt_templates['system_prompt']
else:
    # Иначе, создаем новый словарь с system_prompt
    system_prompt = """
    Ты — Perplexity: профессиональный поисковый ассистент.
    Твоя задача — предоставлять исчерпывающие и точные ответы на запросы пользователей, основываясь на информации из нескольких источников в интернете.
    #... (остальной текст вашего промпта)...
    Начни!
    """
    prompt_templates = {'system_prompt': system_prompt}

agent = CodeAgent(
    model=model,
    tools=[DuckDuckGoSearchTool, final_answer, image_generation_tool],
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)

GradioUI(agent).launch()