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

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

Author: MAsuamah

assemblyai/__version__.py

@@ -1 +1 @@
-__version__ = "0.52.2"
+__version__ = "0.52.3"

assemblyai/streaming/v3/__init__.py

@@ -3,6 +3,7 @@
     BeginEvent,
     Encoding,
     EventMessage,
+    LLMGatewayResponseEvent,
     SpeechModel,
     StreamingClientOptions,
     StreamingError,
@@ -18,6 +19,7 @@
     "BeginEvent",
     "Encoding",
     "EventMessage",
+    "LLMGatewayResponseEvent",
     "SpeechModel",
     "StreamingClient",
     "StreamingClientOptions",

assemblyai/streaming/v3/client.py

@@ -18,6 +18,7 @@
     ErrorEvent,
     EventMessage,
     ForceEndpoint,
+    LLMGatewayResponseEvent,
     OperationMessage,
     StreamingClientOptions,
     StreamingError,
@@ -211,6 +212,8 @@ def _parse_message(self, data: Dict[str, Any]) -> Optional[EventMessage]:
                 return TerminationEvent.model_validate(data)
             elif event_type == StreamingEvents.Turn:
                 return TurnEvent.model_validate(data)
+            elif event_type == StreamingEvents.LLMGatewayResponse:
+                return LLMGatewayResponseEvent.model_validate(data)
             else:
                 return None
         elif "error" in data:

assemblyai/streaming/v3/models.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 from enum import Enum
-from typing import List, Literal, Optional, Union
+from typing import Any, List, Literal, Optional, Union
 
 from pydantic import BaseModel
 
@@ -52,11 +52,19 @@ class ErrorEvent(BaseModel):
     error: str
 
 
+class LLMGatewayResponseEvent(BaseModel):
+    type: Literal["LLMGatewayResponse"] = "LLMGatewayResponse"
+    turn_order: int
+    transcript: str
+    data: Any
+
+
 EventMessage = Union[
     BeginEvent,
     TerminationEvent,
     TurnEvent,
     ErrorEvent,
+    LLMGatewayResponseEvent,
 ]
 
 
@@ -160,3 +168,4 @@ class StreamingEvents(Enum):
     Termination = "Termination"
     Turn = "Turn"
     Error = "Error"
+    LLMGatewayResponse = "LLMGatewayResponse"