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

llama_cloud/types/pipeline_data_source_component.py

@@ -2,6 +2,27 @@
 
 import typing
 
-from .pipeline_data_source_component_one import PipelineDataSourceComponentOne
+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_google_drive_data_source import CloudGoogleDriveDataSource
+from .cloud_jira_data_source import CloudJiraDataSource
+from .cloud_notion_page_data_source import CloudNotionPageDataSource
+from .cloud_one_drive_data_source import CloudOneDriveDataSource
+from .cloud_s_3_data_source import CloudS3DataSource
+from .cloud_sharepoint_data_source import CloudSharepointDataSource
+from .cloud_slack_data_source import CloudSlackDataSource
 
-PipelineDataSourceComponent = typing.Union[typing.Dict[str, typing.Any], PipelineDataSourceComponentOne]
+PipelineDataSourceComponent = typing.Union[
+    typing.Dict[str, typing.Any],
+    CloudS3DataSource,
+    CloudAzStorageBlobDataSource,
+    CloudGoogleDriveDataSource,
+    CloudOneDriveDataSource,
+    CloudSharepointDataSource,
+    CloudSlackDataSource,
+    CloudNotionPageDataSource,
+    CloudConfluenceDataSource,
+    CloudJiraDataSource,
+    CloudBoxDataSource,
+]

llama_cloud/types/pipeline_data_source_create.py

@@ -20,9 +20,7 @@ class PipelineDataSourceCreate(pydantic.BaseModel):
     """
 
     data_source_id: str = pydantic.Field(description="The ID of the data source.")
-    sync_interval: typing.Optional[float] = pydantic.Field(
-        description="The interval at which the data source should be synced."
-    )
+    sync_interval: typing.Optional[float]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/pipeline_deployment.py

@@ -16,16 +16,12 @@ except ImportError:
 
 
 class PipelineDeployment(pydantic.BaseModel):
-    """
-    Base schema model containing common database fields.
-    """
-
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
-    updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
     status: ManagedIngestionStatus = pydantic.Field(description="Status of the pipeline deployment.")
-    started_at: typing.Optional[dt.datetime] = pydantic.Field(description="Time the pipeline deployment started.")
-    ended_at: typing.Optional[dt.datetime] = pydantic.Field(description="Time the pipeline deployment finished.")
+    started_at: typing.Optional[dt.datetime]
+    ended_at: typing.Optional[dt.datetime]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/pipeline_embedding_config.py

@@ -15,8 +15,8 @@ from .open_ai_embedding_config import OpenAiEmbeddingConfig
 from .vertex_ai_embedding_config import VertexAiEmbeddingConfig
 
 
-class PipelineEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
-    type: typing_extensions.Literal["OPENAI_EMBEDDING"]
+class PipelineEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
+    type: typing_extensions.Literal["AZURE_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -24,8 +24,8 @@ class PipelineEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
         allow_population_by_field_name = True
 
 
-class PipelineEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
-    type: typing_extensions.Literal["AZURE_EMBEDDING"]
+class PipelineEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
+    type: typing_extensions.Literal["BEDROCK_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -33,8 +33,8 @@ class PipelineEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
         allow_population_by_field_name = True
 
 
-class PipelineEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInferenceApiEmbeddingConfig):
-    type: typing_extensions.Literal["HUGGINGFACE_API_EMBEDDING"]
+class PipelineEmbeddingConfig_CohereEmbedding(CohereEmbeddingConfig):
+    type: typing_extensions.Literal["COHERE_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -42,8 +42,8 @@ class PipelineEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInferenceApiEmb
         allow_population_by_field_name = True
 
 
-class PipelineEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
-    type: typing_extensions.Literal["BEDROCK_EMBEDDING"]
+class PipelineEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
+    type: typing_extensions.Literal["GEMINI_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -51,8 +51,8 @@ class PipelineEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
         allow_population_by_field_name = True
 
 
-class PipelineEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
-    type: typing_extensions.Literal["GEMINI_EMBEDDING"]
+class PipelineEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInferenceApiEmbeddingConfig):
+    type: typing_extensions.Literal["HUGGINGFACE_API_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -60,8 +60,8 @@ class PipelineEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
         allow_population_by_field_name = True
 
 
-class PipelineEmbeddingConfig_CohereEmbedding(CohereEmbeddingConfig):
-    type: typing_extensions.Literal["COHERE_EMBEDDING"]
+class PipelineEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
+    type: typing_extensions.Literal["OPENAI_EMBEDDING"]
 
     class Config:
         frozen = True
@@ -79,11 +79,11 @@ class PipelineEmbeddingConfig_VertexaiEmbedding(VertexAiEmbeddingConfig):
 
 
 PipelineEmbeddingConfig = typing.Union[
-    PipelineEmbeddingConfig_OpenaiEmbedding,
     PipelineEmbeddingConfig_AzureEmbedding,
-    PipelineEmbeddingConfig_HuggingfaceApiEmbedding,
     PipelineEmbeddingConfig_BedrockEmbedding,
-    PipelineEmbeddingConfig_GeminiEmbedding,
     PipelineEmbeddingConfig_CohereEmbedding,
+    PipelineEmbeddingConfig_GeminiEmbedding,
+    PipelineEmbeddingConfig_HuggingfaceApiEmbedding,
+    PipelineEmbeddingConfig_OpenaiEmbedding,
     PipelineEmbeddingConfig_VertexaiEmbedding,
 ]

llama_cloud/types/pipeline_file.py

@@ -23,27 +23,19 @@ class PipelineFile(pydantic.BaseModel):
     """
 
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
-    updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
     name: typing.Optional[str]
-    file_size: typing.Optional[int] = pydantic.Field(description="Size of the file in bytes")
-    file_type: typing.Optional[str] = pydantic.Field(description="File type (e.g. pdf, docx, etc.)")
+    file_size: typing.Optional[int]
+    file_type: typing.Optional[str]
     project_id: str = pydantic.Field(description="The ID of the project that the file belongs to")
-    last_modified_at: typing.Optional[dt.datetime] = pydantic.Field(description="The last modified time of the file")
-    resource_info: typing.Optional[typing.Dict[str, PipelineFileResourceInfoValue]] = pydantic.Field(
-        description="Resource information for the file"
-    )
-    data_source_id: typing.Optional[str] = pydantic.Field(
-        description="The ID of the data source that the file belongs to"
-    )
-    file_id: typing.Optional[str] = pydantic.Field(description="The ID of the file")
+    last_modified_at: typing.Optional[dt.datetime]
+    resource_info: typing.Optional[typing.Dict[str, typing.Optional[PipelineFileResourceInfoValue]]]
+    data_source_id: typing.Optional[str]
+    file_id: typing.Optional[str]
     pipeline_id: str = pydantic.Field(description="The ID of the pipeline that the file is associated with")
-    custom_metadata: typing.Optional[typing.Dict[str, PipelineFileCustomMetadataValue]] = pydantic.Field(
-        description="Custom metadata for the file"
-    )
-    config_hash: typing.Optional[typing.Dict[str, PipelineFileConfigHashValue]] = pydantic.Field(
-        description="Hashes for the configuration of the pipeline."
-    )
+    custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[PipelineFileCustomMetadataValue]]]
+    config_hash: typing.Optional[typing.Dict[str, typing.Optional[PipelineFileConfigHashValue]]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/pipeline_file_create.py

@@ -21,9 +21,7 @@ class PipelineFileCreate(pydantic.BaseModel):
     """
 
     file_id: str = pydantic.Field(description="The ID of the file")
-    custom_metadata: typing.Optional[typing.Dict[str, PipelineFileCreateCustomMetadataValue]] = pydantic.Field(
-        description="Custom metadata for the file"
-    )
+    custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[PipelineFileCreateCustomMetadataValue]]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/playground_session.py

@@ -23,8 +23,8 @@ class PlaygroundSession(pydantic.BaseModel):
     """
 
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
-    updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
     pipeline_id: str
     user_id: str
     llm_params_id: str

llama_cloud/types/preset_retrieval_params.py

@@ -21,19 +21,16 @@ class PresetRetrievalParams(pydantic.BaseModel):
     Schema for the search params for an retrieval execution that can be preset for a pipeline.
     """
 
-    dense_similarity_top_k: typing.Optional[int] = pydantic.Field(description="Number of nodes for dense retrieval.")
-    sparse_similarity_top_k: typing.Optional[int] = pydantic.Field(description="Number of nodes for sparse retrieval.")
-    enable_reranking: typing.Optional[bool] = pydantic.Field(description="Enable reranking for retrieval")
-    rerank_top_n: typing.Optional[int] = pydantic.Field(description="Number of reranked nodes for returning.")
-    alpha: typing.Optional[float] = pydantic.Field(
-        description="Alpha value for hybrid retrieval to determine the weights between dense and sparse retrieval. 0 is sparse retrieval and 1 is dense retrieval."
-    )
-    search_filters: typing.Optional[MetadataFilters] = pydantic.Field(description="Search filters for retrieval.")
-    files_top_k: typing.Optional[int] = pydantic.Field(
-        description="Number of files to retrieve (only for retrieval mode files_via_metadata and files_via_content)."
-    )
+    dense_similarity_top_k: typing.Optional[int]
+    sparse_similarity_top_k: typing.Optional[int]
+    enable_reranking: typing.Optional[bool]
+    rerank_top_n: typing.Optional[int]
+    alpha: typing.Optional[float]
+    search_filters: typing.Optional[MetadataFilters]
+    files_top_k: typing.Optional[int]
     retrieval_mode: typing.Optional[RetrievalMode] = pydantic.Field(description="The retrieval mode for the query.")
     retrieve_image_nodes: typing.Optional[bool] = pydantic.Field(description="Whether to retrieve image nodes.")
+    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/presigned_url.py

@@ -21,9 +21,7 @@ class PresignedUrl(pydantic.BaseModel):
 
     url: str = pydantic.Field(description="A presigned URL for IO operations against a private file")
     expires_at: dt.datetime = pydantic.Field(description="The time at which the presigned URL expires")
-    form_fields: typing.Optional[typing.Dict[str, str]] = pydantic.Field(
-        description="Form fields for a presigned POST request"
-    )
+    form_fields: 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/project.py

@@ -21,8 +21,8 @@ class Project(pydantic.BaseModel):
 
     name: str
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
-    updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
     ad_hoc_eval_dataset_id: typing.Optional[str]
     organization_id: str = pydantic.Field(description="The Organization ID the project is under.")
     is_default: typing.Optional[bool] = pydantic.Field(

llama_cloud/types/prompt_mixin_prompts.py

@@ -21,7 +21,7 @@ class PromptMixinPrompts(pydantic.BaseModel):
     """
 
     project_id: str = pydantic.Field(description="The ID of the project.")
-    id: typing.Optional[str] = pydantic.Field(description="The ID of the prompt set.")
+    id: typing.Optional[str]
     name: str = pydantic.Field(description="The name of the prompt set.")
     prompts: typing.List[PromptSpec] = pydantic.Field(description="The prompts.")
 

llama_cloud/types/prompt_spec.py

@@ -19,10 +19,8 @@ class PromptSpec(pydantic.BaseModel):
     prompt_key: str = pydantic.Field(description="The key of the prompt in the PromptMixin.")
     prompt_class: str = pydantic.Field(description="The class of the prompt (PromptTemplate or ChatPromptTemplate).")
     prompt_type: str = pydantic.Field(description="The type of prompt.")
-    template: typing.Optional[str] = pydantic.Field(description="The template of the prompt.")
-    message_templates: typing.Optional[typing.List[ChatMessage]] = pydantic.Field(
-        description="The chat message templates of the prompt."
-    )
+    template: typing.Optional[str]
+    message_templates: typing.Optional[typing.List[ChatMessage]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/related_node_info.py

@@ -16,10 +16,6 @@ except ImportError:
 
 
 class RelatedNodeInfo(pydantic.BaseModel):
-    """
-    Base component object to capture class names.
-    """
-
     node_id: str
     node_type: typing.Optional[ObjectType]
     metadata: typing.Optional[typing.Dict[str, typing.Any]]

llama_cloud/types/retrieval_mode.py

@@ -7,10 +7,6 @@ T_Result = typing.TypeVar("T_Result")
 
 
 class RetrievalMode(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
     CHUNKS = "chunks"
     FILES_VIA_METADATA = "files_via_metadata"
     FILES_VIA_CONTENT = "files_via_content"

llama_cloud/types/sentence_splitter.py

@@ -27,14 +27,13 @@ class SentenceSplitter(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/supported_llm_model_names.py

@@ -7,10 +7,6 @@ T_Result = typing.TypeVar("T_Result")
 
 
 class SupportedLlmModelNames(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
     GPT_3_5_TURBO = "GPT_3_5_TURBO"
     GPT_4 = "GPT_4"
     GPT_4_TURBO = "GPT_4_TURBO"

llama_cloud/types/text_node.py

@@ -16,14 +16,8 @@ except ImportError:
 
 
 class TextNode(pydantic.BaseModel):
-    """
-    Base node Object.
-
-    Generic abstract interface for retrievable nodes
-    """
-
     id: typing.Optional[str] = pydantic.Field(alias="id_", description="Unique ID of the node.")
-    embedding: typing.Optional[typing.List[float]] = pydantic.Field(description="Embedding of the node.")
+    embedding: typing.Optional[typing.List[float]]
     extra_info: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(
         description="A flat dictionary of metadata fields"
     )
@@ -38,8 +32,8 @@ class TextNode(pydantic.BaseModel):
     )
     text: typing.Optional[str] = pydantic.Field(description="Text content of the node.")
     mimetype: typing.Optional[str] = pydantic.Field(description="MIME type of the node content.")
-    start_char_idx: typing.Optional[int] = pydantic.Field(description="Start char index of the node.")
-    end_char_idx: typing.Optional[int] = pydantic.Field(description="End char index of the node.")
+    start_char_idx: typing.Optional[int]
+    end_char_idx: typing.Optional[int]
     text_template: typing.Optional[str] = pydantic.Field(
         description="Template for how text is formatted, with {content} and {metadata_str} placeholders."
     )

llama_cloud/types/token_text_splitter.py

@@ -23,7 +23,8 @@ class TokenTextSplitter(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")

llama_cloud/types/transformation_category_names.py

@@ -7,10 +7,6 @@ T_Result = typing.TypeVar("T_Result")
 
 
 class TransformationCategoryNames(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
     NODE_PARSER = "NODE_PARSER"
     EMBEDDING = "EMBEDDING"
 

llama_cloud/types/user_organization.py

@@ -20,20 +20,16 @@ class UserOrganization(pydantic.BaseModel):
     """
 
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
-    updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
     email: str = pydantic.Field(description="The user's email address.")
-    user_id: typing.Optional[str] = pydantic.Field(description="The user's ID.")
+    user_id: typing.Optional[str]
     organization_id: str = pydantic.Field(description="The organization's ID.")
     pending: typing.Optional[bool] = pydantic.Field(
         description="Whether the user's membership is pending account signup."
     )
-    invited_by_user_id: typing.Optional[str] = pydantic.Field(
-        description="The user ID of the user who added the user to the organization."
-    )
-    invited_by_user_email: typing.Optional[str] = pydantic.Field(
-        description="The email address of the user who added the user to the organization."
-    )
+    invited_by_user_id: typing.Optional[str]
+    invited_by_user_email: 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/user_organization_create.py

@@ -19,8 +19,8 @@ class UserOrganizationCreate(pydantic.BaseModel):
     Schema for creating a user's membership to an organization.
     """
 
-    user_id: typing.Optional[str] = pydantic.Field(description="The user's ID.")
-    email: typing.Optional[str] = pydantic.Field(description="The user's email address.")
+    user_id: typing.Optional[str]
+    email: 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/user_organization_delete.py

@@ -19,8 +19,8 @@ class UserOrganizationDelete(pydantic.BaseModel):
     Schema for deleting a user's membership to an organization.
     """
 
-    user_id: typing.Optional[str] = pydantic.Field(description="The user's ID.")
-    email: typing.Optional[str] = pydantic.Field(description="The user's email address.")
+    user_id: typing.Optional[str]
+    email: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{llama_cloud-0.0.17.dist-info → llama_cloud-0.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-cloud
-Version: 0.0.17
+Version: 0.1.0
 Summary: 
 Author: Logan Markewich
 Author-email: logan@runllama.ai