2
2
3
3
from __future__ import annotations
4
4
5
+ import httpx
6
+
5
7
from .data import (
6
8
DataResource ,
7
9
AsyncDataResource ,
18
20
RowsResourceWithStreamingResponse ,
19
21
AsyncRowsResourceWithStreamingResponse ,
20
22
)
23
+ from ..._types import NOT_GIVEN , Body , Query , Headers , NoneType , NotGiven
21
24
from ..._compat import cached_property
22
25
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
+ )
23
32
from .test_results import (
24
33
TestResultsResource ,
25
34
AsyncTestResultsResource ,
28
37
TestResultsResourceWithStreamingResponse ,
29
38
AsyncTestResultsResourceWithStreamingResponse ,
30
39
)
40
+ from ..._base_client import make_request_options
41
+ from ...types .inference_pipeline_retrieve_response import InferencePipelineRetrieveResponse
31
42
32
43
__all__ = ["InferencePipelinesResource" , "AsyncInferencePipelinesResource" ]
33
44
@@ -53,6 +64,77 @@ def with_raw_response(self) -> InferencePipelinesResourceWithRawResponse:
53
64
def with_streaming_response (self ) -> InferencePipelinesResourceWithStreamingResponse :
54
65
return InferencePipelinesResourceWithStreamingResponse (self )
55
66
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
+
56
138
57
139
class AsyncInferencePipelinesResource (AsyncAPIResource ):
58
140
@cached_property
@@ -75,11 +157,89 @@ def with_raw_response(self) -> AsyncInferencePipelinesResourceWithRawResponse:
75
157
def with_streaming_response (self ) -> AsyncInferencePipelinesResourceWithStreamingResponse :
76
158
return AsyncInferencePipelinesResourceWithStreamingResponse (self )
77
159
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
+
78
231
79
232
class InferencePipelinesResourceWithRawResponse :
80
233
def __init__ (self , inference_pipelines : InferencePipelinesResource ) -> None :
81
234
self ._inference_pipelines = inference_pipelines
82
235
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
+
83
243
@cached_property
84
244
def data (self ) -> DataResourceWithRawResponse :
85
245
return DataResourceWithRawResponse (self ._inference_pipelines .data )
@@ -97,6 +257,13 @@ class AsyncInferencePipelinesResourceWithRawResponse:
97
257
def __init__ (self , inference_pipelines : AsyncInferencePipelinesResource ) -> None :
98
258
self ._inference_pipelines = inference_pipelines
99
259
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
+
100
267
@cached_property
101
268
def data (self ) -> AsyncDataResourceWithRawResponse :
102
269
return AsyncDataResourceWithRawResponse (self ._inference_pipelines .data )
@@ -114,6 +281,13 @@ class InferencePipelinesResourceWithStreamingResponse:
114
281
def __init__ (self , inference_pipelines : InferencePipelinesResource ) -> None :
115
282
self ._inference_pipelines = inference_pipelines
116
283
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
+
117
291
@cached_property
118
292
def data (self ) -> DataResourceWithStreamingResponse :
119
293
return DataResourceWithStreamingResponse (self ._inference_pipelines .data )
@@ -131,6 +305,13 @@ class AsyncInferencePipelinesResourceWithStreamingResponse:
131
305
def __init__ (self , inference_pipelines : AsyncInferencePipelinesResource ) -> None :
132
306
self ._inference_pipelines = inference_pipelines
133
307
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
+
134
315
@cached_property
135
316
def data (self ) -> AsyncDataResourceWithStreamingResponse :
136
317
return AsyncDataResourceWithStreamingResponse (self ._inference_pipelines .data )
0 commit comments