scale-gp-beta 0.1.0a41__py3-none-any.whl → 0.1.0a42__py3-none-any.whl

scale_gp_beta/_client.py

@@ -173,6 +173,8 @@ class SGPClient(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._default_stream_cls = Stream
+
     @cached_property
     def responses(self) -> ResponsesResource:
         from .resources.responses import ResponsesResource
@@ -472,6 +474,8 @@ class AsyncSGPClient(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._default_stream_cls = AsyncStream
+
     @cached_property
     def responses(self) -> AsyncResponsesResource:
         from .resources.responses import AsyncResponsesResource

scale_gp_beta/_streaming.py

@@ -56,7 +56,8 @@ class Stream(Generic[_T]):
 
         try:
             for sse in iterator:
-                yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+                if sse.event is None:
+                    yield process_data(data=sse.json(), cast_to=cast_to, response=response)
         finally:
             # Ensure the response is closed even if the consumer doesn't read all data
             response.close()
@@ -120,7 +121,8 @@ class AsyncStream(Generic[_T]):
 
         try:
             async for sse in iterator:
-                yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+                if sse.event is None:
+                    yield process_data(data=sse.json(), cast_to=cast_to, response=response)
         finally:
             # Ensure the response is closed even if the consumer doesn't read all data
             await response.aclose()

scale_gp_beta/_version.py

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

scale_gp_beta/resources/build.py

@@ -18,8 +18,10 @@ from .._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from .._streaming import Stream, AsyncStream
 from ..pagination import SyncCursorPage, AsyncCursorPage
 from .._base_client import AsyncPaginator, make_request_options
+from ..types.stream_chunk import StreamChunk
 from ..types.build_list_response import BuildListResponse
 from ..types.build_cancel_response import BuildCancelResponse
 from ..types.build_create_response import BuildCreateResponse
@@ -236,7 +238,7 @@ class BuildResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> object:
+    ) -> Stream[StreamChunk]:
         """
         Stream build logs via Server-Sent Events (SSE).
 
@@ -259,7 +261,9 @@ class BuildResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=StreamChunk,
+            stream=True,
+            stream_cls=Stream[StreamChunk],
         )
 
 
@@ -471,7 +475,7 @@ class AsyncBuildResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> object:
+    ) -> AsyncStream[StreamChunk]:
         """
         Stream build logs via Server-Sent Events (SSE).
 
@@ -494,7 +498,9 @@ class AsyncBuildResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=StreamChunk,
+            stream=True,
+            stream_cls=AsyncStream[StreamChunk],
         )
 
 

scale_gp_beta/types/__init__.py

@@ -19,6 +19,7 @@ from .evaluation import Evaluation as Evaluation
 from .span_status import SpanStatus as SpanStatus
 from .dataset_item import DatasetItem as DatasetItem
 from .item_locator import ItemLocator as ItemLocator
+from .stream_chunk import StreamChunk as StreamChunk
 from .approval_status import ApprovalStatus as ApprovalStatus
 from .assessment_type import AssessmentType as AssessmentType
 from .component_param import ComponentParam as ComponentParam

scale_gp_beta/types/stream_chunk.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .._models import BaseModel
+
+__all__ = ["StreamChunk"]
+
+
+class StreamChunk(BaseModel):
+    """A single log line from the build process."""
+
+    line: str
+    """The log line content"""

{scale_gp_beta-0.1.0a41.dist-info → scale_gp_beta-0.1.0a42.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: scale-gp-beta
-Version: 0.1.0a41
+Version: 0.1.0a42
 Summary: The official Python library for the Scale GP API
 Project-URL: Homepage, https://github.com/scaleapi/sgp-python-beta
 Project-URL: Repository, https://github.com/scaleapi/sgp-python-beta
@@ -395,6 +395,44 @@ async def main() -> None:
 asyncio.run(main())
 ```
 
+## Streaming responses
+
+We provide support for streaming responses using Server Side Events (SSE).
+
+```python
+from scale_gp_beta import SGPClient
+
+client = SGPClient(
+    account_id="My Account ID",
+)
+
+stream = client.chat.completions.create(
+    messages=[{"foo": "bar"}],
+    model="model",
+    stream=True,
+)
+for completion in stream:
+    print(completion)
+```
+
+The async client uses the exact same interface.
+
+```python
+from scale_gp_beta import AsyncSGPClient
+
+client = AsyncSGPClient(
+    account_id="My Account ID",
+)
+
+stream = await client.chat.completions.create(
+    messages=[{"foo": "bar"}],
+    model="model",
+    stream=True,
+)
+async for completion in stream:
+    print(completion)
+```
+
 ## Using types
 
 Nested request parameters are [TypedDicts](https://docs.python.org/3/library/typing.html#typing.TypedDict). Responses are [Pydantic models](https://docs.pydantic.dev) which also provide helper methods for things like:

{scale_gp_beta-0.1.0a41.dist-info → scale_gp_beta-0.1.0a42.dist-info}/RECORD

@@ -1,6 +1,6 @@
 scale_gp_beta/__init__.py,sha256=MFsKxnaLkZ-Rqv127bA2T0OpHD3me1OnwIom-0I2gB0,2742
 scale_gp_beta/_base_client.py,sha256=J1u-5BLJ_gL10TknCyP420e8kRngnTzGh6bY5AcZez8,73416
-scale_gp_beta/_client.py,sha256=0fFnb2Te5RY5NqhgOOEx_l_218gDReBBnfUPrTtmI6Y,41097
+scale_gp_beta/_client.py,sha256=dzyLzPqwlvAjC0Z3rdSAl7clxuRv0EOyr4hRGrak3vU,41188
 scale_gp_beta/_compat.py,sha256=DQBVORjFb33zch24jzkhM14msvnzY7mmSmgDLaVFUM8,6562
 scale_gp_beta/_constants.py,sha256=S14PFzyN9-I31wiV7SmIlL5Ga0MLHxdvegInGdXH7tM,462
 scale_gp_beta/_exceptions.py,sha256=95GM5CLFtP-QMjjmzsr5ajjZOyEZvyaETfGmqNPR8YM,3226
@@ -9,9 +9,9 @@ scale_gp_beta/_models.py,sha256=R3MpO2z4XhTAnD3ObEks32suRXleF1g7BEgQTOLIxTs,3211
 scale_gp_beta/_qs.py,sha256=craIKyvPktJ94cvf9zn8j8ekG9dWJzhWv0ob34lIOv4,4828
 scale_gp_beta/_resource.py,sha256=siZly_U6D0AOVLAzaOsqUdEFFzVMbWRj-ml30nvRp7E,1118
 scale_gp_beta/_response.py,sha256=GemuybPk0uemovTlGHyHkj-ScYTTDJA0jqH5FQqIPwQ,28852
-scale_gp_beta/_streaming.py,sha256=-_hEXPmI3ouK-z-AqZgV-7E0nsg0hGFbSay_4DS0G4E,10233
+scale_gp_beta/_streaming.py,sha256=xX-SwwToqjKjZCpAuUixrO5qTWgdHzOMErAadnKk4Yw,10317
 scale_gp_beta/_types.py,sha256=91u4yp566AoT5VoDhNGZbmJ213XzpSbyPHgZLCBqAs8,7601
-scale_gp_beta/_version.py,sha256=n4_fu2WdZu0bO0wYwmjuwZm6voO_F8uLlhqB7e7nFJU,174
+scale_gp_beta/_version.py,sha256=ld1qdiUAH4VxEkLUUYjrZLQwiOJh-ooM2jdekXQG_mA,174
 scale_gp_beta/pagination.py,sha256=t-U86PYxl20VRsz8VXOMJJDe7HxkX7ISFMvRNbBNy9s,4054
 scale_gp_beta/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 scale_gp_beta/_utils/__init__.py,sha256=7fch0GT9zpNnErbciSpUNa-SjTxxjY6kxHxKMOM4AGs,2305
@@ -44,7 +44,7 @@ scale_gp_beta/lib/tracing/integrations/openai/openai_span_type_map.py,sha256=WIP
 scale_gp_beta/lib/tracing/integrations/openai/openai_tracing_sgp_processor.py,sha256=o-Fp_V7U9cC1kKP0_750GQInKGlF7KAdpIbGTmyu2tg,5270
 scale_gp_beta/lib/tracing/integrations/openai/utils.py,sha256=DqCb7_QX-OT4y34K9wtoXIv7-Yk_IOvSbQvC96Sgpy8,2183
 scale_gp_beta/resources/__init__.py,sha256=7OBbTu19SH3ZS6kO0VWCPIOUk66prBJpLy6PVP4W8ZA,7440
-scale_gp_beta/resources/build.py,sha256=7m6FhLsasO7aGazAvqvCD_BHmnDicPQspXg4DCram5M,22002
+scale_gp_beta/resources/build.py,sha256=1LNowzczQYRXKN0xqquhsAqqQh78Jxa_Njab1hoXsGs,22276
 scale_gp_beta/resources/completions.py,sha256=wtVhUh5LMwXolICQhpeCZUlzRM-ChLSvLiQm92sO-EQ,30934
 scale_gp_beta/resources/credentials.py,sha256=Eax7ouhJQ61CHy-mQlmBiylBKiNfGhBFoWlBiuHHQH4,32903
 scale_gp_beta/resources/dataset_items.py,sha256=-cJJ1ikJ5mDUGIRkhqTuY652h0EAP5Wa-lmH2gA782w,23278
@@ -63,7 +63,7 @@ scale_gp_beta/resources/chat/completions.py,sha256=yOnyGK2QBaFNL0el5aL1A5T6JmBKo
 scale_gp_beta/resources/files/__init__.py,sha256=VgAtqUimN5Kf_-lmEaNBnu_ApGegKsJQ1zNf-42MXFA,1002
 scale_gp_beta/resources/files/content.py,sha256=QDtcnsK2my4-ndmwyaR6zyQnV1fEjIJ7CSDoyXUcYxg,5762
 scale_gp_beta/resources/files/files.py,sha256=dowTYD-xKhCI17zINYZywUtzv7VwofjV1tY6AfncHjE,24603
-scale_gp_beta/types/__init__.py,sha256=cgPnOBpdkmIF5fyr5uCCXkPewwv4fFdXVPjEq7P3pIc,6899
+scale_gp_beta/types/__init__.py,sha256=DWbXOTnI7vqsLfOJnKHVotfq4PxIJdO5K_fxicy3-I0,6952
 scale_gp_beta/types/approval_status.py,sha256=rI70l7e2ey2ax8tR_sBkziVSirq_QaDDV1bHgNrfCWQ,227
 scale_gp_beta/types/assessment_type.py,sha256=_4V6RZ5wUyU0UYrDjCoIRdaiiZtZVpY6DMsW1QClq-c,271
 scale_gp_beta/types/build_cancel_response.py,sha256=ghlldFT3BpPZbbDa0Ag0Jj4gJKXXe9rDxX9BPN9j29Q,867
@@ -146,6 +146,7 @@ scale_gp_beta/types/span_type.py,sha256=Y4cFVO66V_KJQ2stokdspyBunh4kxgSDt2EZg960
 scale_gp_beta/types/span_update_params.py,sha256=xpk0NKNsw_RAXIZN1U1ddqUHpPE--W9_Y3i1tTZKLZg,574
 scale_gp_beta/types/span_upsert_batch_params.py,sha256=JdXDZapv7kRYYgU5Stltyagtw4PPuCZLUy7Qo8019bQ,1329
 scale_gp_beta/types/span_upsert_batch_response.py,sha256=Gkndmd_cyfodeGaCJu4sF4TsgB22hgHVlmoek3e_Kkc,366
+scale_gp_beta/types/stream_chunk.py,sha256=3D9ElvLwGCPwvymBiLClUs3fhGXIoNTXx0TiS7C91iI,276
 scale_gp_beta/types/chat/__init__.py,sha256=YmEJo3C_C7aRkImUaKB5BG97oOGkV0q3sHimL8cMa4g,688
 scale_gp_beta/types/chat/chat_completion.py,sha256=JM1meIbWU2ImWkSO2m78666ltpUgDzgj1RpN5y3EaZM,16335
 scale_gp_beta/types/chat/chat_completion_chunk.py,sha256=w7kGJcwoGuj_wl9kEACNspEvy8RV-h1wlwMW3dhC-xY,12025
@@ -157,7 +158,7 @@ scale_gp_beta/types/chat/model_definition.py,sha256=OWcZ4StZ4K2wfL7FqHNK-HqxLjy9
 scale_gp_beta/types/files/__init__.py,sha256=OKfJYcKb4NObdiRObqJV_dOyDQ8feXekDUge2o_4pXQ,122
 scale_gp_beta/types/shared/__init__.py,sha256=6-walu4YgOTaOj7wsidywTj67PyBJaNYFqasfiTP9-4,130
 scale_gp_beta/types/shared/identity.py,sha256=4XDoJjsPI4lkwyaYyNstw7OunIzJjVWujPoZPrNdoQA,348
-scale_gp_beta-0.1.0a41.dist-info/METADATA,sha256=_laC_GUGD5XqCQBuz3vNpUPUA953GKs_SnsRqLepACU,29627
-scale_gp_beta-0.1.0a41.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-scale_gp_beta-0.1.0a41.dist-info/licenses/LICENSE,sha256=LvNy55OMOsNOGUqjHCwQ6uYO0IuU9LJp1CJRpoz0abY,11338
-scale_gp_beta-0.1.0a41.dist-info/RECORD,,
+scale_gp_beta-0.1.0a42.dist-info/METADATA,sha256=r3OH5p8Riqmmud4qooNUGofX-AGyFBEcEVWPt0_sbok,30328
+scale_gp_beta-0.1.0a42.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+scale_gp_beta-0.1.0a42.dist-info/licenses/LICENSE,sha256=LvNy55OMOsNOGUqjHCwQ6uYO0IuU9LJp1CJRpoz0abY,11338
+scale_gp_beta-0.1.0a42.dist-info/RECORD,,