llama-cloud 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

llama_cloud/resources/pipelines/client.py

@@ -23,6 +23,7 @@ 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
+from ...types.paginated_list_pipeline_files_response import PaginatedListPipelineFilesResponse
 from ...types.pipeline import Pipeline
 from ...types.pipeline_create import PipelineCreate
 from ...types.pipeline_data_source import PipelineDataSource
@@ -59,22 +60,25 @@ class PipelinesClient:
     def search_pipelines(
         self,
         *,
+        project_id: typing.Optional[str] = None,
         project_name: typing.Optional[str] = None,
         pipeline_name: typing.Optional[str] = None,
         pipeline_type: typing.Optional[PipelineType] = None,
-        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
     ) -> typing.List[Pipeline]:
         """
         Search for pipelines by various parameters.
 
         Parameters:
+            - project_id: typing.Optional[str].
+
             - project_name: typing.Optional[str].
 
             - pipeline_name: typing.Optional[str].
 
             - pipeline_type: typing.Optional[PipelineType].
 
-            - project_id: typing.Optional[str].
+            - organization_id: typing.Optional[str].
         ---
         from llama_cloud import PipelineType
         from llama_cloud.client import LlamaCloud
@@ -91,10 +95,11 @@ class PipelinesClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/pipelines"),
             params=remove_none_from_dict(
                 {
+                    "project_id": project_id,
                     "project_name": project_name,
                     "pipeline_name": pipeline_name,
                     "pipeline_type": pipeline_type,
-                    "project_id": project_id,
+                    "organization_id": organization_id,
                 }
             ),
             headers=self._client_wrapper.get_headers(),
@@ -110,19 +115,27 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def create_pipeline(self, *, project_id: typing.Optional[str] = None, request: PipelineCreate) -> Pipeline:
+    def create_pipeline(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: PipelineCreate,
+    ) -> Pipeline:
         """
         Create a new pipeline for a project.
 
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: PipelineCreate.
         """
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/pipelines"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -137,7 +150,13 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def upsert_pipeline(self, *, project_id: typing.Optional[str] = None, request: PipelineCreate) -> Pipeline:
+    def upsert_pipeline(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: PipelineCreate,
+    ) -> Pipeline:
         """
         Upsert a pipeline for a project.
         Updates if a pipeline with the same name and project_id already exists. Otherwise, creates a new pipeline.
@@ -145,12 +164,14 @@ class PipelinesClient:
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: PipelineCreate.
         """
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/pipelines"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -196,6 +217,7 @@ class PipelinesClient:
         transform_config: typing.Optional[PipelineUpdateTransformConfig] = OMIT,
         configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]] = OMIT,
         data_sink_id: typing.Optional[str] = OMIT,
+        embedding_model_config_id: typing.Optional[str] = OMIT,
         data_sink: typing.Optional[DataSinkCreate] = OMIT,
         preset_retrieval_parameters: typing.Optional[PresetRetrievalParams] = OMIT,
         eval_parameters: typing.Optional[EvalExecutionParams] = OMIT,
@@ -217,6 +239,8 @@ class PipelinesClient:
 
             - data_sink_id: typing.Optional[str].
 
+            - embedding_model_config_id: typing.Optional[str].
+
             - data_sink: typing.Optional[DataSinkCreate].
 
             - preset_retrieval_parameters: typing.Optional[PresetRetrievalParams].
@@ -238,6 +262,8 @@ class PipelinesClient:
             _request["configured_transformations"] = configured_transformations
         if data_sink_id is not OMIT:
             _request["data_sink_id"] = data_sink_id
+        if embedding_model_config_id is not OMIT:
+            _request["embedding_model_config_id"] = embedding_model_config_id
         if data_sink is not OMIT:
             _request["data_sink"] = data_sink
         if preset_retrieval_parameters is not OMIT:
@@ -642,6 +668,62 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def list_pipeline_files_2(
+        self,
+        pipeline_id: str,
+        *,
+        data_source_id: typing.Optional[str] = None,
+        only_manually_uploaded: typing.Optional[bool] = None,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+    ) -> PaginatedListPipelineFilesResponse:
+        """
+        Get files for a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - data_source_id: typing.Optional[str].
+
+            - only_manually_uploaded: typing.Optional[bool].
+
+            - limit: typing.Optional[int].
+
+            - offset: typing.Optional[int].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.list_pipeline_files_2(
+            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}/files2"),
+            params=remove_none_from_dict(
+                {
+                    "data_source_id": data_source_id,
+                    "only_manually_uploaded": only_manually_uploaded,
+                    "limit": limit,
+                    "offset": offset,
+                }
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PaginatedListPipelineFilesResponse, _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_pipeline_file_status(self, pipeline_id: str, file_id: str) -> ManagedIngestionStatusResponse:
         """
         Get status of a file for a pipeline.
@@ -1616,22 +1698,25 @@ class AsyncPipelinesClient:
     async def search_pipelines(
         self,
         *,
+        project_id: typing.Optional[str] = None,
         project_name: typing.Optional[str] = None,
         pipeline_name: typing.Optional[str] = None,
         pipeline_type: typing.Optional[PipelineType] = None,
-        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
     ) -> typing.List[Pipeline]:
         """
         Search for pipelines by various parameters.
 
         Parameters:
+            - project_id: typing.Optional[str].
+
             - project_name: typing.Optional[str].
 
             - pipeline_name: typing.Optional[str].
 
             - pipeline_type: typing.Optional[PipelineType].
 
-            - project_id: typing.Optional[str].
+            - organization_id: typing.Optional[str].
         ---
         from llama_cloud import PipelineType
         from llama_cloud.client import AsyncLlamaCloud
@@ -1648,10 +1733,11 @@ class AsyncPipelinesClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/pipelines"),
             params=remove_none_from_dict(
                 {
+                    "project_id": project_id,
                     "project_name": project_name,
                     "pipeline_name": pipeline_name,
                     "pipeline_type": pipeline_type,
-                    "project_id": project_id,
+                    "organization_id": organization_id,
                 }
             ),
             headers=self._client_wrapper.get_headers(),
@@ -1667,19 +1753,27 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def create_pipeline(self, *, project_id: typing.Optional[str] = None, request: PipelineCreate) -> Pipeline:
+    async def create_pipeline(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: PipelineCreate,
+    ) -> Pipeline:
         """
         Create a new pipeline for a project.
 
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: PipelineCreate.
         """
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/pipelines"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -1694,7 +1788,13 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def upsert_pipeline(self, *, project_id: typing.Optional[str] = None, request: PipelineCreate) -> Pipeline:
+    async def upsert_pipeline(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: PipelineCreate,
+    ) -> Pipeline:
         """
         Upsert a pipeline for a project.
         Updates if a pipeline with the same name and project_id already exists. Otherwise, creates a new pipeline.
@@ -1702,12 +1802,14 @@ class AsyncPipelinesClient:
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: PipelineCreate.
         """
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/pipelines"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -1753,6 +1855,7 @@ class AsyncPipelinesClient:
         transform_config: typing.Optional[PipelineUpdateTransformConfig] = OMIT,
         configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]] = OMIT,
         data_sink_id: typing.Optional[str] = OMIT,
+        embedding_model_config_id: typing.Optional[str] = OMIT,
         data_sink: typing.Optional[DataSinkCreate] = OMIT,
         preset_retrieval_parameters: typing.Optional[PresetRetrievalParams] = OMIT,
         eval_parameters: typing.Optional[EvalExecutionParams] = OMIT,
@@ -1774,6 +1877,8 @@ class AsyncPipelinesClient:
 
             - data_sink_id: typing.Optional[str].
 
+            - embedding_model_config_id: typing.Optional[str].
+
             - data_sink: typing.Optional[DataSinkCreate].
 
             - preset_retrieval_parameters: typing.Optional[PresetRetrievalParams].
@@ -1795,6 +1900,8 @@ class AsyncPipelinesClient:
             _request["configured_transformations"] = configured_transformations
         if data_sink_id is not OMIT:
             _request["data_sink_id"] = data_sink_id
+        if embedding_model_config_id is not OMIT:
+            _request["embedding_model_config_id"] = embedding_model_config_id
         if data_sink is not OMIT:
             _request["data_sink"] = data_sink
         if preset_retrieval_parameters is not OMIT:
@@ -2201,6 +2308,62 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def list_pipeline_files_2(
+        self,
+        pipeline_id: str,
+        *,
+        data_source_id: typing.Optional[str] = None,
+        only_manually_uploaded: typing.Optional[bool] = None,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+    ) -> PaginatedListPipelineFilesResponse:
+        """
+        Get files for a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - data_source_id: typing.Optional[str].
+
+            - only_manually_uploaded: typing.Optional[bool].
+
+            - limit: typing.Optional[int].
+
+            - offset: typing.Optional[int].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.list_pipeline_files_2(
+            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}/files2"),
+            params=remove_none_from_dict(
+                {
+                    "data_source_id": data_source_id,
+                    "only_manually_uploaded": only_manually_uploaded,
+                    "limit": limit,
+                    "offset": offset,
+                }
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PaginatedListPipelineFilesResponse, _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_pipeline_file_status(self, pipeline_id: str, file_id: str) -> ManagedIngestionStatusResponse:
         """
         Get status of a file for a pipeline.

llama_cloud/resources/projects/client.py

@@ -11,6 +11,7 @@ from ...core.remove_none_from_dict import remove_none_from_dict
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.eval_dataset import EvalDataset
 from ...types.http_validation_error import HttpValidationError
+from ...types.interval_usage_and_plan import IntervalUsageAndPlan
 from ...types.local_eval import LocalEval
 from ...types.local_eval_results import LocalEvalResults
 from ...types.local_eval_sets import LocalEvalSets
@@ -248,6 +249,41 @@ class ProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def get_project_usage(
+        self, project_id: typing.Optional[str], *, organization_id: typing.Optional[str] = None
+    ) -> IntervalUsageAndPlan:
+        """
+        Get usage for a project
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.projects.get_project_usage()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/projects/{project_id}/usage"),
+            params=remove_none_from_dict({"organization_id": organization_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(IntervalUsageAndPlan, _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 list_datasets_for_project(self, project_id: str) -> typing.List[EvalDataset]:
         """
         List eval datasets for a project.
@@ -844,6 +880,41 @@ 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_project_usage(
+        self, project_id: typing.Optional[str], *, organization_id: typing.Optional[str] = None
+    ) -> IntervalUsageAndPlan:
+        """
+        Get usage for a project
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.projects.get_project_usage()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/projects/{project_id}/usage"),
+            params=remove_none_from_dict({"organization_id": organization_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(IntervalUsageAndPlan, _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 list_datasets_for_project(self, project_id: str) -> typing.List[EvalDataset]:
         """
         List eval datasets for a project.

llama_cloud/types/__init__.py

@@ -67,6 +67,28 @@ from .data_source_create_custom_metadata_value import DataSourceCreateCustomMeta
 from .data_source_custom_metadata_value import DataSourceCustomMetadataValue
 from .data_source_definition import DataSourceDefinition
 from .element_segmentation_config import ElementSegmentationConfig
+from .embedding_model_config import EmbeddingModelConfig
+from .embedding_model_config_embedding_config import (
+    EmbeddingModelConfigEmbeddingConfig,
+    EmbeddingModelConfigEmbeddingConfig_AzureEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_BedrockEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_CohereEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_GeminiEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_HuggingfaceApiEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_OpenaiEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_VertexaiEmbedding,
+)
+from .embedding_model_config_update import EmbeddingModelConfigUpdate
+from .embedding_model_config_update_embedding_config import (
+    EmbeddingModelConfigUpdateEmbeddingConfig,
+    EmbeddingModelConfigUpdateEmbeddingConfig_AzureEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_BedrockEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_CohereEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_GeminiEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_OpenaiEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_VertexaiEmbedding,
+)
 from .eval_dataset import EvalDataset
 from .eval_dataset_job_params import EvalDatasetJobParams
 from .eval_dataset_job_record import EvalDatasetJobRecord
@@ -93,6 +115,7 @@ from .hugging_face_inference_api_embedding_config import HuggingFaceInferenceApi
 from .hugging_face_inference_api_embedding_token import HuggingFaceInferenceApiEmbeddingToken
 from .ingestion_error_response import IngestionErrorResponse
 from .input_message import InputMessage
+from .interval_usage_and_plan import IntervalUsageAndPlan
 from .job_name_mapping import JobNameMapping
 from .llama_parse_parameters import LlamaParseParameters
 from .llama_parse_supported_file_extensions import LlamaParseSupportedFileExtensions
@@ -126,6 +149,7 @@ 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 .paginated_list_pipeline_files_response import PaginatedListPipelineFilesResponse
 from .parser_languages import ParserLanguages
 from .parsing_history_item import ParsingHistoryItem
 from .parsing_job import ParsingJob
@@ -176,6 +200,7 @@ from .pipeline_transform_config import (
     PipelineTransformConfig_Auto,
 )
 from .pipeline_type import PipelineType
+from .plan import Plan
 from .playground_session import PlaygroundSession
 from .pooling import Pooling
 from .preset_retrieval_params import PresetRetrievalParams
@@ -201,6 +226,7 @@ from .text_node_with_score import TextNodeWithScore
 from .token_chunking_config import TokenChunkingConfig
 from .token_text_splitter import TokenTextSplitter
 from .transformation_category_names import TransformationCategoryNames
+from .usage import Usage
 from .user_organization import UserOrganization
 from .user_organization_create import UserOrganizationCreate
 from .user_organization_delete import UserOrganizationDelete
@@ -275,6 +301,24 @@ __all__ = [
     "DataSourceCustomMetadataValue",
     "DataSourceDefinition",
     "ElementSegmentationConfig",
+    "EmbeddingModelConfig",
+    "EmbeddingModelConfigEmbeddingConfig",
+    "EmbeddingModelConfigEmbeddingConfig_AzureEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_BedrockEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_CohereEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_GeminiEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_HuggingfaceApiEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_OpenaiEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_VertexaiEmbedding",
+    "EmbeddingModelConfigUpdate",
+    "EmbeddingModelConfigUpdateEmbeddingConfig",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_AzureEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_BedrockEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_CohereEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_GeminiEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_OpenaiEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_VertexaiEmbedding",
     "EvalDataset",
     "EvalDatasetJobParams",
     "EvalDatasetJobRecord",
@@ -301,6 +345,7 @@ __all__ = [
     "HuggingFaceInferenceApiEmbeddingToken",
     "IngestionErrorResponse",
     "InputMessage",
+    "IntervalUsageAndPlan",
     "JobNameMapping",
     "LlamaParseParameters",
     "LlamaParseSupportedFileExtensions",
@@ -334,6 +379,7 @@ __all__ = [
     "PageScreenshotNodeWithScore",
     "PageSegmentationConfig",
     "PageSplitterNodeParser",
+    "PaginatedListPipelineFilesResponse",
     "ParserLanguages",
     "ParsingHistoryItem",
     "ParsingJob",
@@ -378,6 +424,7 @@ __all__ = [
     "PipelineTransformConfig_Advanced",
     "PipelineTransformConfig_Auto",
     "PipelineType",
+    "Plan",
     "PlaygroundSession",
     "Pooling",
     "PresetRetrievalParams",
@@ -403,6 +450,7 @@ __all__ = [
     "TokenChunkingConfig",
     "TokenTextSplitter",
     "TransformationCategoryNames",
+    "Usage",
     "UserOrganization",
     "UserOrganizationCreate",
     "UserOrganizationDelete",

llama_cloud/types/base.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class Base(pydantic.BaseModel):
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

llama_cloud/types/cloud_one_drive_data_source.py

@@ -21,6 +21,7 @@ class CloudOneDriveDataSource(pydantic.BaseModel):
     client_id: str = pydantic.Field(description="The client ID to use for authentication.")
     client_secret: str = pydantic.Field(description="The client secret to use for authentication.")
     tenant_id: str = pydantic.Field(description="The tenant ID to use for authentication.")
+    required_exts: typing.Optional[typing.List[str]]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/cloud_postgres_vector_store.py

@@ -19,7 +19,7 @@ class CloudPostgresVectorStore(pydantic.BaseModel):
     database: str
     host: str
     password: str
-    port: str
+    port: int
     user: str
     table_name: str
     schema_name: str

llama_cloud/types/cloud_sharepoint_data_source.py

@@ -23,6 +23,7 @@ class CloudSharepointDataSource(pydantic.BaseModel):
     client_id: str = pydantic.Field(description="The client ID to use for authentication.")
     client_secret: str = pydantic.Field(description="The client secret to use for authentication.")
     tenant_id: str = pydantic.Field(description="The tenant ID to use for authentication.")
+    required_exts: typing.Optional[typing.List[str]]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str: