letta-client 1.0.0a9__py3-none-any.whl → 1.0.0a11__py3-none-any.whl

letta_client/_client.py

@@ -64,7 +64,7 @@ __all__ = [
 ]
 
 ENVIRONMENTS: Dict[str, str] = {
-    "cloud": "https://app.letta.com",
+    "cloud": "https://api.letta.com",
     "local": "http://localhost:8283",
 }
 
@@ -164,6 +164,8 @@ class Letta(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._default_stream_cls = Stream
+
         self.archives = archives.ArchivesResource(self)
         self.tools = tools.ToolsResource(self)
         self.folders = folders.FoldersResource(self)
@@ -401,6 +403,8 @@ class AsyncLetta(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._default_stream_cls = AsyncStream
+
         self.archives = archives.AsyncArchivesResource(self)
         self.tools = tools.AsyncToolsResource(self)
         self.folders = folders.AsyncFoldersResource(self)

letta_client/_streaming.py

@@ -55,7 +55,26 @@ class Stream(Generic[_T]):
         iterator = self._iter_events()
 
         for sse in iterator:
-            yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+            if sse.data.startswith("[DONE]"):
+                break
+
+            if sse.event == "error":
+                body = sse.data
+
+                try:
+                    body = sse.json()
+                    err_msg = f"{body}"
+                except Exception:
+                    err_msg = sse.data or f"Error code: {response.status_code}"
+
+                raise self._client._make_status_error(
+                    err_msg,
+                    body=body,
+                    response=self.response,
+                )
+
+            if sse.event is None:
+                yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 
         # Ensure the entire stream is consumed
         for _sse in iterator:
@@ -119,7 +138,26 @@ class AsyncStream(Generic[_T]):
         iterator = self._iter_events()
 
         async for sse in iterator:
-            yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+            if sse.data.startswith("[DONE]"):
+                break
+
+            if sse.event == "error":
+                body = sse.data
+
+                try:
+                    body = sse.json()
+                    err_msg = f"{body}"
+                except Exception:
+                    err_msg = sse.data or f"Error code: {response.status_code}"
+
+                raise self._client._make_status_error(
+                    err_msg,
+                    body=body,
+                    response=self.response,
+                )
+
+            if sse.event is None:
+                yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 
         # Ensure the entire stream is consumed
         async for _sse in iterator:

letta_client/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "letta_client"
-__version__ = "1.0.0-alpha.9"  # x-release-please-version
+__version__ = "1.0.0-alpha.11"  # x-release-please-version