llama-cloud 0.0.12__py3-none-any.whl → 0.0.14__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,
             ),
         )
         """
@@ -1287,21 +1273,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 +1342,6 @@ class PipelinesClient:
         )
         client.pipelines.chat(
             pipeline_id="string",
-            messages=[],
             data=ChatData(
                 retrieval_parameters=PresetRetrievalParams(
                     search_filters=MetadataFilters(
@@ -1333,7 +1354,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 +1732,6 @@ class AsyncPipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
-            EmbeddingConfig,
-            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -1717,7 +1740,7 @@ class AsyncPipelinesClient:
             PipelineType,
             PresetRetrievalParams,
             RetrievalMode,
-            SupportedEvalLlmModelNames,
+            SupportedLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1726,9 +1749,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 +1761,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 +1800,6 @@ class AsyncPipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
-            EmbeddingConfig,
-            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -1790,7 +1808,7 @@ class AsyncPipelinesClient:
             PipelineType,
             PresetRetrievalParams,
             RetrievalMode,
-            SupportedEvalLlmModelNames,
+            SupportedLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1799,9 +1817,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 +1829,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 +1891,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 +1908,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 +1931,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 +1946,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 +1958,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 +2189,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 +2200,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,
             ),
         )
         """
@@ -2875,21 +2885,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 +2954,6 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.chat(
             pipeline_id="string",
-            messages=[],
             data=ChatData(
                 retrieval_parameters=PresetRetrievalParams(
                     search_filters=MetadataFilters(
@@ -2921,7 +2966,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(

llama_cloud/resources/pipelines/types/__init__.py

@@ -1,6 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .pipeline_file_update_custom_metadata_value import PipelineFileUpdateCustomMetadataValue
+from .pipeline_update_embedding_config import (
+    PipelineUpdateEmbeddingConfig,
+    PipelineUpdateEmbeddingConfig_AzureEmbedding,
+    PipelineUpdateEmbeddingConfig_BedrockEmbedding,
+    PipelineUpdateEmbeddingConfig_CohereEmbedding,
+    PipelineUpdateEmbeddingConfig_GeminiEmbedding,
+    PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
+    PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
+)
 from .pipeline_update_transform_config import (
     PipelineUpdateTransformConfig,
     PipelineUpdateTransformConfig_Advanced,
@@ -9,6 +18,13 @@ from .pipeline_update_transform_config import (
 
 __all__ = [
     "PipelineFileUpdateCustomMetadataValue",
+    "PipelineUpdateEmbeddingConfig",
+    "PipelineUpdateEmbeddingConfig_AzureEmbedding",
+    "PipelineUpdateEmbeddingConfig_BedrockEmbedding",
+    "PipelineUpdateEmbeddingConfig_CohereEmbedding",
+    "PipelineUpdateEmbeddingConfig_GeminiEmbedding",
+    "PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
+    "PipelineUpdateEmbeddingConfig_OpenaiEmbedding",
     "PipelineUpdateTransformConfig",
     "PipelineUpdateTransformConfig_Advanced",
     "PipelineUpdateTransformConfig_Auto",

llama_cloud/resources/pipelines/types/pipeline_update_embedding_config.py

@@ -0,0 +1,78 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from ....types.azure_open_ai_embedding_config import AzureOpenAiEmbeddingConfig
+from ....types.bedrock_embedding_config import BedrockEmbeddingConfig
+from ....types.cohere_embedding_config import CohereEmbeddingConfig
+from ....types.gemini_embedding_config import GeminiEmbeddingConfig
+from ....types.hugging_face_inference_api_embedding_config import HuggingFaceInferenceApiEmbeddingConfig
+from ....types.open_ai_embedding_config import OpenAiEmbeddingConfig
+
+
+class PipelineUpdateEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
+    type: typing_extensions.Literal["OPENAI_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class PipelineUpdateEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
+    type: typing_extensions.Literal["AZURE_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInferenceApiEmbeddingConfig):
+    type: typing_extensions.Literal["HUGGINGFACE_API_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class PipelineUpdateEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
+    type: typing_extensions.Literal["BEDROCK_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class PipelineUpdateEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
+    type: typing_extensions.Literal["GEMINI_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class PipelineUpdateEmbeddingConfig_CohereEmbedding(CohereEmbeddingConfig):
+    type: typing_extensions.Literal["COHERE_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+PipelineUpdateEmbeddingConfig = typing.Union[
+    PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
+    PipelineUpdateEmbeddingConfig_AzureEmbedding,
+    PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
+    PipelineUpdateEmbeddingConfig_BedrockEmbedding,
+    PipelineUpdateEmbeddingConfig_GeminiEmbedding,
+    PipelineUpdateEmbeddingConfig_CohereEmbedding,
+]