Skip to content

Commit 6d2708a

Browse files
feat(api): update via SDK Studio (#272)
1 parent 859bc11 commit 6d2708a

File tree

6 files changed

+429
-1
lines changed

6 files changed

+429
-1
lines changed

.stats.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
configured_endpoints: 10
1+
configured_endpoints: 12

api.md

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,17 @@ Methods:
5252

5353
# InferencePipelines
5454

55+
Types:
56+
57+
```python
58+
from openlayer.types import InferencePipelineRetrieveResponse
59+
```
60+
61+
Methods:
62+
63+
- <code title="get /inference-pipelines/{inferencePipelineId}">client.inference_pipelines.<a href="./src/openlayer/resources/inference_pipelines/inference_pipelines.py">retrieve</a>(inference_pipeline_id) -> <a href="./src/openlayer/types/inference_pipeline_retrieve_response.py">InferencePipelineRetrieveResponse</a></code>
64+
- <code title="delete /inference-pipelines/{inferencePipelineId}">client.inference_pipelines.<a href="./src/openlayer/resources/inference_pipelines/inference_pipelines.py">delete</a>(inference_pipeline_id) -> None</code>
65+
5566
## Data
5667

5768
Types:

src/openlayer/resources/inference_pipelines/inference_pipelines.py

Lines changed: 181 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
from __future__ import annotations
44

5+
import httpx
6+
57
from .data import (
68
DataResource,
79
AsyncDataResource,
@@ -18,8 +20,15 @@
1820
RowsResourceWithStreamingResponse,
1921
AsyncRowsResourceWithStreamingResponse,
2022
)
23+
from ..._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
2124
from ..._compat import cached_property
2225
from ..._resource import SyncAPIResource, AsyncAPIResource
26+
from ..._response import (
27+
to_raw_response_wrapper,
28+
to_streamed_response_wrapper,
29+
async_to_raw_response_wrapper,
30+
async_to_streamed_response_wrapper,
31+
)
2332
from .test_results import (
2433
TestResultsResource,
2534
AsyncTestResultsResource,
@@ -28,6 +37,8 @@
2837
TestResultsResourceWithStreamingResponse,
2938
AsyncTestResultsResourceWithStreamingResponse,
3039
)
40+
from ..._base_client import make_request_options
41+
from ...types.inference_pipeline_retrieve_response import InferencePipelineRetrieveResponse
3142

3243
__all__ = ["InferencePipelinesResource", "AsyncInferencePipelinesResource"]
3344

@@ -53,6 +64,77 @@ def with_raw_response(self) -> InferencePipelinesResourceWithRawResponse:
5364
def with_streaming_response(self) -> InferencePipelinesResourceWithStreamingResponse:
5465
return InferencePipelinesResourceWithStreamingResponse(self)
5566

67+
def retrieve(
68+
self,
69+
inference_pipeline_id: str,
70+
*,
71+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
72+
# The extra values given here take precedence over values defined on the client or passed to this method.
73+
extra_headers: Headers | None = None,
74+
extra_query: Query | None = None,
75+
extra_body: Body | None = None,
76+
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
77+
) -> InferencePipelineRetrieveResponse:
78+
"""
79+
Retrieve inference pipeline.
80+
81+
Args:
82+
extra_headers: Send extra headers
83+
84+
extra_query: Add additional query parameters to the request
85+
86+
extra_body: Add additional JSON properties to the request
87+
88+
timeout: Override the client-level default timeout for this request, in seconds
89+
"""
90+
if not inference_pipeline_id:
91+
raise ValueError(
92+
f"Expected a non-empty value for `inference_pipeline_id` but received {inference_pipeline_id!r}"
93+
)
94+
return self._get(
95+
f"/inference-pipelines/{inference_pipeline_id}",
96+
options=make_request_options(
97+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
98+
),
99+
cast_to=InferencePipelineRetrieveResponse,
100+
)
101+
102+
def delete(
103+
self,
104+
inference_pipeline_id: str,
105+
*,
106+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
107+
# The extra values given here take precedence over values defined on the client or passed to this method.
108+
extra_headers: Headers | None = None,
109+
extra_query: Query | None = None,
110+
extra_body: Body | None = None,
111+
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
112+
) -> None:
113+
"""
114+
Delete inference pipeline.
115+
116+
Args:
117+
extra_headers: Send extra headers
118+
119+
extra_query: Add additional query parameters to the request
120+
121+
extra_body: Add additional JSON properties to the request
122+
123+
timeout: Override the client-level default timeout for this request, in seconds
124+
"""
125+
if not inference_pipeline_id:
126+
raise ValueError(
127+
f"Expected a non-empty value for `inference_pipeline_id` but received {inference_pipeline_id!r}"
128+
)
129+
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
130+
return self._delete(
131+
f"/inference-pipelines/{inference_pipeline_id}",
132+
options=make_request_options(
133+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
134+
),
135+
cast_to=NoneType,
136+
)
137+
56138

57139
class AsyncInferencePipelinesResource(AsyncAPIResource):
58140
@cached_property
@@ -75,11 +157,89 @@ def with_raw_response(self) -> AsyncInferencePipelinesResourceWithRawResponse:
75157
def with_streaming_response(self) -> AsyncInferencePipelinesResourceWithStreamingResponse:
76158
return AsyncInferencePipelinesResourceWithStreamingResponse(self)
77159

160+
async def retrieve(
161+
self,
162+
inference_pipeline_id: str,
163+
*,
164+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
165+
# The extra values given here take precedence over values defined on the client or passed to this method.
166+
extra_headers: Headers | None = None,
167+
extra_query: Query | None = None,
168+
extra_body: Body | None = None,
169+
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
170+
) -> InferencePipelineRetrieveResponse:
171+
"""
172+
Retrieve inference pipeline.
173+
174+
Args:
175+
extra_headers: Send extra headers
176+
177+
extra_query: Add additional query parameters to the request
178+
179+
extra_body: Add additional JSON properties to the request
180+
181+
timeout: Override the client-level default timeout for this request, in seconds
182+
"""
183+
if not inference_pipeline_id:
184+
raise ValueError(
185+
f"Expected a non-empty value for `inference_pipeline_id` but received {inference_pipeline_id!r}"
186+
)
187+
return await self._get(
188+
f"/inference-pipelines/{inference_pipeline_id}",
189+
options=make_request_options(
190+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
191+
),
192+
cast_to=InferencePipelineRetrieveResponse,
193+
)
194+
195+
async def delete(
196+
self,
197+
inference_pipeline_id: str,
198+
*,
199+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
200+
# The extra values given here take precedence over values defined on the client or passed to this method.
201+
extra_headers: Headers | None = None,
202+
extra_query: Query | None = None,
203+
extra_body: Body | None = None,
204+
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
205+
) -> None:
206+
"""
207+
Delete inference pipeline.
208+
209+
Args:
210+
extra_headers: Send extra headers
211+
212+
extra_query: Add additional query parameters to the request
213+
214+
extra_body: Add additional JSON properties to the request
215+
216+
timeout: Override the client-level default timeout for this request, in seconds
217+
"""
218+
if not inference_pipeline_id:
219+
raise ValueError(
220+
f"Expected a non-empty value for `inference_pipeline_id` but received {inference_pipeline_id!r}"
221+
)
222+
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
223+
return await self._delete(
224+
f"/inference-pipelines/{inference_pipeline_id}",
225+
options=make_request_options(
226+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
227+
),
228+
cast_to=NoneType,
229+
)
230+
78231

79232
class InferencePipelinesResourceWithRawResponse:
80233
def __init__(self, inference_pipelines: InferencePipelinesResource) -> None:
81234
self._inference_pipelines = inference_pipelines
82235

236+
self.retrieve = to_raw_response_wrapper(
237+
inference_pipelines.retrieve,
238+
)
239+
self.delete = to_raw_response_wrapper(
240+
inference_pipelines.delete,
241+
)
242+
83243
@cached_property
84244
def data(self) -> DataResourceWithRawResponse:
85245
return DataResourceWithRawResponse(self._inference_pipelines.data)
@@ -97,6 +257,13 @@ class AsyncInferencePipelinesResourceWithRawResponse:
97257
def __init__(self, inference_pipelines: AsyncInferencePipelinesResource) -> None:
98258
self._inference_pipelines = inference_pipelines
99259

260+
self.retrieve = async_to_raw_response_wrapper(
261+
inference_pipelines.retrieve,
262+
)
263+
self.delete = async_to_raw_response_wrapper(
264+
inference_pipelines.delete,
265+
)
266+
100267
@cached_property
101268
def data(self) -> AsyncDataResourceWithRawResponse:
102269
return AsyncDataResourceWithRawResponse(self._inference_pipelines.data)
@@ -114,6 +281,13 @@ class InferencePipelinesResourceWithStreamingResponse:
114281
def __init__(self, inference_pipelines: InferencePipelinesResource) -> None:
115282
self._inference_pipelines = inference_pipelines
116283

284+
self.retrieve = to_streamed_response_wrapper(
285+
inference_pipelines.retrieve,
286+
)
287+
self.delete = to_streamed_response_wrapper(
288+
inference_pipelines.delete,
289+
)
290+
117291
@cached_property
118292
def data(self) -> DataResourceWithStreamingResponse:
119293
return DataResourceWithStreamingResponse(self._inference_pipelines.data)
@@ -131,6 +305,13 @@ class AsyncInferencePipelinesResourceWithStreamingResponse:
131305
def __init__(self, inference_pipelines: AsyncInferencePipelinesResource) -> None:
132306
self._inference_pipelines = inference_pipelines
133307

308+
self.retrieve = async_to_streamed_response_wrapper(
309+
inference_pipelines.retrieve,
310+
)
311+
self.delete = async_to_streamed_response_wrapper(
312+
inference_pipelines.delete,
313+
)
314+
134315
@cached_property
135316
def data(self) -> AsyncDataResourceWithStreamingResponse:
136317
return AsyncDataResourceWithStreamingResponse(self._inference_pipelines.data)

src/openlayer/types/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,3 +6,4 @@
66
from .project_create_params import ProjectCreateParams as ProjectCreateParams
77
from .project_list_response import ProjectListResponse as ProjectListResponse
88
from .project_create_response import ProjectCreateResponse as ProjectCreateResponse
9+
from .inference_pipeline_retrieve_response import InferencePipelineRetrieveResponse as InferencePipelineRetrieveResponse
Lines changed: 61 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,61 @@
1+
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2+
3+
from typing import Optional
4+
from datetime import datetime
5+
from typing_extensions import Literal
6+
7+
from pydantic import Field as FieldInfo
8+
9+
from .._models import BaseModel
10+
11+
__all__ = ["InferencePipelineRetrieveResponse", "Links"]
12+
13+
14+
class Links(BaseModel):
15+
app: str
16+
17+
18+
class InferencePipelineRetrieveResponse(BaseModel):
19+
id: str
20+
"""The inference pipeline id."""
21+
22+
date_created: datetime = FieldInfo(alias="dateCreated")
23+
"""The creation date."""
24+
25+
date_last_evaluated: Optional[datetime] = FieldInfo(alias="dateLastEvaluated", default=None)
26+
"""The last test evaluation date."""
27+
28+
date_last_sample_received: Optional[datetime] = FieldInfo(alias="dateLastSampleReceived", default=None)
29+
"""The last data sample received date."""
30+
31+
date_of_next_evaluation: Optional[datetime] = FieldInfo(alias="dateOfNextEvaluation", default=None)
32+
"""The next test evaluation date."""
33+
34+
date_updated: datetime = FieldInfo(alias="dateUpdated")
35+
"""The last updated date."""
36+
37+
description: Optional[str] = None
38+
"""The inference pipeline description."""
39+
40+
failing_goal_count: int = FieldInfo(alias="failingGoalCount")
41+
"""The number of tests failing."""
42+
43+
links: Links
44+
45+
name: str
46+
"""The inference pipeline name."""
47+
48+
passing_goal_count: int = FieldInfo(alias="passingGoalCount")
49+
"""The number of tests passing."""
50+
51+
project_id: str = FieldInfo(alias="projectId")
52+
"""The project id."""
53+
54+
status: Literal["queued", "running", "paused", "failed", "completed", "unknown"]
55+
"""The status of test evaluation for the inference pipeline."""
56+
57+
status_message: Optional[str] = FieldInfo(alias="statusMessage", default=None)
58+
"""The status message of test evaluation for the inference pipeline."""
59+
60+
total_goal_count: int = FieldInfo(alias="totalGoalCount")
61+
"""The total number of tests."""

0 commit comments

Comments
 (0)