chore: sync sdk code with DeepLearning repo (#192)

Co-authored-by: AssemblyAI <engineering.sdk@assemblyai.com>

Author: bgotthold-aai

assemblyai/__version__.py

@@ -1 +1 @@
-__version__ = "0.61.1"
+__version__ = "0.62.0"

assemblyai/streaming/v3/__init__.py

@@ -14,6 +14,7 @@
     StreamingSessionParameters,
     TerminationEvent,
     TurnEvent,
+    WarningEvent,
     Word,
 )
 
@@ -33,5 +34,6 @@
     "StreamingSessionParameters",
     "TerminationEvent",
     "TurnEvent",
+    "WarningEvent",
     "Word",
 ]

assemblyai/streaming/v3/client.py

@@ -31,6 +31,7 @@
     TerminationEvent,
     TurnEvent,
     UpdateConfiguration,
+    WarningEvent,
 )
 
 logger = logging.getLogger(__name__)
@@ -220,6 +221,8 @@ def _read_message(self) -> None:
 
             if isinstance(message, ErrorEvent):
                 self._handle_error(message)
+            elif isinstance(message, WarningEvent):
+                self._handle_warning(message)
             elif message:
                 self._handle_message(message)
             else:
@@ -250,6 +253,10 @@ def _parse_message(self, data: Dict[str, Any]) -> Optional[EventMessage]:
                 return SpeechStartedEvent.model_validate(data)
             elif event_type == StreamingEvents.LLMGatewayResponse:
                 return LLMGatewayResponseEvent.model_validate(data)
+            elif event_type == StreamingEvents.Error:
+                return ErrorEvent.model_validate(data)
+            elif event_type == StreamingEvents.Warning:
+                return WarningEvent.model_validate(data)
             else:
                 return None
         elif "error" in data:
@@ -267,6 +274,13 @@ def _parse_event_type(message_type: Optional[Any]) -> Optional[StreamingEvents]:
         except KeyError:
             return None
 
+    def _handle_warning(self, warning: WarningEvent):
+        logger.warning(
+            "Streaming warning (code=%s): %s", warning.warning_code, warning.warning
+        )
+        for handler in self._handlers[StreamingEvents.Warning]:
+            handler(self, warning)
+
     def _handle_error(
         self,
         error: Union[
@@ -291,6 +305,7 @@ def _parse_error(
         if isinstance(error, ErrorEvent):
             return StreamingError(
                 message=error.error,
+                code=error.error_code,
             )
         elif isinstance(error, websockets.exceptions.ConnectionClosed):
             if (

assemblyai/streaming/v3/models.py

@@ -55,9 +55,17 @@ class SpeechStartedEvent(BaseModel):
 
 
 class ErrorEvent(BaseModel):
+    type: Literal["Error"] = "Error"
+    error_code: Optional[int] = None
     error: str
 
 
+class WarningEvent(BaseModel):
+    type: Literal["Warning"] = "Warning"
+    warning_code: int
+    warning: str
+
+
 class LLMGatewayResponseEvent(BaseModel):
     type: Literal["LLMGatewayResponse"] = "LLMGatewayResponse"
     turn_order: int
@@ -71,6 +79,7 @@ class LLMGatewayResponseEvent(BaseModel):
     TurnEvent,
     SpeechStartedEvent,
     ErrorEvent,
+    WarningEvent,
     LLMGatewayResponseEvent,
 ]
 
@@ -201,4 +210,5 @@ class StreamingEvents(Enum):
     Turn = "Turn"
     SpeechStarted = "SpeechStarted"
     Error = "Error"
+    Warning = "Warning"
     LLMGatewayResponse = "LLMGatewayResponse"