Anthropic

Anthropic Model class

• Read on GitHub

Full source code

import os
import warnings
from typing import Literal

from minisweagent.models.litellm_model import LitellmModel, LitellmModelConfig
from minisweagent.models.utils.cache_control import set_cache_control
from minisweagent.models.utils.key_per_thread import get_key_per_thread


class AnthropicModelConfig(LitellmModelConfig):
    set_cache_control: Literal["default_end"] | None = "default_end"
    """Set explicit cache control markers, for example for Anthropic models"""


class AnthropicModel(LitellmModel):
    """This class is now only a thin wrapper around the LitellmModel class.
    It is largely kept for backwards compatibility.
    It will not be selected by `get_model` and `get_model_class` unless explicitly specified.
    """

    def __init__(self, *, config_class: type = AnthropicModelConfig, **kwargs):
        super().__init__(config_class=config_class, **kwargs)

    def query(self, messages: list[dict], **kwargs) -> dict:
        api_key = None
        # Legacy only
        if rotating_keys := os.getenv("ANTHROPIC_API_KEYS"):
            warnings.warn(
                "ANTHROPIC_API_KEYS is deprecated and will be removed in the future. "
                "Simply use the ANTHROPIC_API_KEY environment variable instead. "
                "Key rotation is no longer required."
            )
            api_key = get_key_per_thread(rotating_keys.split("::"))
        messages = set_cache_control(messages, mode="default_end")
        return super().query(messages, api_key=api_key, **kwargs)

minisweagent.models.anthropic

AnthropicModelConfig dataclass

AnthropicModelConfig(
    model_name: str,
    model_kwargs: dict[str, Any] = dict(),
    litellm_model_registry: Path | str | None = getenv(
        "LITELLM_MODEL_REGISTRY_PATH"
    ),
    set_cache_control: Literal["default_end"] | None = None,
)

Bases: LitellmModelConfig

set_cache_control class-attribute instance-attribute

set_cache_control: Literal["default_end"] | None = (
    "default_end"
)

Set explicit cache control markers, for example for Anthropic models

AnthropicModel

AnthropicModel(
    *, config_class: type = AnthropicModelConfig, **kwargs
)

Bases: LitellmModel

This class is now only a thin wrapper around the LitellmModel class. It is largely kept for backwards compatibility. It will not be selected by get_model and get_model_class unless explicitly specified.

Source code in src/minisweagent/models/anthropic.py

def __init__(self, *, config_class: type = AnthropicModelConfig, **kwargs):
    super().__init__(config_class=config_class, **kwargs)

query

query(messages: list[dict], **kwargs) -> dict

Source code in src/minisweagent/models/anthropic.py

def query(self, messages: list[dict], **kwargs) -> dict:
    api_key = None
    # Legacy only
    if rotating_keys := os.getenv("ANTHROPIC_API_KEYS"):
        warnings.warn(
            "ANTHROPIC_API_KEYS is deprecated and will be removed in the future. "
            "Simply use the ANTHROPIC_API_KEY environment variable instead. "
            "Key rotation is no longer required."
        )
        api_key = get_key_per_thread(rotating_keys.split("::"))
    messages = set_cache_control(messages, mode="default_end")
    return super().query(messages, api_key=api_key, **kwargs)

bug_report Something broken/unclear?

Open an issue on GitHub!

help Open-ended discussions

Join our Slack!

Our projects

SWE-agent SWE-ReX SWE-smith SWE-bench sb-cli