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

llama_cloud/__init__.py

@@ -41,7 +41,6 @@ from .types import (
     ClassifierRule,
     ClassifyJob,
     ClassifyJobResults,
-    ClassifyJobWithStatus,
     ClassifyParsingConfiguration,
     CloudAstraDbVectorStore,
     CloudAzStorageBlobDataSource,
@@ -51,6 +50,8 @@ from .types import (
     CloudDocument,
     CloudDocumentCreate,
     CloudJiraDataSource,
+    CloudJiraDataSourceV2,
+    CloudJiraDataSourceV2ApiVersion,
     CloudMilvusVectorStore,
     CloudMongoDbAtlasVectorSearch,
     CloudNotionPageDataSource,
@@ -141,10 +142,15 @@ from .types import (
     File,
     FileClassification,
     FileCountByStatusResponse,
+    FileCreate,
+    FileCreatePermissionInfoValue,
+    FileCreateResourceInfoValue,
     FileData,
+    FileFilter,
     FileIdPresignedUrl,
     FileParsePublic,
     FilePermissionInfoValue,
+    FileQueryResponse,
     FileResourceInfoValue,
     FileStoreInfoResponse,
     FileStoreInfoResponseStatus,
@@ -184,6 +190,8 @@ from .types import (
     LLamaParseTransformConfig,
     LegacyParseJobConfig,
     LicenseInfoResponse,
+    LlamaExtractModeAvailability,
+    LlamaExtractModeAvailabilityStatus,
     LlamaExtractSettings,
     LlamaIndexCoreBaseLlmsTypesChatMessage,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem,
@@ -228,6 +236,7 @@ from .types import (
     PaginatedReportResponse,
     PaginatedResponseAgentData,
     PaginatedResponseAggregateGroup,
+    PaginatedResponseClassifyJob,
     PaginatedResponseQuotaConfiguration,
     ParseJobConfig,
     ParseJobConfigPriority,
@@ -392,8 +401,6 @@ from .resources import (
     ExtractStatelessRequestDataSchema,
     ExtractStatelessRequestDataSchemaZeroValue,
     FileCreateFromUrlResourceInfoValue,
-    FileCreatePermissionInfoValue,
-    FileCreateResourceInfoValue,
     PipelineFileUpdateCustomMetadataValue,
     PipelineUpdateEmbeddingConfig,
     PipelineUpdateEmbeddingConfig_AzureEmbedding,
@@ -408,6 +415,7 @@ from .resources import (
     UpdateReportPlanApiV1ReportsReportIdPlanPatchRequestAction,
     admin,
     agent_deployments,
+    alpha,
     beta,
     chat_apps,
     classifier,
@@ -469,7 +477,6 @@ __all__ = [
     "ClassifierRule",
     "ClassifyJob",
     "ClassifyJobResults",
-    "ClassifyJobWithStatus",
     "ClassifyParsingConfiguration",
     "CloudAstraDbVectorStore",
     "CloudAzStorageBlobDataSource",
@@ -479,6 +486,8 @@ __all__ = [
     "CloudDocument",
     "CloudDocumentCreate",
     "CloudJiraDataSource",
+    "CloudJiraDataSourceV2",
+    "CloudJiraDataSourceV2ApiVersion",
     "CloudMilvusVectorStore",
     "CloudMongoDbAtlasVectorSearch",
     "CloudNotionPageDataSource",
@@ -590,13 +599,16 @@ __all__ = [
     "File",
     "FileClassification",
     "FileCountByStatusResponse",
+    "FileCreate",
     "FileCreateFromUrlResourceInfoValue",
     "FileCreatePermissionInfoValue",
     "FileCreateResourceInfoValue",
     "FileData",
+    "FileFilter",
     "FileIdPresignedUrl",
     "FileParsePublic",
     "FilePermissionInfoValue",
+    "FileQueryResponse",
     "FileResourceInfoValue",
     "FileStoreInfoResponse",
     "FileStoreInfoResponseStatus",
@@ -637,6 +649,8 @@ __all__ = [
     "LegacyParseJobConfig",
     "LicenseInfoResponse",
     "LlamaCloudEnvironment",
+    "LlamaExtractModeAvailability",
+    "LlamaExtractModeAvailabilityStatus",
     "LlamaExtractSettings",
     "LlamaIndexCoreBaseLlmsTypesChatMessage",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem",
@@ -681,6 +695,7 @@ __all__ = [
     "PaginatedReportResponse",
     "PaginatedResponseAgentData",
     "PaginatedResponseAggregateGroup",
+    "PaginatedResponseClassifyJob",
     "PaginatedResponseQuotaConfiguration",
     "ParseJobConfig",
     "ParseJobConfigPriority",
@@ -835,6 +850,7 @@ __all__ = [
     "WebhookConfigurationWebhookEventsItem",
     "admin",
     "agent_deployments",
+    "alpha",
     "beta",
     "chat_apps",
     "classifier",

llama_cloud/client.py

@@ -8,6 +8,7 @@ from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import LlamaCloudEnvironment
 from .resources.admin.client import AdminClient, AsyncAdminClient
 from .resources.agent_deployments.client import AgentDeploymentsClient, AsyncAgentDeploymentsClient
+from .resources.alpha.client import AlphaClient, AsyncAlphaClient
 from .resources.beta.client import AsyncBetaClient, BetaClient
 from .resources.chat_apps.client import AsyncChatAppsClient, ChatAppsClient
 from .resources.classifier.client import AsyncClassifierClient, ClassifierClient
@@ -61,6 +62,7 @@ class LlamaCloud:
         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)
+        self.alpha = AlphaClient(client_wrapper=self._client_wrapper)
 
 
 class AsyncLlamaCloud:
@@ -97,6 +99,7 @@ class AsyncLlamaCloud:
         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)
+        self.alpha = AsyncAlphaClient(client_wrapper=self._client_wrapper)
 
 
 def _get_base_url(*, base_url: typing.Optional[str] = None, environment: LlamaCloudEnvironment) -> str:

llama_cloud/resources/__init__.py

@@ -3,6 +3,7 @@
 from . import (
     admin,
     agent_deployments,
+    alpha,
     beta,
     chat_apps,
     classifier,
@@ -33,7 +34,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 +85,6 @@ __all__ = [
     "ExtractStatelessRequestDataSchema",
     "ExtractStatelessRequestDataSchemaZeroValue",
     "FileCreateFromUrlResourceInfoValue",
-    "FileCreatePermissionInfoValue",
-    "FileCreateResourceInfoValue",
     "PipelineFileUpdateCustomMetadataValue",
     "PipelineUpdateEmbeddingConfig",
     "PipelineUpdateEmbeddingConfig_AzureEmbedding",
@@ -100,6 +99,7 @@ __all__ = [
     "UpdateReportPlanApiV1ReportsReportIdPlanPatchRequestAction",
     "admin",
     "agent_deployments",
+    "alpha",
     "beta",
     "chat_apps",
     "classifier",

llama_cloud/resources/admin/client.py

@@ -11,6 +11,7 @@ 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
@@ -79,6 +80,34 @@ class AdminClient:
             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):
@@ -137,3 +166,31 @@ 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_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)

llama_cloud/resources/alpha/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

llama_cloud/resources/alpha/client.py

@@ -0,0 +1,118 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+import urllib.parse
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.remove_none_from_dict import remove_none_from_dict
+from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.http_validation_error import HttpValidationError
+from ...types.parsing_job import ParsingJob
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class AlphaClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def upload_file_v_2(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        configuration: str,
+        file: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = None,
+    ) -> ParsingJob:
+        """
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - configuration: str.
+
+            - file: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+        """
+        _request: typing.Dict[str, typing.Any] = {"configuration": configuration}
+        if file is not OMIT:
+            _request["file"] = file
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v2alpha1/parse/upload"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(_request),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ParsingJob, _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 AsyncAlphaClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def upload_file_v_2(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        configuration: str,
+        file: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = None,
+    ) -> ParsingJob:
+        """
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - configuration: str.
+
+            - file: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+        """
+        _request: typing.Dict[str, typing.Any] = {"configuration": configuration}
+        if file is not OMIT:
+            _request["file"] = file
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v2alpha1/parse/upload"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(_request),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ParsingJob, _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)