Создание системы с агентами ИИ на LangGraph: пошаговое руководство по добавлению постоянства и потоковой передачи

Создание ИИ-агента с функциями сохранения и потоковой передачи

В этом руководстве мы улучшили нашего ИИ-агента, добавив две важные функции: сохранение состояния и потоковую передачу.

Настройка агента

Начнем с создания нашего агента. Мы загрузим необходимые переменные окружения, установим библиотеки и настроим инструмент поиска Tavily. Затем мы определим состояние агента и создадим его.

Добавление функции сохранения

Для сохранения состояния агента мы будем использовать функцию LangGraph, называемую checkpointer. Это позволит сохранять текущее состояние агента после каждого действия. Мы используем SqliteSaver, который легко подключается к базе данных SQLite.

Пример кода для сохранения состояния:

Создайте агента с возможностью сохранения:

    model = ChatGroq(model="Llama-3.3-70b-Specdec")
    bot = Agent(model, [tool], system=prompt, checkpointer=memory)

Добавление потоковой передачи

Потоковая передача необходима для предоставления обновлений в реальном времени. Мы будем фокусироваться на двух типах потоковой передачи:

Потоковые сообщения: промежуточные сообщения о действиях агента.
Потоковые токены: индивидуальные токены ответа ИИ.

Пример потоковой передачи сообщений:

Когда вы задаете вопрос, например, “Какова погода в Техасе?”, агент отправляет запрос и получает ответ в реальном времени.

Работа с Идентификаторами потоков

Идентификатор потока позволяет агенту поддерживать отдельные беседы. Это важно для сохранения контекста при общении с несколькими пользователями.

Заключение

Добавив функции сохранения и потоковой передачи, мы значительно улучшили возможности нашего ИИ-агента. Эти функции важны для создания готовых к производству приложений, особенно для взаимодействий с несколькими пользователями.

Практическое применение ИИ в вашей компании

Если вы хотите, чтобы ваша компания развивалась с помощью ИИ, определите возможности для автоматизации и ключевые показатели эффективности (KPI), которые хотите улучшить. Выбирайте подходящее решение и внедряйте его поэтапно.