llama-cloud 0.0.16__py3-none-any.whl → 0.1.0__py3-none-any.whl

llama_cloud/resources/pipelines/types/pipeline_update_embedding_config.py

@@ -15,8 +15,8 @@ from ....types.open_ai_embedding_config import OpenAiEmbeddingConfig
 from ....types.vertex_ai_embedding_config import VertexAiEmbeddingConfig
 
 
-class PipelineUpdateEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
-    type: typing_extensions.Literal["OPENAI_EMBEDDING"]
+class PipelineUpdateEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
+    type: typing_extensions.Literal["AZURE_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -24,8 +24,8 @@ class PipelineUpdateEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
         allow_population_by_field_name = True
 
 
-class PipelineUpdateEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
-    type: typing_extensions.Literal["AZURE_EMBEDDING"]
+class PipelineUpdateEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
+    type: typing_extensions.Literal["BEDROCK_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -33,8 +33,8 @@ class PipelineUpdateEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
         allow_population_by_field_name = True
 
 
-class PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInferenceApiEmbeddingConfig):
-    type: typing_extensions.Literal["HUGGINGFACE_API_EMBEDDING"]
+class PipelineUpdateEmbeddingConfig_CohereEmbedding(CohereEmbeddingConfig):
+    type: typing_extensions.Literal["COHERE_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -42,8 +42,8 @@ class PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInference
         allow_population_by_field_name = True
 
 
-class PipelineUpdateEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
-    type: typing_extensions.Literal["BEDROCK_EMBEDDING"]
+class PipelineUpdateEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
+    type: typing_extensions.Literal["GEMINI_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -51,8 +51,8 @@ class PipelineUpdateEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
         allow_population_by_field_name = True
 
 
-class PipelineUpdateEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
-    type: typing_extensions.Literal["GEMINI_EMBEDDING"]
+class PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInferenceApiEmbeddingConfig):
+    type: typing_extensions.Literal["HUGGINGFACE_API_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -60,8 +60,8 @@ class PipelineUpdateEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
         allow_population_by_field_name = True
 
 
-class PipelineUpdateEmbeddingConfig_CohereEmbedding(CohereEmbeddingConfig):
-    type: typing_extensions.Literal["COHERE_EMBEDDING"]
+class PipelineUpdateEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
+    type: typing_extensions.Literal["OPENAI_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -79,11 +79,11 @@ class PipelineUpdateEmbeddingConfig_VertexaiEmbedding(VertexAiEmbeddingConfig):
 
 
 PipelineUpdateEmbeddingConfig = typing.Union[
-    PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
     PipelineUpdateEmbeddingConfig_AzureEmbedding,
-    PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineUpdateEmbeddingConfig_BedrockEmbedding,
-    PipelineUpdateEmbeddingConfig_GeminiEmbedding,
     PipelineUpdateEmbeddingConfig_CohereEmbedding,
+    PipelineUpdateEmbeddingConfig_GeminiEmbedding,
+    PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
+    PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
     PipelineUpdateEmbeddingConfig_VertexaiEmbedding,
 ]

llama_cloud/resources/pipelines/types/pipeline_update_transform_config.py

@@ -1,31 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
-
 import typing
 
-import typing_extensions
-
 from ....types.advanced_mode_transform_config import AdvancedModeTransformConfig
 from ....types.auto_transform_config import AutoTransformConfig
 
-
-class PipelineUpdateTransformConfig_Auto(AutoTransformConfig):
-    mode: typing_extensions.Literal["auto"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class PipelineUpdateTransformConfig_Advanced(AdvancedModeTransformConfig):
-    mode: typing_extensions.Literal["advanced"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-PipelineUpdateTransformConfig = typing.Union[PipelineUpdateTransformConfig_Auto, PipelineUpdateTransformConfig_Advanced]
+PipelineUpdateTransformConfig = typing.Union[AutoTransformConfig, AdvancedModeTransformConfig]

llama_cloud/types/__init__.py

@@ -34,6 +34,7 @@ from .cloud_chroma_vector_store import CloudChromaVectorStore
 from .cloud_confluence_data_source import CloudConfluenceDataSource
 from .cloud_document import CloudDocument
 from .cloud_document_create import CloudDocumentCreate
+from .cloud_google_drive_data_source import CloudGoogleDriveDataSource
 from .cloud_jira_data_source import CloudJiraDataSource
 from .cloud_milvus_vector_store import CloudMilvusVectorStore
 from .cloud_mongo_db_atlas_vector_search import CloudMongoDbAtlasVectorSearch
@@ -55,21 +56,15 @@ from .configurable_transformation_definition import ConfigurableTransformationDe
 from .configurable_transformation_names import ConfigurableTransformationNames
 from .configured_transformation_item import ConfiguredTransformationItem
 from .configured_transformation_item_component import ConfiguredTransformationItemComponent
-from .configured_transformation_item_component_one import ConfiguredTransformationItemComponentOne
-from .custom_claims import CustomClaims
 from .data_sink import DataSink
 from .data_sink_component import DataSinkComponent
-from .data_sink_component_one import DataSinkComponentOne
 from .data_sink_create import DataSinkCreate
 from .data_sink_create_component import DataSinkCreateComponent
-from .data_sink_create_component_one import DataSinkCreateComponentOne
 from .data_sink_definition import DataSinkDefinition
 from .data_source import DataSource
 from .data_source_component import DataSourceComponent
-from .data_source_component_one import DataSourceComponentOne
 from .data_source_create import DataSourceCreate
 from .data_source_create_component import DataSourceCreateComponent
-from .data_source_create_component_one import DataSourceCreateComponentOne
 from .data_source_create_custom_metadata_value import DataSourceCreateCustomMetadataValue
 from .data_source_custom_metadata_value import DataSourceCustomMetadataValue
 from .data_source_definition import DataSourceDefinition
@@ -79,6 +74,7 @@ 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_metric import EvalMetric
 from .eval_question import EvalQuestion
 from .eval_question_create import EvalQuestionCreate
 from .eval_question_result import EvalQuestionResult
@@ -121,6 +117,7 @@ from .metadata_filters import MetadataFilters
 from .metadata_filters_filters_item import MetadataFiltersFiltersItem
 from .metric_result import MetricResult
 from .node_parser import NodeParser
+from .node_relationship import NodeRelationship
 from .none_chunking_config import NoneChunkingConfig
 from .none_segmentation_config import NoneSegmentationConfig
 from .object_type import ObjectType
@@ -139,6 +136,7 @@ from .parsing_job_json_result import ParsingJobJsonResult
 from .parsing_job_markdown_result import ParsingJobMarkdownResult
 from .parsing_job_text_result import ParsingJobTextResult
 from .parsing_usage import ParsingUsage
+from .partition_names import PartitionNames
 from .pipeline import Pipeline
 from .pipeline_configuration_hashes import PipelineConfigurationHashes
 from .pipeline_create import PipelineCreate
@@ -152,14 +150,9 @@ from .pipeline_create_embedding_config import (
     PipelineCreateEmbeddingConfig_OpenaiEmbedding,
     PipelineCreateEmbeddingConfig_VertexaiEmbedding,
 )
-from .pipeline_create_transform_config import (
-    PipelineCreateTransformConfig,
-    PipelineCreateTransformConfig_Advanced,
-    PipelineCreateTransformConfig_Auto,
-)
+from .pipeline_create_transform_config import PipelineCreateTransformConfig
 from .pipeline_data_source import PipelineDataSource
 from .pipeline_data_source_component import PipelineDataSourceComponent
-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
@@ -209,7 +202,6 @@ from .text_node_with_score import TextNodeWithScore
 from .token_chunking_config import TokenChunkingConfig
 from .token_text_splitter import TokenTextSplitter
 from .transformation_category_names import TransformationCategoryNames
-from .user import User
 from .user_organization import UserOrganization
 from .user_organization_create import UserOrganizationCreate
 from .user_organization_delete import UserOrganizationDelete
@@ -249,6 +241,7 @@ __all__ = [
     "CloudConfluenceDataSource",
     "CloudDocument",
     "CloudDocumentCreate",
+    "CloudGoogleDriveDataSource",
     "CloudJiraDataSource",
     "CloudMilvusVectorStore",
     "CloudMongoDbAtlasVectorSearch",
@@ -270,21 +263,15 @@ __all__ = [
     "ConfigurableTransformationNames",
     "ConfiguredTransformationItem",
     "ConfiguredTransformationItemComponent",
-    "ConfiguredTransformationItemComponentOne",
-    "CustomClaims",
     "DataSink",
     "DataSinkComponent",
-    "DataSinkComponentOne",
     "DataSinkCreate",
     "DataSinkCreateComponent",
-    "DataSinkCreateComponentOne",
     "DataSinkDefinition",
     "DataSource",
     "DataSourceComponent",
-    "DataSourceComponentOne",
     "DataSourceCreate",
     "DataSourceCreateComponent",
-    "DataSourceCreateComponentOne",
     "DataSourceCreateCustomMetadataValue",
     "DataSourceCustomMetadataValue",
     "DataSourceDefinition",
@@ -294,6 +281,7 @@ __all__ = [
     "EvalDatasetJobRecord",
     "EvalExecutionParams",
     "EvalExecutionParamsOverride",
+    "EvalMetric",
     "EvalQuestion",
     "EvalQuestionCreate",
     "EvalQuestionResult",
@@ -336,6 +324,7 @@ __all__ = [
     "MetadataFiltersFiltersItem",
     "MetricResult",
     "NodeParser",
+    "NodeRelationship",
     "NoneChunkingConfig",
     "NoneSegmentationConfig",
     "ObjectType",
@@ -354,6 +343,7 @@ __all__ = [
     "ParsingJobMarkdownResult",
     "ParsingJobTextResult",
     "ParsingUsage",
+    "PartitionNames",
     "Pipeline",
     "PipelineConfigurationHashes",
     "PipelineCreate",
@@ -366,11 +356,8 @@ __all__ = [
     "PipelineCreateEmbeddingConfig_OpenaiEmbedding",
     "PipelineCreateEmbeddingConfig_VertexaiEmbedding",
     "PipelineCreateTransformConfig",
-    "PipelineCreateTransformConfig_Advanced",
-    "PipelineCreateTransformConfig_Auto",
     "PipelineDataSource",
     "PipelineDataSourceComponent",
-    "PipelineDataSourceComponentOne",
     "PipelineDataSourceCreate",
     "PipelineDataSourceCustomMetadataValue",
     "PipelineDeployment",
@@ -416,7 +403,6 @@ __all__ = [
     "TokenChunkingConfig",
     "TokenTextSplitter",
     "TransformationCategoryNames",
-    "User",
     "UserOrganization",
     "UserOrganizationCreate",
     "UserOrganizationDelete",

llama_cloud/types/azure_open_ai_embedding.py

@@ -15,34 +15,10 @@ except ImportError:
 
 
 class AzureOpenAiEmbedding(pydantic.BaseModel):
-    """
-    OpenAI class for embeddings.
-
-    Args:
-    mode (str): Mode for embedding.
-    Defaults to OpenAIEmbeddingMode.TEXT_SEARCH_MODE.
-    Options are:
-
-            - OpenAIEmbeddingMode.SIMILARITY_MODE
-            - OpenAIEmbeddingMode.TEXT_SEARCH_MODE
-
-        model (str): Model for embedding.
-            Defaults to OpenAIEmbeddingModelType.TEXT_EMBED_ADA_002.
-            Options are:
-
-            - OpenAIEmbeddingModelType.DAVINCI
-            - OpenAIEmbeddingModelType.CURIE
-            - OpenAIEmbeddingModelType.BABBAGE
-            - OpenAIEmbeddingModelType.ADA
-            - OpenAIEmbeddingModelType.TEXT_EMBED_ADA_002
-    """
-
     model_name: typing.Optional[str] = pydantic.Field(description="The name of the embedding model.")
     embed_batch_size: typing.Optional[int] = pydantic.Field(description="The batch size for embedding calls.")
-    callback_manager: typing.Optional[typing.Dict[str, typing.Any]]
-    num_workers: typing.Optional[int] = pydantic.Field(
-        description="The number of workers to use for async embedding calls."
-    )
+    callback_manager: typing.Optional[typing.Any]
+    num_workers: typing.Optional[int]
     additional_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(
         description="Additional kwargs for the OpenAI API."
     )
@@ -51,17 +27,14 @@ class AzureOpenAiEmbedding(pydantic.BaseModel):
     api_version: typing.Optional[str] = pydantic.Field(description="The version for Azure OpenAI API.")
     max_retries: typing.Optional[int] = pydantic.Field(description="Maximum number of retries.")
     timeout: typing.Optional[float] = pydantic.Field(description="Timeout for each request.")
-    default_headers: typing.Optional[typing.Dict[str, str]] = pydantic.Field(
-        description="The default headers for API requests."
-    )
+    default_headers: typing.Optional[typing.Dict[str, typing.Optional[str]]]
     reuse_client: typing.Optional[bool] = pydantic.Field(
         description="Reuse the OpenAI client between requests. When doing anything with large volumes of async API calls, setting this to false can improve stability."
     )
-    dimensions: typing.Optional[int] = pydantic.Field(
-        description="The number of dimensions on the output embedding vectors. Works only with v3 embedding models."
-    )
-    azure_endpoint: typing.Optional[str] = pydantic.Field(description="The Azure endpoint to use.")
-    azure_deployment: typing.Optional[str] = pydantic.Field(description="The Azure deployment to use.")
+    dimensions: typing.Optional[int]
+    azure_endpoint: typing.Optional[str]
+    azure_deployment: typing.Optional[str]
+    azure_ad_token_provider: typing.Optional[str]
     use_azure_ad: bool = pydantic.Field(
         description="Indicates if Microsoft Entra ID (former Azure AD) is used for token authentication"
     )

llama_cloud/types/base_prompt_template.py

@@ -15,23 +15,12 @@ except ImportError:
 
 
 class BasePromptTemplate(pydantic.BaseModel):
-    """
-    Chainable mixin.
-
-    A module that can produce a `QueryComponent` from a set of inputs through
-    `as_query_component`.
-
-    If plugged in directly into a `QueryPipeline`, the `ChainableMixin` will be
-    converted into a `QueryComponent` with default parameters.
-    """
-
     metadata: typing.Dict[str, typing.Any]
     template_vars: typing.List[str]
     kwargs: typing.Dict[str, str]
-    output_parser: typing.Optional[typing.Dict[str, typing.Any]]
-    template_var_mappings: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(
-        description="Template variable mappings (Optional)."
-    )
+    output_parser: typing.Any
+    template_var_mappings: typing.Optional[typing.Dict[str, typing.Any]]
+    function_mappings: typing.Optional[typing.Dict[str, 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/bedrock_embedding.py

@@ -15,25 +15,15 @@ except ImportError:
 
 
 class BedrockEmbedding(pydantic.BaseModel):
-    """
-    Base class for embeddings.
-    """
-
     model_name: str = pydantic.Field(description="The modelId of the Bedrock model to use.")
     embed_batch_size: typing.Optional[int] = pydantic.Field(description="The batch size for embedding calls.")
-    callback_manager: typing.Optional[typing.Dict[str, typing.Any]]
-    num_workers: typing.Optional[int] = pydantic.Field(
-        description="The number of workers to use for async embedding calls."
-    )
-    profile_name: typing.Optional[str] = pydantic.Field(
-        description="The name of aws profile to use. If not given, then the default profile is used."
-    )
-    aws_access_key_id: typing.Optional[str] = pydantic.Field(description="AWS Access Key ID to use")
-    aws_secret_access_key: typing.Optional[str] = pydantic.Field(description="AWS Secret Access Key to use")
-    aws_session_token: typing.Optional[str] = pydantic.Field(description="AWS Session Token to use")
-    region_name: typing.Optional[str] = pydantic.Field(
-        description="AWS region name to use. Uses region configured in AWS CLI if not passed"
-    )
+    callback_manager: typing.Optional[typing.Any]
+    num_workers: typing.Optional[int]
+    profile_name: typing.Optional[str]
+    aws_access_key_id: typing.Optional[str]
+    aws_secret_access_key: typing.Optional[str]
+    aws_session_token: typing.Optional[str]
+    region_name: typing.Optional[str]
     botocore_session: typing.Optional[typing.Any]
     botocore_config: typing.Optional[typing.Any]
     max_retries: typing.Optional[int] = pydantic.Field(description="The maximum number of API retries.")

llama_cloud/types/box_auth_mechanism.py

@@ -7,10 +7,6 @@ T_Result = typing.TypeVar("T_Result")
 
 
 class BoxAuthMechanism(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
     DEVELOPER_TOKEN = "developer_token"
     CCG = "ccg"
 

llama_cloud/types/character_splitter.py

@@ -23,14 +23,13 @@ class CharacterSplitter(pydantic.BaseModel):
         description="Whether or not to consider metadata when splitting."
     )
     include_prev_next_rel: typing.Optional[bool] = pydantic.Field(description="Include prev/next node relationships.")
-    callback_manager: typing.Optional[typing.Dict[str, typing.Any]]
+    callback_manager: typing.Optional[typing.Any]
+    id_func: typing.Optional[str]
     chunk_size: typing.Optional[int] = pydantic.Field(description="The token chunk size for each chunk.")
     chunk_overlap: typing.Optional[int] = pydantic.Field(description="The token overlap of each chunk when splitting.")
     separator: typing.Optional[str] = pydantic.Field(description="Default separator for splitting into words")
     paragraph_separator: typing.Optional[str] = pydantic.Field(description="Separator between paragraphs.")
-    secondary_chunking_regex: typing.Optional[str] = pydantic.Field(
-        description="Backup regex for splitting into sentences."
-    )
+    secondary_chunking_regex: typing.Optional[str]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/chat_data.py

@@ -17,11 +17,6 @@ except ImportError:
 
 
 class ChatData(pydantic.BaseModel):
-    """
-    Base schema model for BaseComponent classes used in the platform.
-    Comes with special serialization logic for types used commonly in platform codebase.
-    """
-
     retrieval_parameters: typing.Optional[PresetRetrievalParams]
     llm_parameters: typing.Optional[LlmParameters]
     class_name: typing.Optional[str]

llama_cloud/types/chat_message.py

@@ -17,18 +17,13 @@ except ImportError:
 
 
 class ChatMessage(pydantic.BaseModel):
-    """
-    Base schema model for BaseComponent classes used in the platform.
-    Comes with special serialization logic for types used commonly in platform codebase.
-    """
-
     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")
+    content: typing.Optional[str]
     additional_kwargs: typing.Optional[typing.Dict[str, str]] = pydantic.Field(
         description="Additional arguments passed to the model"
     )

llama_cloud/types/cloud_az_storage_blob_data_source.py

@@ -15,26 +15,15 @@ except ImportError:
 
 
 class CloudAzStorageBlobDataSource(pydantic.BaseModel):
-    """
-    Base component object to capture class names.
-    """
-
     container_name: str = pydantic.Field(description="The name of the Azure Storage Blob container to read from.")
     account_url: str = pydantic.Field(description="The Azure Storage Blob account URL to use for authentication.")
-    prefix: typing.Optional[str] = pydantic.Field(
-        description="The prefix of the Azure Storage Blob objects to read from. Use this to filter files at the subdirectory level"
-    )
-    account_name: typing.Optional[str] = pydantic.Field(
-        description="The Azure Storage Blob account name to use for authentication."
-    )
-    account_key: typing.Optional[str] = pydantic.Field(
-        description="The Azure Storage Blob account key to use for authentication."
-    )
-    tenant_id: typing.Optional[str] = pydantic.Field(description="The Azure AD tenant ID to use for authentication.")
-    client_id: typing.Optional[str] = pydantic.Field(description="The Azure AD client ID to use for authentication.")
-    client_secret: typing.Optional[str] = pydantic.Field(
-        description="The Azure AD client secret to use for authentication."
-    )
+    blob: typing.Optional[str]
+    prefix: typing.Optional[str]
+    account_name: typing.Optional[str]
+    account_key: typing.Optional[str]
+    tenant_id: typing.Optional[str]
+    client_id: typing.Optional[str]
+    client_secret: typing.Optional[str]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/cloud_box_data_source.py

@@ -16,25 +16,15 @@ except ImportError:
 
 
 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.")
+    folder_id: typing.Optional[str]
     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."
-    )
+    developer_token: typing.Optional[str]
+    client_id: typing.Optional[str]
+    client_secret: typing.Optional[str]
+    user_id: typing.Optional[str]
+    enterprise_id: typing.Optional[str]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/cloud_chroma_vector_store.py

@@ -15,16 +15,12 @@ except ImportError:
 
 
 class CloudChromaVectorStore(pydantic.BaseModel):
-    """
-    Base class for cloud vector stores.
-    """
-
     supports_nested_metadata_filters: typing.Optional[bool]
     collection_name: typing.Optional[str]
     host: typing.Optional[str]
     port: typing.Optional[str]
     ssl: bool
-    headers: typing.Optional[typing.Dict[str, str]]
+    headers: typing.Optional[typing.Dict[str, typing.Optional[str]]]
     persist_dir: typing.Optional[str]
     collection_kwargs: typing.Optional[typing.Dict[str, typing.Any]]
     class_name: typing.Optional[str]

llama_cloud/types/cloud_confluence_data_source.py

@@ -15,20 +15,16 @@ except ImportError:
 
 
 class CloudConfluenceDataSource(pydantic.BaseModel):
-    """
-    Base component object to capture class names.
-    """
-
     server_url: str = pydantic.Field(description="The server URL of the Confluence instance.")
     authentication_mechanism: str = pydantic.Field(
         description="Type of Authentication for connecting to Confluence APIs."
     )
-    user_name: typing.Optional[str] = pydantic.Field(description="The username to use for authentication.")
-    api_token: typing.Optional[str] = pydantic.Field(description="The API token to use for authentication.")
-    space_key: typing.Optional[str] = pydantic.Field(description="The space key to read from.")
-    page_ids: typing.Optional[str] = pydantic.Field(description="The page IDs of the Confluence to read from.")
-    cql: typing.Optional[str] = pydantic.Field(description="The CQL query to use for fetching pages.")
-    label: typing.Optional[str] = pydantic.Field(description="The label to use for fetching pages.")
+    user_name: typing.Optional[str]
+    api_token: typing.Optional[str]
+    space_key: typing.Optional[str]
+    page_ids: typing.Optional[str]
+    cql: typing.Optional[str]
+    label: typing.Optional[str]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/cloud_document.py

@@ -23,9 +23,7 @@ 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]."
-    )
+    page_positions: typing.Optional[typing.List[int]]
     id: str
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/cloud_document_create.py

@@ -23,9 +23,7 @@ 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]."
-    )
+    page_positions: typing.Optional[typing.List[int]]
     id: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/{user.py → cloud_google_drive_data_source.py}

@@ -4,7 +4,6 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .custom_claims import CustomClaims
 
 try:
     import pydantic
@@ -15,11 +14,12 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class User(pydantic.BaseModel):
-    id: str
-    email: str
-    name: typing.Optional[str] = pydantic.Field(description="The user's name.")
-    claims: typing.Optional[CustomClaims] = pydantic.Field(description="The user's custom claims.")
+class CloudGoogleDriveDataSource(pydantic.BaseModel):
+    folder_id: str = pydantic.Field(description="The ID of the Google Drive folder to read from.")
+    service_account_key: typing.Dict[str, typing.Any] = pydantic.Field(
+        description="The service account key JSON to use for authentication."
+    )
+    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_jira_data_source.py

@@ -19,12 +19,10 @@ class CloudJiraDataSource(pydantic.BaseModel):
     Cloud Jira Data Source integrating JiraReader.
     """
 
-    email: typing.Optional[str] = pydantic.Field(description="The email address to use for authentication.")
-    api_token: typing.Optional[str] = pydantic.Field(
-        description="The API/ Access Token used for Basic, PAT and OAuth2 authentication."
-    )
-    server_url: typing.Optional[str] = pydantic.Field(description="The server url for Jira Cloud.")
-    cloud_id: typing.Optional[str] = pydantic.Field(description="The cloud ID, used in case of OAuth2.")
+    email: typing.Optional[str]
+    api_token: typing.Optional[str]
+    server_url: typing.Optional[str]
+    cloud_id: typing.Optional[str]
     authentication_mechanism: str = pydantic.Field(description="Type of Authentication for connecting to Jira APIs.")
     query: str = pydantic.Field(description="JQL (Jira Query Language) query to search.")
     class_name: typing.Optional[str]

llama_cloud/types/cloud_notion_page_data_source.py

@@ -15,13 +15,9 @@ except ImportError:
 
 
 class CloudNotionPageDataSource(pydantic.BaseModel):
-    """
-    Base component object to capture class names.
-    """
-
     integration_token: str = pydantic.Field(description="The integration token to use for authentication.")
-    database_ids: typing.Optional[str] = pydantic.Field(description="The Notion Database Id to read content from.")
-    page_ids: typing.Optional[str] = pydantic.Field(description="The Page ID's of the Notion to read from.")
+    database_ids: typing.Optional[str]
+    page_ids: typing.Optional[str]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str: