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

scale_gp_beta/_base_client.py

@@ -86,6 +86,7 @@ from ._exceptions import (
     APIConnectionError,
     APIResponseValidationError,
 )
+from ._utils._json import openapi_dumps
 
 log: logging.Logger = logging.getLogger(__name__)
 
@@ -554,8 +555,10 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
                 kwargs["content"] = options.content
             elif isinstance(json_data, bytes):
                 kwargs["content"] = json_data
-            else:
-                kwargs["json"] = json_data if is_given(json_data) else None
+            elif not files:
+                # Don't set content when JSON is sent as multipart/form-data,
+                # since httpx's content param overrides other body arguments
+                kwargs["content"] = openapi_dumps(json_data) if is_given(json_data) and json_data is not None else None
             kwargs["files"] = files
         else:
             headers.pop("Content-Type", None)

scale_gp_beta/_client.py

@@ -45,6 +45,7 @@ if TYPE_CHECKING:
         credentials,
         evaluations,
         dataset_items,
+        vector_stores,
         evaluation_items,
         span_assessments,
     )
@@ -61,6 +62,7 @@ if TYPE_CHECKING:
     from .resources.evaluations import EvaluationsResource, AsyncEvaluationsResource
     from .resources.files.files import FilesResource, AsyncFilesResource
     from .resources.dataset_items import DatasetItemsResource, AsyncDatasetItemsResource
+    from .resources.vector_stores import VectorStoresResource, AsyncVectorStoresResource
     from .resources.evaluation_items import EvaluationItemsResource, AsyncEvaluationItemsResource
     from .resources.span_assessments import SpanAssessmentsResource, AsyncSpanAssessmentsResource
 
@@ -265,6 +267,12 @@ class SGPClient(SyncAPIClient):
 
         return BuildResource(self)
 
+    @cached_property
+    def vector_stores(self) -> VectorStoresResource:
+        from .resources.vector_stores import VectorStoresResource
+
+        return VectorStoresResource(self)
+
     @cached_property
     def with_raw_response(self) -> SGPClientWithRawResponse:
         return SGPClientWithRawResponse(self)
@@ -566,6 +574,12 @@ class AsyncSGPClient(AsyncAPIClient):
 
         return AsyncBuildResource(self)
 
+    @cached_property
+    def vector_stores(self) -> AsyncVectorStoresResource:
+        from .resources.vector_stores import AsyncVectorStoresResource
+
+        return AsyncVectorStoresResource(self)
+
     @cached_property
     def with_raw_response(self) -> AsyncSGPClientWithRawResponse:
         return AsyncSGPClientWithRawResponse(self)
@@ -780,6 +794,12 @@ class SGPClientWithRawResponse:
 
         return BuildResourceWithRawResponse(self._client.build)
 
+    @cached_property
+    def vector_stores(self) -> vector_stores.VectorStoresResourceWithRawResponse:
+        from .resources.vector_stores import VectorStoresResourceWithRawResponse
+
+        return VectorStoresResourceWithRawResponse(self._client.vector_stores)
+
 
 class AsyncSGPClientWithRawResponse:
     _client: AsyncSGPClient
@@ -877,6 +897,12 @@ class AsyncSGPClientWithRawResponse:
 
         return AsyncBuildResourceWithRawResponse(self._client.build)
 
+    @cached_property
+    def vector_stores(self) -> vector_stores.AsyncVectorStoresResourceWithRawResponse:
+        from .resources.vector_stores import AsyncVectorStoresResourceWithRawResponse
+
+        return AsyncVectorStoresResourceWithRawResponse(self._client.vector_stores)
+
 
 class SGPClientWithStreamedResponse:
     _client: SGPClient
@@ -974,6 +1000,12 @@ class SGPClientWithStreamedResponse:
 
         return BuildResourceWithStreamingResponse(self._client.build)
 
+    @cached_property
+    def vector_stores(self) -> vector_stores.VectorStoresResourceWithStreamingResponse:
+        from .resources.vector_stores import VectorStoresResourceWithStreamingResponse
+
+        return VectorStoresResourceWithStreamingResponse(self._client.vector_stores)
+
 
 class AsyncSGPClientWithStreamedResponse:
     _client: AsyncSGPClient
@@ -1071,6 +1103,12 @@ class AsyncSGPClientWithStreamedResponse:
 
         return AsyncBuildResourceWithStreamingResponse(self._client.build)
 
+    @cached_property
+    def vector_stores(self) -> vector_stores.AsyncVectorStoresResourceWithStreamingResponse:
+        from .resources.vector_stores import AsyncVectorStoresResourceWithStreamingResponse
+
+        return AsyncVectorStoresResourceWithStreamingResponse(self._client.vector_stores)
+
 
 Client = SGPClient
 

scale_gp_beta/_compat.py

@@ -139,6 +139,7 @@ def model_dump(
     exclude_defaults: bool = False,
     warnings: bool = True,
     mode: Literal["json", "python"] = "python",
+    by_alias: bool | None = None,
 ) -> dict[str, Any]:
     if (not PYDANTIC_V1) or hasattr(model, "model_dump"):
         return model.model_dump(
@@ -148,13 +149,12 @@ def model_dump(
             exclude_defaults=exclude_defaults,
             # warnings are not supported in Pydantic v1
             warnings=True if PYDANTIC_V1 else warnings,
+            by_alias=by_alias,
         )
     return cast(
         "dict[str, Any]",
         model.dict(  # pyright: ignore[reportDeprecated, reportUnnecessaryCast]
-            exclude=exclude,
-            exclude_unset=exclude_unset,
-            exclude_defaults=exclude_defaults,
+            exclude=exclude, exclude_unset=exclude_unset, exclude_defaults=exclude_defaults, by_alias=bool(by_alias)
         ),
     )
 

scale_gp_beta/_utils/_json.py

@@ -0,0 +1,35 @@
+import json
+from typing import Any
+from datetime import datetime
+from typing_extensions import override
+
+import pydantic
+
+from .._compat import model_dump
+
+
+def openapi_dumps(obj: Any) -> bytes:
+    """
+    Serialize an object to UTF-8 encoded JSON bytes.
+
+    Extends the standard json.dumps with support for additional types
+    commonly used in the SDK, such as `datetime`, `pydantic.BaseModel`, etc.
+    """
+    return json.dumps(
+        obj,
+        cls=_CustomEncoder,
+        # Uses the same defaults as httpx's JSON serialization
+        ensure_ascii=False,
+        separators=(",", ":"),
+        allow_nan=False,
+    ).encode()
+
+
+class _CustomEncoder(json.JSONEncoder):
+    @override
+    def default(self, o: Any) -> Any:
+        if isinstance(o, datetime):
+            return o.isoformat()
+        if isinstance(o, pydantic.BaseModel):
+            return model_dump(o, exclude_unset=True, mode="json", by_alias=True)
+        return super().default(o)

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.42"  # x-release-please-version
+__version__ = "0.1.0-alpha.43"  # x-release-please-version

scale_gp_beta/resources/__init__.py

@@ -104,6 +104,14 @@ from .dataset_items import (
     DatasetItemsResourceWithStreamingResponse,
     AsyncDatasetItemsResourceWithStreamingResponse,
 )
+from .vector_stores import (
+    VectorStoresResource,
+    AsyncVectorStoresResource,
+    VectorStoresResourceWithRawResponse,
+    AsyncVectorStoresResourceWithRawResponse,
+    VectorStoresResourceWithStreamingResponse,
+    AsyncVectorStoresResourceWithStreamingResponse,
+)
 from .evaluation_items import (
     EvaluationItemsResource,
     AsyncEvaluationItemsResource,
@@ -212,4 +220,10 @@ __all__ = [
     "AsyncBuildResourceWithRawResponse",
     "BuildResourceWithStreamingResponse",
     "AsyncBuildResourceWithStreamingResponse",
+    "VectorStoresResource",
+    "AsyncVectorStoresResource",
+    "VectorStoresResourceWithRawResponse",
+    "AsyncVectorStoresResourceWithRawResponse",
+    "VectorStoresResourceWithStreamingResponse",
+    "AsyncVectorStoresResourceWithStreamingResponse",
 ]