Function main

tutorials/07_auto_compaction.py:381–499

()

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

def main():
    print("=" * 60)
    print("Tutorial 07: Auto Compaction 自动压缩演示")
    print("=" * 60)

    # --- 构建一个包含很多消息的 session ---
    session = Session()

    # 模拟一段长对话
    conversations = [
        ("user", "帮我创建一个 Flask Web 应用"),
        ("assistant", "好的，让我先创建项目结构。我会用 bash 来创建目录。"),
        ("user", "然后在 src/app.py 里写一个简单的 Hello World"),
        ("assistant", "好的，我来创建 src/app.py 文件。TODO: 还需要添加路由测试。"),
        ("user", "帮我添加一个 /api/users 的 REST 接口"),
        ("assistant", "我来修改 src/app.py，添加用户接口。Next: 需要写单元测试。"),
        ("user", "在 tests/test_app.py 里写测试"),
        ("assistant", "测试已经写好了。所有 3 个测试都通过了。"),
        ("user", "帮我添加数据库连接"),
        ("assistant", "我来配置 SQLAlchemy 数据库连接。Remaining: 迁移脚本还没写。"),
        ("user", "看看当前的 git 状态"),
        ("assistant", "当前有 5 个文件被修改，还没提交。"),
    ]

    for role, text in conversations:
        if role == "user":
            session.messages.append(ConversationMessage.user_text(text))
        else:
            session.messages.append(ConversationMessage.assistant_text(text))

    # 加一些工具使用记录
    session.messages.insert(2, ConversationMessage(
        role="assistant",
        blocks=(
            TextBlock(text="让我创建目录。"),
            ToolUseBlock(id="t1", name="bash", input='{"command": "mkdir -p src tests"}'),
        ),
    ))
    session.messages.insert(3, ConversationMessage.tool_result("t1", "bash", "(directory created)"))

    print(f"\n对话总消息数: {len(session.messages)}")
    print(f"预估 token 数: {estimate_session_tokens(session)}")

    # --- 1. 测试是否需要压缩（token 阈值设得低一点方便演示）---
    config = CompactionConfig(preserve_recent_messages=4, max_estimated_tokens=50)
    print(f"\n需要压缩吗 (阈值={config.max_estimated_tokens} tokens)? {should_compact(session, config)}")

    # --- 2. 执行压缩 ---
    result = compact_session(session, config)

    print(f"\n--- 压缩结果 ---")
    print(f"  被压缩掉的消息数: {result.removed_message_count}")
    print(f"  压缩前消息数: {len(session.messages)}")
    print(f"  压缩后消息数: {len(result.compacted_session.messages)}")
    print(f"  压缩前 token 数: {estimate_session_tokens(session)}")
    print(f"  压缩后 token 数: {estimate_session_tokens(result.compacted_session)}")

    # --- 3. 看看摘要长什么样 ---