Function demo_auto_compaction

tutorials/16_prompt_building_and_compaction.py:948–1007

演示自动压缩

()

def demo_auto_compaction():
    """演示自动压缩"""
    print("\n" + "=" * 60)
    print("自动压缩演示")
    print("=" * 60)

    # 构建一个较长的会话
    messages = [
        ConversationMessage.user_text("请帮我修改 rust/crates/runtime/src/main.rs 中的 bug"),
        ConversationMessage(
            role=MessageRole.ASSISTANT,
            blocks=[
                ContentBlock(block_type="text", text="让我先读取这个文件"),
                ContentBlock(
                    block_type="tool_use", tool_name="Read",
                    tool_id="1", input_str='{"path":"rust/crates/runtime/src/main.rs"}'
                ),
            ],
        ),
        ConversationMessage(
            role=MessageRole.TOOL,
            blocks=[ContentBlock(
                block_type="tool_result", tool_name="Read",
                output="fn main() { println!(\"hello\"); }" * 50,  # 模拟大文件
            )],
        ),
        ConversationMessage(
            role=MessageRole.ASSISTANT,
            blocks=[ContentBlock(
                block_type="text",
                text="我发现了 bug。Next: 需要修改第 42 行的类型错误。"
            )],
        ),
        ConversationMessage.user_text("好的请修改"),
        ConversationMessage(
            role=MessageRole.ASSISTANT,
            blocks=[ContentBlock(
                block_type="text",
                text="已修改完成。todo: 还需要更新测试文件 tests/test_main.rs"
            )],
        ),
    ]

    print(f"\n压缩前:")
    print(f"  消息数: {len(messages)}")
    print(f"  估算 tokens: {estimate_session_tokens(messages)}")

    result = compact_session(
        messages,
        CompactionConfig(preserve_recent_messages=2, max_estimated_tokens=100),
    )

    print(f"\n压缩后:")
    print(f"  消息数: {len(result.compacted_messages)}")
    print(f"  删除了: {result.removed_count} 条消息")
    print(f"  估算 tokens: {estimate_session_tokens(result.compacted_messages)}")

    print(f"\n摘要内容:")