Make mypy happy

Author: antonpirker

sentry_sdk/integrations/anthropic.py

@@ -90,6 +90,10 @@ def _get_responses(content):
 
 def _sentry_patched_create_common(f, *args, **kwargs):
     # type: (Any, *Any, **Any) -> Any
+    integration = sentry_sdk.get_client().get_integration(AnthropicIntegration)
+    if integration is None:
+        return f(*args, **kwargs)
+
     if "messages" not in kwargs:
         return f(*args, **kwargs)
 
@@ -115,8 +119,6 @@ def _sentry_patched_create_common(f, *args, **kwargs):
         span.__exit__(None, None, None)
         raise exc from None
 
-    integration = sentry_sdk.get_client().get_integration(AnthropicIntegration)
-
     with capture_internal_exceptions():
         span.set_data(SPANDATA.AI_MODEL_ID, model)
         span.set_data(SPANDATA.AI_STREAMING, False)

sentry_sdk/integrations/openai.py

@@ -116,6 +116,10 @@ def _calculate_chat_completion_usage(
 
 def _new_chat_completion_common(f, *args, **kwargs):
     # type: (Any, *Any, **Any) -> Any
+    integration = sentry_sdk.get_client().get_integration(OpenAIIntegration)
+    if integration is None:
+        return f(*args, **kwargs)
+
     if "messages" not in kwargs:
         # invalid call (in all versions of openai), let it return error
         return f(*args, **kwargs)
@@ -144,8 +148,6 @@ def _new_chat_completion_common(f, *args, **kwargs):
         span.__exit__(None, None, None)
         raise e from None
 
-    integration = sentry_sdk.get_client().get_integration(OpenAIIntegration)
-
     with capture_internal_exceptions():
         if should_send_default_pii() and integration.include_prompts:
             set_data_normalized(span, SPANDATA.AI_INPUT_MESSAGES, messages)
@@ -271,12 +273,15 @@ async def _sentry_patched_create_async(*args, **kwargs):
 
 def _new_embeddings_create_common(f, *args, **kwargs):
     # type: (Any, *Any, **Any) -> Any
+    integration = sentry_sdk.get_client().get_integration(OpenAIIntegration)
+    if integration is None:
+        return f(*args, **kwargs)
+
     with sentry_sdk.start_span(
         op=consts.OP.OPENAI_EMBEDDINGS_CREATE,
         description="OpenAI Embedding Creation",
         origin=OpenAIIntegration.origin,
     ) as span:
-        integration = sentry_sdk.get_client().get_integration(OpenAIIntegration)
         if "input" in kwargs and (
             should_send_default_pii() and integration.include_prompts
         ):