Туториал 1: Первый агент за 30 минут

import requests
import json

OLLAMA_API = "http://localhost:11434/api/chat"
MODEL = "qwen3.5:4b"


def search_web(query: str) -> str:
    """
    Имитация поиска в интернете.
    В реальном проекте здесь был бы вызов Google Search API,
    DuckDuckGo или векторного поиска.
    """
    database = {
        "погода москва": "В Москве сейчас +18°C, облачно",
        "погода лондон": "В Лондоне +15°C, дождь",
        "столица франции": "Столица Франции — Париж",
        "python создатель": "Python создал Гвидо ван Россум в 1991 году",
        "что такое ai агент": (
            "AI-агент — это программа, которая использует языковую модель "
            "для принятия решений и вызова инструментов"
        ),
    }
    
    # Ищем по ключевым словам
    for key, value in database.items():
        if key in query.lower():
            return value
    
    return f"Информация по запросу '{query}' не найдена в локальной БД"


# Описываем инструмент для модели
TOOLS = [
    {
        "type": "function",
        "function": {
            "name": "search_web",
            "description": (
                "Поиск информации в интернете. "
                "Используй когда нужно узнать факты, погоду, даты."
            ),
            "parameters": {
                "type": "object",
                "properties": {
                    "query": {
                        "type": "string",
                        "description": "Поисковый запрос"
                    }
                },
                "required": ["query"]
            }
        }
    }
]


def agent_loop(user_input: str, max_steps: int = 5) -> str:
    """
    ReAct-цикл агента:
    1. Спросить модель
    2. Если модель вызывает инструмент — выполнить
    3. Вернуть результат модели
    4. Повторять, пока модель не ответит
    """
    
    messages = [
        {
            "role": "system",
            "content": (
                "Ты — полезный ассистент. Если тебе нужно узнать "
                "какую-то информацию — используй search_web. "
                "Не угадывай факты, лучше поищи."
            )
        },
        {"role": "user", "content": user_input}
    ]
    
    for step in range(max_steps):
        print(f"\n  [Шаг {step + 1}] Отправляю запрос модели...")
        
        response = requests.post(OLLAMA_API, json={
            "model": MODEL,
            "messages": messages,
            "tools": TOOLS,
            "stream": False
        })
        
        msg = response.json()["message"]
        messages.append(msg)
        
        # Модель хочет вызвать инструмент?
        if msg.get("tool_calls"):
            for tc in msg["tool_calls"]:
                name = tc["function"]["name"]
                args = json.loads(tc["function"]["arguments"])
                
                print(f"  [Шаг {step + 1}] Вызываю {name}({args})")
                
                # Выполняем инструмент
                result = search_web(**args)
                print(f"  [Шаг {step + 1}] Результат: {result}")
                
                messages.append({
                    "role": "tool",
                    "name": name,
                    "content": json.dumps({"result": result})
                })
        else:
            # Модель решила ответить
            answer = msg["content"]
            print(f"  [Шаг {step + 1}] Ответ модели получен")
            return answer
    
    return "Агент не смог выполнить задачу за отведённое число шагов"


# === Точка входа ===
if __name__ == "__main__":
    print("🧠 Мой первый AI-агент")
    print("=" * 40)
    
    while True:
        user_input = input("\n❓ Спроси что-нибудь (или 'exit'): ")
        if user_input.lower() == "exit":
            break
        
        result = agent_loop(user_input)
        print(f"\n📋 Ответ: {result}")