Jack1808/Claude_Code

1"""Pydantic models for API responses."""2 3from typing import Any, Literal4 5from pydantic import BaseModel6 7from .anthropic import ContentBlockText, ContentBlockThinking, ContentBlockToolUse8 9 10class TokenCountResponse(BaseModel):11    input_tokens: int12 13 14class Usage(BaseModel):15    input_tokens: int16    output_tokens: int17    cache_creation_input_tokens: int = 018    cache_read_input_tokens: int = 019 20 21class MessagesResponse(BaseModel):22    id: str23    model: str24    role: Literal["assistant"] = "assistant"25    content: list[26        ContentBlockText | ContentBlockToolUse | ContentBlockThinking | dict[str, Any]27    ]28    type: Literal["message"] = "message"29    stop_reason: (30        Literal["end_turn", "max_tokens", "stop_sequence", "tool_use"] | None31    ) = None32    stop_sequence: str | None = None33    usage: Usage34