llama-cloud 0.0.13__py3-none-any.whl → 0.0.14__py3-none-any.whl

llama_cloud/types/__init__.py

@@ -17,15 +17,19 @@ from .advanced_mode_transform_config_segmentation_config import (
 )
 from .auto_transform_config import AutoTransformConfig
 from .azure_open_ai_embedding import AzureOpenAiEmbedding
+from .azure_open_ai_embedding_config import AzureOpenAiEmbeddingConfig
 from .base import Base
 from .base_prompt_template import BasePromptTemplate
 from .bedrock_embedding import BedrockEmbedding
+from .bedrock_embedding_config import BedrockEmbeddingConfig
+from .box_auth_mechanism import BoxAuthMechanism
 from .character_chunking_config import CharacterChunkingConfig
 from .character_splitter import CharacterSplitter
 from .chat_data import ChatData
 from .chat_message import ChatMessage
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
 from .cloud_azure_ai_search_vector_store import CloudAzureAiSearchVectorStore
+from .cloud_box_data_source import CloudBoxDataSource
 from .cloud_chroma_vector_store import CloudChromaVectorStore
 from .cloud_confluence_data_source import CloudConfluenceDataSource
 from .cloud_document import CloudDocument
@@ -44,6 +48,7 @@ from .cloud_slack_data_source import CloudSlackDataSource
 from .cloud_weaviate_vector_store import CloudWeaviateVectorStore
 from .code_splitter import CodeSplitter
 from .cohere_embedding import CohereEmbedding
+from .cohere_embedding_config import CohereEmbeddingConfig
 from .configurable_data_sink_names import ConfigurableDataSinkNames
 from .configurable_data_source_names import ConfigurableDataSourceNames
 from .configurable_transformation_definition import ConfigurableTransformationDefinition
@@ -69,16 +74,11 @@ 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_config import EmbeddingConfig
-from .embedding_config_component import EmbeddingConfigComponent
-from .embedding_config_component_one import EmbeddingConfigComponentOne
-from .embedding_config_type import EmbeddingConfigType
 from .eval_dataset import EvalDataset
 from .eval_dataset_job_params import EvalDatasetJobParams
 from .eval_dataset_job_record import EvalDatasetJobRecord
 from .eval_execution_params import EvalExecutionParams
 from .eval_execution_params_override import EvalExecutionParamsOverride
-from .eval_llm_model_data import EvalLlmModelData
 from .eval_question import EvalQuestion
 from .eval_question_create import EvalQuestionCreate
 from .eval_question_result import EvalQuestionResult
@@ -92,14 +92,18 @@ from .file_resource_info_value import FileResourceInfoValue
 from .filter_condition import FilterCondition
 from .filter_operator import FilterOperator
 from .gemini_embedding import GeminiEmbedding
+from .gemini_embedding_config import GeminiEmbeddingConfig
 from .http_validation_error import HttpValidationError
 from .hugging_face_inference_api_embedding import HuggingFaceInferenceApiEmbedding
+from .hugging_face_inference_api_embedding_config import HuggingFaceInferenceApiEmbeddingConfig
 from .hugging_face_inference_api_embedding_token import HuggingFaceInferenceApiEmbeddingToken
 from .ingestion_error_response import IngestionErrorResponse
+from .input_message import InputMessage
 from .job_name_mapping import JobNameMapping
 from .llama_parse_parameters import LlamaParseParameters
 from .llama_parse_supported_file_extensions import LlamaParseSupportedFileExtensions
 from .llm import Llm
+from .llm_model_data import LlmModelData
 from .llm_parameters import LlmParameters
 from .local_eval import LocalEval
 from .local_eval_results import LocalEvalResults
@@ -108,6 +112,7 @@ from .managed_ingestion_status import ManagedIngestionStatus
 from .managed_ingestion_status_response import ManagedIngestionStatusResponse
 from .markdown_element_node_parser import MarkdownElementNodeParser
 from .markdown_node_parser import MarkdownNodeParser
+from .message_annotation import MessageAnnotation
 from .message_role import MessageRole
 from .metadata_filter import MetadataFilter
 from .metadata_filter_value import MetadataFilterValue
@@ -119,6 +124,7 @@ from .none_chunking_config import NoneChunkingConfig
 from .none_segmentation_config import NoneSegmentationConfig
 from .object_type import ObjectType
 from .open_ai_embedding import OpenAiEmbedding
+from .open_ai_embedding_config import OpenAiEmbeddingConfig
 from .organization import Organization
 from .organization_create import OrganizationCreate
 from .page_segmentation_config import PageSegmentationConfig
@@ -132,6 +138,15 @@ from .parsing_job_text_result import ParsingJobTextResult
 from .parsing_usage import ParsingUsage
 from .pipeline import Pipeline
 from .pipeline_create import PipelineCreate
+from .pipeline_create_embedding_config import (
+    PipelineCreateEmbeddingConfig,
+    PipelineCreateEmbeddingConfig_AzureEmbedding,
+    PipelineCreateEmbeddingConfig_BedrockEmbedding,
+    PipelineCreateEmbeddingConfig_CohereEmbedding,
+    PipelineCreateEmbeddingConfig_GeminiEmbedding,
+    PipelineCreateEmbeddingConfig_HuggingfaceApiEmbedding,
+    PipelineCreateEmbeddingConfig_OpenaiEmbedding,
+)
 from .pipeline_create_transform_config import (
     PipelineCreateTransformConfig,
     PipelineCreateTransformConfig_Advanced,
@@ -143,12 +158,27 @@ from .pipeline_data_source_component_one import PipelineDataSourceComponentOne
 from .pipeline_data_source_create import PipelineDataSourceCreate
 from .pipeline_data_source_custom_metadata_value import PipelineDataSourceCustomMetadataValue
 from .pipeline_deployment import PipelineDeployment
+from .pipeline_embedding_config import (
+    PipelineEmbeddingConfig,
+    PipelineEmbeddingConfig_AzureEmbedding,
+    PipelineEmbeddingConfig_BedrockEmbedding,
+    PipelineEmbeddingConfig_CohereEmbedding,
+    PipelineEmbeddingConfig_GeminiEmbedding,
+    PipelineEmbeddingConfig_HuggingfaceApiEmbedding,
+    PipelineEmbeddingConfig_OpenaiEmbedding,
+)
 from .pipeline_file import PipelineFile
 from .pipeline_file_create import PipelineFileCreate
 from .pipeline_file_create_custom_metadata_value import PipelineFileCreateCustomMetadataValue
 from .pipeline_file_custom_metadata_value import PipelineFileCustomMetadataValue
 from .pipeline_file_resource_info_value import PipelineFileResourceInfoValue
+from .pipeline_transform_config import (
+    PipelineTransformConfig,
+    PipelineTransformConfig_Advanced,
+    PipelineTransformConfig_Auto,
+)
 from .pipeline_type import PipelineType
+from .playground_session import PlaygroundSession
 from .pooling import Pooling
 from .preset_retrieval_params import PresetRetrievalParams
 from .presigned_url import PresignedUrl
@@ -164,8 +194,8 @@ from .semantic_chunking_config import SemanticChunkingConfig
 from .sentence_chunking_config import SentenceChunkingConfig
 from .sentence_splitter import SentenceSplitter
 from .status_enum import StatusEnum
-from .supported_eval_llm_model import SupportedEvalLlmModel
-from .supported_eval_llm_model_names import SupportedEvalLlmModelNames
+from .supported_llm_model import SupportedLlmModel
+from .supported_llm_model_names import SupportedLlmModelNames
 from .text_node import TextNode
 from .text_node_relationships_value import TextNodeRelationshipsValue
 from .text_node_with_score import TextNodeWithScore
@@ -193,15 +223,19 @@ __all__ = [
     "AdvancedModeTransformConfigSegmentationConfig_Page",
     "AutoTransformConfig",
     "AzureOpenAiEmbedding",
+    "AzureOpenAiEmbeddingConfig",
     "Base",
     "BasePromptTemplate",
     "BedrockEmbedding",
+    "BedrockEmbeddingConfig",
+    "BoxAuthMechanism",
     "CharacterChunkingConfig",
     "CharacterSplitter",
     "ChatData",
     "ChatMessage",
     "CloudAzStorageBlobDataSource",
     "CloudAzureAiSearchVectorStore",
+    "CloudBoxDataSource",
     "CloudChromaVectorStore",
     "CloudConfluenceDataSource",
     "CloudDocument",
@@ -220,6 +254,7 @@ __all__ = [
     "CloudWeaviateVectorStore",
     "CodeSplitter",
     "CohereEmbedding",
+    "CohereEmbeddingConfig",
     "ConfigurableDataSinkNames",
     "ConfigurableDataSourceNames",
     "ConfigurableTransformationDefinition",
@@ -245,16 +280,11 @@ __all__ = [
     "DataSourceCustomMetadataValue",
     "DataSourceDefinition",
     "ElementSegmentationConfig",
-    "EmbeddingConfig",
-    "EmbeddingConfigComponent",
-    "EmbeddingConfigComponentOne",
-    "EmbeddingConfigType",
     "EvalDataset",
     "EvalDatasetJobParams",
     "EvalDatasetJobRecord",
     "EvalExecutionParams",
     "EvalExecutionParamsOverride",
-    "EvalLlmModelData",
     "EvalQuestion",
     "EvalQuestionCreate",
     "EvalQuestionResult",
@@ -268,14 +298,18 @@ __all__ = [
     "FilterCondition",
     "FilterOperator",
     "GeminiEmbedding",
+    "GeminiEmbeddingConfig",
     "HttpValidationError",
     "HuggingFaceInferenceApiEmbedding",
+    "HuggingFaceInferenceApiEmbeddingConfig",
     "HuggingFaceInferenceApiEmbeddingToken",
     "IngestionErrorResponse",
+    "InputMessage",
     "JobNameMapping",
     "LlamaParseParameters",
     "LlamaParseSupportedFileExtensions",
     "Llm",
+    "LlmModelData",
     "LlmParameters",
     "LocalEval",
     "LocalEvalResults",
@@ -284,6 +318,7 @@ __all__ = [
     "ManagedIngestionStatusResponse",
     "MarkdownElementNodeParser",
     "MarkdownNodeParser",
+    "MessageAnnotation",
     "MessageRole",
     "MetadataFilter",
     "MetadataFilterValue",
@@ -295,6 +330,7 @@ __all__ = [
     "NoneSegmentationConfig",
     "ObjectType",
     "OpenAiEmbedding",
+    "OpenAiEmbeddingConfig",
     "Organization",
     "OrganizationCreate",
     "PageSegmentationConfig",
@@ -308,6 +344,13 @@ __all__ = [
     "ParsingUsage",
     "Pipeline",
     "PipelineCreate",
+    "PipelineCreateEmbeddingConfig",
+    "PipelineCreateEmbeddingConfig_AzureEmbedding",
+    "PipelineCreateEmbeddingConfig_BedrockEmbedding",
+    "PipelineCreateEmbeddingConfig_CohereEmbedding",
+    "PipelineCreateEmbeddingConfig_GeminiEmbedding",
+    "PipelineCreateEmbeddingConfig_HuggingfaceApiEmbedding",
+    "PipelineCreateEmbeddingConfig_OpenaiEmbedding",
     "PipelineCreateTransformConfig",
     "PipelineCreateTransformConfig_Advanced",
     "PipelineCreateTransformConfig_Auto",
@@ -317,12 +360,23 @@ __all__ = [
     "PipelineDataSourceCreate",
     "PipelineDataSourceCustomMetadataValue",
     "PipelineDeployment",
+    "PipelineEmbeddingConfig",
+    "PipelineEmbeddingConfig_AzureEmbedding",
+    "PipelineEmbeddingConfig_BedrockEmbedding",
+    "PipelineEmbeddingConfig_CohereEmbedding",
+    "PipelineEmbeddingConfig_GeminiEmbedding",
+    "PipelineEmbeddingConfig_HuggingfaceApiEmbedding",
+    "PipelineEmbeddingConfig_OpenaiEmbedding",
     "PipelineFile",
     "PipelineFileCreate",
     "PipelineFileCreateCustomMetadataValue",
     "PipelineFileCustomMetadataValue",
     "PipelineFileResourceInfoValue",
+    "PipelineTransformConfig",
+    "PipelineTransformConfig_Advanced",
+    "PipelineTransformConfig_Auto",
     "PipelineType",
+    "PlaygroundSession",
     "Pooling",
     "PresetRetrievalParams",
     "PresignedUrl",
@@ -338,8 +392,8 @@ __all__ = [
     "SentenceChunkingConfig",
     "SentenceSplitter",
     "StatusEnum",
-    "SupportedEvalLlmModel",
-    "SupportedEvalLlmModelNames",
+    "SupportedLlmModel",
+    "SupportedLlmModelNames",
     "TextNode",
     "TextNodeRelationshipsValue",
     "TextNodeWithScore",

llama_cloud/types/{embedding_config.py → azure_open_ai_embedding_config.py}

@@ -4,8 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .embedding_config_component import EmbeddingConfigComponent
-from .embedding_config_type import EmbeddingConfigType
+from .azure_open_ai_embedding import AzureOpenAiEmbedding
 
 try:
     import pydantic
@@ -16,10 +15,9 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class EmbeddingConfig(pydantic.BaseModel):
-    type: typing.Optional[EmbeddingConfigType] = pydantic.Field(description="Type of the embedding model.")
-    component: typing.Optional[EmbeddingConfigComponent] = pydantic.Field(
-        description="Configuration for the transformation."
+class AzureOpenAiEmbeddingConfig(pydantic.BaseModel):
+    component: typing.Optional[AzureOpenAiEmbedding] = pydantic.Field(
+        description="Configuration for the Azure OpenAI embedding model."
     )
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/bedrock_embedding_config.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .bedrock_embedding import BedrockEmbedding
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class BedrockEmbeddingConfig(pydantic.BaseModel):
+    component: typing.Optional[BedrockEmbedding] = pydantic.Field(
+        description="Configuration for the Bedrock embedding model."
+    )
+
+    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/box_auth_mechanism.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class BoxAuthMechanism(str, enum.Enum):
+    """
+    An enumeration.
+    """
+
+    DEVELOPER_TOKEN = "developer_token"
+    CCG = "ccg"
+
+    def visit(self, developer_token: typing.Callable[[], T_Result], ccg: typing.Callable[[], T_Result]) -> T_Result:
+        if self is BoxAuthMechanism.DEVELOPER_TOKEN:
+            return developer_token()
+        if self is BoxAuthMechanism.CCG:
+            return ccg()

llama_cloud/types/chat_data.py

@@ -22,7 +22,7 @@ class ChatData(pydantic.BaseModel):
     Comes with special serialization logic for types used commonly in platform codebase.
     """
 
-    retrieval_parameters: PresetRetrievalParams
+    retrieval_parameters: typing.Optional[PresetRetrievalParams]
     llm_parameters: typing.Optional[LlmParameters]
     class_name: typing.Optional[str]
 

llama_cloud/types/chat_message.py

@@ -4,6 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .message_annotation import MessageAnnotation
 from .message_role import MessageRole
 
 try:
@@ -17,12 +18,21 @@ except ImportError:
 
 class ChatMessage(pydantic.BaseModel):
     """
-    Chat message.
+    Base schema model for BaseComponent classes used in the platform.
+    Comes with special serialization logic for types used commonly in platform codebase.
     """
 
-    role: typing.Optional[MessageRole]
-    content: typing.Optional[typing.Any]
-    additional_kwargs: typing.Optional[typing.Dict[str, typing.Any]]
+    id: str
+    index: int = pydantic.Field(description="The index of the message in the chat.")
+    annotations: typing.Optional[typing.List[MessageAnnotation]] = pydantic.Field(
+        description="Retrieval annotations for the message."
+    )
+    role: MessageRole
+    content: typing.Optional[str] = pydantic.Field(description="Text content of the generation")
+    additional_kwargs: typing.Optional[typing.Dict[str, str]] = pydantic.Field(
+        description="Additional arguments passed to the model"
+    )
+    class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/cloud_azure_ai_search_vector_store.py

@@ -26,6 +26,9 @@ class CloudAzureAiSearchVectorStore(pydantic.BaseModel):
     index_name: typing.Optional[str]
     filterable_metadata_field_keys: typing.Optional[typing.Dict[str, typing.Any]]
     embedding_dimension: typing.Optional[int]
+    client_id: typing.Optional[str]
+    client_secret: typing.Optional[str]
+    tenant_id: typing.Optional[str]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/cloud_box_data_source.py

@@ -0,0 +1,51 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .box_auth_mechanism import BoxAuthMechanism
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class CloudBoxDataSource(pydantic.BaseModel):
+    """
+    Base component object to capture class names.
+    """
+
+    folder_id: typing.Optional[str] = pydantic.Field(description="The ID of the Box folder to read from.")
+    authentication_mechanism: BoxAuthMechanism = pydantic.Field(
+        description="The type of authentication to use (Developer Token or CCG)"
+    )
+    developer_token: typing.Optional[str] = pydantic.Field(
+        description="Developer token for authentication if authentication_mechanism is 'developer_token'."
+    )
+    client_id: typing.Optional[str] = pydantic.Field(
+        description="Box API key used for identifying the application the user is authenticating with"
+    )
+    client_secret: typing.Optional[str] = pydantic.Field(description="Box API secret used for making auth requests.")
+    user_id: typing.Optional[str] = pydantic.Field(description="Box User ID, if provided authenticates as user.")
+    enterprise_id: typing.Optional[str] = pydantic.Field(
+        description="Box Enterprise ID, if provided authenticates as service."
+    )
+    class_name: typing.Optional[str]
+
+    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_document.py

@@ -23,6 +23,9 @@ class CloudDocument(pydantic.BaseModel):
     metadata: typing.Dict[str, typing.Any]
     excluded_embed_metadata_keys: typing.Optional[typing.List[str]]
     excluded_llm_metadata_keys: typing.Optional[typing.List[str]]
+    page_positions: typing.Optional[typing.List[int]] = pydantic.Field(
+        description="indices in the CloudDocument.text where a new page begins. e.g. Second page starts at index specified by page_positions[1]."
+    )
     id: str
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/cloud_document_create.py

@@ -23,6 +23,9 @@ class CloudDocumentCreate(pydantic.BaseModel):
     metadata: typing.Dict[str, typing.Any]
     excluded_embed_metadata_keys: typing.Optional[typing.List[str]]
     excluded_llm_metadata_keys: typing.Optional[typing.List[str]]
+    page_positions: typing.Optional[typing.List[int]] = pydantic.Field(
+        description="indices in the CloudDocument.text where a new page begins. e.g. Second page starts at index specified by page_positions[1]."
+    )
     id: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/cloud_sharepoint_data_source.py

@@ -19,7 +19,8 @@ class CloudSharepointDataSource(pydantic.BaseModel):
     Base component object to capture class names.
     """
 
-    site_name: str = pydantic.Field(description="The name of the SharePoint site to download from.")
+    site_name: typing.Optional[str] = pydantic.Field(description="The name of the SharePoint site to download from.")
+    site_id: typing.Optional[str] = pydantic.Field(description="The ID of the SharePoint site to download from.")
     folder_path: typing.Optional[str] = pydantic.Field(description="The path of the Sharepoint folder to read from.")
     folder_id: typing.Optional[str] = pydantic.Field(description="The ID of the Sharepoint folder to read from.")
     drive_name: typing.Optional[str] = pydantic.Field(description="The name of the Sharepoint drive to read from.")

llama_cloud/types/cohere_embedding_config.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .cohere_embedding import CohereEmbedding
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class CohereEmbeddingConfig(pydantic.BaseModel):
+    component: typing.Optional[CohereEmbedding] = pydantic.Field(
+        description="Configuration for the Cohere embedding model."
+    )
+
+    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/configurable_data_source_names.py

@@ -19,6 +19,7 @@ class ConfigurableDataSourceNames(str, enum.Enum):
     NOTION_PAGE = "NOTION_PAGE"
     CONFLUENCE = "CONFLUENCE"
     JIRA = "JIRA"
+    BOX = "BOX"
 
     def visit(
         self,
@@ -30,6 +31,7 @@ class ConfigurableDataSourceNames(str, enum.Enum):
         notion_page: typing.Callable[[], T_Result],
         confluence: typing.Callable[[], T_Result],
         jira: typing.Callable[[], T_Result],
+        box: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is ConfigurableDataSourceNames.S_3:
             return s_3()
@@ -47,3 +49,5 @@ class ConfigurableDataSourceNames(str, enum.Enum):
             return confluence()
         if self is ConfigurableDataSourceNames.JIRA:
             return jira()
+        if self is ConfigurableDataSourceNames.BOX:
+            return box()

llama_cloud/types/custom_claims.py

@@ -33,9 +33,6 @@ class CustomClaims(pydantic.BaseModel):
     usage_index_max_files_per_pipeline: typing.Optional[int] = pydantic.Field(
         description="The maximum number of files per pipeline the user can index without LlamaParse premium."
     )
-    max_jobs_in_execution: typing.Optional[int] = pydantic.Field(
-        description="The maximum number of jobs the user can have in execution."
-    )
     max_jobs_in_execution_per_job_type: typing.Optional[int] = pydantic.Field(
         description="The maximum number of jobs the user can have in execution per job type."
     )

llama_cloud/types/data_source_component_one.py

@@ -3,6 +3,7 @@
 import typing
 
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
+from .cloud_box_data_source import CloudBoxDataSource
 from .cloud_confluence_data_source import CloudConfluenceDataSource
 from .cloud_jira_data_source import CloudJiraDataSource
 from .cloud_notion_page_data_source import CloudNotionPageDataSource
@@ -20,4 +21,5 @@ DataSourceComponentOne = typing.Union[
     CloudNotionPageDataSource,
     CloudConfluenceDataSource,
     CloudJiraDataSource,
+    CloudBoxDataSource,
 ]

llama_cloud/types/data_source_create_component_one.py

@@ -3,6 +3,7 @@
 import typing
 
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
+from .cloud_box_data_source import CloudBoxDataSource
 from .cloud_confluence_data_source import CloudConfluenceDataSource
 from .cloud_jira_data_source import CloudJiraDataSource
 from .cloud_notion_page_data_source import CloudNotionPageDataSource
@@ -20,4 +21,5 @@ DataSourceCreateComponentOne = typing.Union[
     CloudNotionPageDataSource,
     CloudConfluenceDataSource,
     CloudJiraDataSource,
+    CloudBoxDataSource,
 ]

llama_cloud/types/eval_execution_params.py

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .supported_eval_llm_model_names import SupportedEvalLlmModelNames
+from .supported_llm_model_names import SupportedLlmModelNames
 
 try:
     import pydantic
@@ -20,7 +20,7 @@ class EvalExecutionParams(pydantic.BaseModel):
     Schema for the params for an eval execution.
     """
 
-    llm_model: typing.Optional[SupportedEvalLlmModelNames] = pydantic.Field(
+    llm_model: typing.Optional[SupportedLlmModelNames] = pydantic.Field(
         description="The LLM model to use within eval execution."
     )
     qa_prompt_tmpl: typing.Optional[str] = pydantic.Field(

llama_cloud/types/eval_execution_params_override.py

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .supported_eval_llm_model_names import SupportedEvalLlmModelNames
+from .supported_llm_model_names import SupportedLlmModelNames
 
 try:
     import pydantic
@@ -20,7 +20,7 @@ class EvalExecutionParamsOverride(pydantic.BaseModel):
     Schema for the params override for an eval execution.
     """
 
-    llm_model: typing.Optional[SupportedEvalLlmModelNames] = pydantic.Field(
+    llm_model: typing.Optional[SupportedLlmModelNames] = pydantic.Field(
         description="The LLM model to use within eval execution."
     )
     qa_prompt_tmpl: typing.Optional[str] = pydantic.Field(

llama_cloud/types/filter_operator.py

@@ -23,6 +23,7 @@ class FilterOperator(str, enum.Enum):
     ALL = "all"
     TEXT_MATCH = "text_match"
     CONTAINS = "contains"
+    IS_EMPTY = "is_empty"
 
     def visit(
         self,
@@ -38,6 +39,7 @@ class FilterOperator(str, enum.Enum):
         all: typing.Callable[[], T_Result],
         text_match: typing.Callable[[], T_Result],
         contains: typing.Callable[[], T_Result],
+        is_empty: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is FilterOperator.EQUAL_TO:
             return equal_to()
@@ -63,3 +65,5 @@ class FilterOperator(str, enum.Enum):
             return text_match()
         if self is FilterOperator.CONTAINS:
             return contains()
+        if self is FilterOperator.IS_EMPTY:
+            return is_empty()