feat(api): OpenAPI spec update via Stainless API (#261)

Author: stainless-app[bot]

.github/workflows/ci.yml

@@ -6,6 +6,7 @@ on:
   pull_request:
     branches:
       - main
+      - next
 
 jobs:
   lint:

README.md

@@ -10,7 +10,7 @@ It is generated with [Stainless](https://www.stainlessapi.com/).
 
 ## Documentation
 
-The REST API documentation can be found [on openlayer.com](https://openlayer.com/docs/api-reference/rest/overview). The full API of this library can be found in [api.md](api.md).
+The REST API documentation can be found on [openlayer.com](https://openlayer.com/docs/api-reference/rest/overview). The full API of this library can be found in [api.md](api.md).
 
 ## Installation
 
@@ -33,7 +33,7 @@ client = Openlayer(
 )
 
 data_stream_response = client.inference_pipelines.data.stream(
-    "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
+    inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
     config={
         "input_variable_names": ["user_query"],
         "output_column_name": "output",
@@ -76,7 +76,7 @@ client = AsyncOpenlayer(
 
 async def main() -> None:
     data_stream_response = await client.inference_pipelines.data.stream(
-        "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
+        inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
         config={
             "input_variable_names": ["user_query"],
             "output_column_name": "output",
@@ -128,7 +128,7 @@ client = Openlayer()
 
 try:
     client.inference_pipelines.data.stream(
-        "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
+        inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
         config={
             "input_variable_names": ["user_query"],
             "output_column_name": "output",
@@ -189,7 +189,7 @@ client = Openlayer(
 
 # Or, configure per-request:
 client.with_options(max_retries=5).inference_pipelines.data.stream(
-    "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
+    inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
     config={
         "input_variable_names": ["user_query"],
         "output_column_name": "output",
@@ -230,7 +230,7 @@ client = Openlayer(
 
 # Override per-request:
 client.with_options(timeout=5.0).inference_pipelines.data.stream(
-    "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
+    inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
     config={
         "input_variable_names": ["user_query"],
         "output_column_name": "output",
@@ -287,7 +287,7 @@ from openlayer import Openlayer
 
 client = Openlayer()
 response = client.inference_pipelines.data.with_raw_response.stream(
-    "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
+    inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
     config={
         "input_variable_names": ["user_query"],
         "output_column_name": "output",
@@ -321,7 +321,7 @@ To stream the response body, use `.with_streaming_response` instead, which requi
 
 ```python
 with client.inference_pipelines.data.with_streaming_response.stream(
-    "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
+    inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
     config={
         "input_variable_names": ["user_query"],
         "output_column_name": "output",

requirements-dev.lock

@@ -49,7 +49,7 @@ markdown-it-py==3.0.0
     # via rich
 mdurl==0.1.2
     # via markdown-it-py
-mypy==1.7.1
+mypy==1.10.1
 mypy-extensions==1.0.0
     # via mypy
 nodeenv==1.8.0

src/openlayer/_base_client.py

@@ -879,9 +879,9 @@ def __exit__(
     def _prepare_options(
         self,
         options: FinalRequestOptions,  # noqa: ARG002
-    ) -> None:
+    ) -> FinalRequestOptions:
         """Hook for mutating the given options"""
-        return None
+        return options
 
     def _prepare_request(
         self,
@@ -955,8 +955,13 @@ def _request(
         stream: bool,
         stream_cls: type[_StreamT] | None,
     ) -> ResponseT | _StreamT:
+        # create a copy of the options we were given so that if the
+        # options are mutated later & we then retry, the retries are
+        # given the original options
+        input_options = model_copy(options)
+
         cast_to = self._maybe_override_cast_to(cast_to, options)
-        self._prepare_options(options)
+        options = self._prepare_options(options)
 
         retries = self._remaining_retries(remaining_retries, options)
         request = self._build_request(options)
@@ -979,7 +984,7 @@ def _request(
 
             if retries > 0:
                 return self._retry_request(
-                    options,
+                    input_options,
                     cast_to,
                     retries,
                     stream=stream,
@@ -994,7 +999,7 @@ def _request(
 
             if retries > 0:
                 return self._retry_request(
-                    options,
+                    input_options,
                     cast_to,
                     retries,
                     stream=stream,
@@ -1022,7 +1027,7 @@ def _request(
             if retries > 0 and self._should_retry(err.response):
                 err.response.close()
                 return self._retry_request(
-                    options,
+                    input_options,
                     cast_to,
                     retries,
                     err.response.headers,
@@ -1437,9 +1442,9 @@ async def __aexit__(
     async def _prepare_options(
         self,
         options: FinalRequestOptions,  # noqa: ARG002
-    ) -> None:
+    ) -> FinalRequestOptions:
         """Hook for mutating the given options"""
-        return None
+        return options
 
     async def _prepare_request(
         self,
@@ -1518,8 +1523,13 @@ async def _request(
             # execute it earlier while we are in an async context
             self._platform = await asyncify(get_platform)()
 
+        # create a copy of the options we were given so that if the
+        # options are mutated later & we then retry, the retries are
+        # given the original options
+        input_options = model_copy(options)
+
         cast_to = self._maybe_override_cast_to(cast_to, options)
-        await self._prepare_options(options)
+        options = await self._prepare_options(options)
 
         retries = self._remaining_retries(remaining_retries, options)
         request = self._build_request(options)
@@ -1540,7 +1550,7 @@ async def _request(
 
             if retries > 0:
                 return await self._retry_request(
-                    options,
+                    input_options,
                     cast_to,
                     retries,
                     stream=stream,
@@ -1555,7 +1565,7 @@ async def _request(
 
             if retries > 0:
                 return await self._retry_request(
-                    options,
+                    input_options,
                     cast_to,
                     retries,
                     stream=stream,
@@ -1578,7 +1588,7 @@ async def _request(
             if retries > 0 and self._should_retry(err.response):
                 await err.response.aclose()
                 return await self._retry_request(
-                    options,
+                    input_options,
                     cast_to,
                     retries,
                     err.response.headers,

src/openlayer/_compat.py

@@ -118,10 +118,10 @@ def get_model_fields(model: type[pydantic.BaseModel]) -> dict[str, FieldInfo]:
     return model.__fields__  # type: ignore
 
 
-def model_copy(model: _ModelT) -> _ModelT:
+def model_copy(model: _ModelT, *, deep: bool = False) -> _ModelT:
     if PYDANTIC_V2:
-        return model.model_copy()
-    return model.copy()  # type: ignore
+        return model.model_copy(deep=deep)
+    return model.copy(deep=deep)  # type: ignore
 
 
 def model_json(model: pydantic.BaseModel, *, indent: int | None = None) -> str:

src/openlayer/_models.py

@@ -643,6 +643,14 @@ def validate_type(*, type_: type[_T], value: object) -> _T:
     return cast(_T, _validate_non_model_type(type_=type_, value=value))
 
 
+def set_pydantic_config(typ: Any, config: pydantic.ConfigDict) -> None:
+    """Add a pydantic config for the given type.
+
+    Note: this is a no-op on Pydantic v1.
+    """
+    setattr(typ, "__pydantic_config__", config)  # noqa: B010
+
+
 # our use of subclasssing here causes weirdness for type checkers,
 # so we just pretend that we don't subclass
 if TYPE_CHECKING:

src/openlayer/resources/commits/test_results.py

@@ -19,9 +19,7 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import (
-    make_request_options,
-)
+from ..._base_client import make_request_options
 from ...types.commits import test_result_list_params
 from ...types.commits.test_result_list_response import TestResultListResponse
 

src/openlayer/resources/inference_pipelines/data.py

@@ -19,9 +19,7 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import (
-    make_request_options,
-)
+from ..._base_client import make_request_options
 from ...types.inference_pipelines import data_stream_params
 from ...types.inference_pipelines.data_stream_response import DataStreamResponse
 
@@ -51,13 +49,13 @@ def stream(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> DataStreamResponse:
         """
-        Stream production data to an inference pipeline.
+        Create an inference data point in an inference pipeline.
 
         Args:
           config: Configuration for the data stream. Depends on your **Openlayer project task
               type**.
 
-          rows: A list of entries that represent rows of a csv file
+          rows: A list of inference data points with inputs and outputs
 
           extra_headers: Send extra headers
 
@@ -110,13 +108,13 @@ async def stream(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> DataStreamResponse:
         """
-        Stream production data to an inference pipeline.
+        Create an inference data point in an inference pipeline.
 
         Args:
           config: Configuration for the data stream. Depends on your **Openlayer project task
               type**.
 
-          rows: A list of entries that represent rows of a csv file
+          rows: A list of inference data points with inputs and outputs
 
           extra_headers: Send extra headers
 

src/openlayer/resources/inference_pipelines/test_results.py

@@ -19,9 +19,7 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import (
-    make_request_options,
-)
+from ..._base_client import make_request_options
 from ...types.inference_pipelines import test_result_list_params
 from ...types.inference_pipelines.test_result_list_response import TestResultListResponse
 

src/openlayer/resources/projects/commits.py

@@ -17,9 +17,7 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import (
-    make_request_options,
-)
+from ..._base_client import make_request_options
 from ...types.projects import commit_list_params
 from ...types.projects.commit_list_response import CommitListResponse
 

src/openlayer/resources/projects/inference_pipelines.py

@@ -19,9 +19,7 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import (
-    make_request_options,
-)
+from ..._base_client import make_request_options
 from ...types.projects import inference_pipeline_list_params, inference_pipeline_create_params
 from ...types.projects.inference_pipeline_list_response import InferencePipelineListResponse
 from ...types.projects.inference_pipeline_create_response import InferencePipelineCreateResponse

src/openlayer/resources/projects/projects.py

@@ -29,9 +29,7 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import (
-    make_request_options,
-)
+from ..._base_client import make_request_options
 from .inference_pipelines import (
     InferencePipelinesResource,
     AsyncInferencePipelinesResource,

src/openlayer/types/inference_pipelines/data_stream_params.py

@@ -26,7 +26,7 @@ class DataStreamParams(TypedDict, total=False):
     """
 
     rows: Required[Iterable[Dict[str, object]]]
-    """A list of entries that represent rows of a csv file"""
+    """A list of inference data points with inputs and outputs"""
 
 
 class ConfigLlmDataPrompt(TypedDict, total=False):