Function build_assistant_message

tutorials/10_sse_streaming.py:546–599

把 AssistantEvent 列表组装成一条完整的 AI 消息。 逻辑： 1. 累积 text_delta → 合并成一个 TextBlock 2. 遇到 tool_use → 先把之前累积的文本存起来，再存工具调用 3. 记录 usage 4. 检查是否收到 message_stop 对应源码: conversation.rs:353-390 (build_assistant_message)

(events: list[AssistantEvent])

def build_assistant_message(events: list[AssistantEvent]) -> AssistantMessage:
    """
    把 AssistantEvent 列表组装成一条完整的 AI 消息。

    逻辑：
    1. 累积 text_delta → 合并成一个 TextBlock
    2. 遇到 tool_use → 先把之前累积的文本存起来，再存工具调用
    3. 记录 usage
    4. 检查是否收到 message_stop

    对应源码: conversation.rs:353-390 (build_assistant_message)
    """
    blocks = []
    current_text = ""  # 正在累积的文本
    input_tokens = 0
    output_tokens = 0
    finished = False

    for event in events:
        if event.event_type == AssistantEventType.TEXT_DELTA:
            current_text += event.text

        elif event.event_type == AssistantEventType.TOOL_USE:
            # 遇到工具调用 → 先把之前的文本存起来
            if current_text:
                blocks.append(ContentBlock(block_type="text", text=current_text))
                current_text = ""
            # 再存工具调用
            blocks.append(ContentBlock(
                block_type="tool_use",
                tool_id=event.tool_id,
                tool_name=event.tool_name,
                tool_input=event.tool_input,
            ))

        elif event.event_type == AssistantEventType.USAGE:
            input_tokens = event.input_tokens
            output_tokens = event.output_tokens

        elif event.event_type == AssistantEventType.MESSAGE_STOP:
            finished = True

    # 把最后累积的文本也存起来
    if current_text:
        blocks.append(ContentBlock(block_type="text", text=current_text))

    if not finished:
        print("  [WARNING] 流结束了但没收到 message_stop 事件！")

    return AssistantMessage(
        blocks=blocks,
        input_tokens=input_tokens,
        output_tokens=output_tokens,
    )


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