Skip to content

Commit bf25bfd

Browse files
committed
Apply appropriate method for function wrapping
1 parent ffdc03a commit bf25bfd

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

src/extendable_pydantic/_patch.py

+10-9
Original file line numberDiff line numberDiff line change
@@ -66,12 +66,13 @@ def _create_response_field_wrapper(wrapped, instance, args, kwargs):
6666
_resolve_model_fields_annotation([field])
6767
return field
6868

69-
# fastapi < 0.112.3
70-
wrapt.wrap_function_wrapper(
71-
utils, "create_response_field", _create_response_field_wrapper
72-
)
73-
74-
# fastapi >= 0.112.3
75-
wrapt.wrap_function_wrapper(
76-
utils, "create_model_field", _create_response_field_wrapper
77-
)
69+
if hasattr(utils, "create_response_field"):
70+
# For fastapi < 0.112.3
71+
wrapt.wrap_function_wrapper(
72+
utils, "create_response_field", _create_response_field_wrapper
73+
)
74+
else:
75+
# For fastapi >= 0.112.3
76+
wrapt.wrap_function_wrapper(
77+
utils, "create_model_field", _create_response_field_wrapper
78+
)

0 commit comments

Comments
 (0)