From c4310227e0cab0a16d14fb6d338294b78b927815 Mon Sep 17 00:00:00 2001 From: AssemblyAI Date: Mon, 23 Feb 2026 12:46:55 -0500 Subject: [PATCH] Project import generated by Copybara. GitOrigin-RevId: f2038f0836672b16c4dbdcceba2990a2b6e5ccce --- assemblyai/__version__.py | 2 +- assemblyai/streaming/v3/__init__.py | 2 ++ assemblyai/streaming/v3/client.py | 3 +++ assemblyai/streaming/v3/models.py | 11 ++++++++++- 4 files changed, 16 insertions(+), 2 deletions(-) diff --git a/assemblyai/__version__.py b/assemblyai/__version__.py index 4136999..2133351 100644 --- a/assemblyai/__version__.py +++ b/assemblyai/__version__.py @@ -1 +1 @@ -__version__ = "0.52.2" +__version__ = "0.52.3" diff --git a/assemblyai/streaming/v3/__init__.py b/assemblyai/streaming/v3/__init__.py index 7dd05b0..7deb8f5 100644 --- a/assemblyai/streaming/v3/__init__.py +++ b/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", diff --git a/assemblyai/streaming/v3/client.py b/assemblyai/streaming/v3/client.py index b2b2002..af0afbd 100644 --- a/assemblyai/streaming/v3/client.py +++ b/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: diff --git a/assemblyai/streaming/v3/models.py b/assemblyai/streaming/v3/models.py index b6557c5..e6c63f8 100644 --- a/assemblyai/streaming/v3/models.py +++ b/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"