bidi exception chain

Author: pgrayy

src/strands/experimental/bidi/_async/__init__.py

@@ -2,6 +2,7 @@
 
 from typing import Awaitable, Callable
 
+from ..errors import BidiExceptionChain
 from ._task_group import _TaskGroup
 from ._task_pool import _TaskPool
 
@@ -27,8 +28,4 @@ async def stop_all(*funcs: Callable[..., Awaitable[None]]) -> None:
             exceptions.append(exception)
 
     if exceptions:
-        exceptions.append(RuntimeError("failed stop sequence"))
-        for i in range(1, len(exceptions)):
-            exceptions[i].__context__ = exceptions[i - 1]
-
-        raise exceptions[-1]
+        raise BidiExceptionChain("failed stop sequence", exceptions)

src/strands/experimental/bidi/agent/loop.py

@@ -20,7 +20,7 @@
     BidiInterruptionEvent as BidiInterruptionHookEvent,
 )
 from .._async import _TaskPool, stop_all
-from ..models import BidiModelTimeoutError
+from ..errors import BidiModelTimeoutError
 from ..types.events import (
     BidiConnectionCloseEvent,
     BidiConnectionRestartEvent,

src/strands/experimental/bidi/errors.py

@@ -0,0 +1,45 @@
+"""Custom bidi exceptions."""
+from typing import Any
+
+
+class BidiExceptionChain(Exception):
+    """Chain a list of exceptions together.
+
+    Useful for chaining together exceptions raised across a multi-step workflow (e.g., the bidi `stop` methods).
+    Note, this exception is meant to mimic ExceptionGroup released in Python 3.11.
+
+    - Docs: https://docs.python.org/3/library/exceptions.html#ExceptionGroup
+    """
+
+    def __init__(self, message: str, exceptions: list[Exception]) -> None:
+        """Chain exceptions.
+
+        Args:
+            message: Top-level exception message.
+            exceptions: List of exceptions to chain.
+        """
+        super().__init__(self, message)
+
+        exceptions.append(self)
+        for i in range(1, len(exceptions)):
+            exceptions[i].__context__ = exceptions[i - 1]
+
+
+class BidiModelTimeoutError(Exception):
+    """Model timeout error.
+
+    Bidirectional models are often configured with a connection time limit. Nova sonic for example keeps the connection
+    open for 8 minutes max. Upon receiving a timeout, the agent loop is configured to restart the model connection so as
+    to create a seamless, uninterrupted experience for the user.
+    """
+
+    def __init__(self, message: str, **restart_config: Any) -> None:
+        """Initialize error.
+
+        Args:
+            message: Timeout message from model.
+            **restart_config: Configure restart specific behaviors in the call to model start.
+        """
+        super().__init__(self, message)
+
+        self.restart_config = restart_config

src/strands/experimental/bidi/models/__init__.py

@@ -1,10 +1,9 @@
 """Bidirectional model interfaces and implementations."""
 
-from .model import BidiModel, BidiModelTimeoutError
+from .model import BidiModel
 from .nova_sonic import BidiNovaSonicModel
 
 __all__ = [
     "BidiModel",
-    "BidiModelTimeoutError",
     "BidiNovaSonicModel",
 ]

src/strands/experimental/bidi/models/gemini_live.py

@@ -25,6 +25,7 @@
 from ....types.content import Messages
 from ....types.tools import ToolResult, ToolSpec, ToolUse
 from .._async import stop_all
+from ..errors import BidiModelTimeoutError
 from ..types.events import (
     AudioChannel,
     AudioSampleRate,
@@ -41,7 +42,7 @@
     ModalityUsage,
 )
 from ..types.model import AudioConfig
-from .model import BidiModel, BidiModelTimeoutError
+from .model import BidiModel
 
 logger = logging.getLogger(__name__)
 

src/strands/experimental/bidi/models/model.py

@@ -112,23 +112,3 @@ async def send(
             ```
         """
         ...
-
-
-class BidiModelTimeoutError(Exception):
-    """Model timeout error.
-
-    Bidirectional models are often configured with a connection time limit. Nova sonic for example keeps the connection
-    open for 8 minutes max. Upon receiving a timeout, the agent loop is configured to restart the model connection so as
-    to create a seamless, uninterrupted experience for the user.
-    """
-
-    def __init__(self, message: str, **restart_config: Any) -> None:
-        """Initialize error.
-
-        Args:
-            message: Timeout message from model.
-            **restart_config: Configure restart specific behaviors in the call to model start.
-        """
-        super().__init__(self, message)
-
-        self.restart_config = restart_config

src/strands/experimental/bidi/models/nova_sonic.py

@@ -37,6 +37,7 @@
 from ....types.content import Messages
 from ....types.tools import ToolResult, ToolSpec, ToolUse
 from .._async import stop_all
+from ..errors import BidiModelTimeoutError
 from ..types.events import (
     AudioChannel,
     AudioSampleRate,
@@ -53,7 +54,7 @@
     BidiUsageEvent,
 )
 from ..types.model import AudioConfig
-from .model import BidiModel, BidiModelTimeoutError
+from .model import BidiModel
 
 logger = logging.getLogger(__name__)
 

src/strands/experimental/bidi/models/openai_realtime.py

@@ -19,6 +19,7 @@
 from ....types.content import Messages
 from ....types.tools import ToolResult, ToolSpec, ToolUse
 from .._async import stop_all
+from ..errors import BidiModelTimeoutError
 from ..types.events import (
     AudioSampleRate,
     BidiAudioInputEvent,
@@ -37,7 +38,7 @@
     StopReason,
 )
 from ..types.model import AudioConfig
-from .model import BidiModel, BidiModelTimeoutError
+from .model import BidiModel
 
 logger = logging.getLogger(__name__)
 

src/strands/experimental/bidi/types/events.py

@@ -27,7 +27,7 @@
 from ....types.streaming import ContentBlockDelta
 
 if TYPE_CHECKING:
-    from ..models.model import BidiModelTimeoutError
+    from ..errors import BidiModelTimeoutError
 
 AudioChannel = Literal[1, 2]
 """Number of audio channels.

src/strands/experimental/hooks/events.py

@@ -14,7 +14,7 @@
 
 if TYPE_CHECKING:
     from ..bidi.agent.agent import BidiAgent
-    from ..bidi.models import BidiModelTimeoutError
+    from ..bidi.errors import BidiModelTimeoutError
 
 warnings.warn(
     "BeforeModelCallEvent, AfterModelCallEvent, BeforeToolCallEvent, and AfterToolCallEvent are no longer experimental."