Cost Aware Llm Pipeline

MODEL_SONNET = "claude-sonnet-5"MODEL_HAIKU = "claude-haiku-4-5-20251001"
_SONNET_TEXT_THRESHOLD = 10_000  # 文字数_SONNET_ITEM_THRESHOLD = 30     # アイテム数
def select_model(    text_length: int,    item_count: int,    force_model: str | None = None,) -> str:    """タスクの複雑さに基づいてモデルを選択。"""    if force_model is not None:        return force_model    if text_length >= _SONNET_TEXT_THRESHOLD or item_count >= _SONNET_ITEM_THRESHOLD:        return MODEL_SONNET  # 複雑なタスク    return MODEL_HAIKU  # シンプルなタスク（3〜4倍安価）

from dataclasses import dataclass
@dataclass(frozen=True, slots=True)class CostRecord:    model: str    input_tokens: int    output_tokens: int    cost_usd: float
@dataclass(frozen=True, slots=True)class CostTracker:    budget_limit: float = 1.00    records: tuple[CostRecord, ...] = ()
    def add(self, record: CostRecord) -> "CostTracker":        """追加されたレコードで新しいトラッカーを返す（selfは変更しない）。"""        return CostTracker(            budget_limit=self.budget_limit,            records=(*self.records, record),        )
    @property    def total_cost(self) -> float:        return sum(r.cost_usd for r in self.records)
    @property    def over_budget(self) -> bool:        return self.total_cost > self.budget_limit

from anthropic import (    APIConnectionError,    InternalServerError,    RateLimitError,)
_RETRYABLE_ERRORS = (APIConnectionError, RateLimitError, InternalServerError)_MAX_RETRIES = 3
def call_with_retry(func, *, max_retries: int = _MAX_RETRIES):    """一時的なエラーのみリトライし、それ以外はすぐに失敗する。"""    for attempt in range(max_retries):        try:            return func()        except _RETRYABLE_ERRORS:            if attempt == max_retries - 1:                raise            time.sleep(2 ** attempt)  # 指数バックオフ    # AuthenticationError、BadRequestErrorなど → 即座に例外発生

messages = [    {        "role": "user",        "content": [            {                "type": "text",                "text": system_prompt,                "cache_control": {"type": "ephemeral"},  # これをキャッシュ            },            {                "type": "text",                "text": user_input,  # 可変部分            },        ],    }]

def process(text: str, config: Config, tracker: CostTracker) -> tuple[Result, CostTracker]:    # 1. モデルをルーティング    model = select_model(len(text), estimated_items, config.force_model)
    # 2. 予算を確認    if tracker.over_budget:        raise BudgetExceededError(tracker.total_cost, tracker.budget_limit)
    # 3. リトライ + キャッシングで呼び出し    response = call_with_retry(lambda: client.messages.create(        model=model,        messages=build_cached_messages(system_prompt, text),    ))
    # 4. コストを追跡（不変）    record = CostRecord(model=model, input_tokens=..., output_tokens=..., cost_usd=...)    tracker = tracker.add(record)
    return parse_result(response), tracker