Function main

tutorials/12_session_persistence.py:516–748

()

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

def main():
    print("=" * 60)
    print("Tutorial 12: 会话持久化（JSONL）")
    print("=" * 60)

    # 创建临时目录
    tmpdir = tempfile.mkdtemp(prefix="session_demo_")

    # --- 1. 方案 A：JSON 整文件 ---
    print("\n--- 方案 A: JSON 整文件（源码中的实现）---")

    session = Session()
    session.add_message(ConversationMessage.user_text("帮我写个排序函数"))
    session.add_message(ConversationMessage.assistant_text(
        "好的，我来帮你写一个快速排序。"
    ))

    json_path = os.path.join(tmpdir, "session.json")
    save_session_json(session, json_path)
    print(f"  保存到: {json_path}")

    restored = load_session_json(json_path)
    print(f"  恢复成功！消息数: {len(restored.messages)}")
    for msg in restored.messages:
        text = msg.blocks[0].text if msg.blocks else ""
        print(f"    [{msg.role}] {text[:40]}")

    # --- 2. 方案 B：JSONL 仅追加 ---
    print("\n--- 方案 B: JSONL 仅追加（生产级方案）---")

    storage = JsonlSessionStorage(tmpdir)
    sid = "demo-session-001"

    # 模拟对话
    msgs = [
        ConversationMessage.user_text("帮我分析一下这个 bug"),
        ConversationMessage.assistant_text(
            "好的，让我看看代码。"
        ),
        ConversationMessage.user_text("bug 在 auth.py 第 42 行"),
        ConversationMessage.assistant_text(
            "我看到了，这是一个空指针问题。已修复。"
        ),
    ]

    # 逐条追加（模拟实时对话）
    prev_uuid = ""
    for msg in msgs:
        msg.parent_uuid = prev_uuid
        storage.append_message(sid, msg)
        prev_uuid = msg.uuid
        print(f"  追加: [{msg.role}] {msg.blocks[0].text[:30]}...")

    # 追加元数据
    storage.append_metadata(sid, "custom-title",
                            {"title": "修复 auth.py 空指针 bug"})
    storage.append_metadata(sid, "last-prompt",
                            {"text": "bug 在 auth.py 第 42 行"})