From 3b653aa1618ddb5625a70ba469b51b7ff3307b3b Mon Sep 17 00:00:00 2001
From: Kazuhiro Sera <seratch@openai.com>
Date: Mon, 28 Jul 2025 17:02:25 +0900
Subject: [PATCH] Fix #1277 missing StopAtTools and ToolsToFinalOutputFunction
 in agents module

---
 src/agents/__init__.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/agents/__init__.py b/src/agents/__init__.py
index 7de17efdb..8c97af824 100644
--- a/src/agents/__init__.py
+++ b/src/agents/__init__.py
@@ -5,7 +5,13 @@
 from openai import AsyncOpenAI
 
 from . import _config
-from .agent import Agent, AgentBase, ToolsToFinalOutputFunction, ToolsToFinalOutputResult
+from .agent import (
+    Agent,
+    AgentBase,
+    StopAtTools,
+    ToolsToFinalOutputFunction,
+    ToolsToFinalOutputResult,
+)
 from .agent_output import AgentOutputSchema, AgentOutputSchemaBase
 from .computer import AsyncComputer, Button, Computer, Environment
 from .exceptions import (
@@ -162,6 +168,7 @@ def enable_verbose_stdout_logging():
 __all__ = [
     "Agent",
     "AgentBase",
+    "StopAtTools",
     "ToolsToFinalOutputFunction",
     "ToolsToFinalOutputResult",
     "Runner",