.

Author: alexander-alderman-webb

sentry_sdk/integrations/pydantic_ai/__init__.py

@@ -1,7 +1,7 @@
 import functools
 
 from sentry_sdk.integrations import DidNotEnable, Integration
-from sentry_sdk.utils import capture_internal_exceptions, package_version
+from sentry_sdk.utils import capture_internal_exceptions, parse_version
 
 try:
     import pydantic_ai  # type: ignore # noqa: F401
@@ -10,6 +10,7 @@
     raise DidNotEnable("pydantic-ai not installed")
 
 
+from importlib.metadata import PackageNotFoundError, version
 from typing import TYPE_CHECKING
 
 from .patches import (
@@ -165,8 +166,16 @@ def setup_once() -> None:
             PydanticAIIntegration.are_request_hooks_available = False
 
         # ModelRequestContext.model added in https://github.com/pydantic/pydantic-ai/commit/f1260dfe09907f17688eee1646daf898fc428d4c
-        PYDANTIC_AI_VERSION = package_version("pydantic-ai")
-        if PYDANTIC_AI_VERSION is not None and PYDANTIC_AI_VERSION < (
+        try:
+            PYDANTIC_AI_VERSION = version("pydantic-ai-slim")
+        except PackageNotFoundError:
+            return
+
+        PYDANTIC_AI_VERSION = parse_version(PYDANTIC_AI_VERSION)
+        if PYDANTIC_AI_VERSION is None:
+            return
+
+        if PYDANTIC_AI_VERSION < (
             1,
             73,
         ):