llama-cloud 0.1.17__tar.gz → 0.1.19__tar.gz

{llama_cloud-0.1.17 → llama_cloud-0.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-cloud
-Version: 0.1.17
+Version: 0.1.19
 Summary: 
 License: MIT
 Author: Logan Markewich

{llama_cloud-0.1.17 → llama_cloud-0.1.19}/llama_cloud/__init__.py

@@ -13,6 +13,7 @@ from .types import (
     AdvancedModeTransformConfigSegmentationConfig_None,
     AdvancedModeTransformConfigSegmentationConfig_Page,
     AppSchemaChatChatMessage,
+    AudioBlock,
     AutoTransformConfig,
     AzureOpenAiEmbedding,
     AzureOpenAiEmbeddingConfig,
@@ -21,6 +22,10 @@ from .types import (
     BasePlanName,
     BasePlanPlanFrequency,
     BasePromptTemplate,
+    Batch,
+    BatchItem,
+    BatchPaginatedList,
+    BatchPublicOutput,
     BedrockEmbedding,
     BedrockEmbeddingConfig,
     BillingPeriod,
@@ -98,19 +103,10 @@ from .types import (
     EmbeddingModelConfigUpdateEmbeddingConfig_VertexaiEmbedding,
     EvalExecutionParams,
     ExtractAgent,
-    ExtractAgentCreate,
-    ExtractAgentCreateDataSchema,
-    ExtractAgentCreateDataSchemaZeroValue,
     ExtractAgentDataSchemaValue,
-    ExtractAgentUpdate,
-    ExtractAgentUpdateDataSchema,
-    ExtractAgentUpdateDataSchemaZeroValue,
     ExtractConfig,
     ExtractJob,
     ExtractJobCreate,
-    ExtractJobCreateBatch,
-    ExtractJobCreateBatchDataSchemaOverride,
-    ExtractJobCreateBatchDataSchemaOverrideZeroValue,
     ExtractJobCreateDataSchemaOverride,
     ExtractJobCreateDataSchemaOverrideZeroValue,
     ExtractMode,
@@ -125,14 +121,14 @@ from .types import (
     ExtractRunDataSchemaValue,
     ExtractRunDataZeroValue,
     ExtractRunExtractionMetadataValue,
-    ExtractSchemaValidateRequest,
-    ExtractSchemaValidateRequestDataSchema,
-    ExtractSchemaValidateRequestDataSchemaZeroValue,
     ExtractSchemaValidateResponse,
     ExtractSchemaValidateResponseDataSchemaValue,
     ExtractState,
     ExtractTarget,
+    FailPageMode,
     File,
+    FileCountByStatusResponse,
+    FileParsePublic,
     FilePermissionInfoValue,
     FileResourceInfoValue,
     FilterCondition,
@@ -154,6 +150,7 @@ from .types import (
     LlamaExtractSettings,
     LlamaIndexCoreBaseLlmsTypesChatMessage,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem,
+    LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Audio,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Image,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Text,
     LlamaParseParameters,
@@ -199,9 +196,11 @@ from .types import (
     ParsingJobStructuredResult,
     ParsingJobTextResult,
     ParsingMode,
-    ParsingUsage,
     PartitionNames,
     Permission,
+    PgVectorDistanceMethod,
+    PgVectorHnswSettings,
+    PgVectorVectorType,
     Pipeline,
     PipelineConfigurationHashes,
     PipelineCreate,
@@ -218,6 +217,7 @@ from .types import (
     PipelineDataSourceComponent,
     PipelineDataSourceCreate,
     PipelineDataSourceCustomMetadataValue,
+    PipelineDataSourceStatus,
     PipelineDeployment,
     PipelineEmbeddingConfig,
     PipelineEmbeddingConfig_AzureEmbedding,
@@ -324,6 +324,14 @@ from .resources import (
     EmbeddingModelConfigCreateEmbeddingConfig_HuggingfaceApiEmbedding,
     EmbeddingModelConfigCreateEmbeddingConfig_OpenaiEmbedding,
     EmbeddingModelConfigCreateEmbeddingConfig_VertexaiEmbedding,
+    ExtractAgentCreateDataSchema,
+    ExtractAgentCreateDataSchemaZeroValue,
+    ExtractAgentUpdateDataSchema,
+    ExtractAgentUpdateDataSchemaZeroValue,
+    ExtractJobCreateBatchDataSchemaOverride,
+    ExtractJobCreateBatchDataSchemaOverrideZeroValue,
+    ExtractSchemaValidateRequestDataSchema,
+    ExtractSchemaValidateRequestDataSchemaZeroValue,
     FileCreateFromUrlResourceInfoValue,
     FileCreatePermissionInfoValue,
     FileCreateResourceInfoValue,
@@ -338,6 +346,7 @@ from .resources import (
     PipelineUpdateEmbeddingConfig_VertexaiEmbedding,
     PipelineUpdateTransformConfig,
     UpdateReportPlanApiV1ReportsReportIdPlanPatchRequestAction,
+    beta,
     chat_apps,
     component_definitions,
     data_sinks,
@@ -369,6 +378,7 @@ __all__ = [
     "AdvancedModeTransformConfigSegmentationConfig_None",
     "AdvancedModeTransformConfigSegmentationConfig_Page",
     "AppSchemaChatChatMessage",
+    "AudioBlock",
     "AutoTransformConfig",
     "AzureOpenAiEmbedding",
     "AzureOpenAiEmbeddingConfig",
@@ -377,6 +387,10 @@ __all__ = [
     "BasePlanName",
     "BasePlanPlanFrequency",
     "BasePromptTemplate",
+    "Batch",
+    "BatchItem",
+    "BatchPaginatedList",
+    "BatchPublicOutput",
     "BedrockEmbedding",
     "BedrockEmbeddingConfig",
     "BillingPeriod",
@@ -465,17 +479,14 @@ __all__ = [
     "EmbeddingModelConfigUpdateEmbeddingConfig_VertexaiEmbedding",
     "EvalExecutionParams",
     "ExtractAgent",
-    "ExtractAgentCreate",
     "ExtractAgentCreateDataSchema",
     "ExtractAgentCreateDataSchemaZeroValue",
     "ExtractAgentDataSchemaValue",
-    "ExtractAgentUpdate",
     "ExtractAgentUpdateDataSchema",
     "ExtractAgentUpdateDataSchemaZeroValue",
     "ExtractConfig",
     "ExtractJob",
     "ExtractJobCreate",
-    "ExtractJobCreateBatch",
     "ExtractJobCreateBatchDataSchemaOverride",
     "ExtractJobCreateBatchDataSchemaOverrideZeroValue",
     "ExtractJobCreateDataSchemaOverride",
@@ -492,17 +503,19 @@ __all__ = [
     "ExtractRunDataSchemaValue",
     "ExtractRunDataZeroValue",
     "ExtractRunExtractionMetadataValue",
-    "ExtractSchemaValidateRequest",
     "ExtractSchemaValidateRequestDataSchema",
     "ExtractSchemaValidateRequestDataSchemaZeroValue",
     "ExtractSchemaValidateResponse",
     "ExtractSchemaValidateResponseDataSchemaValue",
     "ExtractState",
     "ExtractTarget",
+    "FailPageMode",
     "File",
+    "FileCountByStatusResponse",
     "FileCreateFromUrlResourceInfoValue",
     "FileCreatePermissionInfoValue",
     "FileCreateResourceInfoValue",
+    "FileParsePublic",
     "FilePermissionInfoValue",
     "FileResourceInfoValue",
     "FilterCondition",
@@ -525,6 +538,7 @@ __all__ = [
     "LlamaExtractSettings",
     "LlamaIndexCoreBaseLlmsTypesChatMessage",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem",
+    "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Audio",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Image",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Text",
     "LlamaParseParameters",
@@ -570,9 +584,11 @@ __all__ = [
     "ParsingJobStructuredResult",
     "ParsingJobTextResult",
     "ParsingMode",
-    "ParsingUsage",
     "PartitionNames",
     "Permission",
+    "PgVectorDistanceMethod",
+    "PgVectorHnswSettings",
+    "PgVectorVectorType",
     "Pipeline",
     "PipelineConfigurationHashes",
     "PipelineCreate",
@@ -589,6 +605,7 @@ __all__ = [
     "PipelineDataSourceComponent",
     "PipelineDataSourceCreate",
     "PipelineDataSourceCustomMetadataValue",
+    "PipelineDataSourceStatus",
     "PipelineDeployment",
     "PipelineEmbeddingConfig",
     "PipelineEmbeddingConfig_AzureEmbedding",
@@ -693,6 +710,7 @@ __all__ = [
     "VertexAiEmbeddingConfig",
     "VertexEmbeddingMode",
     "VertexTextEmbedding",
+    "beta",
     "chat_apps",
     "component_definitions",
     "data_sinks",

{llama_cloud-0.1.17 → llama_cloud-0.1.19}/llama_cloud/client.py

@@ -6,6 +6,7 @@ import httpx
 
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import LlamaCloudEnvironment
+from .resources.beta.client import AsyncBetaClient, BetaClient
 from .resources.chat_apps.client import AsyncChatAppsClient, ChatAppsClient
 from .resources.component_definitions.client import AsyncComponentDefinitionsClient, ComponentDefinitionsClient
 from .resources.data_sinks.client import AsyncDataSinksClient, DataSinksClient
@@ -53,6 +54,7 @@ class LlamaCloud:
         self.chat_apps = ChatAppsClient(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)
 
 
 class AsyncLlamaCloud:
@@ -85,6 +87,7 @@ class AsyncLlamaCloud:
         self.chat_apps = AsyncChatAppsClient(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)
 
 
 def _get_base_url(*, base_url: typing.Optional[str] = None, environment: LlamaCloudEnvironment) -> str:

{llama_cloud-0.1.17 → llama_cloud-0.1.19}/llama_cloud/resources/__init__.py

@@ -1,6 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from . import (
+    beta,
     chat_apps,
     component_definitions,
     data_sinks,
@@ -30,6 +31,16 @@ from .embedding_model_configs import (
     EmbeddingModelConfigCreateEmbeddingConfig_VertexaiEmbedding,
 )
 from .files import FileCreateFromUrlResourceInfoValue, FileCreatePermissionInfoValue, FileCreateResourceInfoValue
+from .llama_extract import (
+    ExtractAgentCreateDataSchema,
+    ExtractAgentCreateDataSchemaZeroValue,
+    ExtractAgentUpdateDataSchema,
+    ExtractAgentUpdateDataSchemaZeroValue,
+    ExtractJobCreateBatchDataSchemaOverride,
+    ExtractJobCreateBatchDataSchemaOverrideZeroValue,
+    ExtractSchemaValidateRequestDataSchema,
+    ExtractSchemaValidateRequestDataSchemaZeroValue,
+)
 from .pipelines import (
     PipelineFileUpdateCustomMetadataValue,
     PipelineUpdateEmbeddingConfig,
@@ -56,6 +67,14 @@ __all__ = [
     "EmbeddingModelConfigCreateEmbeddingConfig_HuggingfaceApiEmbedding",
     "EmbeddingModelConfigCreateEmbeddingConfig_OpenaiEmbedding",
     "EmbeddingModelConfigCreateEmbeddingConfig_VertexaiEmbedding",
+    "ExtractAgentCreateDataSchema",
+    "ExtractAgentCreateDataSchemaZeroValue",
+    "ExtractAgentUpdateDataSchema",
+    "ExtractAgentUpdateDataSchemaZeroValue",
+    "ExtractJobCreateBatchDataSchemaOverride",
+    "ExtractJobCreateBatchDataSchemaOverrideZeroValue",
+    "ExtractSchemaValidateRequestDataSchema",
+    "ExtractSchemaValidateRequestDataSchemaZeroValue",
     "FileCreateFromUrlResourceInfoValue",
     "FileCreatePermissionInfoValue",
     "FileCreateResourceInfoValue",
@@ -70,6 +89,7 @@ __all__ = [
     "PipelineUpdateEmbeddingConfig_VertexaiEmbedding",
     "PipelineUpdateTransformConfig",
     "UpdateReportPlanApiV1ReportsReportIdPlanPatchRequestAction",
+    "beta",
     "chat_apps",
     "component_definitions",
     "data_sinks",

llama_cloud-0.1.19/llama_cloud/resources/beta/client.py

@@ -0,0 +1,371 @@
+# 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.batch import Batch
+from ...types.batch_paginated_list import BatchPaginatedList
+from ...types.batch_public_output import BatchPublicOutput
+from ...types.http_validation_error import HttpValidationError
+from ...types.llama_parse_parameters import LlamaParseParameters
+
+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 BetaClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list_batches(
+        self,
+        *,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+    ) -> BatchPaginatedList:
+        """
+        Parameters:
+            - limit: typing.Optional[int].
+
+            - offset: typing.Optional[int].
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.list_batches()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/batches"),
+            params=remove_none_from_dict(
+                {"limit": limit, "offset": offset, "project_id": project_id, "organization_id": organization_id}
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(BatchPaginatedList, _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 create_batch(
+        self,
+        *,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        tool: str,
+        tool_data: typing.Optional[LlamaParseParameters] = OMIT,
+        input_type: str,
+        input_id: str,
+        output_type: typing.Optional[str] = OMIT,
+        output_id: typing.Optional[str] = OMIT,
+        batch_create_project_id: str,
+        external_id: str,
+        completion_window: typing.Optional[int] = OMIT,
+    ) -> Batch:
+        """
+        Parameters:
+            - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+
+            - tool: str. The tool to be used for all requests in the batch.
+
+            - tool_data: typing.Optional[LlamaParseParameters].
+
+            - input_type: str. The type of input file. Currently only 'datasource' is supported.
+
+            - input_id: str. The ID of the input file for the batch.
+
+            - output_type: typing.Optional[str].
+
+            - output_id: typing.Optional[str].
+
+            - batch_create_project_id: str. The ID of the project to which the batch belongs
+
+            - external_id: str. A developer-provided ID for the batch. This ID will be returned in the response.
+
+            - completion_window: typing.Optional[int]. The time frame within which the batch should be processed. Currently only 24h is supported.
+        ---
+        from llama_cloud import FailPageMode, LlamaParseParameters, ParsingMode
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.create_batch(
+            tool="string",
+            tool_data=LlamaParseParameters(
+                parse_mode=ParsingMode.PARSE_PAGE_WITHOUT_LLM,
+                replace_failed_page_mode=FailPageMode.RAW_TEXT,
+            ),
+            input_type="string",
+            input_id="string",
+            batch_create_project_id="string",
+            external_id="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {
+            "tool": tool,
+            "input_type": input_type,
+            "input_id": input_id,
+            "project_id": batch_create_project_id,
+            "external_id": external_id,
+        }
+        if tool_data is not OMIT:
+            _request["tool_data"] = tool_data
+        if output_type is not OMIT:
+            _request["output_type"] = output_type
+        if output_id is not OMIT:
+            _request["output_id"] = output_id
+        if completion_window is not OMIT:
+            _request["completion_window"] = completion_window
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/batches"),
+            params=remove_none_from_dict({"organization_id": organization_id, "project_id": project_id}),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(Batch, _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_batch(self, batch_id: str, *, organization_id: typing.Optional[str] = None) -> BatchPublicOutput:
+        """
+        Parameters:
+            - batch_id: str.
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.get_batch(
+            batch_id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/batches/{batch_id}"),
+            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(BatchPublicOutput, _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 AsyncBetaClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list_batches(
+        self,
+        *,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+    ) -> BatchPaginatedList:
+        """
+        Parameters:
+            - limit: typing.Optional[int].
+
+            - offset: typing.Optional[int].
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.list_batches()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/batches"),
+            params=remove_none_from_dict(
+                {"limit": limit, "offset": offset, "project_id": project_id, "organization_id": organization_id}
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(BatchPaginatedList, _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 create_batch(
+        self,
+        *,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        tool: str,
+        tool_data: typing.Optional[LlamaParseParameters] = OMIT,
+        input_type: str,
+        input_id: str,
+        output_type: typing.Optional[str] = OMIT,
+        output_id: typing.Optional[str] = OMIT,
+        batch_create_project_id: str,
+        external_id: str,
+        completion_window: typing.Optional[int] = OMIT,
+    ) -> Batch:
+        """
+        Parameters:
+            - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+
+            - tool: str. The tool to be used for all requests in the batch.
+
+            - tool_data: typing.Optional[LlamaParseParameters].
+
+            - input_type: str. The type of input file. Currently only 'datasource' is supported.
+
+            - input_id: str. The ID of the input file for the batch.
+
+            - output_type: typing.Optional[str].
+
+            - output_id: typing.Optional[str].
+
+            - batch_create_project_id: str. The ID of the project to which the batch belongs
+
+            - external_id: str. A developer-provided ID for the batch. This ID will be returned in the response.
+
+            - completion_window: typing.Optional[int]. The time frame within which the batch should be processed. Currently only 24h is supported.
+        ---
+        from llama_cloud import FailPageMode, LlamaParseParameters, ParsingMode
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.create_batch(
+            tool="string",
+            tool_data=LlamaParseParameters(
+                parse_mode=ParsingMode.PARSE_PAGE_WITHOUT_LLM,
+                replace_failed_page_mode=FailPageMode.RAW_TEXT,
+            ),
+            input_type="string",
+            input_id="string",
+            batch_create_project_id="string",
+            external_id="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {
+            "tool": tool,
+            "input_type": input_type,
+            "input_id": input_id,
+            "project_id": batch_create_project_id,
+            "external_id": external_id,
+        }
+        if tool_data is not OMIT:
+            _request["tool_data"] = tool_data
+        if output_type is not OMIT:
+            _request["output_type"] = output_type
+        if output_id is not OMIT:
+            _request["output_id"] = output_id
+        if completion_window is not OMIT:
+            _request["completion_window"] = completion_window
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/batches"),
+            params=remove_none_from_dict({"organization_id": organization_id, "project_id": project_id}),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(Batch, _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_batch(self, batch_id: str, *, organization_id: typing.Optional[str] = None) -> BatchPublicOutput:
+        """
+        Parameters:
+            - batch_id: str.
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.get_batch(
+            batch_id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/batches/{batch_id}"),
+            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(BatchPublicOutput, _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-0.1.17 → llama_cloud-0.1.19}/llama_cloud/resources/chat_apps/client.py

@@ -108,7 +108,7 @@ class ChatAppsClient:
             name="string",
             retriever_id="string",
             llm_config=LlmParameters(
-                model_name=SupportedLlmModelNames.GPT_3_5_TURBO,
+                model_name=SupportedLlmModelNames.GPT_4_O,
             ),
             retrieval_config=PresetCompositeRetrievalParams(
                 mode=CompositeRetrievalMode.ROUTING,
@@ -217,7 +217,7 @@ class ChatAppsClient:
         client.chat_apps.update_chat_app(
             id="string",
             llm_config=LlmParameters(
-                model_name=SupportedLlmModelNames.GPT_3_5_TURBO,
+                model_name=SupportedLlmModelNames.GPT_4_O,
             ),
             retrieval_config=PresetCompositeRetrievalParams(
                 mode=CompositeRetrievalMode.ROUTING,
@@ -401,7 +401,7 @@ class AsyncChatAppsClient:
             name="string",
             retriever_id="string",
             llm_config=LlmParameters(
-                model_name=SupportedLlmModelNames.GPT_3_5_TURBO,
+                model_name=SupportedLlmModelNames.GPT_4_O,
             ),
             retrieval_config=PresetCompositeRetrievalParams(
                 mode=CompositeRetrievalMode.ROUTING,
@@ -510,7 +510,7 @@ class AsyncChatAppsClient:
         await client.chat_apps.update_chat_app(
             id="string",
             llm_config=LlmParameters(
-                model_name=SupportedLlmModelNames.GPT_3_5_TURBO,
+                model_name=SupportedLlmModelNames.GPT_4_O,
             ),
             retrieval_config=PresetCompositeRetrievalParams(
                 mode=CompositeRetrievalMode.ROUTING,