22
33from __future__ import annotations
44
5+ import httpx
6+
57from .data import (
68 DataResource ,
79 AsyncDataResource ,
1820 RowsResourceWithStreamingResponse ,
1921 AsyncRowsResourceWithStreamingResponse ,
2022)
23+ from ..._types import NOT_GIVEN , Body , Query , Headers , NoneType , NotGiven
2124from ..._compat import cached_property
2225from ..._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+ )
2332from .test_results import (
2433 TestResultsResource ,
2534 AsyncTestResultsResource ,
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
57139class 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
79232class 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 )
0 commit comments