from smolagents import LiteLLMModel


ANTHROPIC_MODEL_IDS = [
    "claude-opus-4-20250514",
    "claude-sonnet-4-20250514",
    "claude-3-7-sonnet-latest",
    "claude-3-5-haiku-latest",
    "claude-3-5-sonnet-latest",
    "claude-3-5-sonnet-20240620",
    "claude-3-opus-latest",
    "claude-3-sonnet-20240229",
    "claude-3-haiku-20240307",
]


def get_anthropic_model(model_id, anthropic_api_key):
    if model_id not in ANTHROPIC_MODEL_IDS:
        raise ValueError(f"Model {model_id} not found in Anthropic model IDs")
    model = LiteLLMModel(
        model_id=model_id,
        api_key=anthropic_api_key,
    )
    return model