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

Идеи и обсуждения с помощью ИИ

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

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 ExternalTermination, TextMentionTermination
from autogen_agentchat.teams import RoundRobinGroupChat
from autogen_agentchat.ui import Console
from autogen_core import CancellationToken
from autogen_ext.models.openai import OpenAIChatCompletionClient

from apikey import API_KEY

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)

Завершение

Теперь система готова к использованию. Вы можете улучшить ее, добавив специализированные агенты, реализовав пользовательские условия завершения и расширив команду.