llama-cloud 0.1.34__py3-none-any.whl → 0.1.36__py3-none-any.whl

llama_cloud/__init__.py

@@ -38,7 +38,12 @@ from .types import (
     ChatData,
     ChunkMode,
     ClassificationResult,
-    ClassifyResponse,
+    ClassifierRule,
+    ClassifyJob,
+    ClassifyJobResults,
+    ClassifyJobWithStatus,
+    ClassifyParsingConfiguration,
+    CloudAstraDbVectorStore,
     CloudAzStorageBlobDataSource,
     CloudAzureAiSearchVectorStore,
     CloudBoxDataSource,
@@ -76,6 +81,7 @@ from .types import (
     DataSourceCreateCustomMetadataValue,
     DataSourceCustomMetadataValue,
     DataSourceReaderVersionMetadata,
+    DataSourceReaderVersionMetadataReaderVersion,
     DataSourceUpdateDispatcherConfig,
     DeleteParams,
     DocumentBlock,
@@ -131,12 +137,17 @@ from .types import (
     ExtractState,
     ExtractTarget,
     FailPageMode,
+    FailureHandlingConfig,
     File,
+    FileClassification,
     FileCountByStatusResponse,
+    FileData,
     FileIdPresignedUrl,
     FileParsePublic,
     FilePermissionInfoValue,
     FileResourceInfoValue,
+    FileStoreInfoResponse,
+    FileStoreInfoResponseStatus,
     FilterCondition,
     FilterOperation,
     FilterOperationEq,
@@ -188,12 +199,15 @@ from .types import (
     LoadFilesJobConfig,
     ManagedIngestionStatus,
     ManagedIngestionStatusResponse,
+    ManagedOpenAiEmbedding,
+    ManagedOpenAiEmbeddingConfig,
     MessageAnnotation,
     MessageRole,
     MetadataFilter,
     MetadataFilterValue,
     MetadataFilters,
     MetadataFiltersFiltersItem,
+    MultimodalParseResolution,
     NodeRelationship,
     NoneChunkingConfig,
     NoneSegmentationConfig,
@@ -214,6 +228,7 @@ from .types import (
     PaginatedReportResponse,
     PaginatedResponseAgentData,
     PaginatedResponseAggregateGroup,
+    PaginatedResponseQuotaConfiguration,
     ParseJobConfig,
     ParseJobConfigPriority,
     ParsePlanLevel,
@@ -254,6 +269,7 @@ from .types import (
     PipelineEmbeddingConfig_CohereEmbedding,
     PipelineEmbeddingConfig_GeminiEmbedding,
     PipelineEmbeddingConfig_HuggingfaceApiEmbedding,
+    PipelineEmbeddingConfig_ManagedOpenaiEmbedding,
     PipelineEmbeddingConfig_OpenaiEmbedding,
     PipelineEmbeddingConfig_VertexaiEmbedding,
     PipelineFile,
@@ -285,6 +301,11 @@ from .types import (
     Project,
     ProjectCreate,
     PromptConf,
+    QuotaConfiguration,
+    QuotaConfigurationConfigurationType,
+    QuotaConfigurationStatus,
+    QuotaRateLimitConfigurationValue,
+    QuotaRateLimitConfigurationValueDenominatorUnits,
     ReRankConfig,
     ReRankerType,
     RecurringCreditGrant,
@@ -328,6 +349,7 @@ from .types import (
     TextNodeRelationshipsValue,
     TextNodeWithScore,
     TokenChunkingConfig,
+    UpdateUserResponse,
     UsageAndPlan,
     UsageMetricResponse,
     UsageResponse,
@@ -337,6 +359,7 @@ from .types import (
     UserOrganizationCreate,
     UserOrganizationDelete,
     UserOrganizationRole,
+    UserSummary,
     ValidationError,
     ValidationErrorLocItem,
     VertexAiEmbeddingConfig,
@@ -366,6 +389,8 @@ from .resources import (
     ExtractJobCreateBatchDataSchemaOverrideZeroValue,
     ExtractSchemaValidateRequestDataSchema,
     ExtractSchemaValidateRequestDataSchemaZeroValue,
+    ExtractStatelessRequestDataSchema,
+    ExtractStatelessRequestDataSchemaZeroValue,
     FileCreateFromUrlResourceInfoValue,
     FileCreatePermissionInfoValue,
     FileCreateResourceInfoValue,
@@ -399,6 +424,7 @@ from .resources import (
     projects,
     reports,
     retrievers,
+    users,
 )
 from .environment import LlamaCloudEnvironment
 
@@ -440,7 +466,12 @@ __all__ = [
     "ChatData",
     "ChunkMode",
     "ClassificationResult",
-    "ClassifyResponse",
+    "ClassifierRule",
+    "ClassifyJob",
+    "ClassifyJobResults",
+    "ClassifyJobWithStatus",
+    "ClassifyParsingConfiguration",
+    "CloudAstraDbVectorStore",
     "CloudAzStorageBlobDataSource",
     "CloudAzureAiSearchVectorStore",
     "CloudBoxDataSource",
@@ -479,6 +510,7 @@ __all__ = [
     "DataSourceCreateCustomMetadataValue",
     "DataSourceCustomMetadataValue",
     "DataSourceReaderVersionMetadata",
+    "DataSourceReaderVersionMetadataReaderVersion",
     "DataSourceUpdateComponent",
     "DataSourceUpdateCustomMetadataValue",
     "DataSourceUpdateDispatcherConfig",
@@ -550,17 +582,24 @@ __all__ = [
     "ExtractSchemaValidateResponse",
     "ExtractSchemaValidateResponseDataSchemaValue",
     "ExtractState",
+    "ExtractStatelessRequestDataSchema",
+    "ExtractStatelessRequestDataSchemaZeroValue",
     "ExtractTarget",
     "FailPageMode",
+    "FailureHandlingConfig",
     "File",
+    "FileClassification",
     "FileCountByStatusResponse",
     "FileCreateFromUrlResourceInfoValue",
     "FileCreatePermissionInfoValue",
     "FileCreateResourceInfoValue",
+    "FileData",
     "FileIdPresignedUrl",
     "FileParsePublic",
     "FilePermissionInfoValue",
     "FileResourceInfoValue",
+    "FileStoreInfoResponse",
+    "FileStoreInfoResponseStatus",
     "FilterCondition",
     "FilterOperation",
     "FilterOperationEq",
@@ -613,12 +652,15 @@ __all__ = [
     "LoadFilesJobConfig",
     "ManagedIngestionStatus",
     "ManagedIngestionStatusResponse",
+    "ManagedOpenAiEmbedding",
+    "ManagedOpenAiEmbeddingConfig",
     "MessageAnnotation",
     "MessageRole",
     "MetadataFilter",
     "MetadataFilterValue",
     "MetadataFilters",
     "MetadataFiltersFiltersItem",
+    "MultimodalParseResolution",
     "NodeRelationship",
     "NoneChunkingConfig",
     "NoneSegmentationConfig",
@@ -639,6 +681,7 @@ __all__ = [
     "PaginatedReportResponse",
     "PaginatedResponseAgentData",
     "PaginatedResponseAggregateGroup",
+    "PaginatedResponseQuotaConfiguration",
     "ParseJobConfig",
     "ParseJobConfigPriority",
     "ParsePlanLevel",
@@ -679,6 +722,7 @@ __all__ = [
     "PipelineEmbeddingConfig_CohereEmbedding",
     "PipelineEmbeddingConfig_GeminiEmbedding",
     "PipelineEmbeddingConfig_HuggingfaceApiEmbedding",
+    "PipelineEmbeddingConfig_ManagedOpenaiEmbedding",
     "PipelineEmbeddingConfig_OpenaiEmbedding",
     "PipelineEmbeddingConfig_VertexaiEmbedding",
     "PipelineFile",
@@ -720,6 +764,11 @@ __all__ = [
     "Project",
     "ProjectCreate",
     "PromptConf",
+    "QuotaConfiguration",
+    "QuotaConfigurationConfigurationType",
+    "QuotaConfigurationStatus",
+    "QuotaRateLimitConfigurationValue",
+    "QuotaRateLimitConfigurationValueDenominatorUnits",
     "ReRankConfig",
     "ReRankerType",
     "RecurringCreditGrant",
@@ -766,6 +815,7 @@ __all__ = [
     "TokenChunkingConfig",
     "UnprocessableEntityError",
     "UpdateReportPlanApiV1ReportsReportIdPlanPatchRequestAction",
+    "UpdateUserResponse",
     "UsageAndPlan",
     "UsageMetricResponse",
     "UsageResponse",
@@ -775,6 +825,7 @@ __all__ = [
     "UserOrganizationCreate",
     "UserOrganizationDelete",
     "UserOrganizationRole",
+    "UserSummary",
     "ValidationError",
     "ValidationErrorLocItem",
     "VertexAiEmbeddingConfig",
@@ -800,4 +851,5 @@ __all__ = [
     "projects",
     "reports",
     "retrievers",
+    "users",
 ]

llama_cloud/client.py

@@ -24,6 +24,7 @@ from .resources.pipelines.client import AsyncPipelinesClient, PipelinesClient
 from .resources.projects.client import AsyncProjectsClient, ProjectsClient
 from .resources.reports.client import AsyncReportsClient, ReportsClient
 from .resources.retrievers.client import AsyncRetrieversClient, RetrieversClient
+from .resources.users.client import AsyncUsersClient, UsersClient
 
 
 class LlamaCloud:
@@ -56,6 +57,7 @@ class LlamaCloud:
         self.agent_deployments = AgentDeploymentsClient(client_wrapper=self._client_wrapper)
         self.classifier = ClassifierClient(client_wrapper=self._client_wrapper)
         self.admin = AdminClient(client_wrapper=self._client_wrapper)
+        self.users = UsersClient(client_wrapper=self._client_wrapper)
         self.llama_extract = LlamaExtractClient(client_wrapper=self._client_wrapper)
         self.reports = ReportsClient(client_wrapper=self._client_wrapper)
         self.beta = BetaClient(client_wrapper=self._client_wrapper)
@@ -91,6 +93,7 @@ class AsyncLlamaCloud:
         self.agent_deployments = AsyncAgentDeploymentsClient(client_wrapper=self._client_wrapper)
         self.classifier = AsyncClassifierClient(client_wrapper=self._client_wrapper)
         self.admin = AsyncAdminClient(client_wrapper=self._client_wrapper)
+        self.users = AsyncUsersClient(client_wrapper=self._client_wrapper)
         self.llama_extract = AsyncLlamaExtractClient(client_wrapper=self._client_wrapper)
         self.reports = AsyncReportsClient(client_wrapper=self._client_wrapper)
         self.beta = AsyncBetaClient(client_wrapper=self._client_wrapper)

llama_cloud/resources/__init__.py

@@ -19,6 +19,7 @@ from . import (
     projects,
     reports,
     retrievers,
+    users,
 )
 from .data_sinks import DataSinkUpdateComponent
 from .data_sources import DataSourceUpdateComponent, DataSourceUpdateCustomMetadataValue
@@ -42,6 +43,8 @@ from .llama_extract import (
     ExtractJobCreateBatchDataSchemaOverrideZeroValue,
     ExtractSchemaValidateRequestDataSchema,
     ExtractSchemaValidateRequestDataSchemaZeroValue,
+    ExtractStatelessRequestDataSchema,
+    ExtractStatelessRequestDataSchemaZeroValue,
 )
 from .pipelines import (
     PipelineFileUpdateCustomMetadataValue,
@@ -78,6 +81,8 @@ __all__ = [
     "ExtractJobCreateBatchDataSchemaOverrideZeroValue",
     "ExtractSchemaValidateRequestDataSchema",
     "ExtractSchemaValidateRequestDataSchemaZeroValue",
+    "ExtractStatelessRequestDataSchema",
+    "ExtractStatelessRequestDataSchemaZeroValue",
     "FileCreateFromUrlResourceInfoValue",
     "FileCreatePermissionInfoValue",
     "FileCreateResourceInfoValue",
@@ -111,4 +116,5 @@ __all__ = [
     "projects",
     "reports",
     "retrievers",
+    "users",
 ]

llama_cloud/resources/admin/client.py

@@ -8,6 +8,7 @@ from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.remove_none_from_dict import remove_none_from_dict
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.file_store_info_response import FileStoreInfoResponse
 from ...types.http_validation_error import HttpValidationError
 from ...types.license_info_response import LicenseInfoResponse
 
@@ -53,6 +54,31 @@ class AdminClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def get_file_store_info(self) -> FileStoreInfoResponse:
+        """
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.admin.get_file_store_info()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/admin/filestores/info"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(FileStoreInfoResponse, _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 AsyncAdminClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -86,3 +112,28 @@ class AsyncAdminClient:
         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_store_info(self) -> FileStoreInfoResponse:
+        """
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.admin.get_file_store_info()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/admin/filestores/info"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(FileStoreInfoResponse, _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)