|
14 | 14 | DocumentGetResponse, |
15 | 15 | DocumentListResponse, |
16 | 16 | DocumentUpdateResponse, |
| 17 | + DocumentBatchAddResponse, |
17 | 18 | DocumentDeleteBulkResponse, |
18 | 19 | DocumentUploadFileResponse, |
19 | | - DocumentBatchCreateResponse, |
20 | 20 | DocumentListProcessingResponse, |
21 | 21 | ) |
22 | 22 |
|
@@ -236,16 +236,16 @@ def test_streaming_response_add(self, client: Supermemory) -> None: |
236 | 236 |
|
237 | 237 | @pytest.mark.skip(reason="Prism tests are disabled") |
238 | 238 | @parametrize |
239 | | - def test_method_batch_create(self, client: Supermemory) -> None: |
240 | | - document = client.documents.batch_create( |
| 239 | + def test_method_batch_add(self, client: Supermemory) -> None: |
| 240 | + document = client.documents.batch_add( |
241 | 241 | documents=[{"content": "This is a detailed article about machine learning concepts..."}], |
242 | 242 | ) |
243 | | - assert_matches_type(DocumentBatchCreateResponse, document, path=["response"]) |
| 243 | + assert_matches_type(DocumentBatchAddResponse, document, path=["response"]) |
244 | 244 |
|
245 | 245 | @pytest.mark.skip(reason="Prism tests are disabled") |
246 | 246 | @parametrize |
247 | | - def test_method_batch_create_with_all_params(self, client: Supermemory) -> None: |
248 | | - document = client.documents.batch_create( |
| 247 | + def test_method_batch_add_with_all_params(self, client: Supermemory) -> None: |
| 248 | + document = client.documents.batch_add( |
249 | 249 | documents=[ |
250 | 250 | { |
251 | 251 | "content": "This is a detailed article about machine learning concepts...", |
@@ -274,31 +274,31 @@ def test_method_batch_create_with_all_params(self, client: Supermemory) -> None: |
274 | 274 | "tag_2": "machine-learning", |
275 | 275 | }, |
276 | 276 | ) |
277 | | - assert_matches_type(DocumentBatchCreateResponse, document, path=["response"]) |
| 277 | + assert_matches_type(DocumentBatchAddResponse, document, path=["response"]) |
278 | 278 |
|
279 | 279 | @pytest.mark.skip(reason="Prism tests are disabled") |
280 | 280 | @parametrize |
281 | | - def test_raw_response_batch_create(self, client: Supermemory) -> None: |
282 | | - response = client.documents.with_raw_response.batch_create( |
| 281 | + def test_raw_response_batch_add(self, client: Supermemory) -> None: |
| 282 | + response = client.documents.with_raw_response.batch_add( |
283 | 283 | documents=[{"content": "This is a detailed article about machine learning concepts..."}], |
284 | 284 | ) |
285 | 285 |
|
286 | 286 | assert response.is_closed is True |
287 | 287 | assert response.http_request.headers.get("X-Stainless-Lang") == "python" |
288 | 288 | document = response.parse() |
289 | | - assert_matches_type(DocumentBatchCreateResponse, document, path=["response"]) |
| 289 | + assert_matches_type(DocumentBatchAddResponse, document, path=["response"]) |
290 | 290 |
|
291 | 291 | @pytest.mark.skip(reason="Prism tests are disabled") |
292 | 292 | @parametrize |
293 | | - def test_streaming_response_batch_create(self, client: Supermemory) -> None: |
294 | | - with client.documents.with_streaming_response.batch_create( |
| 293 | + def test_streaming_response_batch_add(self, client: Supermemory) -> None: |
| 294 | + with client.documents.with_streaming_response.batch_add( |
295 | 295 | documents=[{"content": "This is a detailed article about machine learning concepts..."}], |
296 | 296 | ) as response: |
297 | 297 | assert not response.is_closed |
298 | 298 | assert response.http_request.headers.get("X-Stainless-Lang") == "python" |
299 | 299 |
|
300 | 300 | document = response.parse() |
301 | | - assert_matches_type(DocumentBatchCreateResponse, document, path=["response"]) |
| 301 | + assert_matches_type(DocumentBatchAddResponse, document, path=["response"]) |
302 | 302 |
|
303 | 303 | assert cast(Any, response.is_closed) is True |
304 | 304 |
|
@@ -671,16 +671,16 @@ async def test_streaming_response_add(self, async_client: AsyncSupermemory) -> N |
671 | 671 |
|
672 | 672 | @pytest.mark.skip(reason="Prism tests are disabled") |
673 | 673 | @parametrize |
674 | | - async def test_method_batch_create(self, async_client: AsyncSupermemory) -> None: |
675 | | - document = await async_client.documents.batch_create( |
| 674 | + async def test_method_batch_add(self, async_client: AsyncSupermemory) -> None: |
| 675 | + document = await async_client.documents.batch_add( |
676 | 676 | documents=[{"content": "This is a detailed article about machine learning concepts..."}], |
677 | 677 | ) |
678 | | - assert_matches_type(DocumentBatchCreateResponse, document, path=["response"]) |
| 678 | + assert_matches_type(DocumentBatchAddResponse, document, path=["response"]) |
679 | 679 |
|
680 | 680 | @pytest.mark.skip(reason="Prism tests are disabled") |
681 | 681 | @parametrize |
682 | | - async def test_method_batch_create_with_all_params(self, async_client: AsyncSupermemory) -> None: |
683 | | - document = await async_client.documents.batch_create( |
| 682 | + async def test_method_batch_add_with_all_params(self, async_client: AsyncSupermemory) -> None: |
| 683 | + document = await async_client.documents.batch_add( |
684 | 684 | documents=[ |
685 | 685 | { |
686 | 686 | "content": "This is a detailed article about machine learning concepts...", |
@@ -709,31 +709,31 @@ async def test_method_batch_create_with_all_params(self, async_client: AsyncSupe |
709 | 709 | "tag_2": "machine-learning", |
710 | 710 | }, |
711 | 711 | ) |
712 | | - assert_matches_type(DocumentBatchCreateResponse, document, path=["response"]) |
| 712 | + assert_matches_type(DocumentBatchAddResponse, document, path=["response"]) |
713 | 713 |
|
714 | 714 | @pytest.mark.skip(reason="Prism tests are disabled") |
715 | 715 | @parametrize |
716 | | - async def test_raw_response_batch_create(self, async_client: AsyncSupermemory) -> None: |
717 | | - response = await async_client.documents.with_raw_response.batch_create( |
| 716 | + async def test_raw_response_batch_add(self, async_client: AsyncSupermemory) -> None: |
| 717 | + response = await async_client.documents.with_raw_response.batch_add( |
718 | 718 | documents=[{"content": "This is a detailed article about machine learning concepts..."}], |
719 | 719 | ) |
720 | 720 |
|
721 | 721 | assert response.is_closed is True |
722 | 722 | assert response.http_request.headers.get("X-Stainless-Lang") == "python" |
723 | 723 | document = await response.parse() |
724 | | - assert_matches_type(DocumentBatchCreateResponse, document, path=["response"]) |
| 724 | + assert_matches_type(DocumentBatchAddResponse, document, path=["response"]) |
725 | 725 |
|
726 | 726 | @pytest.mark.skip(reason="Prism tests are disabled") |
727 | 727 | @parametrize |
728 | | - async def test_streaming_response_batch_create(self, async_client: AsyncSupermemory) -> None: |
729 | | - async with async_client.documents.with_streaming_response.batch_create( |
| 728 | + async def test_streaming_response_batch_add(self, async_client: AsyncSupermemory) -> None: |
| 729 | + async with async_client.documents.with_streaming_response.batch_add( |
730 | 730 | documents=[{"content": "This is a detailed article about machine learning concepts..."}], |
731 | 731 | ) as response: |
732 | 732 | assert not response.is_closed |
733 | 733 | assert response.http_request.headers.get("X-Stainless-Lang") == "python" |
734 | 734 |
|
735 | 735 | document = await response.parse() |
736 | | - assert_matches_type(DocumentBatchCreateResponse, document, path=["response"]) |
| 736 | + assert_matches_type(DocumentBatchAddResponse, document, path=["response"]) |
737 | 737 |
|
738 | 738 | assert cast(Any, response.is_closed) is True |
739 | 739 |
|
|
0 commit comments