99
1010from openlayer import Openlayer , AsyncOpenlayer
1111from tests .utils import assert_matches_type
12- from openlayer .types .inference_pipelines import RowStreamResponse
12+ from openlayer .types .inference_pipelines import RowUpdateResponse
1313
1414base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
1515
@@ -18,17 +18,17 @@ class TestRows:
1818 parametrize = pytest .mark .parametrize ("client" , [False , True ], indirect = True , ids = ["loose" , "strict" ])
1919
2020 @parametrize
21- def test_method_stream (self , client : Openlayer ) -> None :
22- row = client .inference_pipelines .rows .stream (
21+ def test_method_update (self , client : Openlayer ) -> None :
22+ row = client .inference_pipelines .rows .update (
2323 inference_pipeline_id = "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
2424 inference_id = "inferenceId" ,
2525 row = {},
2626 )
27- assert_matches_type (RowStreamResponse , row , path = ["response" ])
27+ assert_matches_type (RowUpdateResponse , row , path = ["response" ])
2828
2929 @parametrize
30- def test_method_stream_with_all_params (self , client : Openlayer ) -> None :
31- row = client .inference_pipelines .rows .stream (
30+ def test_method_update_with_all_params (self , client : Openlayer ) -> None :
31+ row = client .inference_pipelines .rows .update (
3232 inference_pipeline_id = "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
3333 inference_id = "inferenceId" ,
3434 row = {},
@@ -40,11 +40,11 @@ def test_method_stream_with_all_params(self, client: Openlayer) -> None:
4040 "human_feedback_column_name" : "human_feedback" ,
4141 },
4242 )
43- assert_matches_type (RowStreamResponse , row , path = ["response" ])
43+ assert_matches_type (RowUpdateResponse , row , path = ["response" ])
4444
4545 @parametrize
46- def test_raw_response_stream (self , client : Openlayer ) -> None :
47- response = client .inference_pipelines .rows .with_raw_response .stream (
46+ def test_raw_response_update (self , client : Openlayer ) -> None :
47+ response = client .inference_pipelines .rows .with_raw_response .update (
4848 inference_pipeline_id = "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
4949 inference_id = "inferenceId" ,
5050 row = {},
@@ -53,11 +53,11 @@ def test_raw_response_stream(self, client: Openlayer) -> None:
5353 assert response .is_closed is True
5454 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
5555 row = response .parse ()
56- assert_matches_type (RowStreamResponse , row , path = ["response" ])
56+ assert_matches_type (RowUpdateResponse , row , path = ["response" ])
5757
5858 @parametrize
59- def test_streaming_response_stream (self , client : Openlayer ) -> None :
60- with client .inference_pipelines .rows .with_streaming_response .stream (
59+ def test_streaming_response_update (self , client : Openlayer ) -> None :
60+ with client .inference_pipelines .rows .with_streaming_response .update (
6161 inference_pipeline_id = "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
6262 inference_id = "inferenceId" ,
6363 row = {},
@@ -66,14 +66,14 @@ def test_streaming_response_stream(self, client: Openlayer) -> None:
6666 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
6767
6868 row = response .parse ()
69- assert_matches_type (RowStreamResponse , row , path = ["response" ])
69+ assert_matches_type (RowUpdateResponse , row , path = ["response" ])
7070
7171 assert cast (Any , response .is_closed ) is True
7272
7373 @parametrize
74- def test_path_params_stream (self , client : Openlayer ) -> None :
74+ def test_path_params_update (self , client : Openlayer ) -> None :
7575 with pytest .raises (ValueError , match = r"Expected a non-empty value for `inference_pipeline_id` but received ''" ):
76- client .inference_pipelines .rows .with_raw_response .stream (
76+ client .inference_pipelines .rows .with_raw_response .update (
7777 inference_pipeline_id = "" ,
7878 inference_id = "inferenceId" ,
7979 row = {},
@@ -84,17 +84,17 @@ class TestAsyncRows:
8484 parametrize = pytest .mark .parametrize ("async_client" , [False , True ], indirect = True , ids = ["loose" , "strict" ])
8585
8686 @parametrize
87- async def test_method_stream (self , async_client : AsyncOpenlayer ) -> None :
88- row = await async_client .inference_pipelines .rows .stream (
87+ async def test_method_update (self , async_client : AsyncOpenlayer ) -> None :
88+ row = await async_client .inference_pipelines .rows .update (
8989 inference_pipeline_id = "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
9090 inference_id = "inferenceId" ,
9191 row = {},
9292 )
93- assert_matches_type (RowStreamResponse , row , path = ["response" ])
93+ assert_matches_type (RowUpdateResponse , row , path = ["response" ])
9494
9595 @parametrize
96- async def test_method_stream_with_all_params (self , async_client : AsyncOpenlayer ) -> None :
97- row = await async_client .inference_pipelines .rows .stream (
96+ async def test_method_update_with_all_params (self , async_client : AsyncOpenlayer ) -> None :
97+ row = await async_client .inference_pipelines .rows .update (
9898 inference_pipeline_id = "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
9999 inference_id = "inferenceId" ,
100100 row = {},
@@ -106,11 +106,11 @@ async def test_method_stream_with_all_params(self, async_client: AsyncOpenlayer)
106106 "human_feedback_column_name" : "human_feedback" ,
107107 },
108108 )
109- assert_matches_type (RowStreamResponse , row , path = ["response" ])
109+ assert_matches_type (RowUpdateResponse , row , path = ["response" ])
110110
111111 @parametrize
112- async def test_raw_response_stream (self , async_client : AsyncOpenlayer ) -> None :
113- response = await async_client .inference_pipelines .rows .with_raw_response .stream (
112+ async def test_raw_response_update (self , async_client : AsyncOpenlayer ) -> None :
113+ response = await async_client .inference_pipelines .rows .with_raw_response .update (
114114 inference_pipeline_id = "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
115115 inference_id = "inferenceId" ,
116116 row = {},
@@ -119,11 +119,11 @@ async def test_raw_response_stream(self, async_client: AsyncOpenlayer) -> None:
119119 assert response .is_closed is True
120120 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
121121 row = await response .parse ()
122- assert_matches_type (RowStreamResponse , row , path = ["response" ])
122+ assert_matches_type (RowUpdateResponse , row , path = ["response" ])
123123
124124 @parametrize
125- async def test_streaming_response_stream (self , async_client : AsyncOpenlayer ) -> None :
126- async with async_client .inference_pipelines .rows .with_streaming_response .stream (
125+ async def test_streaming_response_update (self , async_client : AsyncOpenlayer ) -> None :
126+ async with async_client .inference_pipelines .rows .with_streaming_response .update (
127127 inference_pipeline_id = "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
128128 inference_id = "inferenceId" ,
129129 row = {},
@@ -132,14 +132,14 @@ async def test_streaming_response_stream(self, async_client: AsyncOpenlayer) ->
132132 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
133133
134134 row = await response .parse ()
135- assert_matches_type (RowStreamResponse , row , path = ["response" ])
135+ assert_matches_type (RowUpdateResponse , row , path = ["response" ])
136136
137137 assert cast (Any , response .is_closed ) is True
138138
139139 @parametrize
140- async def test_path_params_stream (self , async_client : AsyncOpenlayer ) -> None :
140+ async def test_path_params_update (self , async_client : AsyncOpenlayer ) -> None :
141141 with pytest .raises (ValueError , match = r"Expected a non-empty value for `inference_pipeline_id` but received ''" ):
142- await async_client .inference_pipelines .rows .with_raw_response .stream (
142+ await async_client .inference_pipelines .rows .with_raw_response .update (
143143 inference_pipeline_id = "" ,
144144 inference_id = "inferenceId" ,
145145 row = {},
0 commit comments