llama-cloud 0.1.3__tar.gz → 0.1.5__tar.gz

{llama_cloud-0.1.3 → llama_cloud-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-cloud
-Version: 0.1.3
+Version: 0.1.5
 Summary: 
 License: MIT
 Author: Logan Markewich
@@ -12,7 +12,6 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: httpx (>=0.20.0)
 Requires-Dist: pydantic (>=1.10)
 Description-Content-Type: text/markdown

{llama_cloud-0.1.3 → llama_cloud-0.1.5}/llama_cloud/__init__.py

@@ -64,6 +64,24 @@ from .types import (
     DataSourceCustomMetadataValue,
     DataSourceDefinition,
     ElementSegmentationConfig,
+    EmbeddingModelConfig,
+    EmbeddingModelConfigEmbeddingConfig,
+    EmbeddingModelConfigEmbeddingConfig_AzureEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_BedrockEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_CohereEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_GeminiEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_HuggingfaceApiEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_OpenaiEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_VertexaiEmbedding,
+    EmbeddingModelConfigUpdate,
+    EmbeddingModelConfigUpdateEmbeddingConfig,
+    EmbeddingModelConfigUpdateEmbeddingConfig_AzureEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_BedrockEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_CohereEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_GeminiEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_OpenaiEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_VertexaiEmbedding,
     EvalDataset,
     EvalDatasetJobParams,
     EvalDatasetJobRecord,
@@ -90,6 +108,7 @@ from .types import (
     HuggingFaceInferenceApiEmbeddingToken,
     IngestionErrorResponse,
     InputMessage,
+    IntervalUsageAndPlan,
     JobNameMapping,
     LlamaParseParameters,
     LlamaParseSupportedFileExtensions,
@@ -123,6 +142,7 @@ from .types import (
     PageScreenshotNodeWithScore,
     PageSegmentationConfig,
     PageSplitterNodeParser,
+    PaginatedListPipelineFilesResponse,
     ParserLanguages,
     ParsingHistoryItem,
     ParsingJob,
@@ -167,6 +187,7 @@ from .types import (
     PipelineTransformConfig_Advanced,
     PipelineTransformConfig_Auto,
     PipelineType,
+    Plan,
     PlaygroundSession,
     Pooling,
     PresetRetrievalParams,
@@ -192,6 +213,7 @@ from .types import (
     TokenChunkingConfig,
     TokenTextSplitter,
     TransformationCategoryNames,
+    Usage,
     UserOrganization,
     UserOrganizationCreate,
     UserOrganizationDelete,
@@ -207,8 +229,17 @@ from .resources import (
     DataSinkUpdateComponent,
     DataSourceUpdateComponent,
     DataSourceUpdateCustomMetadataValue,
+    EmbeddingModelConfigCreateEmbeddingConfig,
+    EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_BedrockEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_CohereEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_GeminiEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_HuggingfaceApiEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_OpenaiEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_VertexaiEmbedding,
     ExtractionSchemaCreateDataSchemaValue,
     ExtractionSchemaUpdateDataSchemaValue,
+    FileCreateFromUrlResourceInfoValue,
     FileCreateResourceInfoValue,
     PipelineFileUpdateCustomMetadataValue,
     PipelineUpdateEmbeddingConfig,
@@ -223,6 +254,7 @@ from .resources import (
     component_definitions,
     data_sinks,
     data_sources,
+    embedding_model_configs,
     evals,
     extraction,
     files,
@@ -300,6 +332,32 @@ __all__ = [
     "DataSourceUpdateComponent",
     "DataSourceUpdateCustomMetadataValue",
     "ElementSegmentationConfig",
+    "EmbeddingModelConfig",
+    "EmbeddingModelConfigCreateEmbeddingConfig",
+    "EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_BedrockEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_CohereEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_GeminiEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_HuggingfaceApiEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_OpenaiEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_VertexaiEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig",
+    "EmbeddingModelConfigEmbeddingConfig_AzureEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_BedrockEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_CohereEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_GeminiEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_HuggingfaceApiEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_OpenaiEmbedding",
+    "EmbeddingModelConfigEmbeddingConfig_VertexaiEmbedding",
+    "EmbeddingModelConfigUpdate",
+    "EmbeddingModelConfigUpdateEmbeddingConfig",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_AzureEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_BedrockEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_CohereEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_GeminiEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_OpenaiEmbedding",
+    "EmbeddingModelConfigUpdateEmbeddingConfig_VertexaiEmbedding",
     "EvalDataset",
     "EvalDatasetJobParams",
     "EvalDatasetJobRecord",
@@ -317,6 +375,7 @@ __all__ = [
     "ExtractionSchemaDataSchemaValue",
     "ExtractionSchemaUpdateDataSchemaValue",
     "File",
+    "FileCreateFromUrlResourceInfoValue",
     "FileCreateResourceInfoValue",
     "FileResourceInfoValue",
     "FilterCondition",
@@ -329,6 +388,7 @@ __all__ = [
     "HuggingFaceInferenceApiEmbeddingToken",
     "IngestionErrorResponse",
     "InputMessage",
+    "IntervalUsageAndPlan",
     "JobNameMapping",
     "LlamaCloudEnvironment",
     "LlamaParseParameters",
@@ -363,6 +423,7 @@ __all__ = [
     "PageScreenshotNodeWithScore",
     "PageSegmentationConfig",
     "PageSplitterNodeParser",
+    "PaginatedListPipelineFilesResponse",
     "ParserLanguages",
     "ParsingHistoryItem",
     "ParsingJob",
@@ -417,6 +478,7 @@ __all__ = [
     "PipelineUpdateEmbeddingConfig_OpenaiEmbedding",
     "PipelineUpdateEmbeddingConfig_VertexaiEmbedding",
     "PipelineUpdateTransformConfig",
+    "Plan",
     "PlaygroundSession",
     "Pooling",
     "PresetRetrievalParams",
@@ -443,6 +505,7 @@ __all__ = [
     "TokenTextSplitter",
     "TransformationCategoryNames",
     "UnprocessableEntityError",
+    "Usage",
     "UserOrganization",
     "UserOrganizationCreate",
     "UserOrganizationDelete",
@@ -455,6 +518,7 @@ __all__ = [
     "component_definitions",
     "data_sinks",
     "data_sources",
+    "embedding_model_configs",
     "evals",
     "extraction",
     "files",

{llama_cloud-0.1.3 → llama_cloud-0.1.5}/llama_cloud/client.py

@@ -9,6 +9,7 @@ from .environment import LlamaCloudEnvironment
 from .resources.component_definitions.client import AsyncComponentDefinitionsClient, ComponentDefinitionsClient
 from .resources.data_sinks.client import AsyncDataSinksClient, DataSinksClient
 from .resources.data_sources.client import AsyncDataSourcesClient, DataSourcesClient
+from .resources.embedding_model_configs.client import AsyncEmbeddingModelConfigsClient, EmbeddingModelConfigsClient
 from .resources.evals.client import AsyncEvalsClient, EvalsClient
 from .resources.extraction.client import AsyncExtractionClient, ExtractionClient
 from .resources.files.client import AsyncFilesClient, FilesClient
@@ -35,6 +36,7 @@ class LlamaCloud:
         )
         self.data_sinks = DataSinksClient(client_wrapper=self._client_wrapper)
         self.data_sources = DataSourcesClient(client_wrapper=self._client_wrapper)
+        self.embedding_model_configs = EmbeddingModelConfigsClient(client_wrapper=self._client_wrapper)
         self.organizations = OrganizationsClient(client_wrapper=self._client_wrapper)
         self.projects = ProjectsClient(client_wrapper=self._client_wrapper)
         self.files = FilesClient(client_wrapper=self._client_wrapper)
@@ -62,6 +64,7 @@ class AsyncLlamaCloud:
         )
         self.data_sinks = AsyncDataSinksClient(client_wrapper=self._client_wrapper)
         self.data_sources = AsyncDataSourcesClient(client_wrapper=self._client_wrapper)
+        self.embedding_model_configs = AsyncEmbeddingModelConfigsClient(client_wrapper=self._client_wrapper)
         self.organizations = AsyncOrganizationsClient(client_wrapper=self._client_wrapper)
         self.projects = AsyncProjectsClient(client_wrapper=self._client_wrapper)
         self.files = AsyncFilesClient(client_wrapper=self._client_wrapper)

{llama_cloud-0.1.3 → llama_cloud-0.1.5}/llama_cloud/resources/__init__.py

@@ -4,6 +4,7 @@ from . import (
     component_definitions,
     data_sinks,
     data_sources,
+    embedding_model_configs,
     evals,
     extraction,
     files,
@@ -14,8 +15,18 @@ from . import (
 )
 from .data_sinks import DataSinkUpdateComponent
 from .data_sources import DataSourceUpdateComponent, DataSourceUpdateCustomMetadataValue
+from .embedding_model_configs import (
+    EmbeddingModelConfigCreateEmbeddingConfig,
+    EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_BedrockEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_CohereEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_GeminiEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_HuggingfaceApiEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_OpenaiEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_VertexaiEmbedding,
+)
 from .extraction import ExtractionSchemaCreateDataSchemaValue, ExtractionSchemaUpdateDataSchemaValue
-from .files import FileCreateResourceInfoValue
+from .files import FileCreateFromUrlResourceInfoValue, FileCreateResourceInfoValue
 from .pipelines import (
     PipelineFileUpdateCustomMetadataValue,
     PipelineUpdateEmbeddingConfig,
@@ -33,8 +44,17 @@ __all__ = [
     "DataSinkUpdateComponent",
     "DataSourceUpdateComponent",
     "DataSourceUpdateCustomMetadataValue",
+    "EmbeddingModelConfigCreateEmbeddingConfig",
+    "EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_BedrockEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_CohereEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_GeminiEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_HuggingfaceApiEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_OpenaiEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_VertexaiEmbedding",
     "ExtractionSchemaCreateDataSchemaValue",
     "ExtractionSchemaUpdateDataSchemaValue",
+    "FileCreateFromUrlResourceInfoValue",
     "FileCreateResourceInfoValue",
     "PipelineFileUpdateCustomMetadataValue",
     "PipelineUpdateEmbeddingConfig",
@@ -49,6 +69,7 @@ __all__ = [
     "component_definitions",
     "data_sinks",
     "data_sources",
+    "embedding_model_configs",
     "evals",
     "extraction",
     "files",

{llama_cloud-0.1.3 → llama_cloud-0.1.5}/llama_cloud/resources/data_sinks/client.py

@@ -31,13 +31,16 @@ class DataSinksClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def list_data_sinks(self, *, project_id: typing.Optional[str] = None) -> typing.List[DataSink]:
+    def list_data_sinks(
+        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    ) -> typing.List[DataSink]:
         """
         List data sinks for a given project.
-        If project_id is not provided, uses the default project.
 
         Parameters:
             - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -49,7 +52,7 @@ class DataSinksClient:
         _response = self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -282,13 +285,16 @@ class AsyncDataSinksClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def list_data_sinks(self, *, project_id: typing.Optional[str] = None) -> typing.List[DataSink]:
+    async def list_data_sinks(
+        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    ) -> typing.List[DataSink]:
         """
         List data sinks for a given project.
-        If project_id is not provided, uses the default project.
 
         Parameters:
             - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -300,7 +306,7 @@ class AsyncDataSinksClient:
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )

llama_cloud-0.1.5/llama_cloud/resources/embedding_model_configs/__init__.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .types import (
+    EmbeddingModelConfigCreateEmbeddingConfig,
+    EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_BedrockEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_CohereEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_GeminiEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_HuggingfaceApiEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_OpenaiEmbedding,
+    EmbeddingModelConfigCreateEmbeddingConfig_VertexaiEmbedding,
+)
+
+__all__ = [
+    "EmbeddingModelConfigCreateEmbeddingConfig",
+    "EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_BedrockEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_CohereEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_GeminiEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_HuggingfaceApiEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_OpenaiEmbedding",
+    "EmbeddingModelConfigCreateEmbeddingConfig_VertexaiEmbedding",
+]