AttributeError for python 3.14: 'typing.Union' object has no attribute __discriminator__ and no __dict__ for setting new attributes #1

src/anthropic/_models.py

@@ -3,6 +3,7 @@
 import os
 import inspect
 from typing import TYPE_CHECKING, Any, Type, Union, Generic, TypeVar, Callable, Optional, cast
+from weakref import WeakKeyDictionary
 from datetime import date, datetime
 from typing_extensions import (
     List,
@@ -584,9 +585,7 @@ def construct_type(*, value: object, type_: object, metadata: Optional[List[Any]
     return value
 
 
-@runtime_checkable
-class CachedDiscriminatorType(Protocol):
-    __discriminator__: DiscriminatorDetails
+_DISCRIMINATOR_CACHE: "WeakKeyDictionary[type, DiscriminatorDetails]" = WeakKeyDictionary()
 
 
 class DiscriminatorDetails:
@@ -631,8 +630,9 @@ def __init__(
 
 
 def _build_discriminated_union_meta(*, union: type, meta_annotations: tuple[Any, ...]) -> DiscriminatorDetails | None:
-    if isinstance(union, CachedDiscriminatorType):
-        return union.__discriminator__
+    cached_discriminator = _DISCRIMINATOR_CACHE.get(union)
+    if cached_discriminator is not None:
+        return cached_discriminator
 
     discriminator_field_name: str | None = None
 
@@ -685,7 +685,7 @@ def _build_discriminated_union_meta(*, union: type, meta_annotations: tuple[Any,
         discriminator_field=discriminator_field_name,
         discriminator_alias=discriminator_alias,
     )
-    cast(CachedDiscriminatorType, union).__discriminator__ = details
+    _DISCRIMINATOR_CACHE[union] = details
     return details
 
 

src/anthropic/types/beta/messages/batch_create_params.py

@@ -23,7 +23,6 @@ class BatchCreateParams(TypedDict, total=False):
     """Optional header to specify the beta version(s) you want to use."""
 
 
-
 class Request(TypedDict, total=False):
     custom_id: Required[str]
     """Developer-provided ID created for each request in a Message Batch.

src/anthropic/types/messages/batch_create_params.py

@@ -18,7 +18,6 @@ class BatchCreateParams(TypedDict, total=False):
     """
 
 
-
 class Request(TypedDict, total=False):
     custom_id: Required[str]
     """Developer-provided ID created for each request in a Message Batch.

tests/test_models.py

@@ -9,7 +9,7 @@
 
 from anthropic._utils import PropertyInfo
 from anthropic._compat import PYDANTIC_V1, parse_obj, model_dump, model_json
-from anthropic._models import BaseModel, construct_type
+from anthropic._models import _DISCRIMINATOR_CACHE, BaseModel, construct_type
 
 
 class BasicModel(BaseModel):
@@ -809,7 +809,7 @@ class B(BaseModel):
 
     UnionType = cast(Any, Union[A, B])
 
-    assert not hasattr(UnionType, "__discriminator__")
+    assert _DISCRIMINATOR_CACHE.get(UnionType) is None
 
     m = construct_type(
         value={"type": "b", "data": "foo"}, type_=cast(Any, Annotated[UnionType, PropertyInfo(discriminator="type")])
@@ -818,7 +818,7 @@ class B(BaseModel):
     assert m.type == "b"
     assert m.data == "foo"  # type: ignore[comparison-overlap]
 
-    discriminator = UnionType.__discriminator__
+    discriminator = _DISCRIMINATOR_CACHE.get(UnionType)
     assert discriminator is not None
 
     m = construct_type(
@@ -830,7 +830,7 @@ class B(BaseModel):
 
     # if the discriminator details object stays the same between invocations then
     # we hit the cache
-    assert UnionType.__discriminator__ is discriminator
+    assert _DISCRIMINATOR_CACHE.get(UnionType) is discriminator
 
 
 @pytest.mark.skipif(PYDANTIC_V1, reason="TypeAliasType is not supported in Pydantic v1")