livellm 1.3.0__tar.gz → 1.3.5__tar.gz

{livellm-1.3.0 → livellm-1.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livellm
-Version: 1.3.0
+Version: 1.3.5
 Summary: Python client for the LiveLLM Server
 Project-URL: Homepage, https://github.com/qalby-tech/livellm-client-py
 Project-URL: Repository, https://github.com/qalby-tech/livellm-client-py

{livellm-1.3.0 → livellm-1.3.5}/livellm/livellm.py

@@ -493,7 +493,8 @@ class LivellmWsClient(BaseLivellmClient):
     def __init__(
         self, 
         base_url: str, 
-        timeout: Optional[float] = None
+        timeout: Optional[float] = None,
+        max_size: Optional[int] = None
     ):
         # Convert HTTP(S) URL to WS(S) URL
         base_url = base_url.rstrip("/")
@@ -511,6 +512,7 @@ class LivellmWsClient(BaseLivellmClient):
         self.websocket = None
         # Lazily-created clients
         self._transcription = None
+        self.max_size = max_size or 1024 * 1024 * 10 # 10MB is default max size
         
     async def connect(self):
         """Establish WebSocket connection."""
@@ -520,7 +522,8 @@ class LivellmWsClient(BaseLivellmClient):
         self.websocket = await websockets.connect(
             self.base_url, 
             open_timeout=self.timeout,
-            close_timeout=self.timeout
+            close_timeout=self.timeout,
+            max_size=self.max_size
         )
                 
         return self.websocket

{livellm-1.3.0 → livellm-1.3.5}/livellm/models/audio/speak.py

@@ -28,9 +28,17 @@ class EncodedSpeakResponse(BaseModel):
     content_type: SpeakMimeType = Field(..., description="The content type of the audio")
     sample_rate: int = Field(..., description="The sample rate of the audio")
 
-    @field_validator('audio', mode='after')
+    @field_validator("audio", mode="after")
     @classmethod
     def validate_audio(cls, v: bytes | str) -> bytes:
-        if isinstance(v, bytes):
-            return base64.b64decode(v) # decode from base64 string to bytes
-        return v # if bytes, assume it's already a base64 decoded bytes
+        """
+        Ensure that `audio` is always returned as raw bytes.
+
+        - If the server returns a base64-encoded *string*, decode it.
+        - If the server already returned raw bytes, pass them through.
+        """
+        if isinstance(v, str):
+            # Server sent base64-encoded string → decode to raw bytes
+            return base64.b64decode(v)
+        # Already bytes → assume it's raw audio
+        return v

{livellm-1.3.0 → livellm-1.3.5}/livellm/transcripton.py

@@ -10,11 +10,12 @@ import json
 
 
 class TranscriptionWsClient:
-    def __init__(self, base_url: str, timeout: Optional[float] = None):
+    def __init__(self, base_url: str, timeout: Optional[float] = None, max_size: Optional[int] = None):
         self.base_url = base_url.rstrip("/")
         self.url = f"{base_url}/livellm/ws/transcription"
         self.timeout = timeout
         self.websocket = None
+        self.max_size = max_size or 1024 * 1024 * 10 # 10MB is default max size
     
     async def connect(self):
         """
@@ -23,7 +24,8 @@ class TranscriptionWsClient:
         self.websocket = await websockets.connect(
             self.url,
             open_timeout=self.timeout,
-            close_timeout=self.timeout
+            close_timeout=self.timeout,
+            max_size=self.max_size
         )
     
     async def disconnect(self):

{livellm-1.3.0 → livellm-1.3.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "livellm"
-version = "1.3.0"
+version = "1.3.5"
 description = "Python client for the LiveLLM Server"
 readme = "README.md"
 requires-python = ">=3.10"