fix(litellm): Prevent module shadowing

Author: alexander-alderman-webb

sentry_sdk/integrations/litellm.py

@@ -19,6 +19,7 @@
 
 try:
     import litellm  # type: ignore[import-not-found]
+    from litellm import get_llm_provider
 except ImportError:
     raise DidNotEnable("LiteLLM not installed")
 
@@ -45,7 +46,7 @@ def _input_callback(kwargs: "Dict[str, Any]") -> None:
     # Get key parameters
     full_model = kwargs.get("model", "")
     try:
-        model, provider, _, _ = litellm.get_llm_provider(full_model)
+        model, provider, _, _ = get_llm_provider(full_model)
     except Exception:
         model = full_model
         provider = "unknown"

tests/test_shadowed_module.py

@@ -31,7 +31,6 @@ def pytest_generate_tests(metafunc):
             - {
                 "clickhouse_driver",
                 "grpc",
-                "litellm",
                 "opentelemetry",
                 "pure_eval",
                 "ray",