?HAMA

Navigation

Breadcrumbs

Create

Edit File Size: 1.45 KB

//opt/hc_python/lib64/python3.12/site-packages/sentry_sdk/integrations/openai_agents/patches/runner.py

Text Base64

from functools import wraps

import sentry_sdk

from ..spans import agent_workflow_span
from ..utils import _capture_exception

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from typing import Any, Callable

def _create_run_wrapper(original_func):
    # type: (Callable[..., Any]) -> Callable[..., Any]
    """
    Wraps the agents.Runner.run methods to create a root span for the agent workflow runs.

Note agents.Runner.run_sync() is a wrapper around agents.Runner.run(),
    so it does not need to be wrapped separately.
    """

@wraps(original_func)
    async def wrapper(*args, **kwargs):
        # type: (*Any, **Any) -> Any
        # Isolate each workflow so that when agents are run in asyncio tasks they
        # don't touch each other's scopes
        with sentry_sdk.isolation_scope():
            agent = args[0]
            with agent_workflow_span(agent):
                result = None
                try:
                    result = await original_func(*args, **kwargs)
                    return result
                except Exception as exc:
                    _capture_exception(exc)

# It could be that there is a "invoke agent" span still open
                    current_span = sentry_sdk.get_current_span()
                    if current_span is not None and current_span.timestamp is None:
                        current_span.__exit__(None, None, None)

raise exc from None

return wrapper

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
agent_run.py	5.64 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
error_tracing.py	2.57 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
models.py	1.37 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
runner.py	1.45 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
tools.py	3.18 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
__init__.py	293 B	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive

Directory Contents