|
14 | 14 | from ..core.serialization import convert_and_respect_annotation_metadata |
15 | 15 | from .types.create_collection_response import CreateCollectionResponse |
16 | 16 | from .types.delete_collection_response import DeleteCollectionResponse |
| 17 | +from .types.list_documents_in_collection_response import ListDocumentsInCollectionResponse |
17 | 18 | from .types.mongo_collection_sync_config import MongoCollectionSyncConfig |
18 | 19 | from .types.mongo_collection_sync_response import MongoCollectionSyncResponse |
19 | 20 |
|
@@ -115,6 +116,47 @@ def remove_documents_from_collection( |
115 | 116 | raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) |
116 | 117 | raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) |
117 | 118 |
|
| 119 | + def list_documents_in_collection( |
| 120 | + self, *, collection_id: uuid.UUID, request_options: typing.Optional[RequestOptions] = None |
| 121 | + ) -> HttpResponse[ListDocumentsInCollectionResponse]: |
| 122 | + """ |
| 123 | + List documents in a collection |
| 124 | +
|
| 125 | + Parameters |
| 126 | + ---------- |
| 127 | + collection_id : uuid.UUID |
| 128 | + The ID of the document collection to list documents from. |
| 129 | +
|
| 130 | + request_options : typing.Optional[RequestOptions] |
| 131 | + Request-specific configuration. |
| 132 | +
|
| 133 | + Returns |
| 134 | + ------- |
| 135 | + HttpResponse[ListDocumentsInCollectionResponse] |
| 136 | + """ |
| 137 | + _response = self._client_wrapper.httpx_client.request( |
| 138 | + "v0/documentCollections/listDocumentsInCollection", |
| 139 | + method="GET", |
| 140 | + params={ |
| 141 | + "collectionId": collection_id, |
| 142 | + }, |
| 143 | + request_options=request_options, |
| 144 | + ) |
| 145 | + try: |
| 146 | + if 200 <= _response.status_code < 300: |
| 147 | + _data = typing.cast( |
| 148 | + ListDocumentsInCollectionResponse, |
| 149 | + parse_obj_as( |
| 150 | + type_=ListDocumentsInCollectionResponse, # type: ignore |
| 151 | + object_=_response.json(), |
| 152 | + ), |
| 153 | + ) |
| 154 | + return HttpResponse(response=_response, data=_data) |
| 155 | + _response_json = _response.json() |
| 156 | + except JSONDecodeError: |
| 157 | + raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) |
| 158 | + raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) |
| 159 | + |
118 | 160 | def create_collection( |
119 | 161 | self, |
120 | 162 | *, |
@@ -416,6 +458,47 @@ async def remove_documents_from_collection( |
416 | 458 | raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) |
417 | 459 | raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) |
418 | 460 |
|
| 461 | + async def list_documents_in_collection( |
| 462 | + self, *, collection_id: uuid.UUID, request_options: typing.Optional[RequestOptions] = None |
| 463 | + ) -> AsyncHttpResponse[ListDocumentsInCollectionResponse]: |
| 464 | + """ |
| 465 | + List documents in a collection |
| 466 | +
|
| 467 | + Parameters |
| 468 | + ---------- |
| 469 | + collection_id : uuid.UUID |
| 470 | + The ID of the document collection to list documents from. |
| 471 | +
|
| 472 | + request_options : typing.Optional[RequestOptions] |
| 473 | + Request-specific configuration. |
| 474 | +
|
| 475 | + Returns |
| 476 | + ------- |
| 477 | + AsyncHttpResponse[ListDocumentsInCollectionResponse] |
| 478 | + """ |
| 479 | + _response = await self._client_wrapper.httpx_client.request( |
| 480 | + "v0/documentCollections/listDocumentsInCollection", |
| 481 | + method="GET", |
| 482 | + params={ |
| 483 | + "collectionId": collection_id, |
| 484 | + }, |
| 485 | + request_options=request_options, |
| 486 | + ) |
| 487 | + try: |
| 488 | + if 200 <= _response.status_code < 300: |
| 489 | + _data = typing.cast( |
| 490 | + ListDocumentsInCollectionResponse, |
| 491 | + parse_obj_as( |
| 492 | + type_=ListDocumentsInCollectionResponse, # type: ignore |
| 493 | + object_=_response.json(), |
| 494 | + ), |
| 495 | + ) |
| 496 | + return AsyncHttpResponse(response=_response, data=_data) |
| 497 | + _response_json = _response.json() |
| 498 | + except JSONDecodeError: |
| 499 | + raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) |
| 500 | + raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) |
| 501 | + |
419 | 502 | async def create_collection( |
420 | 503 | self, |
421 | 504 | *, |
|
0 commit comments