Add document to allowed cacheable_types for anthropic

Author: marpulli

pydantic_ai_slim/pydantic_ai/models/anthropic.py

@@ -776,7 +776,7 @@ def _add_cache_control_to_last_param(params: list[BetaContentBlockParam], ttl: L
 
         # Only certain types support cache_control
         # See https://docs.anthropic.com/en/docs/build-with-claude/prompt-caching#what-can-be-cached
-        cacheable_types = {'text', 'tool_use', 'server_tool_use', 'image', 'tool_result'}
+        cacheable_types = {'text', 'tool_use', 'server_tool_use', 'image', 'tool_result', 'document'}
         # Cast needed because BetaContentBlockParam is a union including response Block types (Pydantic models)
         # that don't support dict operations, even though at runtime we only have request Param types (TypedDicts).
         last_param = cast(dict[str, Any], params[-1])

tests/models/test_anthropic.py

@@ -419,20 +419,6 @@ async def test_cache_point_in_otel_message_parts(allow_model_requests: None):
     )
 
 
-def test_cache_control_unsupported_param_type():
-    """Test that cache control raises error for unsupported param types."""
-
-    from pydantic_ai.exceptions import UserError
-    from pydantic_ai.models.anthropic import AnthropicModel
-
-    # Create a list with an unsupported param type (document)
-    # We'll use a mock document block param
-    params: list[dict[str, Any]] = [{'type': 'document', 'source': {'data': 'test'}}]
-
-    with pytest.raises(UserError, match='Cache control not supported for param type: document'):
-        AnthropicModel._add_cache_control_to_last_param(params)  # type: ignore[arg-type]  # Testing internal method
-
-
 async def test_anthropic_cache_tools(allow_model_requests: None):
     """Test that anthropic_cache_tool_definitions adds cache_control to last tool."""
     c = completion_message(