1717 async_to_streamed_response_wrapper ,
1818)
1919from ..._base_client import make_request_options
20- from ...types .inference_pipelines import row_create_params , row_update_params
21- from ...types .inference_pipelines .row_create_response import RowCreateResponse
20+ from ...types .inference_pipelines import row_list_params , row_update_params
21+ from ...types .inference_pipelines .row_list_response import RowListResponse
2222from ...types .inference_pipelines .row_update_response import RowUpdateResponse
2323
2424__all__ = ["RowsResource" , "AsyncRowsResource" ]
@@ -44,15 +44,66 @@ def with_streaming_response(self) -> RowsResourceWithStreamingResponse:
4444 """
4545 return RowsResourceWithStreamingResponse (self )
4646
47- def create (
47+ def update (
48+ self ,
49+ inference_pipeline_id : str ,
50+ * ,
51+ inference_id : str ,
52+ row : object ,
53+ config : Optional [row_update_params .Config ] | Omit = omit ,
54+ # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
55+ # The extra values given here take precedence over values defined on the client or passed to this method.
56+ extra_headers : Headers | None = None ,
57+ extra_query : Query | None = None ,
58+ extra_body : Body | None = None ,
59+ timeout : float | httpx .Timeout | None | NotGiven = not_given ,
60+ ) -> RowUpdateResponse :
61+ """
62+ Update an inference data point in an inference pipeline.
63+
64+ Args:
65+ inference_id: Specify the inference id as a query param.
66+
67+ extra_headers: Send extra headers
68+
69+ extra_query: Add additional query parameters to the request
70+
71+ extra_body: Add additional JSON properties to the request
72+
73+ timeout: Override the client-level default timeout for this request, in seconds
74+ """
75+ if not inference_pipeline_id :
76+ raise ValueError (
77+ f"Expected a non-empty value for `inference_pipeline_id` but received { inference_pipeline_id !r} "
78+ )
79+ return self ._put (
80+ f"/inference-pipelines/{ inference_pipeline_id } /rows" ,
81+ body = maybe_transform (
82+ {
83+ "row" : row ,
84+ "config" : config ,
85+ },
86+ row_update_params .RowUpdateParams ,
87+ ),
88+ options = make_request_options (
89+ extra_headers = extra_headers ,
90+ extra_query = extra_query ,
91+ extra_body = extra_body ,
92+ timeout = timeout ,
93+ query = maybe_transform ({"inference_id" : inference_id }, row_update_params .RowUpdateParams ),
94+ ),
95+ cast_to = RowUpdateResponse ,
96+ )
97+
98+ def list (
4899 self ,
49100 inference_pipeline_id : str ,
50101 * ,
51102 asc : bool | Omit = omit ,
52103 page : int | Omit = omit ,
53104 per_page : int | Omit = omit ,
54105 sort_column : str | Omit = omit ,
55- column_filters : Optional [Iterable [row_create_params .ColumnFilter ]] | Omit = omit ,
106+ column_filters : Optional [Iterable [row_list_params .ColumnFilter ]] | Omit = omit ,
56107 exclude_row_id_list : Optional [Iterable [int ]] | Omit = omit ,
57108 not_search_query_and : Optional [SequenceNotStr [str ]] | Omit = omit ,
58109 not_search_query_or : Optional [SequenceNotStr [str ]] | Omit = omit ,
@@ -65,7 +116,7 @@ def create(
65116 extra_query : Query | None = None ,
66117 extra_body : Body | None = None ,
67118 timeout : float | httpx .Timeout | None | NotGiven = not_given ,
68- ) -> RowCreateResponse :
119+ ) -> RowListResponse :
69120 """
70121 A list of rows for an inference pipeline.
71122
@@ -102,7 +153,7 @@ def create(
102153 "search_query_and" : search_query_and ,
103154 "search_query_or" : search_query_or ,
104155 },
105- row_create_params . RowCreateParams ,
156+ row_list_params . RowListParams ,
106157 ),
107158 options = make_request_options (
108159 extra_headers = extra_headers ,
@@ -116,13 +167,34 @@ def create(
116167 "per_page" : per_page ,
117168 "sort_column" : sort_column ,
118169 },
119- row_create_params . RowCreateParams ,
170+ row_list_params . RowListParams ,
120171 ),
121172 ),
122- cast_to = RowCreateResponse ,
173+ cast_to = RowListResponse ,
123174 )
124175
125- def update (
176+
177+ class AsyncRowsResource (AsyncAPIResource ):
178+ @cached_property
179+ def with_raw_response (self ) -> AsyncRowsResourceWithRawResponse :
180+ """
181+ This property can be used as a prefix for any HTTP method call to return
182+ the raw response object instead of the parsed content.
183+
184+ For more information, see https://www.github.com/openlayer-ai/openlayer-python#accessing-raw-response-data-eg-headers
185+ """
186+ return AsyncRowsResourceWithRawResponse (self )
187+
188+ @cached_property
189+ def with_streaming_response (self ) -> AsyncRowsResourceWithStreamingResponse :
190+ """
191+ An alternative to `.with_raw_response` that doesn't eagerly read the response body.
192+
193+ For more information, see https://www.github.com/openlayer-ai/openlayer-python#with_streaming_response
194+ """
195+ return AsyncRowsResourceWithStreamingResponse (self )
196+
197+ async def update (
126198 self ,
127199 inference_pipeline_id : str ,
128200 * ,
@@ -154,9 +226,9 @@ def update(
154226 raise ValueError (
155227 f"Expected a non-empty value for `inference_pipeline_id` but received { inference_pipeline_id !r} "
156228 )
157- return self ._put (
229+ return await self ._put (
158230 f"/inference-pipelines/{ inference_pipeline_id } /rows" ,
159- body = maybe_transform (
231+ body = await async_maybe_transform (
160232 {
161233 "row" : row ,
162234 "config" : config ,
@@ -168,41 +240,20 @@ def update(
168240 extra_query = extra_query ,
169241 extra_body = extra_body ,
170242 timeout = timeout ,
171- query = maybe_transform ({"inference_id" : inference_id }, row_update_params .RowUpdateParams ),
243+ query = await async_maybe_transform ({"inference_id" : inference_id }, row_update_params .RowUpdateParams ),
172244 ),
173245 cast_to = RowUpdateResponse ,
174246 )
175247
176-
177- class AsyncRowsResource (AsyncAPIResource ):
178- @cached_property
179- def with_raw_response (self ) -> AsyncRowsResourceWithRawResponse :
180- """
181- This property can be used as a prefix for any HTTP method call to return
182- the raw response object instead of the parsed content.
183-
184- For more information, see https://www.github.com/openlayer-ai/openlayer-python#accessing-raw-response-data-eg-headers
185- """
186- return AsyncRowsResourceWithRawResponse (self )
187-
188- @cached_property
189- def with_streaming_response (self ) -> AsyncRowsResourceWithStreamingResponse :
190- """
191- An alternative to `.with_raw_response` that doesn't eagerly read the response body.
192-
193- For more information, see https://www.github.com/openlayer-ai/openlayer-python#with_streaming_response
194- """
195- return AsyncRowsResourceWithStreamingResponse (self )
196-
197- async def create (
248+ async def list (
198249 self ,
199250 inference_pipeline_id : str ,
200251 * ,
201252 asc : bool | Omit = omit ,
202253 page : int | Omit = omit ,
203254 per_page : int | Omit = omit ,
204255 sort_column : str | Omit = omit ,
205- column_filters : Optional [Iterable [row_create_params .ColumnFilter ]] | Omit = omit ,
256+ column_filters : Optional [Iterable [row_list_params .ColumnFilter ]] | Omit = omit ,
206257 exclude_row_id_list : Optional [Iterable [int ]] | Omit = omit ,
207258 not_search_query_and : Optional [SequenceNotStr [str ]] | Omit = omit ,
208259 not_search_query_or : Optional [SequenceNotStr [str ]] | Omit = omit ,
@@ -215,7 +266,7 @@ async def create(
215266 extra_query : Query | None = None ,
216267 extra_body : Body | None = None ,
217268 timeout : float | httpx .Timeout | None | NotGiven = not_given ,
218- ) -> RowCreateResponse :
269+ ) -> RowListResponse :
219270 """
220271 A list of rows for an inference pipeline.
221272
@@ -252,7 +303,7 @@ async def create(
252303 "search_query_and" : search_query_and ,
253304 "search_query_or" : search_query_or ,
254305 },
255- row_create_params . RowCreateParams ,
306+ row_list_params . RowListParams ,
256307 ),
257308 options = make_request_options (
258309 extra_headers = extra_headers ,
@@ -266,107 +317,56 @@ async def create(
266317 "per_page" : per_page ,
267318 "sort_column" : sort_column ,
268319 },
269- row_create_params . RowCreateParams ,
320+ row_list_params . RowListParams ,
270321 ),
271322 ),
272- cast_to = RowCreateResponse ,
273- )
274-
275- async def update (
276- self ,
277- inference_pipeline_id : str ,
278- * ,
279- inference_id : str ,
280- row : object ,
281- config : Optional [row_update_params .Config ] | Omit = omit ,
282- # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
283- # The extra values given here take precedence over values defined on the client or passed to this method.
284- extra_headers : Headers | None = None ,
285- extra_query : Query | None = None ,
286- extra_body : Body | None = None ,
287- timeout : float | httpx .Timeout | None | NotGiven = not_given ,
288- ) -> RowUpdateResponse :
289- """
290- Update an inference data point in an inference pipeline.
291-
292- Args:
293- inference_id: Specify the inference id as a query param.
294-
295- extra_headers: Send extra headers
296-
297- extra_query: Add additional query parameters to the request
298-
299- extra_body: Add additional JSON properties to the request
300-
301- timeout: Override the client-level default timeout for this request, in seconds
302- """
303- if not inference_pipeline_id :
304- raise ValueError (
305- f"Expected a non-empty value for `inference_pipeline_id` but received { inference_pipeline_id !r} "
306- )
307- return await self ._put (
308- f"/inference-pipelines/{ inference_pipeline_id } /rows" ,
309- body = await async_maybe_transform (
310- {
311- "row" : row ,
312- "config" : config ,
313- },
314- row_update_params .RowUpdateParams ,
315- ),
316- options = make_request_options (
317- extra_headers = extra_headers ,
318- extra_query = extra_query ,
319- extra_body = extra_body ,
320- timeout = timeout ,
321- query = await async_maybe_transform ({"inference_id" : inference_id }, row_update_params .RowUpdateParams ),
322- ),
323- cast_to = RowUpdateResponse ,
323+ cast_to = RowListResponse ,
324324 )
325325
326326
327327class RowsResourceWithRawResponse :
328328 def __init__ (self , rows : RowsResource ) -> None :
329329 self ._rows = rows
330330
331- self .create = to_raw_response_wrapper (
332- rows .create ,
333- )
334331 self .update = to_raw_response_wrapper (
335332 rows .update ,
336333 )
334+ self .list = to_raw_response_wrapper (
335+ rows .list ,
336+ )
337337
338338
339339class AsyncRowsResourceWithRawResponse :
340340 def __init__ (self , rows : AsyncRowsResource ) -> None :
341341 self ._rows = rows
342342
343- self .create = async_to_raw_response_wrapper (
344- rows .create ,
345- )
346343 self .update = async_to_raw_response_wrapper (
347344 rows .update ,
348345 )
346+ self .list = async_to_raw_response_wrapper (
347+ rows .list ,
348+ )
349349
350350
351351class RowsResourceWithStreamingResponse :
352352 def __init__ (self , rows : RowsResource ) -> None :
353353 self ._rows = rows
354354
355- self .create = to_streamed_response_wrapper (
356- rows .create ,
357- )
358355 self .update = to_streamed_response_wrapper (
359356 rows .update ,
360357 )
358+ self .list = to_streamed_response_wrapper (
359+ rows .list ,
360+ )
361361
362362
363363class AsyncRowsResourceWithStreamingResponse :
364364 def __init__ (self , rows : AsyncRowsResource ) -> None :
365365 self ._rows = rows
366366
367- self .create = async_to_streamed_response_wrapper (
368- rows .create ,
369- )
370367 self .update = async_to_streamed_response_wrapper (
371368 rows .update ,
372369 )
370+ self .list = async_to_streamed_response_wrapper (
371+ rows .list ,
372+ )
0 commit comments