c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python 0.1.0a7__tar.gz → 0.1.0a8__tar.gz

c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.1.0-alpha.8"
+}

{c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a7 → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8}/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## 0.1.0-alpha.8 (2025-06-27)
+
+Full Changelog: [v0.1.0-alpha.7...v0.1.0-alpha.8](https://github.com/digitalocean/gradientai-python/compare/v0.1.0-alpha.7...v0.1.0-alpha.8)
+
+### Features
+
+* **client:** setup streaming ([3fd6e57](https://github.com/digitalocean/gradientai-python/commit/3fd6e575f6f5952860e42d8c1fa22ccb0b10c623))
+
 ## 0.1.0-alpha.7 (2025-06-27)
 
 Full Changelog: [v0.1.0-alpha.6...v0.1.0-alpha.7](https://github.com/digitalocean/gradientai-python/compare/v0.1.0-alpha.6...v0.1.0-alpha.7)

{c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a7 → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python
-Version: 0.1.0a7
+Version: 0.1.0a8
 Summary: The official Python library for GradientAI
 Project-URL: Homepage, https://github.com/digitalocean/gradientai-python
 Project-URL: Repository, https://github.com/digitalocean/gradientai-python

{c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a7 → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8}/api.md

@@ -65,7 +65,7 @@ Methods:
 Types:
 
 ```python
-from gradientai.types.agents.chat import CompletionCreateResponse
+from gradientai.types.agents.chat import ChatCompletionChunk, CompletionCreateResponse
 ```
 
 Methods:

{c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a7 → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python"
-version = "0.1.0-alpha.7"
+version = "0.1.0-alpha.8"
 description = "The official Python library for GradientAI"
 dynamic = ["readme"]
 license = "Apache-2.0"

{c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a7 → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8}/src/gradientai/_client.py

@@ -117,6 +117,8 @@ class GradientAI(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._default_stream_cls = Stream
+
     @cached_property
     def agents(self) -> AgentsResource:
         from .resources.agents import AgentsResource
@@ -355,6 +357,8 @@ class AsyncGradientAI(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._default_stream_cls = AsyncStream
+
     @cached_property
     def agents(self) -> AsyncAgentsResource:
         from .resources.agents import AsyncAgentsResource

{c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a7 → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8}/src/gradientai/_streaming.py

@@ -9,7 +9,8 @@ from typing_extensions import Self, Protocol, TypeGuard, override, get_origin, r
 
 import httpx
 
-from ._utils import extract_type_var_from_base
+from ._utils import is_mapping, extract_type_var_from_base
+from ._exceptions import APIError
 
 if TYPE_CHECKING:
     from ._client import GradientAI, AsyncGradientAI
@@ -55,7 +56,25 @@ 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
+
+            data = sse.json()
+            if is_mapping(data) and data.get("error"):
+                message = None
+                error = data.get("error")
+                if is_mapping(error):
+                    message = error.get("message")
+                if not message or not isinstance(message, str):
+                    message = "An error occurred during streaming"
+
+                raise APIError(
+                    message=message,
+                    request=self.response.request,
+                    body=data["error"],
+                )
+
+            yield process_data(data=data, cast_to=cast_to, response=response)
 
         # Ensure the entire stream is consumed
         for _sse in iterator:
@@ -119,7 +138,25 @@ 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
+
+            data = sse.json()
+            if is_mapping(data) and data.get("error"):
+                message = None
+                error = data.get("error")
+                if is_mapping(error):
+                    message = error.get("message")
+                if not message or not isinstance(message, str):
+                    message = "An error occurred during streaming"
+
+                raise APIError(
+                    message=message,
+                    request=self.response.request,
+                    body=data["error"],
+                )
+
+            yield process_data(data=data, cast_to=cast_to, response=response)
 
         # Ensure the entire stream is consumed
         async for _sse in iterator:

{c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a7 → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8}/src/gradientai/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "gradientai"
-__version__ = "0.1.0-alpha.7"  # x-release-please-version
+__version__ = "0.1.0-alpha.8"  # x-release-please-version