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

llama_cloud/__init__.py

@@ -31,6 +31,7 @@ from .types import (
     CloudConfluenceDataSource,
     CloudDocument,
     CloudDocumentCreate,
+    CloudGoogleDriveDataSource,
     CloudJiraDataSource,
     CloudMilvusVectorStore,
     CloudMongoDbAtlasVectorSearch,
@@ -79,6 +80,7 @@ from .types import (
     EvalQuestion,
     EvalQuestionCreate,
     EvalQuestionResult,
+    ExtendVertexTextEmbedding,
     ExtractionJob,
     ExtractionResult,
     ExtractionResultDataValue,
@@ -124,6 +126,8 @@ from .types import (
     OpenAiEmbeddingConfig,
     Organization,
     OrganizationCreate,
+    PageScreenshotMetadata,
+    PageScreenshotNodeWithScore,
     PageSegmentationConfig,
     PageSplitterNodeParser,
     ParserLanguages,
@@ -134,6 +138,7 @@ from .types import (
     ParsingJobTextResult,
     ParsingUsage,
     Pipeline,
+    PipelineConfigurationHashes,
     PipelineCreate,
     PipelineCreateEmbeddingConfig,
     PipelineCreateEmbeddingConfig_AzureEmbedding,
@@ -142,6 +147,7 @@ from .types import (
     PipelineCreateEmbeddingConfig_GeminiEmbedding,
     PipelineCreateEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineCreateEmbeddingConfig_OpenaiEmbedding,
+    PipelineCreateEmbeddingConfig_VertexaiEmbedding,
     PipelineCreateTransformConfig,
     PipelineCreateTransformConfig_Advanced,
     PipelineCreateTransformConfig_Auto,
@@ -158,7 +164,9 @@ from .types import (
     PipelineEmbeddingConfig_GeminiEmbedding,
     PipelineEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineEmbeddingConfig_OpenaiEmbedding,
+    PipelineEmbeddingConfig_VertexaiEmbedding,
     PipelineFile,
+    PipelineFileConfigHashValue,
     PipelineFileCreate,
     PipelineFileCreateCustomMetadataValue,
     PipelineFileCustomMetadataValue,
@@ -197,6 +205,8 @@ from .types import (
     UserOrganizationDelete,
     ValidationError,
     ValidationErrorLocItem,
+    VertexAiEmbeddingConfig,
+    VertexEmbeddingMode,
 )
 from .errors import UnprocessableEntityError
 from .resources import (
@@ -216,6 +226,7 @@ from .resources import (
     PipelineUpdateEmbeddingConfig_GeminiEmbedding,
     PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
+    PipelineUpdateEmbeddingConfig_VertexaiEmbedding,
     PipelineUpdateTransformConfig,
     PipelineUpdateTransformConfig_Advanced,
     PipelineUpdateTransformConfig_Auto,
@@ -264,6 +275,7 @@ __all__ = [
     "CloudConfluenceDataSource",
     "CloudDocument",
     "CloudDocumentCreate",
+    "CloudGoogleDriveDataSource",
     "CloudJiraDataSource",
     "CloudMilvusVectorStore",
     "CloudMongoDbAtlasVectorSearch",
@@ -317,6 +329,7 @@ __all__ = [
     "EvalQuestion",
     "EvalQuestionCreate",
     "EvalQuestionResult",
+    "ExtendVertexTextEmbedding",
     "ExtractionJob",
     "ExtractionResult",
     "ExtractionResultDataValue",
@@ -366,6 +379,8 @@ __all__ = [
     "OpenAiEmbeddingConfig",
     "Organization",
     "OrganizationCreate",
+    "PageScreenshotMetadata",
+    "PageScreenshotNodeWithScore",
     "PageSegmentationConfig",
     "PageSplitterNodeParser",
     "ParserLanguages",
@@ -376,6 +391,7 @@ __all__ = [
     "ParsingJobTextResult",
     "ParsingUsage",
     "Pipeline",
+    "PipelineConfigurationHashes",
     "PipelineCreate",
     "PipelineCreateEmbeddingConfig",
     "PipelineCreateEmbeddingConfig_AzureEmbedding",
@@ -384,6 +400,7 @@ __all__ = [
     "PipelineCreateEmbeddingConfig_GeminiEmbedding",
     "PipelineCreateEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineCreateEmbeddingConfig_OpenaiEmbedding",
+    "PipelineCreateEmbeddingConfig_VertexaiEmbedding",
     "PipelineCreateTransformConfig",
     "PipelineCreateTransformConfig_Advanced",
     "PipelineCreateTransformConfig_Auto",
@@ -400,7 +417,9 @@ __all__ = [
     "PipelineEmbeddingConfig_GeminiEmbedding",
     "PipelineEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineEmbeddingConfig_OpenaiEmbedding",
+    "PipelineEmbeddingConfig_VertexaiEmbedding",
     "PipelineFile",
+    "PipelineFileConfigHashValue",
     "PipelineFileCreate",
     "PipelineFileCreateCustomMetadataValue",
     "PipelineFileCustomMetadataValue",
@@ -417,6 +436,7 @@ __all__ = [
     "PipelineUpdateEmbeddingConfig_GeminiEmbedding",
     "PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineUpdateEmbeddingConfig_OpenaiEmbedding",
+    "PipelineUpdateEmbeddingConfig_VertexaiEmbedding",
     "PipelineUpdateTransformConfig",
     "PipelineUpdateTransformConfig_Advanced",
     "PipelineUpdateTransformConfig_Auto",
@@ -451,6 +471,8 @@ __all__ = [
     "UserOrganizationDelete",
     "ValidationError",
     "ValidationErrorLocItem",
+    "VertexAiEmbeddingConfig",
+    "VertexEmbeddingMode",
     "auth",
     "component_definitions",
     "data_sinks",

llama_cloud/resources/__init__.py

@@ -26,6 +26,7 @@ from .pipelines import (
     PipelineUpdateEmbeddingConfig_GeminiEmbedding,
     PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
+    PipelineUpdateEmbeddingConfig_VertexaiEmbedding,
     PipelineUpdateTransformConfig,
     PipelineUpdateTransformConfig_Advanced,
     PipelineUpdateTransformConfig_Auto,
@@ -48,6 +49,7 @@ __all__ = [
     "PipelineUpdateEmbeddingConfig_GeminiEmbedding",
     "PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineUpdateEmbeddingConfig_OpenaiEmbedding",
+    "PipelineUpdateEmbeddingConfig_VertexaiEmbedding",
     "PipelineUpdateTransformConfig",
     "PipelineUpdateTransformConfig_Advanced",
     "PipelineUpdateTransformConfig_Auto",

llama_cloud/resources/data_sources/types/data_source_update_component_one.py

@@ -5,6 +5,7 @@ import typing
 from ....types.cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
 from ....types.cloud_box_data_source import CloudBoxDataSource
 from ....types.cloud_confluence_data_source import CloudConfluenceDataSource
+from ....types.cloud_google_drive_data_source import CloudGoogleDriveDataSource
 from ....types.cloud_jira_data_source import CloudJiraDataSource
 from ....types.cloud_notion_page_data_source import CloudNotionPageDataSource
 from ....types.cloud_one_drive_data_source import CloudOneDriveDataSource
@@ -15,6 +16,7 @@ from ....types.cloud_slack_data_source import CloudSlackDataSource
 DataSourceUpdateComponentOne = typing.Union[
     CloudS3DataSource,
     CloudAzStorageBlobDataSource,
+    CloudGoogleDriveDataSource,
     CloudOneDriveDataSource,
     CloudSharepointDataSource,
     CloudSlackDataSource,

llama_cloud/resources/files/client.py

@@ -12,6 +12,7 @@ from ...core.remove_none_from_dict import remove_none_from_dict
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.file import File
 from ...types.http_validation_error import HttpValidationError
+from ...types.page_screenshot_metadata import PageScreenshotMetadata
 from ...types.presigned_url import PresignedUrl
 from .types.file_create_resource_info_value import FileCreateResourceInfoValue
 
@@ -288,6 +289,85 @@ class FilesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def list_file_page_screenshots(
+        self, id: str, *, project_id: typing.Optional[str] = None
+    ) -> typing.List[PageScreenshotMetadata]:
+        """
+        List metadata for all screenshots of pages from a file.
+
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.files.list_file_page_screenshots(
+            id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}/page_screenshots"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[PageScreenshotMetadata], _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_file_page_screenshot(
+        self, id: str, page_index: int, *, project_id: typing.Optional[str] = None
+    ) -> typing.Any:
+        """
+        Get screenshot of a page from a file.
+
+        Parameters:
+            - id: str.
+
+            - page_index: int.
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.files.get_file_page_screenshot(
+            id="string",
+            page_index=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}/page_screenshots/{page_index}"
+            ),
+            params=remove_none_from_dict({"project_id": project_id}),
+            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)
+
 
 class AsyncFilesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -549,3 +629,82 @@ class AsyncFilesClient:
         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_file_page_screenshots(
+        self, id: str, *, project_id: typing.Optional[str] = None
+    ) -> typing.List[PageScreenshotMetadata]:
+        """
+        List metadata for all screenshots of pages from a file.
+
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.files.list_file_page_screenshots(
+            id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}/page_screenshots"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[PageScreenshotMetadata], _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_file_page_screenshot(
+        self, id: str, page_index: int, *, project_id: typing.Optional[str] = None
+    ) -> typing.Any:
+        """
+        Get screenshot of a page from a file.
+
+        Parameters:
+            - id: str.
+
+            - page_index: int.
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.files.get_file_page_screenshot(
+            id="string",
+            page_index=1,
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}/page_screenshots/{page_index}"
+            ),
+            params=remove_none_from_dict({"project_id": project_id}),
+            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)

llama_cloud/resources/parsing/client.py

@@ -118,7 +118,12 @@ class ParsingClient:
         vendor_multimodal_api_key: str,
         page_prefix: str,
         page_suffix: str,
+        webhook_url: str,
         take_screenshot: bool,
+        disable_ocr: bool,
+        disable_reconstruction: bool,
+        input_s_3_path: str,
+        output_s_3_path_prefix: str,
         file: typing.IO,
     ) -> ParsingJob:
         """
@@ -159,8 +164,18 @@ class ParsingClient:
 
             - page_suffix: str.
 
+            - webhook_url: str.
+
             - take_screenshot: bool.
 
+            - disable_ocr: bool.
+
+            - disable_reconstruction: bool.
+
+            - input_s_3_path: str.
+
+            - output_s_3_path_prefix: str.
+
             - file: typing.IO.
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -185,7 +200,12 @@ class ParsingClient:
                     "vendor_multimodal_api_key": vendor_multimodal_api_key,
                     "page_prefix": page_prefix,
                     "page_suffix": page_suffix,
+                    "webhook_url": webhook_url,
                     "take_screenshot": take_screenshot,
+                    "disable_ocr": disable_ocr,
+                    "disable_reconstruction": disable_reconstruction,
+                    "input_s3_path": input_s_3_path,
+                    "output_s3_path_prefix": output_s_3_path_prefix,
                 }
             ),
             files={"file": file},
@@ -647,7 +667,12 @@ class AsyncParsingClient:
         vendor_multimodal_api_key: str,
         page_prefix: str,
         page_suffix: str,
+        webhook_url: str,
         take_screenshot: bool,
+        disable_ocr: bool,
+        disable_reconstruction: bool,
+        input_s_3_path: str,
+        output_s_3_path_prefix: str,
         file: typing.IO,
     ) -> ParsingJob:
         """
@@ -688,8 +713,18 @@ class AsyncParsingClient:
 
             - page_suffix: str.
 
+            - webhook_url: str.
+
             - take_screenshot: bool.
 
+            - disable_ocr: bool.
+
+            - disable_reconstruction: bool.
+
+            - input_s_3_path: str.
+
+            - output_s_3_path_prefix: str.
+
             - file: typing.IO.
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -714,7 +749,12 @@ class AsyncParsingClient:
                     "vendor_multimodal_api_key": vendor_multimodal_api_key,
                     "page_prefix": page_prefix,
                     "page_suffix": page_suffix,
+                    "webhook_url": webhook_url,
                     "take_screenshot": take_screenshot,
+                    "disable_ocr": disable_ocr,
+                    "disable_reconstruction": disable_reconstruction,
+                    "input_s3_path": input_s_3_path,
+                    "output_s3_path_prefix": output_s_3_path_prefix,
                 }
             ),
             files={"file": file},

llama_cloud/resources/pipelines/__init__.py

@@ -9,6 +9,7 @@ from .types import (
     PipelineUpdateEmbeddingConfig_GeminiEmbedding,
     PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
+    PipelineUpdateEmbeddingConfig_VertexaiEmbedding,
     PipelineUpdateTransformConfig,
     PipelineUpdateTransformConfig_Advanced,
     PipelineUpdateTransformConfig_Auto,
@@ -23,6 +24,7 @@ __all__ = [
     "PipelineUpdateEmbeddingConfig_GeminiEmbedding",
     "PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineUpdateEmbeddingConfig_OpenaiEmbedding",
+    "PipelineUpdateEmbeddingConfig_VertexaiEmbedding",
     "PipelineUpdateTransformConfig",
     "PipelineUpdateTransformConfig_Advanced",
     "PipelineUpdateTransformConfig_Auto",

llama_cloud/resources/pipelines/client.py

@@ -1212,6 +1212,7 @@ class PipelinesClient:
         search_filters: typing.Optional[MetadataFilters] = OMIT,
         files_top_k: typing.Optional[int] = OMIT,
         retrieval_mode: typing.Optional[RetrievalMode] = OMIT,
+        retrieve_image_nodes: typing.Optional[bool] = OMIT,
         query: str,
     ) -> RetrieveResults:
         """
@@ -1236,6 +1237,8 @@ class PipelinesClient:
 
             - retrieval_mode: typing.Optional[RetrievalMode]. The retrieval mode for the query.
 
+            - retrieve_image_nodes: typing.Optional[bool]. Whether to retrieve image nodes.
+
             - query: str. The query to retrieve against.
         ---
         from llama_cloud import FilterCondition, MetadataFilters, RetrievalMode
@@ -1271,6 +1274,8 @@ class PipelinesClient:
             _request["files_top_k"] = files_top_k
         if retrieval_mode is not OMIT:
             _request["retrieval_mode"] = retrieval_mode
+        if retrieve_image_nodes is not OMIT:
+            _request["retrieve_image_nodes"] = retrieve_image_nodes
         _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"),
@@ -1418,6 +1423,7 @@ class PipelinesClient:
             MetadataFilters,
             PresetRetrievalParams,
             RetrievalMode,
+            SupportedLlmModelNames,
         )
         from llama_cloud.client import LlamaCloud
 
@@ -1434,7 +1440,9 @@ class PipelinesClient:
                     ),
                     retrieval_mode=RetrievalMode.CHUNKS,
                 ),
-                llm_parameters=LlmParameters(),
+                llm_parameters=LlmParameters(
+                    model_name=SupportedLlmModelNames.GPT_3_5_TURBO,
+                ),
             ),
         )
         """
@@ -2910,6 +2918,7 @@ class AsyncPipelinesClient:
         search_filters: typing.Optional[MetadataFilters] = OMIT,
         files_top_k: typing.Optional[int] = OMIT,
         retrieval_mode: typing.Optional[RetrievalMode] = OMIT,
+        retrieve_image_nodes: typing.Optional[bool] = OMIT,
         query: str,
     ) -> RetrieveResults:
         """
@@ -2934,6 +2943,8 @@ class AsyncPipelinesClient:
 
             - retrieval_mode: typing.Optional[RetrievalMode]. The retrieval mode for the query.
 
+            - retrieve_image_nodes: typing.Optional[bool]. Whether to retrieve image nodes.
+
             - query: str. The query to retrieve against.
         ---
         from llama_cloud import FilterCondition, MetadataFilters, RetrievalMode
@@ -2969,6 +2980,8 @@ class AsyncPipelinesClient:
             _request["files_top_k"] = files_top_k
         if retrieval_mode is not OMIT:
             _request["retrieval_mode"] = retrieval_mode
+        if retrieve_image_nodes is not OMIT:
+            _request["retrieve_image_nodes"] = retrieve_image_nodes
         _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"),
@@ -3116,6 +3129,7 @@ class AsyncPipelinesClient:
             MetadataFilters,
             PresetRetrievalParams,
             RetrievalMode,
+            SupportedLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -3132,7 +3146,9 @@ class AsyncPipelinesClient:
                     ),
                     retrieval_mode=RetrievalMode.CHUNKS,
                 ),
-                llm_parameters=LlmParameters(),
+                llm_parameters=LlmParameters(
+                    model_name=SupportedLlmModelNames.GPT_3_5_TURBO,
+                ),
             ),
         )
         """

llama_cloud/resources/pipelines/types/__init__.py

@@ -9,6 +9,7 @@ from .pipeline_update_embedding_config import (
     PipelineUpdateEmbeddingConfig_GeminiEmbedding,
     PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
+    PipelineUpdateEmbeddingConfig_VertexaiEmbedding,
 )
 from .pipeline_update_transform_config import (
     PipelineUpdateTransformConfig,
@@ -25,6 +26,7 @@ __all__ = [
     "PipelineUpdateEmbeddingConfig_GeminiEmbedding",
     "PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineUpdateEmbeddingConfig_OpenaiEmbedding",
+    "PipelineUpdateEmbeddingConfig_VertexaiEmbedding",
     "PipelineUpdateTransformConfig",
     "PipelineUpdateTransformConfig_Advanced",
     "PipelineUpdateTransformConfig_Auto",

llama_cloud/resources/pipelines/types/pipeline_update_embedding_config.py

@@ -12,6 +12,7 @@ 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
+from ....types.vertex_ai_embedding_config import VertexAiEmbeddingConfig
 
 
 class PipelineUpdateEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
@@ -68,6 +69,15 @@ class PipelineUpdateEmbeddingConfig_CohereEmbedding(CohereEmbeddingConfig):
         allow_population_by_field_name = True
 
 
+class PipelineUpdateEmbeddingConfig_VertexaiEmbedding(VertexAiEmbeddingConfig):
+    type: typing_extensions.Literal["VERTEXAI_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
 PipelineUpdateEmbeddingConfig = typing.Union[
     PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
     PipelineUpdateEmbeddingConfig_AzureEmbedding,
@@ -75,4 +85,5 @@ PipelineUpdateEmbeddingConfig = typing.Union[
     PipelineUpdateEmbeddingConfig_BedrockEmbedding,
     PipelineUpdateEmbeddingConfig_GeminiEmbedding,
     PipelineUpdateEmbeddingConfig_CohereEmbedding,
+    PipelineUpdateEmbeddingConfig_VertexaiEmbedding,
 ]