Respect schema_generator config value in TypeAdapter (#10300)

Author: sydney-runkle

pydantic/type_adapter.py

@@ -98,7 +98,9 @@ def _get_schema(type_: Any, config_wrapper: _config.ConfigWrapper, parent_depth:
  local_ns = _typing_extra.parent_frame_namespace(parent_depth=parent_depth)
  global_ns = sys._getframe(max(parent_depth - 1, 1)).f_globals.copy()
  global_ns.update(local_ns or {})
- gen = _generate_schema.GenerateSchema(config_wrapper, types_namespace=global_ns, typevars_map={})
+ gen = (config_wrapper.schema_generator or _generate_schema.GenerateSchema)(
+ config_wrapper, types_namespace=global_ns, typevars_map={}
+ )
  schema = gen.generate_schema(type_)
  schema = gen.clean_schema(schema)
  return schema

tests/test_type_adapter.py

@@ -5,10 +5,11 @@
 from typing import Any, Dict, ForwardRef, Generic, List, NamedTuple, Optional, Tuple, TypeVar, Union
 
 import pytest
-from pydantic_core import ValidationError
+from pydantic_core import ValidationError, core_schema
 from typing_extensions import Annotated, TypeAlias, TypedDict
 
 from pydantic import BaseModel, Field, TypeAdapter, ValidationInfo, create_model, field_validator
+from pydantic._internal._generate_schema import GenerateSchema
 from pydantic._internal._typing_extra import annotated_type
 from pydantic.config import ConfigDict
 from pydantic.dataclasses import dataclass as pydantic_dataclass
@@ -567,3 +568,11 @@ def test_core_schema_respects_defer_build(model: Any, config: ConfigDict, method
  assert type_adapter._core_schema is not None, 'Should be initialized after the usage'
  assert type_adapter._validator is not None, 'Should be initialized after the usage'
  assert type_adapter._serializer is not None, 'Should be initialized after the usage'
+
+
+def test_custom_schema_gen_respected() -> None:
+ class LaxStrGenerator(GenerateSchema):
+ def str_schema(self) -> core_schema.CoreSchema:
+ return core_schema.no_info_plain_validator_function(str)
+
+ assert TypeAdapter(str, config=ConfigDict(schema_generator=LaxStrGenerator)).validate_python(1) == '1'