Создание системы агентов для генерации идей с AutoGen: разработка ИИ-агентов для мозгового штурма и обсуждения идей

Создание системы идей с помощью AutoGen

Процесс генерации идей часто требует много времени на анализ и обсуждение. Что если мы создадим два LLM, которые будут генерировать идеи и затем обсуждать их? Это звучит интересно, не так ли? Этот урок показывает, как создать решение на базе ИИ с двумя агентами LLM, которые взаимодействуют через структурированный диалог.

1. Установка и настройка

Сначала установите необходимые пакеты:

pip install -U autogen-agentchat
pip install autogen-ext[openai]

2. Основные компоненты

Давайте рассмотрим ключевые компоненты AutoGen, которые делают эту систему идей эффективной:

• RoundRobinGroupChat: Управляет командой агентов по очереди, обеспечивая структурированное взаимодействие.
• TextMentionTermination: Завершает разговор при обнаружении определенного ключевого слова (например, «ЗАВЕРШИТЬ»).
• AssistantAgent: Представляет участника с определенной ролью, генерируя контекстуальные ответы.

3. Создание команды агентов

Создайте двух специализированных агентов с различными ролями:

import asyncio
from autogen_agentchat.agents import AssistantAgent
from autogen_agentchat.base import TaskResult
from autogen_agentchat.conditions import TextMentionTermination
from autogen_agentchat.teams import RoundRobinGroupChat
from autogen_ext.models.openai import OpenAIChatCompletionClient

# Создайте клиента модели OpenAI.
model_client = OpenAIChatCompletionClient(
    model="gpt-4o-mini",
    api_key=API_KEY,
)

# Создайте основного агента.
primary_agent = AssistantAgent(
    "участник1",
    model_client=model_client,
    system_message="Вы участник сессии генерации идей. Вам будет предложено сформулировать идеи, которые затем будут обсуждены с другим участником.",
)

# Создайте агента-критика.
critic_agent = AssistantAgent(
    "участник2",
    model_client=model_client,
    system_message="Вы участник сессии генерации идей. Вам нужно будет оценить идеи вашего партнера и помочь сузить их.",
)

# Определите условие завершения.
text_termination = TextMentionTermination("ЗАВЕРШИТЬ")

# Создайте команду с основным и критическим агентами.
team = RoundRobinGroupChat([primary_agent, critic_agent], termination_condition=text_termination)

4. Запуск команды

Запустите команду с асинхронной обработкой:

result = await team.run(task="Генерация идей для применения ИИ в здравоохранении.")
print(result)

5. Мониторинг взаимодействий

Вы также можете отслеживать дебаты в реальном времени:

await team.reset()  # Сброс команды для новой задачи.
async for message in team.run_stream(task="Генерация идей для применения ИИ в здравоохранении."):
    if isinstance(message, TaskResult):
        print("Причина остановки:", message.stop_reason)
    else:
        print(message)

AutoGen также предоставляет функцию для визуализации взаимодействий в более удобном формате.

Идеи для улучшения системы

• Добавление специализированных агентов (медицинские эксперты, технические валидаторы).
• Реализация пользовательских условий завершения.
• Создание простого интерфейса с помощью streamlit.
• Добавление новых участников в команду.

Как внедрить ИИ в вашу компанию

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

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

Попробуйте AI Sales Bot. Это ИИ ассистент для продаж, который помогает отвечать на вопросы клиентов и генерировать контент для отдела продаж.

Узнайте, как ИИ может изменить процесс продаж в вашей компании с решением от saile.ru. Будущее уже здесь!