AutoGen与NiceGUI集成：在Web界面中捕获代理响应的教程

import autogen
from nicegui import ui, context
from uuid import uuid4
import asyncio

# AutoGen Configuration
config_list = [
    {
        'model': 'gpt-4', # 替换为您的模型
        'api_key': 'YOUR_API_KEY' # 替换为您的API密钥
    }
]
llm_config = {
    'seed': 42,
    'config_list': config_list,
    'temperature': 0.2
}

# Initialize AutoGen Agents
assistant = autogen.AssistantAgent(name='Albert', llm_config=llm_config)
user_proxy = autogen.UserProxyAgent(
    name='user_proxy',
    human_input_mode="NEVER",
    max_consecutive_auto_reply=10, # 增加最大连续回复次数以允许更长的对话
    is_termination_msg=lambda x: x.get("content", "").rstrip().endswith("TERMINATE"),
    code_execution_config={"work_dir": "web"},
    llm_config=llm_config
)

# 用于存储NiceGUI聊天消息的全局列表
nicegui_messages = []

# 定义回调函数，捕获代理消息并添加到NiceGUI消息列表
def capture_agent_message(recipient, messages, sender, config):
    """
    当代理接收到消息时，此回调函数将被调用。
    它将消息内容和发送者添加到NiceGUI的全局消息列表中。
    """
    if messages and messages[-1]:
        latest_message = messages[-1]
        name = latest_message.get('name', sender.name) # 尝试获取消息中的名称，否则使用发送者名称
        content = latest_message.get('content', '')

        # 避免重复添加，并过滤掉可能不需要显示给用户的内部消息
        # 这里可以根据实际需求进行更复杂的过滤
        if content and name not in ["TERMINATE", "exit"]: # 简单的过滤示例
            # 确保在NiceGUI的事件循环中更新UI
            asyncio.create_task(add_message_to_nicegui(name, content))
    return False, None # 返回False表示不阻止消息传递，None表示没有额外回复

async def add_message_to_nicegui(name, content):
    """异步将消息添加到NiceGUI的显示列表并刷新UI"""
    nicegui_messages.append((name, content))
    # 刷新NiceGUI UI，如果当前有活跃的客户端连接
    if context.get_client().has_socket_connection:
        chat_messages.refresh()

# 将回调函数注册到user_proxy和assistant，以便捕获它们之间的所有消息
# 注册到user_proxy，捕获它收到的来自assistant的消息
user_proxy.register_reply(
    [autogen.Agent, None], # 捕获所有Agent发送给user_proxy的消息
    reply_func=capture_agent_message,
    config={"callback": None},
)
# 注册到assistant，捕获它收到的来自user_proxy的消息
assistant.register_reply(
    [autogen.Agent, None], # 捕获所有Agent发送给assistant的消息
    reply_func=capture_agent_message,
    config={"callback": None},
)

@ui.page('/')
def main():
    user_id = str(uuid4())  # Unique ID for each user session

    @ui.refreshable
    def chat_messages():
        for name, text in nicegui_messages:
            ui.chat_message(text=text, name=name, sent=name == 'You')
        if context.get_client().has_socket_connection:
            ui.run_javascript('setTimeout(() => window.scrollTo(0, document.body.scrollHeight), 0)')

    async def send():
        user_message = task_input.value
        if not user_message:
            return

        nicegui_messages.append(('You', user_message))  # Append user's message to the messages list
        chat_messages.refresh()  # Refresh chat messages to display the latest message
        task_input.value = ''  # Clear the input field after sending the message

        try:
            # 启动会话，但我们不再依赖其返回值来获取每条消息
            # 而是通过注册的回调函数来捕获
            await user_proxy.initiate_chat(assistant, message=user_message)
        except Exception as e:
            await add_message_to_nicegui('System Error', f"Error: {e}")
        finally:
            chat_messages.refresh() # 确保在会话结束后刷新一次，以防万一

    with ui.scroll_area().classes('w-full h-60 p-3 bg-white overflow-auto'):
        chat_messages()

    with ui.footer().style('position: fixed; left: 0; bottom: 0; width: 100%; background: white; padding: 10px; box-shadow: 0 -2px 5px rgba(0,0,0,0.1);'):
        task_input = ui.input(placeholder='输入你的消息...').props('clearable').classes('w-full').on('keydown.enter', send)
        ui.button('发送', on_click=send)

ui.run(title='Chat with Albert')