livellm 1.3.0__py3-none-any.whl → 1.3.6__py3-none-any.whl

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
@@ -561,11 +564,11 @@ class LivellmWsClient(BaseLivellmClient):
             
             if response.status == WsStatus.ERROR:
                 raise Exception(f"WebSocket stream failed: {response.error}")
-            
-            yield response
-            
+                        
             if response.status == WsStatus.SUCCESS:
                 break
+            
+            yield response
     
     # Implement abstract methods from BaseLivellmClient
     

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/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.dist-info → livellm-1.3.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livellm
-Version: 1.3.0
+Version: 1.3.6
 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.dist-info → livellm-1.3.6.dist-info}/RECORD

@@ -1,7 +1,7 @@
 livellm/__init__.py,sha256=p2Szx7PELGYi-PTnSNnRPGVbU438ZBTFXYAQoMToUfE,440
-livellm/livellm.py,sha256=tbkMZFhl46Mbp-gJFsRBxahXU35wyVYmcDH1bAtnUPc,32135
+livellm/livellm.py,sha256=Z3UpG8rxQXTW9qT4EPi6b7qOx3xHL-i-Mz5fKX24QPM,32303
 livellm/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-livellm/transcripton.py,sha256=4BVCzhm304aqv81BDWAvYYKqxJ1YJU7r7POLbbXFhpk,4047
+livellm/transcripton.py,sha256=2ttPzc8A6uQL5VouLLb4aj7Q-cMCx3V1HXH2qhpUMfM,4195
 livellm/models/__init__.py,sha256=tCX_Q3ALjO0jbqhLFsVoITSm1AV--3pF5ZZc_l0VC1o,1447
 livellm/models/common.py,sha256=YqRwP6ChWbRdoen4MU6RO4u6HeM0mQJbXiiRV4DuauM,1740
 livellm/models/fallback.py,sha256=zGG_MjdbaTx0fqKZTEg3ullej-CJznPfwaon0jEvRvI,1170
@@ -12,9 +12,9 @@ livellm/models/agent/agent.py,sha256=-UcGv5Bzw5ALmWX4lIqpbWqMVjCsjBc0KIE6_JKbCXM
 livellm/models/agent/chat.py,sha256=zGfeEHx0luwq23pqWF1megcuEDUl6IhV4keLJeZry_A,1028
 livellm/models/agent/tools.py,sha256=wVWfx6_jxL3IcmX_Nt_PonZ3RQLtpfqJnszHz32BQiU,1403
 livellm/models/audio/__init__.py,sha256=sz2NxCOfFGVvp-XQUsdgOR_TYBO1Wb-8LLXaZDEiAZk,282
-livellm/models/audio/speak.py,sha256=ZOHmTfQsqbm-8Nb2Ee1SfNpCbwxk-Mcvm6Km1W63ymA,1617
+livellm/models/audio/speak.py,sha256=lDITZ7fiLRuDhA-LxCPQ6Yraxr33B6Lg7VyR4CkuGk8,1872
 livellm/models/audio/transcribe.py,sha256=Leji2lk5zfq4GE-fw-z2dZR8BuijzW8TJ12GHw_UZJY,2085
-livellm-1.3.0.dist-info/METADATA,sha256=c1sJdLRtKRKWn7e6ZPRaE2ITROHC8DLDVJ_tovZqMMc,18701
-livellm-1.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-livellm-1.3.0.dist-info/licenses/LICENSE,sha256=yapGO2C_00ymEx6TADdbU8Oyc1bWOrZY-fjP-agmFL4,1071
-livellm-1.3.0.dist-info/RECORD,,
+livellm-1.3.6.dist-info/METADATA,sha256=e7TYTCh3xtNS8POgIxe4gADiBS2a_cs6IlL6by0kJsw,18701
+livellm-1.3.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+livellm-1.3.6.dist-info/licenses/LICENSE,sha256=yapGO2C_00ymEx6TADdbU8Oyc1bWOrZY-fjP-agmFL4,1071
+livellm-1.3.6.dist-info/RECORD,,