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

llama_cloud/resources/files/client.py

@@ -31,7 +31,7 @@ class FilesClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def read_file(self, id: str, *, project_id: typing.Optional[str] = None) -> File:
+    def get_file(self, id: str, *, project_id: typing.Optional[str] = None) -> File:
         """
         Read File metadata objects.
 
@@ -45,7 +45,7 @@ class FilesClient:
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.files.read_file(
+        client.files.get_file(
             id="string",
         )
         """
@@ -101,7 +101,7 @@ class FilesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def read_files(self, *, project_id: typing.Optional[str] = None) -> typing.List[File]:
+    def list_files(self, *, project_id: typing.Optional[str] = None) -> typing.List[File]:
         """
         Read File metadata objects.
 
@@ -113,7 +113,7 @@ class FilesClient:
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.files.read_files()
+        client.files.list_files()
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
@@ -293,7 +293,7 @@ class AsyncFilesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def read_file(self, id: str, *, project_id: typing.Optional[str] = None) -> File:
+    async def get_file(self, id: str, *, project_id: typing.Optional[str] = None) -> File:
         """
         Read File metadata objects.
 
@@ -307,7 +307,7 @@ class AsyncFilesClient:
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.files.read_file(
+        await client.files.get_file(
             id="string",
         )
         """
@@ -363,7 +363,7 @@ class AsyncFilesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def read_files(self, *, project_id: typing.Optional[str] = None) -> typing.List[File]:
+    async def list_files(self, *, project_id: typing.Optional[str] = None) -> typing.List[File]:
         """
         Read File metadata objects.
 
@@ -375,7 +375,7 @@ class AsyncFilesClient:
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.files.read_files()
+        await client.files.list_files()
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",

llama_cloud/resources/parsing/client.py

@@ -111,6 +111,8 @@ class ParsingClient:
         gpt_4_o_api_key: str,
         do_not_unroll_columns: bool,
         page_separator: str,
+        bounding_box: str,
+        target_pages: str,
         file: typing.IO,
     ) -> ParsingJob:
         """
@@ -137,6 +139,10 @@ class ParsingClient:
 
             - page_separator: str.
 
+            - bounding_box: str.
+
+            - target_pages: str.
+
             - file: typing.IO.
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -154,6 +160,8 @@ class ParsingClient:
                     "gpt4o_api_key": gpt_4_o_api_key,
                     "do_not_unroll_columns": do_not_unroll_columns,
                     "page_separator": page_separator,
+                    "bounding_box": bounding_box,
+                    "target_pages": target_pages,
                 }
             ),
             files={"file": file},
@@ -576,6 +584,8 @@ class AsyncParsingClient:
         gpt_4_o_api_key: str,
         do_not_unroll_columns: bool,
         page_separator: str,
+        bounding_box: str,
+        target_pages: str,
         file: typing.IO,
     ) -> ParsingJob:
         """
@@ -602,6 +612,10 @@ class AsyncParsingClient:
 
             - page_separator: str.
 
+            - bounding_box: str.
+
+            - target_pages: str.
+
             - file: typing.IO.
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -619,6 +633,8 @@ class AsyncParsingClient:
                     "gpt4o_api_key": gpt_4_o_api_key,
                     "do_not_unroll_columns": do_not_unroll_columns,
                     "page_separator": page_separator,
+                    "bounding_box": bounding_box,
+                    "target_pages": target_pages,
                 }
             ),
             files={"file": file},

llama_cloud/resources/pipelines/client.py

@@ -644,25 +644,38 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_files_for_pipeline(self, pipeline_id: str) -> typing.List[PipelineFile]:
+    def list_pipeline_files(
+        self,
+        pipeline_id: str,
+        *,
+        data_source_id: typing.Optional[str] = None,
+        only_manually_uploaded: typing.Optional[bool] = None,
+    ) -> typing.List[PipelineFile]:
         """
         Get files for a pipeline.
 
         Parameters:
             - pipeline_id: str.
+
+            - data_source_id: typing.Optional[str].
+
+            - only_manually_uploaded: typing.Optional[bool].
         ---
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.pipelines.get_files_for_pipeline(
+        client.pipelines.list_pipeline_files(
             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}/files"),
+            params=remove_none_from_dict(
+                {"data_source_id": data_source_id, "only_manually_uploaded": only_manually_uploaded}
+            ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -837,7 +850,66 @@ 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_sources(self, pipeline_id: str) -> typing.List[PipelineDataSource]:
+    def import_pipeline_metadata(self, pipeline_id: str, *, upload_file: typing.IO) -> typing.Dict[str, str]:
+        """
+        Import metadata for a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - upload_file: typing.IO.
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/metadata"),
+            data=jsonable_encoder({}),
+            files={"upload_file": upload_file},
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Dict[str, str], _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_files_metadata(self, pipeline_id: str) -> None:
+        """
+        Delete metadata for all files in a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.delete_pipeline_files_metadata(
+            pipeline_id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "DELETE",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/metadata"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def list_pipeline_data_sources(self, pipeline_id: str) -> typing.List[PipelineDataSource]:
         """
         Get data sources for a pipeline.
 
@@ -849,7 +921,7 @@ class PipelinesClient:
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.pipelines.get_pipeline_data_sources(
+        client.pipelines.list_pipeline_data_sources(
             pipeline_id="string",
         )
         """
@@ -1064,7 +1136,7 @@ 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_jobs(self, pipeline_id: str) -> typing.List[PipelineDeployment]:
+    def list_pipeline_jobs(self, pipeline_id: str) -> typing.List[PipelineDeployment]:
         """
         Get jobs for a pipeline.
 
@@ -1076,7 +1148,7 @@ class PipelinesClient:
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.pipelines.get_pipeline_jobs(
+        client.pipelines.list_pipeline_jobs(
             pipeline_id="string",
         )
         """
@@ -1968,25 +2040,38 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_files_for_pipeline(self, pipeline_id: str) -> typing.List[PipelineFile]:
+    async def list_pipeline_files(
+        self,
+        pipeline_id: str,
+        *,
+        data_source_id: typing.Optional[str] = None,
+        only_manually_uploaded: typing.Optional[bool] = None,
+    ) -> typing.List[PipelineFile]:
         """
         Get files for a pipeline.
 
         Parameters:
             - pipeline_id: str.
+
+            - data_source_id: typing.Optional[str].
+
+            - only_manually_uploaded: typing.Optional[bool].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.pipelines.get_files_for_pipeline(
+        await client.pipelines.list_pipeline_files(
             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}/files"),
+            params=remove_none_from_dict(
+                {"data_source_id": data_source_id, "only_manually_uploaded": only_manually_uploaded}
+            ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -2161,7 +2246,66 @@ 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_sources(self, pipeline_id: str) -> typing.List[PipelineDataSource]:
+    async def import_pipeline_metadata(self, pipeline_id: str, *, upload_file: typing.IO) -> typing.Dict[str, str]:
+        """
+        Import metadata for a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - upload_file: typing.IO.
+        """
+        _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}/metadata"),
+            data=jsonable_encoder({}),
+            files={"upload_file": upload_file},
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Dict[str, str], _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_files_metadata(self, pipeline_id: str) -> None:
+        """
+        Delete metadata for all files in a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.delete_pipeline_files_metadata(
+            pipeline_id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "DELETE",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/metadata"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def list_pipeline_data_sources(self, pipeline_id: str) -> typing.List[PipelineDataSource]:
         """
         Get data sources for a pipeline.
 
@@ -2173,7 +2317,7 @@ class AsyncPipelinesClient:
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.pipelines.get_pipeline_data_sources(
+        await client.pipelines.list_pipeline_data_sources(
             pipeline_id="string",
         )
         """
@@ -2388,7 +2532,7 @@ 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_jobs(self, pipeline_id: str) -> typing.List[PipelineDeployment]:
+    async def list_pipeline_jobs(self, pipeline_id: str) -> typing.List[PipelineDeployment]:
         """
         Get jobs for a pipeline.
 
@@ -2400,7 +2544,7 @@ class AsyncPipelinesClient:
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.pipelines.get_pipeline_jobs(
+        await client.pipelines.list_pipeline_jobs(
             pipeline_id="string",
         )
         """

llama_cloud/resources/projects/client.py

@@ -238,9 +238,9 @@ class ProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_datasets_for_project(self, project_id: str) -> typing.List[EvalDataset]:
+    def list_datasets_for_project(self, project_id: str) -> typing.List[EvalDataset]:
         """
-        Get all eval datasets for a project.
+        List eval datasets for a project.
 
         Parameters:
             - project_id: str.
@@ -250,7 +250,7 @@ class ProjectsClient:
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.projects.get_datasets_for_project(
+        client.projects.list_datasets_for_project(
             project_id="string",
         )
         """
@@ -353,9 +353,9 @@ class ProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_local_evals_for_project(self, project_id: str) -> typing.List[LocalEvalResults]:
+    def list_local_evals_for_project(self, project_id: str) -> typing.List[LocalEvalResults]:
         """
-        Get all local eval results for a project.
+        List local eval results for a project.
 
         Parameters:
             - project_id: str.
@@ -365,7 +365,7 @@ class ProjectsClient:
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.projects.get_local_evals_for_project(
+        client.projects.list_local_evals_for_project(
             project_id="string",
         )
         """
@@ -385,9 +385,9 @@ class ProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_local_eval_sets_for_project(self, project_id: str) -> typing.List[LocalEvalSets]:
+    def list_local_eval_sets_for_project(self, project_id: str) -> typing.List[LocalEvalSets]:
         """
-        Get all local eval sets for a project.
+        List local eval sets for a project.
 
         Parameters:
             - project_id: str.
@@ -397,7 +397,7 @@ class ProjectsClient:
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.projects.get_local_eval_sets_for_project(
+        client.projects.list_local_eval_sets_for_project(
             project_id="string",
         )
         """
@@ -457,9 +457,9 @@ class ProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_promptmixin_prompts(self, project_id: str) -> typing.List[PromptMixinPrompts]:
+    def list_promptmixin_prompts(self, project_id: str) -> typing.List[PromptMixinPrompts]:
         """
-        Get all PromptMixin prompt sets for a project.
+        List PromptMixin prompt sets for a project.
 
         Parameters:
             - project_id: str.
@@ -469,7 +469,7 @@ class ProjectsClient:
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.projects.get_promptmixin_prompts(
+        client.projects.list_promptmixin_prompts(
             project_id="string",
         )
         """
@@ -824,9 +824,9 @@ class AsyncProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_datasets_for_project(self, project_id: str) -> typing.List[EvalDataset]:
+    async def list_datasets_for_project(self, project_id: str) -> typing.List[EvalDataset]:
         """
-        Get all eval datasets for a project.
+        List eval datasets for a project.
 
         Parameters:
             - project_id: str.
@@ -836,7 +836,7 @@ class AsyncProjectsClient:
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.projects.get_datasets_for_project(
+        await client.projects.list_datasets_for_project(
             project_id="string",
         )
         """
@@ -939,9 +939,9 @@ class AsyncProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_local_evals_for_project(self, project_id: str) -> typing.List[LocalEvalResults]:
+    async def list_local_evals_for_project(self, project_id: str) -> typing.List[LocalEvalResults]:
         """
-        Get all local eval results for a project.
+        List local eval results for a project.
 
         Parameters:
             - project_id: str.
@@ -951,7 +951,7 @@ class AsyncProjectsClient:
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.projects.get_local_evals_for_project(
+        await client.projects.list_local_evals_for_project(
             project_id="string",
         )
         """
@@ -971,9 +971,9 @@ class AsyncProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_local_eval_sets_for_project(self, project_id: str) -> typing.List[LocalEvalSets]:
+    async def list_local_eval_sets_for_project(self, project_id: str) -> typing.List[LocalEvalSets]:
         """
-        Get all local eval sets for a project.
+        List local eval sets for a project.
 
         Parameters:
             - project_id: str.
@@ -983,7 +983,7 @@ class AsyncProjectsClient:
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.projects.get_local_eval_sets_for_project(
+        await client.projects.list_local_eval_sets_for_project(
             project_id="string",
         )
         """
@@ -1043,9 +1043,9 @@ class AsyncProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_promptmixin_prompts(self, project_id: str) -> typing.List[PromptMixinPrompts]:
+    async def list_promptmixin_prompts(self, project_id: str) -> typing.List[PromptMixinPrompts]:
         """
-        Get all PromptMixin prompt sets for a project.
+        List PromptMixin prompt sets for a project.
 
         Parameters:
             - project_id: str.
@@ -1055,7 +1055,7 @@ class AsyncProjectsClient:
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.projects.get_promptmixin_prompts(
+        await client.projects.list_promptmixin_prompts(
             project_id="string",
         )
         """

llama_cloud/types/__init__.py

@@ -6,17 +6,18 @@ from .base_prompt_template import BasePromptTemplate
 from .bedrock_embedding import BedrockEmbedding
 from .chat_message import ChatMessage
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
+from .cloud_azure_ai_search_vector_store import CloudAzureAiSearchVectorStore
 from .cloud_chroma_vector_store import CloudChromaVectorStore
 from .cloud_document import CloudDocument
 from .cloud_document_create import CloudDocumentCreate
-from .cloud_gcs_data_source import CloudGcsDataSource
-from .cloud_google_drive_data_source import CloudGoogleDriveDataSource
+from .cloud_notion_page_data_source import CloudNotionPageDataSource
 from .cloud_one_drive_data_source import CloudOneDriveDataSource
 from .cloud_pinecone_vector_store import CloudPineconeVectorStore
 from .cloud_postgres_vector_store import CloudPostgresVectorStore
 from .cloud_qdrant_vector_store import CloudQdrantVectorStore
 from .cloud_s_3_data_source import CloudS3DataSource
 from .cloud_sharepoint_data_source import CloudSharepointDataSource
+from .cloud_slack_data_source import CloudSlackDataSource
 from .cloud_weaviate_vector_store import CloudWeaviateVectorStore
 from .code_splitter import CodeSplitter
 from .cohere_embedding import CohereEmbedding
@@ -52,6 +53,10 @@ from .eval_llm_model_data import EvalLlmModelData
 from .eval_question import EvalQuestion
 from .eval_question_create import EvalQuestionCreate
 from .eval_question_result import EvalQuestionResult
+from .extraction_result import ExtractionResult
+from .extraction_result_data_value import ExtractionResultDataValue
+from .extraction_schema import ExtractionSchema
+from .extraction_schema_data_schema_value import ExtractionSchemaDataSchemaValue
 from .file import File
 from .file_resource_info_value import FileResourceInfoValue
 from .filter_condition import FilterCondition
@@ -132,17 +137,18 @@ __all__ = [
     "BedrockEmbedding",
     "ChatMessage",
     "CloudAzStorageBlobDataSource",
+    "CloudAzureAiSearchVectorStore",
     "CloudChromaVectorStore",
     "CloudDocument",
     "CloudDocumentCreate",
-    "CloudGcsDataSource",
-    "CloudGoogleDriveDataSource",
+    "CloudNotionPageDataSource",
     "CloudOneDriveDataSource",
     "CloudPineconeVectorStore",
     "CloudPostgresVectorStore",
     "CloudQdrantVectorStore",
     "CloudS3DataSource",
     "CloudSharepointDataSource",
+    "CloudSlackDataSource",
     "CloudWeaviateVectorStore",
     "CodeSplitter",
     "CohereEmbedding",
@@ -178,6 +184,10 @@ __all__ = [
     "EvalQuestion",
     "EvalQuestionCreate",
     "EvalQuestionResult",
+    "ExtractionResult",
+    "ExtractionResultDataValue",
+    "ExtractionSchema",
+    "ExtractionSchemaDataSchemaValue",
     "File",
     "FileResourceInfoValue",
     "FilterCondition",

llama_cloud/types/azure_open_ai_embedding.py

@@ -62,6 +62,9 @@ class AzureOpenAiEmbedding(pydantic.BaseModel):
     )
     azure_endpoint: typing.Optional[str] = pydantic.Field(description="The Azure endpoint to use.")
     azure_deployment: typing.Optional[str] = pydantic.Field(description="The Azure deployment to use.")
+    use_azure_ad: bool = pydantic.Field(
+        description="Indicates if Microsoft Entra ID (former Azure AD) is used for token authentication"
+    )
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/{cloud_gcs_data_source.py → cloud_azure_ai_search_vector_store.py}

@@ -14,16 +14,18 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class CloudGcsDataSource(pydantic.BaseModel):
+class CloudAzureAiSearchVectorStore(pydantic.BaseModel):
     """
-    Base component object to capture class names.
+    Cloud Azure AI Search Vector Store.
     """
 
-    bucket: str = pydantic.Field(description="The name of the GCS bucket to read from.")
-    prefix: typing.Optional[str] = pydantic.Field(description="The prefix of the GCS objects to read from.")
-    service_account_key: typing.Dict[str, typing.Any] = pydantic.Field(
-        description="The service account key JSON to use for authentication."
-    )
+    supports_nested_metadata_filters: typing.Optional[bool]
+    search_service_api_key: str
+    search_service_endpoint: str
+    search_service_api_version: typing.Optional[str]
+    index_name: typing.Optional[str]
+    filterable_metadata_field_keys: typing.Optional[typing.List[str]]
+    embedding_dimension: typing.Optional[int]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str: