from opencode_sdk import OpencodeClient

# 初始化客户端 - 使用 /api 前缀
client = OpencodeClient(
    base_url="http://192.168.77.28:8001",
    directory="/data/seo/workspace"
)

# 创建一个新会话
print("📝 创建会话...")
session = client.sessions.create(title="测试会话")
print(f"✅ 会话已创建，ID: {session.id}")

# 发送一条消息
print("\n💬 发送消息...")
response = client.sessions.prompt(
    session_id=session.id,
    parts=[{"type": "text", "text": "当前时间"}],
    agent="build",
    model={
        "modelID": "gpt-5-nano",
        "providerID": "opencode"
    },
    variant="low"
)

print(f"✅ 收到响应!")
print(f"消息ID: {response.id}")
print(f"角色: {response.role}")
print(f"时间: {response.time}")
print(f"模型: {response.model_id} ({response.provider_id})")
print(f"令牌: 输入={response.tokens.input}, 输出={response.tokens.output}, 推理={response.tokens.reasoning}")
print(f"部分 ({len(response.parts)}):")
for i, part in enumerate(response.parts):
    print(f"  [{i}] 类型: {part.type}")
    if hasattr(part, 'text') and part.text:
        text_preview = part.text[:100] + "..." if len(part.text) > 100 else part.text
        print(f"      文本: {text_preview}")
    if hasattr(part, 'reason'):
        print(f"      原因: {part.reason}")
    print(f"      ID: {part.id}")
print()


# 列出所有会话
print("\n📋 列出所有会话...")
sessions = client.sessions.list()
print(f"✅ 共有 {len(sessions)} 个会话")
for s in sessions[:5]:  # 只显示前5个
    print(f"  - {s.title} (ID: {s.id})")

import asyncio
from opencode_sdk import OpencodeClient


async def stream_chat():
    # 使用你的服务器地址
    client = OpencodeClient(base_url="http://192.168.77.28:8001")
    
    print("创建会话...")
    # 创建会话
    session = client.sessions.create(
        title="流式对话",
        directory="/data/seo/workspace"
    )
    
    print(f"会话已创建: {session.id}\n")
    print("AI 响应:\n")
    
    # 发送消息并接收流式响应
    try:
        async for event in client.events.subscribe_session(
            session_id=session.id,
            parts=[{"type": "text", "text": "当前时间"}],
            directory="/data/seo/workspace",
            agent="build",
            model={
                "modelID": "gpt-5-nano",
                "providerID": "opencode"
            },
            variant="low"
        ):
            # 只打印流式文本的增量
            if event.type == "message.part.delta":
                print(event.properties.delta, end="", flush=True)
    except Exception as e:
        print(f"\n\n错误: {type(e).__name__}: {e}")
        import traceback
        traceback.print_exc()
    
    print("\n\n会话完成，正在清理...")
    client.sessions.delete(session.id)


# 运行
asyncio.run(stream_chat())

async for event in client.events.subscribe_session(
    session_id="session_id",
    parts=[{"type": "text", "text": "消息内容"}],
    directory="/data/seo/workspace",
    agent="build",
    model={"modelID": "gpt-5-nano", "providerID": "opencode"},
    variant="low"
):
    if event.type == "message.part.delta":
        # 实时打印 AI 生成的文本
        print(event.properties.delta, end="", flush=True)
    elif event.type == "session.idle":
        # 会话完成
        break

async for event in client.events.subscribe_global():
    print(f"全局事件: {event.type}")
    
    if event.type == "session.created":
        # 访问会话信息
        info = event.properties.info
        print(f"新会话: {info.get('title')} (ID: {info.get('id')})")
    elif event.type == "message.part.delta":
        # 消息增量
        print(f"消息: {event.properties.delta}")

# 订阅全局事件
async for event in client.events.subscribe():
    print(f"事件: {event.type}")

# 订阅特定会话事件
async for event in client.events.subscribe(session_id="session_id"):
    print(f"会话事件: {event.type}")