|
4 | 4 |
|
5 | 5 | from apiclient import APIClient |
6 | 6 | from pydantic import AfterValidator, BaseModel, ConfigDict |
7 | | -from pydantic._internal import _generate_schema, _typing_extra |
| 7 | +from pydantic._internal import _generate_schema |
8 | 8 | from pydantic._internal._config import ConfigWrapper |
9 | 9 | from pydantic._internal._validate_call import ValidateCallWrapper as PydanticValidateCallWrapper |
10 | 10 | from pydantic.plugin._schema_validator import create_schema_validator |
11 | 11 | from typing_extensions import Annotated |
12 | 12 |
|
| 13 | +try: # pragma: no cover |
| 14 | + from pydantic._internal._typing_extra import get_module_ns_of as get_module |
| 15 | +except ImportError: # pragma: no cover |
| 16 | + from pydantic._internal._typing_extra import add_module_globals as get_module # type: ignore[attr-defined,no-redef] |
| 17 | + |
| 18 | + |
13 | 19 | AnyCallableT = TypeVar('AnyCallableT', bound=Callable[..., Any]) |
14 | 20 | T = TypeVar('T', bound=APIClient) |
15 | 21 |
|
@@ -48,7 +54,7 @@ def __init__( |
48 | 54 | self.__module__ = function.__module__ |
49 | 55 | self.__doc__ = function.__doc__ |
50 | 56 |
|
51 | | - namespace = _typing_extra.add_module_globals(function, None) |
| 57 | + namespace = get_module(function) |
52 | 58 | config_wrapper = ConfigWrapper(config) |
53 | 59 | gen_schema = _generate_schema.GenerateSchema(config_wrapper, namespace) |
54 | 60 | schema = gen_schema.clean_schema(gen_schema.generate_schema(function)) |
|
0 commit comments