# Author: Simon-Pierre Boucher — simon-pierre.boucher@uqo.ca
"""Anthropic Claude adapter (official SDK, structured outputs via output_config.format, adaptive thinking)."""
from __future__ import annotations

import os
from typing import Any

from pydantic import BaseModel

from .base import LLMProvider, json_schema_for


class AnthropicProvider(LLMProvider):
    name = "anthropic"

    def __init__(self, **kw):
        super().__init__(**kw)
        import anthropic

        if not os.environ.get("ANTHROPIC_API_KEY"):
            raise RuntimeError("ANTHROPIC_API_KEY not set")
        self.client = anthropic.AsyncAnthropic(max_retries=self.max_retries, timeout=self.timeout_s)

    async def _raw_complete(self, *, system: str, user: str, schema: type[BaseModel], model: str, effort: str, max_tokens: int) -> tuple[str, dict[str, Any]]:
        resp = await self.client.messages.create(
            model=model,
            max_tokens=max_tokens,
            system=[{"type": "text", "text": system, "cache_control": {"type": "ephemeral"}}],
            messages=[{"role": "user", "content": user}],
            thinking={"type": "adaptive"},
            output_config={"effort": effort, "format": {"type": "json_schema", "schema": json_schema_for(schema)}},
        )
        if resp.stop_reason == "refusal":
            raise RuntimeError(f"refusal: {getattr(resp, 'stop_details', None)}")
        text = next((b.text for b in resp.content if b.type == "text"), "")
        u = resp.usage
        return text, {
            "input_tokens": u.input_tokens + (u.cache_read_input_tokens or 0) + (u.cache_creation_input_tokens or 0),
            "output_tokens": u.output_tokens,
            "cached_tokens": u.cache_read_input_tokens or 0,
            "request_id": getattr(resp, "_request_id", None),
            "stop_reason": resp.stop_reason,
        }