Function main

tutorials/02_session_and_messages.py:309–417

()

# ============================================================

def main():
    print("=" * 60)
    print("Tutorial 02: Session 与消息模型演示")
    print("=" * 60)

    # --- 1. 构建一段完整对话 ---
    print("\n--- 构建一段对话 ---")

    session = Session()

    # 用户说话
    session.messages.append(ConversationMessage.user_text("2+2 等于几？"))
    print(f"添加消息: {session.messages[-1]}")

    # 助手回复（包含文字 + 工具调用）
    session.messages.append(ConversationMessage.assistant_with_tool_use(
        text="让我算一下。",
        tool_id="tool-1",
        tool_name="add",
        tool_input="2,2",
        usage=TokenUsage(input_tokens=20, output_tokens=6),
    ))
    print(f"添加消息: {session.messages[-1]}")

    # 工具返回结果
    session.messages.append(ConversationMessage.tool_result(
        tool_use_id="tool-1",
        tool_name="add",
        output="4",
    ))
    print(f"添加消息: {session.messages[-1]}")

    # 助手最终回答
    session.messages.append(ConversationMessage.assistant_text(
        text="2 + 2 的答案是 4。",
        usage=TokenUsage(input_tokens=30, output_tokens=8),
    ))
    print(f"添加消息: {session.messages[-1]}")

    # --- 2. 查看消息内部结构 ---
    print("\n--- 查看第二条消息的内部结构（助手的工具调用）---")
    msg = session.messages[1]
    print(f"  角色: {msg.role}")
    print(f"  内容块数量: {len(msg.blocks)}")
    for i, block in enumerate(msg.blocks):
        print(f"  块 {i}: type={block.type}, 内容={block}")
    print(f"  Token 使用: {msg.usage}")

    # --- 3. 保存和恢复 ---
    print("\n--- 保存和恢复 Session ---")
    # 创建临时文件来演示
    tmp_path = os.path.join(tempfile.gettempdir(), "tutorial_session.json")
    session.save_to_file(tmp_path)
    print(f"  已保存到: {tmp_path}")

    # 读回来
    restored = Session.load_from_file(tmp_path)
    print(f"  已恢复: {len(restored.messages)} 条消息")