|
25 | 25 | ClassVar, |
26 | 26 | Protocol, |
27 | 27 | Required, |
| 28 | + Annotated, |
28 | 29 | ParamSpec, |
| 30 | + TypeAlias, |
29 | 31 | TypedDict, |
30 | 32 | TypeGuard, |
31 | 33 | final, |
|
80 | 82 | from ._constants import RAW_RESPONSE_HEADER |
81 | 83 |
|
82 | 84 | if TYPE_CHECKING: |
| 85 | + from pydantic import GetCoreSchemaHandler, ValidatorFunctionWrapHandler |
| 86 | + from pydantic_core import CoreSchema, core_schema |
83 | 87 | from pydantic_core.core_schema import ModelField, ModelSchema, LiteralSchema, ModelFieldsSchema |
| 88 | +else: |
| 89 | + try: |
| 90 | + from pydantic_core import CoreSchema, core_schema |
| 91 | + except ImportError: |
| 92 | + CoreSchema = None |
| 93 | + core_schema = None |
84 | 94 |
|
85 | 95 | __all__ = ["BaseModel", "GenericModel"] |
86 | 96 |
|
@@ -399,6 +409,76 @@ def model_dump_json( |
399 | 409 | ) |
400 | 410 |
|
401 | 411 |
|
| 412 | +class _EagerIterable(list[_T], Generic[_T]): |
| 413 | + """ |
| 414 | + Accepts any Iterable[T] input (including generators), consumes it |
| 415 | + eagerly, and validates all items upfront. |
| 416 | +
|
| 417 | + Validation preserves the original container type where possible |
| 418 | + (e.g. a set[T] stays a set[T]). Serialization (model_dump / JSON) |
| 419 | + always emits a list — round-tripping through model_dump() will not |
| 420 | + restore the original container type. |
| 421 | + """ |
| 422 | + |
| 423 | + @classmethod |
| 424 | + def __get_pydantic_core_schema__( |
| 425 | + cls, |
| 426 | + source_type: Any, |
| 427 | + handler: GetCoreSchemaHandler, |
| 428 | + ) -> CoreSchema: |
| 429 | + (item_type,) = get_args(source_type) or (Any,) |
| 430 | + item_schema: CoreSchema = handler.generate_schema(item_type) |
| 431 | + list_of_items_schema: CoreSchema = core_schema.list_schema(item_schema) |
| 432 | + |
| 433 | + return core_schema.no_info_wrap_validator_function( |
| 434 | + cls._validate, |
| 435 | + list_of_items_schema, |
| 436 | + serialization=core_schema.plain_serializer_function_ser_schema( |
| 437 | + cls._serialize, |
| 438 | + info_arg=False, |
| 439 | + ), |
| 440 | + ) |
| 441 | + |
| 442 | + @staticmethod |
| 443 | + def _validate(v: Iterable[_T], handler: "ValidatorFunctionWrapHandler") -> Any: |
| 444 | + original_type: type[Any] = type(v) |
| 445 | + |
| 446 | + # Normalize to list so list_schema can validate each item |
| 447 | + if isinstance(v, list): |
| 448 | + items: list[_T] = v |
| 449 | + else: |
| 450 | + try: |
| 451 | + items = list(v) |
| 452 | + except TypeError as e: |
| 453 | + raise TypeError("Value is not iterable") from e |
| 454 | + |
| 455 | + # Validate items against the inner schema |
| 456 | + validated: list[_T] = handler(items) |
| 457 | + |
| 458 | + # Reconstruct original container type |
| 459 | + if original_type is list: |
| 460 | + return validated |
| 461 | + # str(list) produces the list's repr, not a string built from items, |
| 462 | + # so skip reconstruction for str and its subclasses. |
| 463 | + if issubclass(original_type, str): |
| 464 | + return validated |
| 465 | + try: |
| 466 | + return original_type(validated) |
| 467 | + except (TypeError, ValueError): |
| 468 | + # If the type cannot be reconstructed, just return the validated list |
| 469 | + return validated |
| 470 | + |
| 471 | + @staticmethod |
| 472 | + def _serialize(v: Iterable[_T]) -> list[_T]: |
| 473 | + """Always serialize as a list so Pydantic's JSON encoder is happy.""" |
| 474 | + if isinstance(v, list): |
| 475 | + return v |
| 476 | + return list(v) |
| 477 | + |
| 478 | + |
| 479 | +EagerIterable: TypeAlias = Annotated[Iterable[_T], _EagerIterable] |
| 480 | + |
| 481 | + |
402 | 482 | def _construct_field(value: object, field: FieldInfo, key: str) -> object: |
403 | 483 | if value is None: |
404 | 484 | return field_get_default(field) |
|
![stainless-app[bot]](https://avatars.githubusercontent.com/in/378072?v=4&size=40){kind=avatar}
0 commit comments