remove exception chain

Author: pgrayy

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

@@ -2,7 +2,6 @@
 
 from typing import Awaitable, Callable
 
-from ..errors import BidiExceptionChain
 from ._task_group import _TaskGroup
 from ._task_pool import _TaskPool
 
@@ -18,14 +17,14 @@ async def stop_all(*funcs: Callable[..., Awaitable[None]]) -> None:
         funcs: Stop functions to call in sequence.
 
     Raises:
-        BidiExceptionChain: If any stop function raises an exception.
+        RuntimeError: If any stop function raises an exception.
     """
     exceptions = []
     for func in funcs:
         try:
             await func()
         except Exception as exception:
-            exceptions.append(exception)
+            exceptions.append({"func_name": func.__name__, "exception": repr(exception)})
 
     if exceptions:
-        raise BidiExceptionChain("failed stop sequence", exceptions)
+        raise RuntimeError(f"exceptions={exceptions} | failed stop sequence")

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

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

src/strands/experimental/bidi/errors.py

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

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

@@ -25,7 +25,6 @@
 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,
@@ -42,7 +41,7 @@
     ModalityUsage,
 )
 from ..types.model import AudioConfig
-from .model import BidiModel
+from .model import BidiModel, BidiModelTimeoutError
 
 logger = logging.getLogger(__name__)
 

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

@@ -112,3 +112,23 @@ 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/model.py

@@ -37,7 +37,6 @@
 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,
@@ -54,7 +53,7 @@
     BidiUsageEvent,
 )
 from ..types.model import AudioConfig
-from .model import BidiModel
+from .model import BidiModel, BidiModelTimeoutError
 
 logger = logging.getLogger(__name__)
 

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

@@ -19,7 +19,6 @@
 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,
@@ -38,7 +37,7 @@
     StopReason,
 )
 from ..types.model import AudioConfig
-from .model import BidiModel
+from .model import BidiModel, BidiModelTimeoutError
 
 logger = logging.getLogger(__name__)
 

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

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

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

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