Class DemoApiClient

tutorials/05_agentic_loop_complete.py:393–440

模拟 API 客户端 —— 按剧本回复。 场景：用户问 "当前目录有什么文件？" AI 第一次回复：要用 bash 工具执行 ls AI 第二次回复：根据工具结果总结

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

class DemoApiClient:
    """
    模拟 API 客户端 —— 按剧本回复。

    场景：用户问 "当前目录有什么文件？"
    AI 第一次回复：要用 bash 工具执行 ls
    AI 第二次回复：根据工具结果总结
    """
    def __init__(self):
        self.call_count = 0

    def stream(self, request: ApiRequest) -> list[AssistantEvent]:
        self.call_count += 1

        if self.call_count == 1:
            return [
                AssistantEvent(event_type="text_delta", text="让我看看当前目录。"),
                AssistantEvent(
                    event_type="tool_use",
                    tool_id="tool-001",
                    tool_name="bash",
                    tool_input='{"command": "ls"}',
                ),
                AssistantEvent(
                    event_type="usage",
                    usage=TokenUsage(input_tokens=50, output_tokens=12),
                ),
                AssistantEvent(event_type="message_stop"),
            ]
        else:
            # 从对话历史里找到工具结果
            tool_output = ""
            for msg in request.messages:
                for block in msg.blocks:
                    if isinstance(block, ToolResultBlock):
                        tool_output = block.output

            return [
                AssistantEvent(
                    event_type="text_delta",
                    text=f"当前目录的文件有:\n{tool_output}",
                ),
                AssistantEvent(
                    event_type="usage",
                    usage=TokenUsage(input_tokens=80, output_tokens=20),
                ),
                AssistantEvent(event_type="message_stop"),
            ]


def main():