Skip to content

Commit 675f015

Browse files
feat(api): update POST /rows to use list method name
1 parent 9c293e6 commit 675f015

File tree

7 files changed

+196
-199
lines changed

7 files changed

+196
-199
lines changed

.stats.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
configured_endpoints: 28
22
openapi_spec_hash: 5f7962599290c70cb47c05c3b29fdbd8
3-
config_hash: 38b7fc7dfc9af970e9e1cb7b81ccc56b
3+
config_hash: f1c0c034bd832878eb47146c51afdd55

api.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -150,13 +150,13 @@ Methods:
150150
Types:
151151

152152
```python
153-
from openlayer.types.inference_pipelines import RowCreateResponse, RowUpdateResponse
153+
from openlayer.types.inference_pipelines import RowUpdateResponse, RowListResponse
154154
```
155155

156156
Methods:
157157

158-
- <code title="post /inference-pipelines/{inferencePipelineId}/rows">client.inference_pipelines.rows.<a href="./src/openlayer/resources/inference_pipelines/rows.py">create</a>(inference_pipeline_id, \*\*<a href="src/openlayer/types/inference_pipelines/row_create_params.py">params</a>) -> <a href="./src/openlayer/types/inference_pipelines/row_create_response.py">RowCreateResponse</a></code>
159158
- <code title="put /inference-pipelines/{inferencePipelineId}/rows">client.inference_pipelines.rows.<a href="./src/openlayer/resources/inference_pipelines/rows.py">update</a>(inference_pipeline_id, \*\*<a href="src/openlayer/types/inference_pipelines/row_update_params.py">params</a>) -> <a href="./src/openlayer/types/inference_pipelines/row_update_response.py">RowUpdateResponse</a></code>
159+
- <code title="post /inference-pipelines/{inferencePipelineId}/rows">client.inference_pipelines.rows.<a href="./src/openlayer/resources/inference_pipelines/rows.py">list</a>(inference_pipeline_id, \*\*<a href="src/openlayer/types/inference_pipelines/row_list_params.py">params</a>) -> <a href="./src/openlayer/types/inference_pipelines/row_list_response.py">RowListResponse</a></code>
160160

161161
## TestResults
162162

src/openlayer/resources/inference_pipelines/rows.py

Lines changed: 102 additions & 102 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@
1717
async_to_streamed_response_wrapper,
1818
)
1919
from ..._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
2222
from ...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

327327
class 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

339339
class 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

351351
class 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

363363
class 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+
)

src/openlayer/types/inference_pipelines/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@
22

33
from __future__ import annotations
44

5-
from .row_create_params import RowCreateParams as RowCreateParams
5+
from .row_list_params import RowListParams as RowListParams
6+
from .row_list_response import RowListResponse as RowListResponse
67
from .row_update_params import RowUpdateParams as RowUpdateParams
78
from .data_stream_params import DataStreamParams as DataStreamParams
8-
from .row_create_response import RowCreateResponse as RowCreateResponse
99
from .row_update_response import RowUpdateResponse as RowUpdateResponse
1010
from .data_stream_response import DataStreamResponse as DataStreamResponse
1111
from .test_result_list_params import TestResultListParams as TestResultListParams

src/openlayer/types/inference_pipelines/row_create_params.py renamed to src/openlayer/types/inference_pipelines/row_list_params.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,15 +9,15 @@
99
from ..._utils import PropertyInfo
1010

1111
__all__ = [
12-
"RowCreateParams",
12+
"RowListParams",
1313
"ColumnFilter",
1414
"ColumnFilterSetColumnFilter",
1515
"ColumnFilterNumericColumnFilter",
1616
"ColumnFilterStringColumnFilter",
1717
]
1818

1919

20-
class RowCreateParams(TypedDict, total=False):
20+
class RowListParams(TypedDict, total=False):
2121
asc: bool
2222
"""Whether or not to sort on the sortColumn in ascending order."""
2323

src/openlayer/types/inference_pipelines/row_create_response.py renamed to src/openlayer/types/inference_pipelines/row_list_response.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,12 +4,12 @@
44

55
from ..._models import BaseModel
66

7-
__all__ = ["RowCreateResponse", "Item"]
7+
__all__ = ["RowListResponse", "Item"]
88

99

1010
class Item(BaseModel):
1111
openlayer_row_id: int
1212

1313

14-
class RowCreateResponse(BaseModel):
14+
class RowListResponse(BaseModel):
1515
items: List[Item]

0 commit comments

Comments
 (0)