mirascope 2.0.0a1__py3-none-any.whl → 2.0.0a3__py3-none-any.whl

mirascope/llm/{clients → providers}/anthropic/_utils/encode.py

@@ -17,9 +17,9 @@ from ....formatting import (
     resolve_format,
 )
 from ....messages import AssistantMessage, Message, UserMessage
-from ....tools import FORMAT_TOOL_NAME, BaseToolkit, ToolSchema
+from ....tools import FORMAT_TOOL_NAME, AnyToolSchema, BaseToolkit
 from ...base import Params, _utils as _base_utils
-from ..model_ids import AnthropicModelId
+from ..model_id import AnthropicModelId, model_name
 
 DEFAULT_MAX_TOKENS = 16000
 
@@ -33,7 +33,7 @@ def encode_image_mime_type(
     if mime_type in ("image/jpeg", "image/png", "image/gif", "image/webp"):
         return mime_type
     raise FeatureNotSupportedError(
-        feature=f"Image with mime_type: {mime_type}", provider="anthropic"
+        feature=f"Image with mime_type: {mime_type}", provider_id="anthropic"
     )  # pragma: no cover
 
 
@@ -59,13 +59,22 @@ def _encode_content(
     """Convert mirascope content to Anthropic content format."""
 
     if len(content) == 1 and content[0].type == "text":
+        if not content[0].text:
+            raise FeatureNotSupportedError(
+                "empty message content",
+                "anthropic",
+                message="Anthropic does not support empty message content.",
+            )
         return content[0].text
 
     blocks: list[anthropic_types.ContentBlockParam] = []
 
     for part in content:
         if part.type == "text":
-            blocks.append(anthropic_types.TextBlockParam(type="text", text=part.text))
+            if part.text:
+                blocks.append(
+                    anthropic_types.TextBlockParam(type="text", text=part.text)
+                )
         elif part.type == "image":
             source: (
                 anthropic_types.Base64ImageSourceParam
@@ -116,6 +125,13 @@ def _encode_content(
         else:
             raise NotImplementedError(f"Unsupported content type: {part.type}")
 
+    if not blocks:
+        raise FeatureNotSupportedError(
+            "empty message content",
+            "anthropic",
+            message="Anthropic does not support empty message content.",
+        )
+
     return blocks
 
 
@@ -136,7 +152,7 @@ def _encode_message(
 
     if (
         message.role == "assistant"
-        and message.provider == "anthropic"
+        and message.provider_id == "anthropic"
         and message.model_id == model_id
         and message.raw_message
         and not encode_thoughts
@@ -149,7 +165,7 @@ def _encode_message(
 
 
 @lru_cache(maxsize=128)
-def _convert_tool_to_tool_param(tool: ToolSchema) -> anthropic_types.ToolParam:
+def _convert_tool_to_tool_param(tool: AnyToolSchema) -> anthropic_types.ToolParam:
     """Convert a single Mirascope tool to Anthropic tool format with caching."""
     schema_dict = tool.parameters.model_dump(by_alias=True, exclude_none=True)
     schema_dict["type"] = "object"
@@ -164,21 +180,26 @@ def encode_request(
     *,
     model_id: AnthropicModelId,
     messages: Sequence[Message],
-    tools: Sequence[ToolSchema] | BaseToolkit | None,
+    tools: Sequence[AnyToolSchema] | BaseToolkit[AnyToolSchema] | None,
     format: type[FormattableT] | Format[FormattableT] | None,
     params: Params,
 ) -> tuple[Sequence[Message], Format[FormattableT] | None, MessageCreateKwargs]:
     """Prepares a request for the `Anthropic.messages.create` method."""
+    if not model_id.startswith("anthropic/"):  # pragma: no cover
+        raise ValueError(
+            f"Model ID must start with 'anthropic/' prefix, got: {model_id}"
+        )
+
     kwargs: MessageCreateKwargs = MessageCreateKwargs(
         {
-            "model": model_id,
+            "model": model_name(model_id),
             "max_tokens": DEFAULT_MAX_TOKENS,
         }
     )
     encode_thoughts = False
 
     with _base_utils.ensure_all_params_accessed(
-        params=params, provider="anthropic", unsupported_params=["seed"]
+        params=params, provider_id="anthropic", unsupported_params=["seed"]
     ) as param_accessor:
         if param_accessor.temperature is not None:
             kwargs["temperature"] = param_accessor.temperature
@@ -206,7 +227,7 @@ def encode_request(
     if format is not None:
         if format.mode == "strict":
             raise FormattingModeNotSupportedError(
-                formatting_mode="strict", provider="anthropic"
+                formatting_mode="strict", provider_id="anthropic"
             )
         elif format.mode == "tool":
             format_tool_schema = _formatting_utils.create_tool_schema(format)

mirascope/llm/providers/anthropic/model_id.py

@@ -0,0 +1,40 @@
+"""Anthropic registered LLM models."""
+
+from typing import Literal, TypeAlias
+
+AnthropicModelId: TypeAlias = (
+    Literal[
+        "anthropic/claude-3-7-sonnet-latest",
+        "anthropic/claude-3-7-sonnet-20250219",
+        "anthropic/claude-3-5-haiku-latest",
+        "anthropic/claude-3-5-haiku-20241022",
+        "anthropic/claude-haiku-4-5",
+        "anthropic/claude-haiku-4-5-20251001",
+        "anthropic/claude-sonnet-4-20250514",
+        "anthropic/claude-sonnet-4-0",
+        "anthropic/claude-4-sonnet-20250514",
+        "anthropic/claude-sonnet-4-5",
+        "anthropic/claude-sonnet-4-5-20250929",
+        "anthropic/claude-opus-4-0",
+        "anthropic/claude-opus-4-20250514",
+        "anthropic/claude-4-opus-20250514",
+        "anthropic/claude-opus-4-1-20250805",
+        "anthropic/claude-3-opus-latest",
+        "anthropic/claude-3-opus-20240229",
+        "anthropic/claude-3-haiku-20240307",
+    ]
+    | str
+)
+"""The Anthropic model ids registered with Mirascope."""
+
+
+def model_name(model_id: AnthropicModelId) -> str:
+    """Extract the anthropic model name from the ModelId
+
+    Args:
+        model_id: Full model ID (e.g. "anthropic/claude-sonnet-4-5")
+
+    Returns:
+        Provider-specific model ID (e.g. "claude-sonnet-4-5")
+    """
+    return model_id.removeprefix("anthropic/")