Skip to content

Commit 964da56

Browse files
committed
🐛 fix new pydantic imports fix #406
1 parent c68f888 commit 964da56

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

apiclient_pydantic/serializers.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,12 +4,18 @@
44

55
from apiclient import APIClient
66
from pydantic import AfterValidator, BaseModel, ConfigDict
7-
from pydantic._internal import _generate_schema, _typing_extra
7+
from pydantic._internal import _generate_schema
88
from pydantic._internal._config import ConfigWrapper
99
from pydantic._internal._validate_call import ValidateCallWrapper as PydanticValidateCallWrapper
1010
from pydantic.plugin._schema_validator import create_schema_validator
1111
from typing_extensions import Annotated
1212

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+
1319
AnyCallableT = TypeVar('AnyCallableT', bound=Callable[..., Any])
1420
T = TypeVar('T', bound=APIClient)
1521

@@ -48,7 +54,7 @@ def __init__(
4854
self.__module__ = function.__module__
4955
self.__doc__ = function.__doc__
5056

51-
namespace = _typing_extra.add_module_globals(function, None)
57+
namespace = get_module(function)
5258
config_wrapper = ConfigWrapper(config)
5359
gen_schema = _generate_schema.GenerateSchema(config_wrapper, namespace)
5460
schema = gen_schema.clean_schema(gen_schema.generate_schema(function))

0 commit comments

Comments
 (0)