llama-cloud 0.0.9__py3-none-any.whl → 0.0.11__py3-none-any.whl

llama_cloud/resources/pipelines/client.py

@@ -9,11 +9,13 @@ from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
 from ...core.remove_none_from_dict import remove_none_from_dict
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
-from ...types.chat_params import ChatParams
+from ...types.chat_data import ChatData
+from ...types.chat_message import ChatMessage
 from ...types.cloud_document import CloudDocument
 from ...types.cloud_document_create import CloudDocumentCreate
 from ...types.configured_transformation_item import ConfiguredTransformationItem
 from ...types.data_sink_create import DataSinkCreate
+from ...types.embedding_config import EmbeddingConfig
 from ...types.eval_dataset_job_record import EvalDatasetJobRecord
 from ...types.eval_execution_params import EvalExecutionParams
 from ...types.eval_execution_params_override import EvalExecutionParamsOverride
@@ -33,6 +35,8 @@ from ...types.pipeline_type import PipelineType
 from ...types.preset_retrieval_params import PresetRetrievalParams
 from ...types.retrieval_mode import RetrievalMode
 from ...types.retrieve_results import RetrieveResults
+from ...types.text_node import TextNode
+from ...types.transform_config import TransformConfig
 from .types.pipeline_file_update_custom_metadata_value import PipelineFileUpdateCustomMetadataValue
 
 try:
@@ -115,8 +119,11 @@ class PipelinesClient:
             - request: PipelineCreate.
         ---
         from llama_cloud import (
+            AutoTransformConfig,
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -126,6 +133,8 @@ class PipelinesClient:
             PresetRetrievalParams,
             RetrievalMode,
             SupportedEvalLlmModelNames,
+            TransformConfig,
+            TransformConfigMode,
         )
         from llama_cloud.client import LlamaCloud
 
@@ -134,6 +143,13 @@ class PipelinesClient:
         )
         client.pipelines.create_pipeline(
             request=PipelineCreate(
+                embedding_config=EmbeddingConfig(
+                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
+                ),
+                transform_config=TransformConfig(
+                    mode=TransformConfigMode.AUTO,
+                    config=AutoTransformConfig(),
+                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -183,8 +199,11 @@ class PipelinesClient:
             - request: PipelineCreate.
         ---
         from llama_cloud import (
+            AutoTransformConfig,
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -194,6 +213,8 @@ class PipelinesClient:
             PresetRetrievalParams,
             RetrievalMode,
             SupportedEvalLlmModelNames,
+            TransformConfig,
+            TransformConfigMode,
         )
         from llama_cloud.client import LlamaCloud
 
@@ -202,6 +223,13 @@ class PipelinesClient:
         )
         client.pipelines.upsert_pipeline(
             request=PipelineCreate(
+                embedding_config=EmbeddingConfig(
+                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
+                ),
+                transform_config=TransformConfig(
+                    mode=TransformConfigMode.AUTO,
+                    config=AutoTransformConfig(),
+                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -276,12 +304,13 @@ class PipelinesClient:
         self,
         pipeline_id: str,
         *,
+        embedding_config: typing.Optional[EmbeddingConfig] = OMIT,
+        transform_config: typing.Optional[TransformConfig] = OMIT,
         configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]] = OMIT,
         data_sink_id: typing.Optional[str] = OMIT,
         data_sink: typing.Optional[DataSinkCreate] = OMIT,
         preset_retrieval_parameters: typing.Optional[PresetRetrievalParams] = OMIT,
         eval_parameters: typing.Optional[EvalExecutionParams] = OMIT,
-        llama_parse_enabled: typing.Optional[bool] = OMIT,
         llama_parse_parameters: typing.Optional[LlamaParseParameters] = OMIT,
         name: typing.Optional[str] = OMIT,
         managed_pipeline_id: typing.Optional[str] = OMIT,
@@ -292,7 +321,11 @@ class PipelinesClient:
         Parameters:
             - pipeline_id: str.
 
-            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]].
+            - embedding_config: typing.Optional[EmbeddingConfig]. Configuration for the embedding model.
+
+            - transform_config: typing.Optional[TransformConfig]. Configuration for the transformation.
+
+            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]]. List of configured transformations.
 
             - data_sink_id: typing.Optional[str]. Data sink ID. When provided instead of data_sink, the data sink will be looked up by ID.
 
@@ -302,8 +335,6 @@ class PipelinesClient:
 
             - eval_parameters: typing.Optional[EvalExecutionParams]. Eval parameters for the pipeline.
 
-            - llama_parse_enabled: typing.Optional[bool]. Whether to use LlamaParse during pipeline execution.
-
             - llama_parse_parameters: typing.Optional[LlamaParseParameters]. Settings that can be configured for how to use LlamaParse to parse files within a LlamaCloud pipeline.
 
             - name: typing.Optional[str].
@@ -311,8 +342,11 @@ class PipelinesClient:
             - managed_pipeline_id: typing.Optional[str]. The ID of the ManagedPipeline this playground pipeline is linked to.
         ---
         from llama_cloud import (
+            AutoTransformConfig,
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -320,6 +354,8 @@ class PipelinesClient:
             PresetRetrievalParams,
             RetrievalMode,
             SupportedEvalLlmModelNames,
+            TransformConfig,
+            TransformConfigMode,
         )
         from llama_cloud.client import LlamaCloud
 
@@ -328,6 +364,13 @@ class PipelinesClient:
         )
         client.pipelines.update_existing_pipeline(
             pipeline_id="string",
+            embedding_config=EmbeddingConfig(
+                type=EmbeddingConfigType.OPENAI_EMBEDDING,
+            ),
+            transform_config=TransformConfig(
+                mode=TransformConfigMode.AUTO,
+                config=AutoTransformConfig(),
+            ),
             data_sink=DataSinkCreate(
                 name="string",
                 sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -346,6 +389,10 @@ class PipelinesClient:
         )
         """
         _request: typing.Dict[str, typing.Any] = {}
+        if embedding_config is not OMIT:
+            _request["embedding_config"] = embedding_config
+        if transform_config is not OMIT:
+            _request["transform_config"] = transform_config
         if configured_transformations is not OMIT:
             _request["configured_transformations"] = configured_transformations
         if data_sink_id is not OMIT:
@@ -356,8 +403,6 @@ class PipelinesClient:
             _request["preset_retrieval_parameters"] = preset_retrieval_parameters
         if eval_parameters is not OMIT:
             _request["eval_parameters"] = eval_parameters
-        if llama_parse_enabled is not OMIT:
-            _request["llama_parse_enabled"] = llama_parse_enabled
         if llama_parse_parameters is not OMIT:
             _request["llama_parse_parameters"] = llama_parse_parameters
         if name is not OMIT:
@@ -477,6 +522,38 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def copy_pipeline(self, pipeline_id: str) -> Pipeline:
+        """
+        Copy a pipeline by ID.
+
+        Parameters:
+            - pipeline_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.copy_pipeline(
+            pipeline_id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/copy"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(Pipeline, _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)
+
     def get_eval_dataset_executions(self, pipeline_id: str, eval_dataset_id: str) -> typing.List[EvalDatasetJobRecord]:
         """
         Get the status of an EvalDatasetExecution.
@@ -1231,17 +1308,26 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def chat(self, pipeline_id: str, *, retrieval_parameters: PresetRetrievalParams, data: ChatParams) -> typing.Any:
+    def chat(
+        self,
+        pipeline_id: str,
+        *,
+        messages: typing.List[ChatMessage],
+        data: ChatData,
+        class_name: typing.Optional[str] = OMIT,
+    ) -> typing.Any:
         """
         Parameters:
             - pipeline_id: str.
 
-            - retrieval_parameters: PresetRetrievalParams.
+            - messages: typing.List[ChatMessage].
+
+            - data: ChatData.
 
-            - data: ChatParams.
+            - class_name: typing.Optional[str].
         ---
         from llama_cloud import (
-            ChatParams,
+            ChatData,
             FilterCondition,
             MetadataFilters,
             PresetRetrievalParams,
@@ -1254,22 +1340,25 @@ class PipelinesClient:
         )
         client.pipelines.chat(
             pipeline_id="string",
-            retrieval_parameters=PresetRetrievalParams(
-                search_filters=MetadataFilters(
-                    filters=[],
-                    condition=FilterCondition.AND,
+            messages=[],
+            data=ChatData(
+                retrieval_parameters=PresetRetrievalParams(
+                    search_filters=MetadataFilters(
+                        filters=[],
+                        condition=FilterCondition.AND,
+                    ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
                 ),
-                retrieval_mode=RetrievalMode.CHUNKS,
-            ),
-            data=ChatParams(
-                messages=[],
             ),
         )
         """
+        _request: typing.Dict[str, typing.Any] = {"messages": messages, "data": data}
+        if class_name is not OMIT:
+            _request["class_name"] = class_name
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/chat"),
-            json=jsonable_encoder({"retrieval_parameters": retrieval_parameters, "data": data}),
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1284,7 +1373,13 @@ class PipelinesClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def list_pipeline_documents(
-        self, pipeline_id: str, *, skip: typing.Optional[int] = None, limit: typing.Optional[int] = None
+        self,
+        pipeline_id: str,
+        *,
+        skip: typing.Optional[int] = None,
+        limit: typing.Optional[int] = None,
+        file_id: typing.Optional[str] = None,
+        only_direct_upload: typing.Optional[bool] = None,
     ) -> typing.List[CloudDocument]:
         """
         Return a list of documents for a pipeline.
@@ -1295,6 +1390,10 @@ class PipelinesClient:
             - skip: typing.Optional[int].
 
             - limit: typing.Optional[int].
+
+            - file_id: typing.Optional[str].
+
+            - only_direct_upload: typing.Optional[bool].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -1310,7 +1409,9 @@ class PipelinesClient:
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/documents"
             ),
-            params=remove_none_from_dict({"skip": skip, "limit": limit}),
+            params=remove_none_from_dict(
+                {"skip": skip, "limit": limit, "file_id": file_id, "only_direct_upload": only_direct_upload}
+            ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1516,6 +1617,44 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def list_pipeline_document_chunks(self, pipeline_id: str, document_id: str) -> typing.List[TextNode]:
+        """
+        Return a list of chunks for a pipeline document.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - document_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.list_pipeline_document_chunks(
+            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}/chunks",
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[TextNode], _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):
@@ -1585,8 +1724,11 @@ class AsyncPipelinesClient:
             - request: PipelineCreate.
         ---
         from llama_cloud import (
+            AutoTransformConfig,
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -1596,6 +1738,8 @@ class AsyncPipelinesClient:
             PresetRetrievalParams,
             RetrievalMode,
             SupportedEvalLlmModelNames,
+            TransformConfig,
+            TransformConfigMode,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1604,6 +1748,13 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.create_pipeline(
             request=PipelineCreate(
+                embedding_config=EmbeddingConfig(
+                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
+                ),
+                transform_config=TransformConfig(
+                    mode=TransformConfigMode.AUTO,
+                    config=AutoTransformConfig(),
+                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -1653,8 +1804,11 @@ class AsyncPipelinesClient:
             - request: PipelineCreate.
         ---
         from llama_cloud import (
+            AutoTransformConfig,
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -1664,6 +1818,8 @@ class AsyncPipelinesClient:
             PresetRetrievalParams,
             RetrievalMode,
             SupportedEvalLlmModelNames,
+            TransformConfig,
+            TransformConfigMode,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1672,6 +1828,13 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.upsert_pipeline(
             request=PipelineCreate(
+                embedding_config=EmbeddingConfig(
+                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
+                ),
+                transform_config=TransformConfig(
+                    mode=TransformConfigMode.AUTO,
+                    config=AutoTransformConfig(),
+                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -1746,12 +1909,13 @@ class AsyncPipelinesClient:
         self,
         pipeline_id: str,
         *,
+        embedding_config: typing.Optional[EmbeddingConfig] = OMIT,
+        transform_config: typing.Optional[TransformConfig] = OMIT,
         configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]] = OMIT,
         data_sink_id: typing.Optional[str] = OMIT,
         data_sink: typing.Optional[DataSinkCreate] = OMIT,
         preset_retrieval_parameters: typing.Optional[PresetRetrievalParams] = OMIT,
         eval_parameters: typing.Optional[EvalExecutionParams] = OMIT,
-        llama_parse_enabled: typing.Optional[bool] = OMIT,
         llama_parse_parameters: typing.Optional[LlamaParseParameters] = OMIT,
         name: typing.Optional[str] = OMIT,
         managed_pipeline_id: typing.Optional[str] = OMIT,
@@ -1762,7 +1926,11 @@ class AsyncPipelinesClient:
         Parameters:
             - pipeline_id: str.
 
-            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]].
+            - embedding_config: typing.Optional[EmbeddingConfig]. Configuration for the embedding model.
+
+            - transform_config: typing.Optional[TransformConfig]. Configuration for the transformation.
+
+            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]]. List of configured transformations.
 
             - data_sink_id: typing.Optional[str]. Data sink ID. When provided instead of data_sink, the data sink will be looked up by ID.
 
@@ -1772,8 +1940,6 @@ class AsyncPipelinesClient:
 
             - eval_parameters: typing.Optional[EvalExecutionParams]. Eval parameters for the pipeline.
 
-            - llama_parse_enabled: typing.Optional[bool]. Whether to use LlamaParse during pipeline execution.
-
             - llama_parse_parameters: typing.Optional[LlamaParseParameters]. Settings that can be configured for how to use LlamaParse to parse files within a LlamaCloud pipeline.
 
             - name: typing.Optional[str].
@@ -1781,8 +1947,11 @@ class AsyncPipelinesClient:
             - managed_pipeline_id: typing.Optional[str]. The ID of the ManagedPipeline this playground pipeline is linked to.
         ---
         from llama_cloud import (
+            AutoTransformConfig,
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -1790,6 +1959,8 @@ class AsyncPipelinesClient:
             PresetRetrievalParams,
             RetrievalMode,
             SupportedEvalLlmModelNames,
+            TransformConfig,
+            TransformConfigMode,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1798,6 +1969,13 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.update_existing_pipeline(
             pipeline_id="string",
+            embedding_config=EmbeddingConfig(
+                type=EmbeddingConfigType.OPENAI_EMBEDDING,
+            ),
+            transform_config=TransformConfig(
+                mode=TransformConfigMode.AUTO,
+                config=AutoTransformConfig(),
+            ),
             data_sink=DataSinkCreate(
                 name="string",
                 sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -1816,6 +1994,10 @@ class AsyncPipelinesClient:
         )
         """
         _request: typing.Dict[str, typing.Any] = {}
+        if embedding_config is not OMIT:
+            _request["embedding_config"] = embedding_config
+        if transform_config is not OMIT:
+            _request["transform_config"] = transform_config
         if configured_transformations is not OMIT:
             _request["configured_transformations"] = configured_transformations
         if data_sink_id is not OMIT:
@@ -1826,8 +2008,6 @@ class AsyncPipelinesClient:
             _request["preset_retrieval_parameters"] = preset_retrieval_parameters
         if eval_parameters is not OMIT:
             _request["eval_parameters"] = eval_parameters
-        if llama_parse_enabled is not OMIT:
-            _request["llama_parse_enabled"] = llama_parse_enabled
         if llama_parse_parameters is not OMIT:
             _request["llama_parse_parameters"] = llama_parse_parameters
         if name is not OMIT:
@@ -1947,6 +2127,38 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def copy_pipeline(self, pipeline_id: str) -> Pipeline:
+        """
+        Copy a pipeline by ID.
+
+        Parameters:
+            - pipeline_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.copy_pipeline(
+            pipeline_id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/copy"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(Pipeline, _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)
+
     async def get_eval_dataset_executions(
         self, pipeline_id: str, eval_dataset_id: str
     ) -> typing.List[EvalDatasetJobRecord]:
@@ -2704,18 +2916,25 @@ class AsyncPipelinesClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def chat(
-        self, pipeline_id: str, *, retrieval_parameters: PresetRetrievalParams, data: ChatParams
+        self,
+        pipeline_id: str,
+        *,
+        messages: typing.List[ChatMessage],
+        data: ChatData,
+        class_name: typing.Optional[str] = OMIT,
     ) -> typing.Any:
         """
         Parameters:
             - pipeline_id: str.
 
-            - retrieval_parameters: PresetRetrievalParams.
+            - messages: typing.List[ChatMessage].
+
+            - data: ChatData.
 
-            - data: ChatParams.
+            - class_name: typing.Optional[str].
         ---
         from llama_cloud import (
-            ChatParams,
+            ChatData,
             FilterCondition,
             MetadataFilters,
             PresetRetrievalParams,
@@ -2728,22 +2947,25 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.chat(
             pipeline_id="string",
-            retrieval_parameters=PresetRetrievalParams(
-                search_filters=MetadataFilters(
-                    filters=[],
-                    condition=FilterCondition.AND,
+            messages=[],
+            data=ChatData(
+                retrieval_parameters=PresetRetrievalParams(
+                    search_filters=MetadataFilters(
+                        filters=[],
+                        condition=FilterCondition.AND,
+                    ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
                 ),
-                retrieval_mode=RetrievalMode.CHUNKS,
-            ),
-            data=ChatParams(
-                messages=[],
             ),
         )
         """
+        _request: typing.Dict[str, typing.Any] = {"messages": messages, "data": data}
+        if class_name is not OMIT:
+            _request["class_name"] = class_name
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/chat"),
-            json=jsonable_encoder({"retrieval_parameters": retrieval_parameters, "data": data}),
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -2758,7 +2980,13 @@ class AsyncPipelinesClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def list_pipeline_documents(
-        self, pipeline_id: str, *, skip: typing.Optional[int] = None, limit: typing.Optional[int] = None
+        self,
+        pipeline_id: str,
+        *,
+        skip: typing.Optional[int] = None,
+        limit: typing.Optional[int] = None,
+        file_id: typing.Optional[str] = None,
+        only_direct_upload: typing.Optional[bool] = None,
     ) -> typing.List[CloudDocument]:
         """
         Return a list of documents for a pipeline.
@@ -2769,6 +2997,10 @@ class AsyncPipelinesClient:
             - skip: typing.Optional[int].
 
             - limit: typing.Optional[int].
+
+            - file_id: typing.Optional[str].
+
+            - only_direct_upload: typing.Optional[bool].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -2784,7 +3016,9 @@ class AsyncPipelinesClient:
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/documents"
             ),
-            params=remove_none_from_dict({"skip": skip, "limit": limit}),
+            params=remove_none_from_dict(
+                {"skip": skip, "limit": limit, "file_id": file_id, "only_direct_upload": only_direct_upload}
+            ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -2989,3 +3223,41 @@ 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 list_pipeline_document_chunks(self, pipeline_id: str, document_id: str) -> typing.List[TextNode]:
+        """
+        Return a list of chunks for a pipeline document.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - document_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.list_pipeline_document_chunks(
+            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}/chunks",
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[TextNode], _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)