"""
简单 Agent 示例。

演示如何使用 PyOpenClaw 创建一个基本的 Agent。
"""

import asyncio
import os
from dotenv import load_dotenv

from pyopenclaw.config.settings import AgentConfig
from pyopenclaw.core.agent import Agent
from pyopenclaw.core.llm_client import create_llm_client
from pyopenclaw.core.tool_registry import ToolRegistry
from pyopenclaw.tools.file_tools import ReadTool, WriteTool


async def main():
    """运行简单 Agent 示例"""
    # 加载环境变量
    load_dotenv()
    
    # 检查 API Key
    api_key = os.getenv("OPENAI_API_KEY")
    if not api_key:
        print("错误: 请设置 OPENAI_API_KEY 环境变量")
        print("示例: export OPENAI_API_KEY='your-api-key'")
        return
    
    # 1. 创建 Agent 配置
    config = AgentConfig(
        name="小白",
        model="openai/gpt-4o",
        system_prompt="你是一个有用的 AI 助手，名叫小白。你是一只忠诚的电子宠物狗。",
        tools=["read", "write"],
        max_iterations=10,
    )
    
    # 2. 创建 LLM 客户端
    llm_client = create_llm_client(
        provider="openai",
        model="gpt-4o",
        api_key=api_key,
    )
    
    # 3. 创建工具注册表并注册工具
    tool_registry = ToolRegistry()
    tool_registry.register(ReadTool())
    tool_registry.register(WriteTool())
    
    # 4. 创建 Agent
    agent = Agent(
        config=config,
        llm_client=llm_client,
        tool_registry=tool_registry,
    )
    
    print("=" * 50)
    print("PyOpenClaw 简单示例")
    print("=" * 50)
    print(f"Agent 名称: {config.name}")
    print(f"模型: {config.model}")
    print(f"可用工具: {config.tools}")
    print("=" * 50)
    
    # 5. 运行 Agent
    user_input = "你好，请介绍一下你自己"
    print(f"\n用户: {user_input}\n")
    print("Agent 回复:")
    print("-" * 50)
    
    async for event in agent.run(user_input):
        event_type = event.get("type")
        
        if event_type == "content":
            # 内容输出
            print(event["content"])
            if event.get("done"):
                break
        
        elif event_type == "tool_call":
            # 工具调用
            print(f"\n[调用工具: {event['name']}]")
            print(f"参数: {event['arguments']}")
        
        elif event_type == "tool_result":
            # 工具结果
            result = event["result"]
            if len(result) > 100:
                result = result[:100] + "..."
            print(f"结果: {result}\n")
        
        elif event_type == "error":
            # 错误
            print(f"\n错误: {event['content']}")
            break
    
    print("-" * 50)
    print("\n运行完成！")


if __name__ == "__main__":
    asyncio.run(main())