Barebone of new integration

Author: antonpirker

.github/workflows/test-integrations-ai.yml

@@ -66,6 +66,10 @@ jobs:
         run: |
           set -x # print commands that are executed
           ./scripts/runtox.sh "py${{ matrix.python-version }}-openai-latest"
+      - name: Test openai_agents latest
+        run: |
+          set -x # print commands that are executed
+          ./scripts/runtox.sh "py${{ matrix.python-version }}-openai_agents-latest"
       - name: Test huggingface_hub latest
         run: |
           set -x # print commands that are executed
@@ -141,6 +145,10 @@ jobs:
         run: |
           set -x # print commands that are executed
           ./scripts/runtox.sh --exclude-latest "py${{ matrix.python-version }}-openai"
+      - name: Test openai_agents pinned
+        run: |
+          set -x # print commands that are executed
+          ./scripts/runtox.sh --exclude-latest "py${{ matrix.python-version }}-openai_agents"
       - name: Test huggingface_hub pinned
         run: |
           set -x # print commands that are executed

scripts/populate_tox/config.py

@@ -139,6 +139,9 @@
     "loguru": {
         "package": "loguru",
     },
+    "openai_agents": {
+        "package": "openai-agents",
+    },
     "openfeature": {
         "package": "openfeature-sdk",
     },

scripts/populate_tox/tox.jinja

@@ -397,6 +397,7 @@ setenv =
     litestar: TESTPATH=tests/integrations/litestar
     loguru: TESTPATH=tests/integrations/loguru
     openai: TESTPATH=tests/integrations/openai
+    openai_agents: TESTPATH=tests/integrations/openai_agents
     openfeature: TESTPATH=tests/integrations/openfeature
     opentelemetry: TESTPATH=tests/integrations/opentelemetry
     potel: TESTPATH=tests/integrations/opentelemetry

scripts/split_tox_gh_actions/split_tox_gh_actions.py

@@ -59,6 +59,7 @@
         "cohere",
         "langchain",
         "openai",
+        "openai_agents",
         "huggingface_hub",
     ],
     "Cloud": [

sentry_sdk/integrations/__init__.py

@@ -145,6 +145,7 @@ def iter_default_integrations(with_auto_enabling_integrations):
     "launchdarkly": (9, 8, 0),
     "loguru": (0, 7, 0),
     "openai": (1, 0, 0),
+    "openai_agents": (0, 0, 16),
     "openfeature": (0, 7, 1),
     "quart": (0, 16, 0),
     "ray": (2, 7, 0),

sentry_sdk/integrations/openai_agents.py

@@ -0,0 +1,36 @@
+import sentry_sdk
+from sentry_sdk.integrations import DidNotEnable, Integration
+from sentry_sdk.utils import event_from_exception
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from typing import Any
+
+try:
+    from agents import Agent  # noqa: F401
+except ImportError:
+    raise DidNotEnable("OpenAI Agents not installed")
+
+
+class OpenAIAgentsIntegration(Integration):
+    identifier = "openai_agents"
+    origin = f"auto.ai.{identifier}"
+
+    def __init__(self):
+        pass
+
+    @staticmethod
+    def setup_once():
+        # type: () -> None
+        pass
+
+
+def _capture_exception(exc):
+    # type: (Any) -> None
+    event, hint = event_from_exception(
+        exc,
+        client_options=sentry_sdk.get_client().options,
+        mechanism={"type": OpenAIAgentsIntegration.identifier, "handled": False},
+    )
+    sentry_sdk.capture_event(event, hint=hint)

tests/integrations/openai_agents/test_openai_agents.py

@@ -0,0 +1,7 @@
+from sentry_sdk.integrations.openai_agents import OpenAIAgentsIntegration
+
+
+def test_basic(sentry_init):
+    sentry_init(integrations=[OpenAIAgentsIntegration()])
+
+    assert True