llama-cloud 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

llama_cloud/resources/pipelines/client.py

@@ -18,6 +18,7 @@ from ...types.eval_execution_params import EvalExecutionParams
 from ...types.eval_execution_params_override import EvalExecutionParamsOverride
 from ...types.eval_question_result import EvalQuestionResult
 from ...types.http_validation_error import HttpValidationError
+from ...types.managed_ingestion_status import ManagedIngestionStatus
 from ...types.metadata_filters import MetadataFilters
 from ...types.pipeline import Pipeline
 from ...types.pipeline_create import PipelineCreate
@@ -33,6 +34,9 @@ from ...types.retrieve_results import RetrieveResults
 from .types.pipeline_file_update_custom_metadata_value import PipelineFileUpdateCustomMetadataValue
 
 try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
     import pydantic.v1 as pydantic  # type: ignore
 except ImportError:
     import pydantic  # type: ignore
@@ -62,10 +66,10 @@ class PipelinesClient:
 
             - pipeline_type: typing.Optional[PipelineType].
         ---
-        from platform import PipelineType
-        from platform.client import PlatformApi
+        from llama_cloud import PipelineType
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -102,7 +106,7 @@ class PipelinesClient:
 
             - request: PipelineCreate.
         ---
-        from platform import (
+        from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
             EvalExecutionParams,
@@ -113,9 +117,9 @@ class PipelinesClient:
             PresetRetrievalParams,
             SupportedEvalLlmModelNames,
         )
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -167,7 +171,7 @@ class PipelinesClient:
 
             - request: PipelineCreate.
         ---
-        from platform import (
+        from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
             EvalExecutionParams,
@@ -178,9 +182,9 @@ class PipelinesClient:
             PresetRetrievalParams,
             SupportedEvalLlmModelNames,
         )
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -233,9 +237,9 @@ class PipelinesClient:
 
             - with_managed_ingestion_status: typing.Optional[bool].
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -295,7 +299,7 @@ class PipelinesClient:
 
             - managed_pipeline_id: typing.Optional[str]. The ID of the ManagedPipeline this playground pipeline is linked to.
         ---
-        from platform import (
+        from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
             EvalExecutionParams,
@@ -304,9 +308,9 @@ class PipelinesClient:
             PresetRetrievalParams,
             SupportedEvalLlmModelNames,
         )
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -368,9 +372,9 @@ class PipelinesClient:
         Parameters:
             - pipeline_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -401,9 +405,9 @@ class PipelinesClient:
         Parameters:
             - pipeline_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -436,9 +440,9 @@ class PipelinesClient:
 
             - eval_dataset_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -486,10 +490,10 @@ class PipelinesClient:
 
             - params: typing.Optional[EvalExecutionParamsOverride]. The parameters for the eval execution that will override the ones set in the pipeline.
         ---
-        from platform import EvalExecutionParamsOverride, SupportedEvalLlmModelNames
-        from platform.client import PlatformApi
+        from llama_cloud import EvalExecutionParamsOverride, SupportedEvalLlmModelNames
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -539,9 +543,9 @@ class PipelinesClient:
 
             - eval_dataset_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -582,9 +586,9 @@ class PipelinesClient:
 
             - eval_dataset_execution_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -620,9 +624,9 @@ class PipelinesClient:
         Parameters:
             - pipeline_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -657,9 +661,9 @@ class PipelinesClient:
 
             - request: typing.List[PipelineFileCreate].
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -694,9 +698,9 @@ class PipelinesClient:
 
             - file_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -740,9 +744,9 @@ class PipelinesClient:
 
             - custom_metadata: typing.Optional[typing.Dict[str, PipelineFileUpdateCustomMetadataValue]]. Custom metadata for the file
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -782,9 +786,9 @@ class PipelinesClient:
 
             - file_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -818,9 +822,9 @@ class PipelinesClient:
         Parameters:
             - pipeline_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -857,9 +861,9 @@ class PipelinesClient:
 
             - request: typing.List[PipelineDataSourceCreate].
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -896,9 +900,9 @@ class PipelinesClient:
 
             - data_source_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -935,9 +939,9 @@ class PipelinesClient:
 
             - data_source_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -997,10 +1001,10 @@ class PipelinesClient:
 
             - query: str. The query to retrieve against.
         ---
-        from platform import FilterCondition, MetadataFilters
-        from platform.client import PlatformApi
+        from llama_cloud import FilterCondition, MetadataFilters
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1050,9 +1054,9 @@ class PipelinesClient:
         Parameters:
             - pipeline_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1085,9 +1089,9 @@ class PipelinesClient:
 
             - job_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1127,9 +1131,9 @@ class PipelinesClient:
 
             - limit: typing.Optional[int].
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1156,28 +1160,26 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def create_pipeline_document(self, pipeline_id: str, *, request: CloudDocumentCreate) -> CloudDocument:
+    def create_batch_pipeline_documents(
+        self, pipeline_id: str, *, request: typing.List[CloudDocumentCreate]
+    ) -> typing.List[CloudDocument]:
         """
-        Create a new document for a pipeline.
+        Batch create documents for a pipeline.
 
         Parameters:
             - pipeline_id: str.
 
-            - request: CloudDocumentCreate.
+            - request: typing.List[CloudDocumentCreate].
         ---
-        from platform import CloudDocumentCreate
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.pipelines.create_pipeline_document(
+        client.pipelines.create_batch_pipeline_documents(
             pipeline_id="string",
-            request=CloudDocumentCreate(
-                text="string",
-                metadata={"string": {}},
-            ),
+            request=[],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -1190,7 +1192,7 @@ class PipelinesClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(CloudDocument, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.List[CloudDocument], _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -1199,28 +1201,26 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def upsert_pipeline_document(self, pipeline_id: str, *, request: CloudDocumentCreate) -> CloudDocument:
+    def upsert_batch_pipeline_documents(
+        self, pipeline_id: str, *, request: typing.List[CloudDocumentCreate]
+    ) -> typing.List[CloudDocument]:
         """
-        Create or update a document for a pipeline.
+        Batch create or update a document for a pipeline.
 
         Parameters:
             - pipeline_id: str.
 
-            - request: CloudDocumentCreate.
+            - request: typing.List[CloudDocumentCreate].
         ---
-        from platform import CloudDocumentCreate
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.pipelines.upsert_pipeline_document(
+        client.pipelines.upsert_batch_pipeline_documents(
             pipeline_id="string",
-            request=CloudDocumentCreate(
-                text="string",
-                metadata={"string": {}},
-            ),
+            request=[],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -1233,7 +1233,7 @@ class PipelinesClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(CloudDocument, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.List[CloudDocument], _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -1251,9 +1251,9 @@ class PipelinesClient:
 
             - document_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1289,9 +1289,9 @@ class PipelinesClient:
 
             - document_id: str.
         ---
-        from platform.client import PlatformApi
+        from llama_cloud.client import LlamaCloud
 
-        client = PlatformApi(
+        client = LlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1318,6 +1318,45 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def get_pipeline_document_status(self, pipeline_id: str, document_id: str) -> ManagedIngestionStatus:
+        """
+        Return a single document for a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - document_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.pipelines.get_pipeline_document_status(
+            pipeline_id="string",
+            document_id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/",
+                f"api/v1/pipelines/{pipeline_id}/documents/{document_id}/status",
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ManagedIngestionStatus, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
 
 class AsyncPipelinesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -1340,10 +1379,10 @@ class AsyncPipelinesClient:
 
             - pipeline_type: typing.Optional[PipelineType].
         ---
-        from platform import PipelineType
-        from platform.client import AsyncPlatformApi
+        from llama_cloud import PipelineType
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1380,7 +1419,7 @@ class AsyncPipelinesClient:
 
             - request: PipelineCreate.
         ---
-        from platform import (
+        from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
             EvalExecutionParams,
@@ -1391,9 +1430,9 @@ class AsyncPipelinesClient:
             PresetRetrievalParams,
             SupportedEvalLlmModelNames,
         )
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1445,7 +1484,7 @@ class AsyncPipelinesClient:
 
             - request: PipelineCreate.
         ---
-        from platform import (
+        from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
             EvalExecutionParams,
@@ -1456,9 +1495,9 @@ class AsyncPipelinesClient:
             PresetRetrievalParams,
             SupportedEvalLlmModelNames,
         )
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1511,9 +1550,9 @@ class AsyncPipelinesClient:
 
             - with_managed_ingestion_status: typing.Optional[bool].
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1573,7 +1612,7 @@ class AsyncPipelinesClient:
 
             - managed_pipeline_id: typing.Optional[str]. The ID of the ManagedPipeline this playground pipeline is linked to.
         ---
-        from platform import (
+        from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
             EvalExecutionParams,
@@ -1582,9 +1621,9 @@ class AsyncPipelinesClient:
             PresetRetrievalParams,
             SupportedEvalLlmModelNames,
         )
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1646,9 +1685,9 @@ class AsyncPipelinesClient:
         Parameters:
             - pipeline_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1679,9 +1718,9 @@ class AsyncPipelinesClient:
         Parameters:
             - pipeline_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1716,9 +1755,9 @@ class AsyncPipelinesClient:
 
             - eval_dataset_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1766,10 +1805,10 @@ class AsyncPipelinesClient:
 
             - params: typing.Optional[EvalExecutionParamsOverride]. The parameters for the eval execution that will override the ones set in the pipeline.
         ---
-        from platform import EvalExecutionParamsOverride, SupportedEvalLlmModelNames
-        from platform.client import AsyncPlatformApi
+        from llama_cloud import EvalExecutionParamsOverride, SupportedEvalLlmModelNames
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1819,9 +1858,9 @@ class AsyncPipelinesClient:
 
             - eval_dataset_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1862,9 +1901,9 @@ class AsyncPipelinesClient:
 
             - eval_dataset_execution_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1900,9 +1939,9 @@ class AsyncPipelinesClient:
         Parameters:
             - pipeline_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1937,9 +1976,9 @@ class AsyncPipelinesClient:
 
             - request: typing.List[PipelineFileCreate].
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -1974,9 +2013,9 @@ class AsyncPipelinesClient:
 
             - file_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2020,9 +2059,9 @@ class AsyncPipelinesClient:
 
             - custom_metadata: typing.Optional[typing.Dict[str, PipelineFileUpdateCustomMetadataValue]]. Custom metadata for the file
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2062,9 +2101,9 @@ class AsyncPipelinesClient:
 
             - file_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2098,9 +2137,9 @@ class AsyncPipelinesClient:
         Parameters:
             - pipeline_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2137,9 +2176,9 @@ class AsyncPipelinesClient:
 
             - request: typing.List[PipelineDataSourceCreate].
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2176,9 +2215,9 @@ class AsyncPipelinesClient:
 
             - data_source_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2215,9 +2254,9 @@ class AsyncPipelinesClient:
 
             - data_source_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2277,10 +2316,10 @@ class AsyncPipelinesClient:
 
             - query: str. The query to retrieve against.
         ---
-        from platform import FilterCondition, MetadataFilters
-        from platform.client import AsyncPlatformApi
+        from llama_cloud import FilterCondition, MetadataFilters
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2330,9 +2369,9 @@ class AsyncPipelinesClient:
         Parameters:
             - pipeline_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2365,9 +2404,9 @@ class AsyncPipelinesClient:
 
             - job_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2407,9 +2446,9 @@ class AsyncPipelinesClient:
 
             - limit: typing.Optional[int].
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2436,28 +2475,26 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def create_pipeline_document(self, pipeline_id: str, *, request: CloudDocumentCreate) -> CloudDocument:
+    async def create_batch_pipeline_documents(
+        self, pipeline_id: str, *, request: typing.List[CloudDocumentCreate]
+    ) -> typing.List[CloudDocument]:
         """
-        Create a new document for a pipeline.
+        Batch create documents for a pipeline.
 
         Parameters:
             - pipeline_id: str.
 
-            - request: CloudDocumentCreate.
+            - request: typing.List[CloudDocumentCreate].
         ---
-        from platform import CloudDocumentCreate
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
-        await client.pipelines.create_pipeline_document(
+        await client.pipelines.create_batch_pipeline_documents(
             pipeline_id="string",
-            request=CloudDocumentCreate(
-                text="string",
-                metadata={"string": {}},
-            ),
+            request=[],
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -2470,7 +2507,7 @@ class AsyncPipelinesClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(CloudDocument, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.List[CloudDocument], _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -2479,28 +2516,26 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def upsert_pipeline_document(self, pipeline_id: str, *, request: CloudDocumentCreate) -> CloudDocument:
+    async def upsert_batch_pipeline_documents(
+        self, pipeline_id: str, *, request: typing.List[CloudDocumentCreate]
+    ) -> typing.List[CloudDocument]:
         """
-        Create or update a document for a pipeline.
+        Batch create or update a document for a pipeline.
 
         Parameters:
             - pipeline_id: str.
 
-            - request: CloudDocumentCreate.
+            - request: typing.List[CloudDocumentCreate].
         ---
-        from platform import CloudDocumentCreate
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
-        await client.pipelines.upsert_pipeline_document(
+        await client.pipelines.upsert_batch_pipeline_documents(
             pipeline_id="string",
-            request=CloudDocumentCreate(
-                text="string",
-                metadata={"string": {}},
-            ),
+            request=[],
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -2513,7 +2548,7 @@ class AsyncPipelinesClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(CloudDocument, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.List[CloudDocument], _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -2531,9 +2566,9 @@ class AsyncPipelinesClient:
 
             - document_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2569,9 +2604,9 @@ class AsyncPipelinesClient:
 
             - document_id: str.
         ---
-        from platform.client import AsyncPlatformApi
+        from llama_cloud.client import AsyncLlamaCloud
 
-        client = AsyncPlatformApi(
+        client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
             base_url="https://yourhost.com/path/to/api",
         )
@@ -2597,3 +2632,42 @@ class AsyncPipelinesClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def get_pipeline_document_status(self, pipeline_id: str, document_id: str) -> ManagedIngestionStatus:
+        """
+        Return a single document for a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - document_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        await client.pipelines.get_pipeline_document_status(
+            pipeline_id="string",
+            document_id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/",
+                f"api/v1/pipelines/{pipeline_id}/documents/{document_id}/status",
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ManagedIngestionStatus, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)