llama-cloud 0.1.35__py3-none-any.whl → 0.1.37__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,
@@ -132,13 +137,22 @@ from .types import (
     ExtractState,
     ExtractTarget,
     FailPageMode,
+    FailureHandlingConfig,
     File,
+    FileClassification,
     FileCountByStatusResponse,
+    FileCreate,
+    FileCreatePermissionInfoValue,
+    FileCreateResourceInfoValue,
     FileData,
+    FileFilter,
     FileIdPresignedUrl,
     FileParsePublic,
     FilePermissionInfoValue,
+    FileQueryResponse,
     FileResourceInfoValue,
+    FileStoreInfoResponse,
+    FileStoreInfoResponseStatus,
     FilterCondition,
     FilterOperation,
     FilterOperationEq,
@@ -175,6 +189,8 @@ from .types import (
     LLamaParseTransformConfig,
     LegacyParseJobConfig,
     LicenseInfoResponse,
+    LlamaExtractModeAvailability,
+    LlamaExtractModeAvailabilityStatus,
     LlamaExtractSettings,
     LlamaIndexCoreBaseLlmsTypesChatMessage,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem,
@@ -383,8 +399,6 @@ from .resources import (
     ExtractStatelessRequestDataSchema,
     ExtractStatelessRequestDataSchemaZeroValue,
     FileCreateFromUrlResourceInfoValue,
-    FileCreatePermissionInfoValue,
-    FileCreateResourceInfoValue,
     PipelineFileUpdateCustomMetadataValue,
     PipelineUpdateEmbeddingConfig,
     PipelineUpdateEmbeddingConfig_AzureEmbedding,
@@ -457,7 +471,12 @@ __all__ = [
     "ChatData",
     "ChunkMode",
     "ClassificationResult",
-    "ClassifyResponse",
+    "ClassifierRule",
+    "ClassifyJob",
+    "ClassifyJobResults",
+    "ClassifyJobWithStatus",
+    "ClassifyParsingConfiguration",
+    "CloudAstraDbVectorStore",
     "CloudAzStorageBlobDataSource",
     "CloudAzureAiSearchVectorStore",
     "CloudBoxDataSource",
@@ -572,16 +591,23 @@ __all__ = [
     "ExtractStatelessRequestDataSchemaZeroValue",
     "ExtractTarget",
     "FailPageMode",
+    "FailureHandlingConfig",
     "File",
+    "FileClassification",
     "FileCountByStatusResponse",
+    "FileCreate",
     "FileCreateFromUrlResourceInfoValue",
     "FileCreatePermissionInfoValue",
     "FileCreateResourceInfoValue",
     "FileData",
+    "FileFilter",
     "FileIdPresignedUrl",
     "FileParsePublic",
     "FilePermissionInfoValue",
+    "FileQueryResponse",
     "FileResourceInfoValue",
+    "FileStoreInfoResponse",
+    "FileStoreInfoResponseStatus",
     "FilterCondition",
     "FilterOperation",
     "FilterOperationEq",
@@ -619,6 +645,8 @@ __all__ = [
     "LegacyParseJobConfig",
     "LicenseInfoResponse",
     "LlamaCloudEnvironment",
+    "LlamaExtractModeAvailability",
+    "LlamaExtractModeAvailabilityStatus",
     "LlamaExtractSettings",
     "LlamaIndexCoreBaseLlmsTypesChatMessage",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem",

llama_cloud/resources/__init__.py

@@ -33,7 +33,7 @@ from .embedding_model_configs import (
     EmbeddingModelConfigCreateEmbeddingConfig_OpenaiEmbedding,
     EmbeddingModelConfigCreateEmbeddingConfig_VertexaiEmbedding,
 )
-from .files import FileCreateFromUrlResourceInfoValue, FileCreatePermissionInfoValue, FileCreateResourceInfoValue
+from .files import FileCreateFromUrlResourceInfoValue
 from .llama_extract import (
     ExtractAgentCreateDataSchema,
     ExtractAgentCreateDataSchemaZeroValue,
@@ -84,8 +84,6 @@ __all__ = [
     "ExtractStatelessRequestDataSchema",
     "ExtractStatelessRequestDataSchemaZeroValue",
     "FileCreateFromUrlResourceInfoValue",
-    "FileCreatePermissionInfoValue",
-    "FileCreateResourceInfoValue",
     "PipelineFileUpdateCustomMetadataValue",
     "PipelineUpdateEmbeddingConfig",
     "PipelineUpdateEmbeddingConfig_AzureEmbedding",

llama_cloud/resources/admin/client.py

@@ -8,8 +8,10 @@ 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
+from ...types.llama_extract_mode_availability import LlamaExtractModeAvailability
 
 try:
     import pydantic
@@ -53,6 +55,59 @@ 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)
+
+    def get_llamaextract_features(self) -> typing.List[LlamaExtractModeAvailability]:
+        """
+        Get LlamaExtract feature availability based on available models.
+
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.admin.get_llamaextract_features()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/admin/llamaextract/features"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[LlamaExtractModeAvailability], _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 +141,56 @@ 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)
+
+    async def get_llamaextract_features(self) -> typing.List[LlamaExtractModeAvailability]:
+        """
+        Get LlamaExtract feature availability based on available models.
+
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.admin.get_llamaextract_features()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/admin/llamaextract/features"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[LlamaExtractModeAvailability], _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)