"""Type stubs for langchain_anthropic package."""

from typing import Any

from langchain_core.language_models import BaseLanguageModel
from langchain_core.messages import AIMessage

class ChatAnthropic(BaseLanguageModel[AIMessage]):
    """Anthropic chat model wrapper."""

    def __init__(
        self,
        *,
        model: str = ...,
        temperature: float = ...,
        max_tokens_to_sample: int | None = None,
        top_k: int | None = None,
        top_p: float | None = None,
        timeout: float | None = None,
        max_retries: int = ...,
        api_key: Any = None,
        base_url: str | None = None,
        anthropic_proxy: str | None = None,
        default_headers: Any = None,
        streaming: bool = ...,
        **kwargs: Any,
    ) -> None: ...

__all__ = ["ChatAnthropic"]
