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

llama_cloud/__init__.py

@@ -79,6 +79,7 @@ from .types import (
     EvalQuestion,
     EvalQuestionCreate,
     EvalQuestionResult,
+    ExtendVertexTextEmbedding,
     ExtractionJob,
     ExtractionResult,
     ExtractionResultDataValue,
@@ -124,6 +125,8 @@ from .types import (
     OpenAiEmbeddingConfig,
     Organization,
     OrganizationCreate,
+    PageScreenshotMetadata,
+    PageScreenshotNodeWithScore,
     PageSegmentationConfig,
     PageSplitterNodeParser,
     ParserLanguages,
@@ -134,6 +137,7 @@ from .types import (
     ParsingJobTextResult,
     ParsingUsage,
     Pipeline,
+    PipelineConfigurationHashes,
     PipelineCreate,
     PipelineCreateEmbeddingConfig,
     PipelineCreateEmbeddingConfig_AzureEmbedding,
@@ -142,6 +146,7 @@ from .types import (
     PipelineCreateEmbeddingConfig_GeminiEmbedding,
     PipelineCreateEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineCreateEmbeddingConfig_OpenaiEmbedding,
+    PipelineCreateEmbeddingConfig_VertexaiEmbedding,
     PipelineCreateTransformConfig,
     PipelineCreateTransformConfig_Advanced,
     PipelineCreateTransformConfig_Auto,
@@ -158,7 +163,9 @@ from .types import (
     PipelineEmbeddingConfig_GeminiEmbedding,
     PipelineEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineEmbeddingConfig_OpenaiEmbedding,
+    PipelineEmbeddingConfig_VertexaiEmbedding,
     PipelineFile,
+    PipelineFileConfigHashValue,
     PipelineFileCreate,
     PipelineFileCreateCustomMetadataValue,
     PipelineFileCustomMetadataValue,
@@ -197,6 +204,8 @@ from .types import (
     UserOrganizationDelete,
     ValidationError,
     ValidationErrorLocItem,
+    VertexAiEmbeddingConfig,
+    VertexEmbeddingMode,
 )
 from .errors import UnprocessableEntityError
 from .resources import (
@@ -216,6 +225,7 @@ from .resources import (
     PipelineUpdateEmbeddingConfig_GeminiEmbedding,
     PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
+    PipelineUpdateEmbeddingConfig_VertexaiEmbedding,
     PipelineUpdateTransformConfig,
     PipelineUpdateTransformConfig_Advanced,
     PipelineUpdateTransformConfig_Auto,
@@ -317,6 +327,7 @@ __all__ = [
     "EvalQuestion",
     "EvalQuestionCreate",
     "EvalQuestionResult",
+    "ExtendVertexTextEmbedding",
     "ExtractionJob",
     "ExtractionResult",
     "ExtractionResultDataValue",
@@ -366,6 +377,8 @@ __all__ = [
     "OpenAiEmbeddingConfig",
     "Organization",
     "OrganizationCreate",
+    "PageScreenshotMetadata",
+    "PageScreenshotNodeWithScore",
     "PageSegmentationConfig",
     "PageSplitterNodeParser",
     "ParserLanguages",
@@ -376,6 +389,7 @@ __all__ = [
     "ParsingJobTextResult",
     "ParsingUsage",
     "Pipeline",
+    "PipelineConfigurationHashes",
     "PipelineCreate",
     "PipelineCreateEmbeddingConfig",
     "PipelineCreateEmbeddingConfig_AzureEmbedding",
@@ -384,6 +398,7 @@ __all__ = [
     "PipelineCreateEmbeddingConfig_GeminiEmbedding",
     "PipelineCreateEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineCreateEmbeddingConfig_OpenaiEmbedding",
+    "PipelineCreateEmbeddingConfig_VertexaiEmbedding",
     "PipelineCreateTransformConfig",
     "PipelineCreateTransformConfig_Advanced",
     "PipelineCreateTransformConfig_Auto",
@@ -400,7 +415,9 @@ __all__ = [
     "PipelineEmbeddingConfig_GeminiEmbedding",
     "PipelineEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineEmbeddingConfig_OpenaiEmbedding",
+    "PipelineEmbeddingConfig_VertexaiEmbedding",
     "PipelineFile",
+    "PipelineFileConfigHashValue",
     "PipelineFileCreate",
     "PipelineFileCreateCustomMetadataValue",
     "PipelineFileCustomMetadataValue",
@@ -417,6 +434,7 @@ __all__ = [
     "PipelineUpdateEmbeddingConfig_GeminiEmbedding",
     "PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineUpdateEmbeddingConfig_OpenaiEmbedding",
+    "PipelineUpdateEmbeddingConfig_VertexaiEmbedding",
     "PipelineUpdateTransformConfig",
     "PipelineUpdateTransformConfig_Advanced",
     "PipelineUpdateTransformConfig_Auto",
@@ -451,6 +469,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/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,
 ]

llama_cloud/types/__init__.py

@@ -82,6 +82,7 @@ from .eval_execution_params_override import EvalExecutionParamsOverride
 from .eval_question import EvalQuestion
 from .eval_question_create import EvalQuestionCreate
 from .eval_question_result import EvalQuestionResult
+from .extend_vertex_text_embedding import ExtendVertexTextEmbedding
 from .extraction_job import ExtractionJob
 from .extraction_result import ExtractionResult
 from .extraction_result_data_value import ExtractionResultDataValue
@@ -127,6 +128,8 @@ from .open_ai_embedding import OpenAiEmbedding
 from .open_ai_embedding_config import OpenAiEmbeddingConfig
 from .organization import Organization
 from .organization_create import OrganizationCreate
+from .page_screenshot_metadata import PageScreenshotMetadata
+from .page_screenshot_node_with_score import PageScreenshotNodeWithScore
 from .page_segmentation_config import PageSegmentationConfig
 from .page_splitter_node_parser import PageSplitterNodeParser
 from .parser_languages import ParserLanguages
@@ -137,6 +140,7 @@ from .parsing_job_markdown_result import ParsingJobMarkdownResult
 from .parsing_job_text_result import ParsingJobTextResult
 from .parsing_usage import ParsingUsage
 from .pipeline import Pipeline
+from .pipeline_configuration_hashes import PipelineConfigurationHashes
 from .pipeline_create import PipelineCreate
 from .pipeline_create_embedding_config import (
     PipelineCreateEmbeddingConfig,
@@ -146,6 +150,7 @@ from .pipeline_create_embedding_config import (
     PipelineCreateEmbeddingConfig_GeminiEmbedding,
     PipelineCreateEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineCreateEmbeddingConfig_OpenaiEmbedding,
+    PipelineCreateEmbeddingConfig_VertexaiEmbedding,
 )
 from .pipeline_create_transform_config import (
     PipelineCreateTransformConfig,
@@ -166,8 +171,10 @@ from .pipeline_embedding_config import (
     PipelineEmbeddingConfig_GeminiEmbedding,
     PipelineEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineEmbeddingConfig_OpenaiEmbedding,
+    PipelineEmbeddingConfig_VertexaiEmbedding,
 )
 from .pipeline_file import PipelineFile
+from .pipeline_file_config_hash_value import PipelineFileConfigHashValue
 from .pipeline_file_create import PipelineFileCreate
 from .pipeline_file_create_custom_metadata_value import PipelineFileCreateCustomMetadataValue
 from .pipeline_file_custom_metadata_value import PipelineFileCustomMetadataValue
@@ -208,6 +215,8 @@ from .user_organization_create import UserOrganizationCreate
 from .user_organization_delete import UserOrganizationDelete
 from .validation_error import ValidationError
 from .validation_error_loc_item import ValidationErrorLocItem
+from .vertex_ai_embedding_config import VertexAiEmbeddingConfig
+from .vertex_embedding_mode import VertexEmbeddingMode
 
 __all__ = [
     "AdvancedModeTransformConfig",
@@ -288,6 +297,7 @@ __all__ = [
     "EvalQuestion",
     "EvalQuestionCreate",
     "EvalQuestionResult",
+    "ExtendVertexTextEmbedding",
     "ExtractionJob",
     "ExtractionResult",
     "ExtractionResultDataValue",
@@ -333,6 +343,8 @@ __all__ = [
     "OpenAiEmbeddingConfig",
     "Organization",
     "OrganizationCreate",
+    "PageScreenshotMetadata",
+    "PageScreenshotNodeWithScore",
     "PageSegmentationConfig",
     "PageSplitterNodeParser",
     "ParserLanguages",
@@ -343,6 +355,7 @@ __all__ = [
     "ParsingJobTextResult",
     "ParsingUsage",
     "Pipeline",
+    "PipelineConfigurationHashes",
     "PipelineCreate",
     "PipelineCreateEmbeddingConfig",
     "PipelineCreateEmbeddingConfig_AzureEmbedding",
@@ -351,6 +364,7 @@ __all__ = [
     "PipelineCreateEmbeddingConfig_GeminiEmbedding",
     "PipelineCreateEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineCreateEmbeddingConfig_OpenaiEmbedding",
+    "PipelineCreateEmbeddingConfig_VertexaiEmbedding",
     "PipelineCreateTransformConfig",
     "PipelineCreateTransformConfig_Advanced",
     "PipelineCreateTransformConfig_Auto",
@@ -367,7 +381,9 @@ __all__ = [
     "PipelineEmbeddingConfig_GeminiEmbedding",
     "PipelineEmbeddingConfig_HuggingfaceApiEmbedding",
     "PipelineEmbeddingConfig_OpenaiEmbedding",
+    "PipelineEmbeddingConfig_VertexaiEmbedding",
     "PipelineFile",
+    "PipelineFileConfigHashValue",
     "PipelineFileCreate",
     "PipelineFileCreateCustomMetadataValue",
     "PipelineFileCustomMetadataValue",
@@ -406,4 +422,6 @@ __all__ = [
     "UserOrganizationDelete",
     "ValidationError",
     "ValidationErrorLocItem",
+    "VertexAiEmbeddingConfig",
+    "VertexEmbeddingMode",
 ]

llama_cloud/types/cloud_az_storage_blob_data_source.py

@@ -21,9 +21,8 @@ class CloudAzStorageBlobDataSource(pydantic.BaseModel):
 
     container_name: str = pydantic.Field(description="The name of the Azure Storage Blob container to read from.")
     account_url: str = pydantic.Field(description="The Azure Storage Blob account URL to use for authentication.")
-    blob: typing.Optional[str] = pydantic.Field(description="The blob name to read from.")
     prefix: typing.Optional[str] = pydantic.Field(
-        description="The prefix of the Azure Storage Blob objects to read from."
+        description="The prefix of the Azure Storage Blob objects to read from. Use this to filter files at the subdirectory level"
     )
     account_name: typing.Optional[str] = pydantic.Field(
         description="The Azure Storage Blob account name to use for authentication."

llama_cloud/types/cloud_postgres_vector_store.py

@@ -20,17 +20,15 @@ class CloudPostgresVectorStore(pydantic.BaseModel):
     """
 
     supports_nested_metadata_filters: typing.Optional[bool]
-    connection_string: str
-    async_connection_string: str
+    database: str
+    host: str
+    password: str
+    port: str
+    user: str
     table_name: str
     schema_name: str
     embed_dim: int
-    hybrid_search: bool
-    text_search_config: str
-    cache_ok: bool
-    perform_setup: bool
-    debug: bool
-    use_jsonb: bool
+    hybrid_search: typing.Optional[bool]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str: