llama-cloud 0.0.13__py3-none-any.whl → 0.0.15__py3-none-any.whl

llama_cloud/resources/pipelines/client.py

@@ -10,17 +10,16 @@ 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
 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
 from ...types.eval_question_result import EvalQuestionResult
 from ...types.http_validation_error import HttpValidationError
+from ...types.input_message import InputMessage
 from ...types.llama_parse_parameters import LlamaParseParameters
 from ...types.managed_ingestion_status_response import ManagedIngestionStatusResponse
 from ...types.metadata_filters import MetadataFilters
@@ -32,11 +31,13 @@ from ...types.pipeline_deployment import PipelineDeployment
 from ...types.pipeline_file import PipelineFile
 from ...types.pipeline_file_create import PipelineFileCreate
 from ...types.pipeline_type import PipelineType
+from ...types.playground_session import PlaygroundSession
 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
+from .types.pipeline_update_embedding_config import PipelineUpdateEmbeddingConfig
 from .types.pipeline_update_transform_config import PipelineUpdateTransformConfig
 
 try:
@@ -121,8 +122,6 @@ class PipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
-            EmbeddingConfig,
-            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -131,7 +130,7 @@ class PipelinesClient:
             PipelineType,
             PresetRetrievalParams,
             RetrievalMode,
-            SupportedEvalLlmModelNames,
+            SupportedLlmModelNames,
         )
         from llama_cloud.client import LlamaCloud
 
@@ -140,9 +139,6 @@ class PipelinesClient:
         )
         client.pipelines.create_pipeline(
             request=PipelineCreate(
-                embedding_config=EmbeddingConfig(
-                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
-                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -155,7 +151,7 @@ class PipelinesClient:
                     retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
-                    llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
+                    llm_model=SupportedLlmModelNames.GPT_3_5_TURBO,
                 ),
                 llama_parse_parameters=LlamaParseParameters(),
                 name="string",
@@ -194,8 +190,6 @@ class PipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
-            EmbeddingConfig,
-            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -204,7 +198,7 @@ class PipelinesClient:
             PipelineType,
             PresetRetrievalParams,
             RetrievalMode,
-            SupportedEvalLlmModelNames,
+            SupportedLlmModelNames,
         )
         from llama_cloud.client import LlamaCloud
 
@@ -213,9 +207,6 @@ class PipelinesClient:
         )
         client.pipelines.upsert_pipeline(
             request=PipelineCreate(
-                embedding_config=EmbeddingConfig(
-                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
-                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -228,7 +219,7 @@ class PipelinesClient:
                     retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
-                    llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
+                    llm_model=SupportedLlmModelNames.GPT_3_5_TURBO,
                 ),
                 llama_parse_parameters=LlamaParseParameters(),
                 name="string",
@@ -290,7 +281,7 @@ class PipelinesClient:
         self,
         pipeline_id: str,
         *,
-        embedding_config: typing.Optional[EmbeddingConfig] = OMIT,
+        embedding_config: typing.Optional[PipelineUpdateEmbeddingConfig] = OMIT,
         transform_config: typing.Optional[PipelineUpdateTransformConfig] = OMIT,
         configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]] = OMIT,
         data_sink_id: typing.Optional[str] = OMIT,
@@ -307,11 +298,11 @@ class PipelinesClient:
         Parameters:
             - pipeline_id: str.
 
-            - embedding_config: typing.Optional[EmbeddingConfig]. Configuration for the embedding model.
+            - embedding_config: typing.Optional[PipelineUpdateEmbeddingConfig]. Configuration for the embedding model.
 
             - transform_config: typing.Optional[PipelineUpdateTransformConfig]. Configuration for the transformation.
 
-            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]]. List of configured transformations.
+            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]]. Deprecated, use embedding_config or transform_config instead. configured transformations for the pipeline.
 
             - data_sink_id: typing.Optional[str]. Data sink ID. When provided instead of data_sink, the data sink will be looked up by ID.
 
@@ -330,15 +321,13 @@ class PipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
-            EmbeddingConfig,
-            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
             MetadataFilters,
             PresetRetrievalParams,
             RetrievalMode,
-            SupportedEvalLlmModelNames,
+            SupportedLlmModelNames,
         )
         from llama_cloud.client import LlamaCloud
 
@@ -347,9 +336,6 @@ class PipelinesClient:
         )
         client.pipelines.update_existing_pipeline(
             pipeline_id="string",
-            embedding_config=EmbeddingConfig(
-                type=EmbeddingConfigType.OPENAI_EMBEDDING,
-            ),
             data_sink=DataSinkCreate(
                 name="string",
                 sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -362,7 +348,7 @@ class PipelinesClient:
                 retrieval_mode=RetrievalMode.CHUNKS,
             ),
             eval_parameters=EvalExecutionParams(
-                llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
+                llm_model=SupportedLlmModelNames.GPT_3_5_TURBO,
             ),
             llama_parse_parameters=LlamaParseParameters(),
         )
@@ -591,7 +577,7 @@ class PipelinesClient:
 
             - params: typing.Optional[EvalExecutionParamsOverride]. The parameters for the eval execution that will override the ones set in the pipeline.
         ---
-        from llama_cloud import EvalExecutionParamsOverride, SupportedEvalLlmModelNames
+        from llama_cloud import EvalExecutionParamsOverride, SupportedLlmModelNames
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
@@ -602,7 +588,7 @@ class PipelinesClient:
             eval_dataset_id="string",
             eval_question_ids=[],
             params=EvalExecutionParamsOverride(
-                llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
+                llm_model=SupportedLlmModelNames.GPT_3_5_TURBO,
             ),
         )
         """
@@ -1054,6 +1040,52 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def update_pipeline_data_source(
+        self, pipeline_id: str, data_source_id: str, *, sync_interval: typing.Optional[float] = OMIT
+    ) -> PipelineDataSource:
+        """
+        Update the configuration of a data source in a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - data_source_id: str.
+
+            - sync_interval: typing.Optional[float]. The interval at which the data source should be synced.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.update_pipeline_data_source(
+            pipeline_id="string",
+            data_source_id="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {}
+        if sync_interval is not OMIT:
+            _request["sync_interval"] = sync_interval
+        _response = self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/",
+                f"api/v1/pipelines/{pipeline_id}/data-sources/{data_source_id}",
+            ),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PipelineDataSource, _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 delete_pipeline_data_source(self, pipeline_id: str, data_source_id: str) -> None:
         """
         Delete a data source from a pipeline.
@@ -1130,6 +1162,44 @@ 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_data_source_status(self, pipeline_id: str, data_source_id: str) -> ManagedIngestionStatusResponse:
+        """
+        Get the status of a data source for a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - data_source_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.get_pipeline_data_source_status(
+            pipeline_id="string",
+            data_source_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}/data-sources/{data_source_id}/status",
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ManagedIngestionStatusResponse, _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 run_search(
         self,
         pipeline_id: str,
@@ -1287,21 +1357,57 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def get_playground_session(self, pipeline_id: str) -> PlaygroundSession:
+        """
+        Get a playground session for a user and pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.get_playground_session(
+            pipeline_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}/playground-session"
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PlaygroundSession, _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 chat(
         self,
         pipeline_id: str,
         *,
-        messages: typing.List[ChatMessage],
-        data: ChatData,
+        messages: typing.Optional[typing.List[InputMessage]] = OMIT,
+        data: typing.Optional[ChatData] = OMIT,
         class_name: typing.Optional[str] = OMIT,
     ) -> typing.Any:
         """
+        Make a retrieval query + chat completion for a managed pipeline.
+
         Parameters:
             - pipeline_id: str.
 
-            - messages: typing.List[ChatMessage].
+            - messages: typing.Optional[typing.List[InputMessage]].
 
-            - data: ChatData.
+            - data: typing.Optional[ChatData].
 
             - class_name: typing.Optional[str].
         ---
@@ -1320,7 +1426,6 @@ class PipelinesClient:
         )
         client.pipelines.chat(
             pipeline_id="string",
-            messages=[],
             data=ChatData(
                 retrieval_parameters=PresetRetrievalParams(
                     search_filters=MetadataFilters(
@@ -1333,7 +1438,11 @@ class PipelinesClient:
             ),
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"messages": messages, "data": data}
+        _request: typing.Dict[str, typing.Any] = {}
+        if messages is not OMIT:
+            _request["messages"] = messages
+        if data is not OMIT:
+            _request["data"] = data
         if class_name is not OMIT:
             _request["class_name"] = class_name
         _response = self._client_wrapper.httpx_client.request(
@@ -1707,8 +1816,6 @@ class AsyncPipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
-            EmbeddingConfig,
-            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -1717,7 +1824,7 @@ class AsyncPipelinesClient:
             PipelineType,
             PresetRetrievalParams,
             RetrievalMode,
-            SupportedEvalLlmModelNames,
+            SupportedLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1726,9 +1833,6 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.create_pipeline(
             request=PipelineCreate(
-                embedding_config=EmbeddingConfig(
-                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
-                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -1741,7 +1845,7 @@ class AsyncPipelinesClient:
                     retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
-                    llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
+                    llm_model=SupportedLlmModelNames.GPT_3_5_TURBO,
                 ),
                 llama_parse_parameters=LlamaParseParameters(),
                 name="string",
@@ -1780,8 +1884,6 @@ class AsyncPipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
-            EmbeddingConfig,
-            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -1790,7 +1892,7 @@ class AsyncPipelinesClient:
             PipelineType,
             PresetRetrievalParams,
             RetrievalMode,
-            SupportedEvalLlmModelNames,
+            SupportedLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1799,9 +1901,6 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.upsert_pipeline(
             request=PipelineCreate(
-                embedding_config=EmbeddingConfig(
-                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
-                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -1814,7 +1913,7 @@ class AsyncPipelinesClient:
                     retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
-                    llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
+                    llm_model=SupportedLlmModelNames.GPT_3_5_TURBO,
                 ),
                 llama_parse_parameters=LlamaParseParameters(),
                 name="string",
@@ -1876,7 +1975,7 @@ class AsyncPipelinesClient:
         self,
         pipeline_id: str,
         *,
-        embedding_config: typing.Optional[EmbeddingConfig] = OMIT,
+        embedding_config: typing.Optional[PipelineUpdateEmbeddingConfig] = OMIT,
         transform_config: typing.Optional[PipelineUpdateTransformConfig] = OMIT,
         configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]] = OMIT,
         data_sink_id: typing.Optional[str] = OMIT,
@@ -1893,11 +1992,11 @@ class AsyncPipelinesClient:
         Parameters:
             - pipeline_id: str.
 
-            - embedding_config: typing.Optional[EmbeddingConfig]. Configuration for the embedding model.
+            - embedding_config: typing.Optional[PipelineUpdateEmbeddingConfig]. Configuration for the embedding model.
 
             - transform_config: typing.Optional[PipelineUpdateTransformConfig]. Configuration for the transformation.
 
-            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]]. List of configured transformations.
+            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]]. Deprecated, use embedding_config or transform_config instead. configured transformations for the pipeline.
 
             - data_sink_id: typing.Optional[str]. Data sink ID. When provided instead of data_sink, the data sink will be looked up by ID.
 
@@ -1916,15 +2015,13 @@ class AsyncPipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
-            EmbeddingConfig,
-            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
             MetadataFilters,
             PresetRetrievalParams,
             RetrievalMode,
-            SupportedEvalLlmModelNames,
+            SupportedLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1933,9 +2030,6 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.update_existing_pipeline(
             pipeline_id="string",
-            embedding_config=EmbeddingConfig(
-                type=EmbeddingConfigType.OPENAI_EMBEDDING,
-            ),
             data_sink=DataSinkCreate(
                 name="string",
                 sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -1948,7 +2042,7 @@ class AsyncPipelinesClient:
                 retrieval_mode=RetrievalMode.CHUNKS,
             ),
             eval_parameters=EvalExecutionParams(
-                llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
+                llm_model=SupportedLlmModelNames.GPT_3_5_TURBO,
             ),
             llama_parse_parameters=LlamaParseParameters(),
         )
@@ -2179,7 +2273,7 @@ class AsyncPipelinesClient:
 
             - params: typing.Optional[EvalExecutionParamsOverride]. The parameters for the eval execution that will override the ones set in the pipeline.
         ---
-        from llama_cloud import EvalExecutionParamsOverride, SupportedEvalLlmModelNames
+        from llama_cloud import EvalExecutionParamsOverride, SupportedLlmModelNames
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
@@ -2190,7 +2284,7 @@ class AsyncPipelinesClient:
             eval_dataset_id="string",
             eval_question_ids=[],
             params=EvalExecutionParamsOverride(
-                llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
+                llm_model=SupportedLlmModelNames.GPT_3_5_TURBO,
             ),
         )
         """
@@ -2642,6 +2736,52 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def update_pipeline_data_source(
+        self, pipeline_id: str, data_source_id: str, *, sync_interval: typing.Optional[float] = OMIT
+    ) -> PipelineDataSource:
+        """
+        Update the configuration of a data source in a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - data_source_id: str.
+
+            - sync_interval: typing.Optional[float]. The interval at which the data source should be synced.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.update_pipeline_data_source(
+            pipeline_id="string",
+            data_source_id="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {}
+        if sync_interval is not OMIT:
+            _request["sync_interval"] = sync_interval
+        _response = await self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/",
+                f"api/v1/pipelines/{pipeline_id}/data-sources/{data_source_id}",
+            ),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PipelineDataSource, _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 delete_pipeline_data_source(self, pipeline_id: str, data_source_id: str) -> None:
         """
         Delete a data source from a pipeline.
@@ -2718,6 +2858,46 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def get_pipeline_data_source_status(
+        self, pipeline_id: str, data_source_id: str
+    ) -> ManagedIngestionStatusResponse:
+        """
+        Get the status of a data source for a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - data_source_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.get_pipeline_data_source_status(
+            pipeline_id="string",
+            data_source_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}/data-sources/{data_source_id}/status",
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ManagedIngestionStatusResponse, _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 run_search(
         self,
         pipeline_id: str,
@@ -2875,21 +3055,57 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def get_playground_session(self, pipeline_id: str) -> PlaygroundSession:
+        """
+        Get a playground session for a user and pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.get_playground_session(
+            pipeline_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}/playground-session"
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PlaygroundSession, _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 chat(
         self,
         pipeline_id: str,
         *,
-        messages: typing.List[ChatMessage],
-        data: ChatData,
+        messages: typing.Optional[typing.List[InputMessage]] = OMIT,
+        data: typing.Optional[ChatData] = OMIT,
         class_name: typing.Optional[str] = OMIT,
     ) -> typing.Any:
         """
+        Make a retrieval query + chat completion for a managed pipeline.
+
         Parameters:
             - pipeline_id: str.
 
-            - messages: typing.List[ChatMessage].
+            - messages: typing.Optional[typing.List[InputMessage]].
 
-            - data: ChatData.
+            - data: typing.Optional[ChatData].
 
             - class_name: typing.Optional[str].
         ---
@@ -2908,7 +3124,6 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.chat(
             pipeline_id="string",
-            messages=[],
             data=ChatData(
                 retrieval_parameters=PresetRetrievalParams(
                     search_filters=MetadataFilters(
@@ -2921,7 +3136,11 @@ class AsyncPipelinesClient:
             ),
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"messages": messages, "data": data}
+        _request: typing.Dict[str, typing.Any] = {}
+        if messages is not OMIT:
+            _request["messages"] = messages
+        if data is not OMIT:
+            _request["data"] = data
         if class_name is not OMIT:
             _request["class_name"] = class_name
         _response = await self._client_wrapper.httpx_client.request(