llama-cloud 0.0.16__tar.gz → 0.0.17__tar.gz

{llama_cloud-0.0.16 → llama_cloud-0.0.17}/PKG-INFO

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

{llama_cloud-0.0.16 → llama_cloud-0.0.17}/llama_cloud/__init__.py

@@ -31,6 +31,7 @@ from .types import (
     CloudConfluenceDataSource,
     CloudDocument,
     CloudDocumentCreate,
+    CloudGoogleDriveDataSource,
     CloudJiraDataSource,
     CloudMilvusVectorStore,
     CloudMongoDbAtlasVectorSearch,
@@ -274,6 +275,7 @@ __all__ = [
     "CloudConfluenceDataSource",
     "CloudDocument",
     "CloudDocumentCreate",
+    "CloudGoogleDriveDataSource",
     "CloudJiraDataSource",
     "CloudMilvusVectorStore",
     "CloudMongoDbAtlasVectorSearch",

{llama_cloud-0.0.16 → llama_cloud-0.0.17}/llama_cloud/resources/data_sources/types/data_source_update_component_one.py

@@ -5,6 +5,7 @@ import typing
 from ....types.cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
 from ....types.cloud_box_data_source import CloudBoxDataSource
 from ....types.cloud_confluence_data_source import CloudConfluenceDataSource
+from ....types.cloud_google_drive_data_source import CloudGoogleDriveDataSource
 from ....types.cloud_jira_data_source import CloudJiraDataSource
 from ....types.cloud_notion_page_data_source import CloudNotionPageDataSource
 from ....types.cloud_one_drive_data_source import CloudOneDriveDataSource
@@ -15,6 +16,7 @@ from ....types.cloud_slack_data_source import CloudSlackDataSource
 DataSourceUpdateComponentOne = typing.Union[
     CloudS3DataSource,
     CloudAzStorageBlobDataSource,
+    CloudGoogleDriveDataSource,
     CloudOneDriveDataSource,
     CloudSharepointDataSource,
     CloudSlackDataSource,

{llama_cloud-0.0.16 → llama_cloud-0.0.17}/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
@@ -249,6 +250,7 @@ __all__ = [
     "CloudConfluenceDataSource",
     "CloudDocument",
     "CloudDocumentCreate",
+    "CloudGoogleDriveDataSource",
     "CloudJiraDataSource",
     "CloudMilvusVectorStore",
     "CloudMongoDbAtlasVectorSearch",

llama_cloud-0.0.17/llama_cloud/types/cloud_google_drive_data_source.py

@@ -0,0 +1,39 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class CloudGoogleDriveDataSource(pydantic.BaseModel):
+    """
+    Base component object to capture class names.
+    """
+
+    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}
+        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-0.0.16 → llama_cloud-0.0.17}/llama_cloud/types/configurable_data_source_names.py

@@ -13,6 +13,7 @@ class ConfigurableDataSourceNames(str, enum.Enum):
 
     S_3 = "S3"
     AZURE_STORAGE_BLOB = "AZURE_STORAGE_BLOB"
+    GOOGLE_DRIVE = "GOOGLE_DRIVE"
     MICROSOFT_ONEDRIVE = "MICROSOFT_ONEDRIVE"
     MICROSOFT_SHAREPOINT = "MICROSOFT_SHAREPOINT"
     SLACK = "SLACK"
@@ -25,6 +26,7 @@ class ConfigurableDataSourceNames(str, enum.Enum):
         self,
         s_3: typing.Callable[[], T_Result],
         azure_storage_blob: typing.Callable[[], T_Result],
+        google_drive: typing.Callable[[], T_Result],
         microsoft_onedrive: typing.Callable[[], T_Result],
         microsoft_sharepoint: typing.Callable[[], T_Result],
         slack: typing.Callable[[], T_Result],
@@ -37,6 +39,8 @@ class ConfigurableDataSourceNames(str, enum.Enum):
             return s_3()
         if self is ConfigurableDataSourceNames.AZURE_STORAGE_BLOB:
             return azure_storage_blob()
+        if self is ConfigurableDataSourceNames.GOOGLE_DRIVE:
+            return google_drive()
         if self is ConfigurableDataSourceNames.MICROSOFT_ONEDRIVE:
             return microsoft_onedrive()
         if self is ConfigurableDataSourceNames.MICROSOFT_SHAREPOINT:

{llama_cloud-0.0.16 → llama_cloud-0.0.17}/llama_cloud/types/data_source_component_one.py

@@ -5,6 +5,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_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
@@ -15,6 +16,7 @@ from .cloud_slack_data_source import CloudSlackDataSource
 DataSourceComponentOne = typing.Union[
     CloudS3DataSource,
     CloudAzStorageBlobDataSource,
+    CloudGoogleDriveDataSource,
     CloudOneDriveDataSource,
     CloudSharepointDataSource,
     CloudSlackDataSource,

{llama_cloud-0.0.16 → llama_cloud-0.0.17}/llama_cloud/types/data_source_create_component_one.py

@@ -5,6 +5,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_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
@@ -15,6 +16,7 @@ from .cloud_slack_data_source import CloudSlackDataSource
 DataSourceCreateComponentOne = typing.Union[
     CloudS3DataSource,
     CloudAzStorageBlobDataSource,
+    CloudGoogleDriveDataSource,
     CloudOneDriveDataSource,
     CloudSharepointDataSource,
     CloudSlackDataSource,

{llama_cloud-0.0.16 → llama_cloud-0.0.17}/llama_cloud/types/pipeline_data_source_component_one.py

@@ -5,6 +5,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_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
@@ -15,6 +16,7 @@ from .cloud_slack_data_source import CloudSlackDataSource
 PipelineDataSourceComponentOne = typing.Union[
     CloudS3DataSource,
     CloudAzStorageBlobDataSource,
+    CloudGoogleDriveDataSource,
     CloudOneDriveDataSource,
     CloudSharepointDataSource,
     CloudSlackDataSource,

{llama_cloud-0.0.16 → llama_cloud-0.0.17}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "llama-cloud"
-version = "0.0.16"
+version = "0.0.17"
 description = ""
 authors = ["Logan Markewich <logan@runllama.ai>"]
 readme = "README.md"