Check FastAPI version and apply appropriate method for function wrapping

src/extendable_pydantic/_patch.py

@@ -16,6 +16,8 @@
     from typing_extensions import Annotated
 
 import wrapt
+import pkg_resources
+from packaging import version
 from extendable import context
 from pydantic import TypeAdapter
 
@@ -66,12 +68,17 @@ def _create_response_field_wrapper(wrapped, instance, args, kwargs):
         _resolve_model_fields_annotation([field])
         return field
 
-    # fastapi <  0.112.3
-    wrapt.wrap_function_wrapper(
-        utils, "create_response_field", _create_response_field_wrapper
-    )
+    # Get the installed version of FastAPI
+    fastapi_version = pkg_resources.get_distribution("fastapi").version
 
-    # fastapi >= 0.112.3
-    wrapt.wrap_function_wrapper(
-        utils, "create_model_field", _create_response_field_wrapper
-    )
+    # Compare the version and call the appropriate method
+    if version.parse(fastapi_version) < version.parse("0.112.3"):
+        # For fastapi < 0.112.3
+        wrapt.wrap_function_wrapper(
+            utils, "create_response_field", _create_response_field_wrapper
+        )
+    else:
+        # For fastapi >= 0.112.3
+        wrapt.wrap_function_wrapper(
+            utils, "create_model_field", _create_response_field_wrapper
+        )