llama-cloud 0.0.8__py3-none-any.whl → 0.0.10__py3-none-any.whl

llama_cloud/resources/pipelines/client.py

@@ -9,6 +9,8 @@ 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_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
@@ -30,7 +32,9 @@ from ...types.pipeline_file import PipelineFile
 from ...types.pipeline_file_create import PipelineFileCreate
 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.pipeline_file_update_custom_metadata_value import PipelineFileUpdateCustomMetadataValue
 
 try:
@@ -52,19 +56,22 @@ class PipelinesClient:
     def search_pipelines(
         self,
         *,
-        project_name: str,
+        project_name: typing.Optional[str] = None,
         pipeline_name: typing.Optional[str] = None,
         pipeline_type: typing.Optional[PipelineType] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[Pipeline]:
         """
         Search for pipelines by various parameters.
 
         Parameters:
-            - project_name: str.
+            - project_name: typing.Optional[str].
 
             - pipeline_name: typing.Optional[str].
 
             - pipeline_type: typing.Optional[PipelineType].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import PipelineType
         from llama_cloud.client import LlamaCloud
@@ -73,7 +80,6 @@ class PipelinesClient:
             token="YOUR_TOKEN",
         )
         client.pipelines.search_pipelines(
-            project_name="string",
             pipeline_type=PipelineType.PLAYGROUND,
         )
         """
@@ -81,7 +87,12 @@ class PipelinesClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/pipelines"),
             params=remove_none_from_dict(
-                {"project_name": project_name, "pipeline_name": pipeline_name, "pipeline_type": pipeline_type}
+                {
+                    "project_name": project_name,
+                    "pipeline_name": pipeline_name,
+                    "pipeline_type": pipeline_type,
+                    "project_id": project_id,
+                }
             ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -115,6 +126,7 @@ class PipelinesClient:
             PipelineCreate,
             PipelineType,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import LlamaCloud
@@ -133,6 +145,7 @@ class PipelinesClient:
                         filters=[],
                         condition=FilterCondition.AND,
                     ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
                     llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -181,6 +194,7 @@ class PipelinesClient:
             PipelineCreate,
             PipelineType,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import LlamaCloud
@@ -199,6 +213,7 @@ class PipelinesClient:
                         filters=[],
                         condition=FilterCondition.AND,
                     ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
                     llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -268,7 +283,6 @@ class PipelinesClient:
         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,
@@ -289,8 +303,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].
@@ -305,6 +317,7 @@ class PipelinesClient:
             LlamaParseParameters,
             MetadataFilters,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import LlamaCloud
@@ -323,6 +336,7 @@ class PipelinesClient:
                     filters=[],
                     condition=FilterCondition.AND,
                 ),
+                retrieval_mode=RetrievalMode.CHUNKS,
             ),
             eval_parameters=EvalExecutionParams(
                 llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -341,8 +355,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:
@@ -462,6 +474,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.
@@ -1069,6 +1113,8 @@ class PipelinesClient:
         rerank_top_n: typing.Optional[int] = OMIT,
         alpha: typing.Optional[float] = OMIT,
         search_filters: typing.Optional[MetadataFilters] = OMIT,
+        files_top_k: typing.Optional[int] = OMIT,
+        retrieval_mode: typing.Optional[RetrievalMode] = OMIT,
         query: str,
     ) -> RetrieveResults:
         """
@@ -1089,9 +1135,13 @@ class PipelinesClient:
 
             - search_filters: typing.Optional[MetadataFilters]. Search filters for retrieval.
 
+            - files_top_k: typing.Optional[int]. Number of files to retrieve (only for retrieval mode files_via_metadata and files_via_content).
+
+            - retrieval_mode: typing.Optional[RetrievalMode]. The retrieval mode for the query.
+
             - query: str. The query to retrieve against.
         ---
-        from llama_cloud import FilterCondition, MetadataFilters
+        from llama_cloud import FilterCondition, MetadataFilters, RetrievalMode
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
@@ -1103,6 +1153,7 @@ class PipelinesClient:
                 filters=[],
                 condition=FilterCondition.AND,
             ),
+            retrieval_mode=RetrievalMode.CHUNKS,
             query="string",
         )
         """
@@ -1119,6 +1170,10 @@ class PipelinesClient:
             _request["alpha"] = alpha
         if search_filters is not OMIT:
             _request["search_filters"] = search_filters
+        if files_top_k is not OMIT:
+            _request["files_top_k"] = files_top_k
+        if retrieval_mode is not OMIT:
+            _request["retrieval_mode"] = retrieval_mode
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/retrieve"),
@@ -1205,8 +1260,78 @@ 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,
+        *,
+        messages: typing.List[ChatMessage],
+        data: ChatData,
+        class_name: typing.Optional[str] = OMIT,
+    ) -> typing.Any:
+        """
+        Parameters:
+            - pipeline_id: str.
+
+            - messages: typing.List[ChatMessage].
+
+            - data: ChatData.
+
+            - class_name: typing.Optional[str].
+        ---
+        from llama_cloud import (
+            ChatData,
+            FilterCondition,
+            MetadataFilters,
+            PresetRetrievalParams,
+            RetrievalMode,
+        )
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.chat(
+            pipeline_id="string",
+            messages=[],
+            data=ChatData(
+                retrieval_parameters=PresetRetrievalParams(
+                    search_filters=MetadataFilters(
+                        filters=[],
+                        condition=FilterCondition.AND,
+                    ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
+                ),
+            ),
+        )
+        """
+        _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(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Any, _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 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.
@@ -1217,6 +1342,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
 
@@ -1232,7 +1361,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,
         )
@@ -1438,6 +1569,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):
@@ -1446,19 +1615,22 @@ class AsyncPipelinesClient:
     async def search_pipelines(
         self,
         *,
-        project_name: str,
+        project_name: typing.Optional[str] = None,
         pipeline_name: typing.Optional[str] = None,
         pipeline_type: typing.Optional[PipelineType] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[Pipeline]:
         """
         Search for pipelines by various parameters.
 
         Parameters:
-            - project_name: str.
+            - project_name: typing.Optional[str].
 
             - pipeline_name: typing.Optional[str].
 
             - pipeline_type: typing.Optional[PipelineType].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import PipelineType
         from llama_cloud.client import AsyncLlamaCloud
@@ -1467,7 +1639,6 @@ class AsyncPipelinesClient:
             token="YOUR_TOKEN",
         )
         await client.pipelines.search_pipelines(
-            project_name="string",
             pipeline_type=PipelineType.PLAYGROUND,
         )
         """
@@ -1475,7 +1646,12 @@ class AsyncPipelinesClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/pipelines"),
             params=remove_none_from_dict(
-                {"project_name": project_name, "pipeline_name": pipeline_name, "pipeline_type": pipeline_type}
+                {
+                    "project_name": project_name,
+                    "pipeline_name": pipeline_name,
+                    "pipeline_type": pipeline_type,
+                    "project_id": project_id,
+                }
             ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -1509,6 +1685,7 @@ class AsyncPipelinesClient:
             PipelineCreate,
             PipelineType,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
@@ -1527,6 +1704,7 @@ class AsyncPipelinesClient:
                         filters=[],
                         condition=FilterCondition.AND,
                     ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
                     llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -1575,6 +1753,7 @@ class AsyncPipelinesClient:
             PipelineCreate,
             PipelineType,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
@@ -1593,6 +1772,7 @@ class AsyncPipelinesClient:
                         filters=[],
                         condition=FilterCondition.AND,
                     ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
                     llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -1662,7 +1842,6 @@ class AsyncPipelinesClient:
         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,
@@ -1683,8 +1862,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].
@@ -1699,6 +1876,7 @@ class AsyncPipelinesClient:
             LlamaParseParameters,
             MetadataFilters,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
@@ -1717,6 +1895,7 @@ class AsyncPipelinesClient:
                     filters=[],
                     condition=FilterCondition.AND,
                 ),
+                retrieval_mode=RetrievalMode.CHUNKS,
             ),
             eval_parameters=EvalExecutionParams(
                 llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -1735,8 +1914,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:
@@ -1856,6 +2033,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]:
@@ -2465,6 +2674,8 @@ class AsyncPipelinesClient:
         rerank_top_n: typing.Optional[int] = OMIT,
         alpha: typing.Optional[float] = OMIT,
         search_filters: typing.Optional[MetadataFilters] = OMIT,
+        files_top_k: typing.Optional[int] = OMIT,
+        retrieval_mode: typing.Optional[RetrievalMode] = OMIT,
         query: str,
     ) -> RetrieveResults:
         """
@@ -2485,9 +2696,13 @@ class AsyncPipelinesClient:
 
             - search_filters: typing.Optional[MetadataFilters]. Search filters for retrieval.
 
+            - files_top_k: typing.Optional[int]. Number of files to retrieve (only for retrieval mode files_via_metadata and files_via_content).
+
+            - retrieval_mode: typing.Optional[RetrievalMode]. The retrieval mode for the query.
+
             - query: str. The query to retrieve against.
         ---
-        from llama_cloud import FilterCondition, MetadataFilters
+        from llama_cloud import FilterCondition, MetadataFilters, RetrievalMode
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
@@ -2499,6 +2714,7 @@ class AsyncPipelinesClient:
                 filters=[],
                 condition=FilterCondition.AND,
             ),
+            retrieval_mode=RetrievalMode.CHUNKS,
             query="string",
         )
         """
@@ -2515,6 +2731,10 @@ class AsyncPipelinesClient:
             _request["alpha"] = alpha
         if search_filters is not OMIT:
             _request["search_filters"] = search_filters
+        if files_top_k is not OMIT:
+            _request["files_top_k"] = files_top_k
+        if retrieval_mode is not OMIT:
+            _request["retrieval_mode"] = retrieval_mode
         _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}/retrieve"),
@@ -2601,8 +2821,78 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def chat(
+        self,
+        pipeline_id: str,
+        *,
+        messages: typing.List[ChatMessage],
+        data: ChatData,
+        class_name: typing.Optional[str] = OMIT,
+    ) -> typing.Any:
+        """
+        Parameters:
+            - pipeline_id: str.
+
+            - messages: typing.List[ChatMessage].
+
+            - data: ChatData.
+
+            - class_name: typing.Optional[str].
+        ---
+        from llama_cloud import (
+            ChatData,
+            FilterCondition,
+            MetadataFilters,
+            PresetRetrievalParams,
+            RetrievalMode,
+        )
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.chat(
+            pipeline_id="string",
+            messages=[],
+            data=ChatData(
+                retrieval_parameters=PresetRetrievalParams(
+                    search_filters=MetadataFilters(
+                        filters=[],
+                        condition=FilterCondition.AND,
+                    ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
+                ),
+            ),
+        )
+        """
+        _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(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Any, _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 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.
@@ -2613,6 +2903,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
 
@@ -2628,7 +2922,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,
         )
@@ -2833,3 +3129,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)