chore(internal): codegen related update

This commit is contained in:
stainless-app[bot] 2025-11-11 03:37:10 +00:00
parent cc65c0df9a
commit fc0f0178bf
5 changed files with 19 additions and 43 deletions

View file

@ -9,7 +9,7 @@ from pydantic import Field
from opencode_ai._utils import PropertyInfo
from opencode_ai._compat import PYDANTIC_V1, parse_obj, model_dump, model_json
from opencode_ai._models import BaseModel, construct_type
from opencode_ai._models import DISCRIMINATOR_CACHE, BaseModel, construct_type
class BasicModel(BaseModel):
@ -809,7 +809,7 @@ def test_discriminated_unions_invalid_data_uses_cache() -> None:
UnionType = cast(Any, Union[A, B])
assert not hasattr(UnionType, "__discriminator__")
assert not DISCRIMINATOR_CACHE.get(UnionType)
m = construct_type(
value={"type": "b", "data": "foo"}, type_=cast(Any, Annotated[UnionType, PropertyInfo(discriminator="type")])
@ -818,7 +818,7 @@ def test_discriminated_unions_invalid_data_uses_cache() -> None:
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 @@ def test_discriminated_unions_invalid_data_uses_cache() -> None:
# 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")