Function main

tutorials/05_agentic_loop_complete.py:443–566

()

def main():
    print("=" * 60)
    print("Tutorial 05: 完整的 ConversationRuntime 演示")
    print("=" * 60)

    # --- 组装引擎 ---
    session = Session()

    api_client = DemoApiClient()

    tool_executor = (
        ToolExecutor()
        .register("bash", lambda input_str: "file1.py\nfile2.py\nREADME.md")
        .register("read_file", lambda input_str: "file content here")
    )

    permission_policy = (
        PermissionPolicy(PermissionMode.WORKSPACE_WRITE)
        .with_tool_requirement("bash", PermissionMode.DANGER_FULL_ACCESS)
        .with_tool_requirement("read_file", PermissionMode.READ_ONLY)
    )

    runtime = ConversationRuntime(
        session=session,
        api_client=api_client,
        tool_executor=tool_executor,
        permission_policy=permission_policy,
        system_prompt=["You are a helpful coding assistant."],
    )

    # --- 场景 A: 有 Prompter，bash 会被询问并通过 ---
    print("\n--- 场景 A: 使用 AutoAllowPrompter ---")
    summary = runtime.run_turn(
        "当前目录有什么文件？",
        prompter=AutoAllowPrompter(),
    )

    print(f"\n  循环轮次: {summary.iterations}")
    print(f"  助手消息数: {len(summary.assistant_messages)}")
    print(f"  工具结果数: {len(summary.tool_results)}")
    print(f"  Token 使用: {runtime.usage_tracker.summary()}")

    # 打印完整对话
    print("\n  --- 完整对话记录 ---")
    for i, msg in enumerate(session.messages):
        role = msg.role.upper()
        for block in msg.blocks:
            if isinstance(block, TextBlock):
                preview = block.text[:60].replace('\n', ' ')
                print(f"  [{i}] {role}: {preview}")
            elif isinstance(block, ToolUseBlock):
                print(f"  [{i}] {role}: [调用工具 {block.name}({block.input})]")
            elif isinstance(block, ToolResultBlock):
                status = "ERROR" if block.is_error else "OK"
                print(f"  [{i}] {role}: [工具结果 {block.tool_name}: {status}] {block.output[:40]}")

    # --- 场景 B: 没有 Prompter，bash 会被拒绝 ---
    print("\n\n--- 场景 B: 不提供 Prompter，bash 将被拒绝 ---")