diff --git a/src/agents/extensions/sandbox/runloop/sandbox.py b/src/agents/extensions/sandbox/runloop/sandbox.py index 31004017f2..aa199b2283 100644 --- a/src/agents/extensions/sandbox/runloop/sandbox.py +++ b/src/agents/extensions/sandbox/runloop/sandbox.py @@ -22,7 +22,7 @@ from collections.abc import Sequence from dataclasses import dataclass from pathlib import Path, PurePosixPath -from typing import TYPE_CHECKING, Any, Literal, cast +from typing import TYPE_CHECKING, Any, Literal, TypeAlias, cast from urllib.parse import urlsplit from pydantic import BaseModel, Field @@ -71,9 +71,9 @@ logger = logging.getLogger(__name__) -RunloopAfterIdle = _RunloopSdkAfterIdle -RunloopLaunchParameters = _RunloopSdkLaunchParameters -RunloopUserParameters = _RunloopSdkUserParameters +RunloopAfterIdle: TypeAlias = _RunloopSdkAfterIdle +RunloopLaunchParameters: TypeAlias = _RunloopSdkLaunchParameters +RunloopUserParameters: TypeAlias = _RunloopSdkUserParameters @dataclass(frozen=True) @@ -1387,7 +1387,7 @@ def _runloop_launch_parameters_payload( launch_parameters: RunloopLaunchParameters | None, user_parameters: RunloopUserParameters | None, ) -> dict[str, object] | None: - payload = ( + payload: dict[str, object] = ( launch_parameters.to_dict(mode="json", exclude_none=True, exclude_defaults=True) if launch_parameters is not None else {} diff --git a/tests/test_run_step_execution.py b/tests/test_run_step_execution.py index 53d3b88164..fa6e6e748a 100644 --- a/tests/test_run_step_execution.py +++ b/tests/test_run_step_execution.py @@ -1463,8 +1463,9 @@ async def _second_tool() -> str: ] loop = asyncio.get_running_loop() previous_task_factory = loop.get_task_factory() - eager_task_factory = cast(Any, asyncio.eager_task_factory) - loop.set_task_factory(eager_task_factory) + eager_task_factory = getattr(asyncio, "eager_task_factory", None) + assert eager_task_factory is not None + loop.set_task_factory(cast(Any, eager_task_factory)) try: (