llama-cloud 0.1.20__py3-none-any.whl → 0.1.22__py3-none-any.whl

llama_cloud/__init__.py

@@ -60,57 +60,21 @@ from .types import (
     CompositeRetrievedTextNodeWithScore,
     ConfigurableDataSinkNames,
     ConfigurableDataSourceNames,
-    ConfigurableTransformationDefinition,
-    ConfigurableTransformationNames,
     CreditType,
     DataSink,
     DataSinkComponent,
-    DataSinkComponent_AzureAiSearch,
-    DataSinkComponent_Milvus,
-    DataSinkComponent_MongodbAtlas,
-    DataSinkComponent_Pinecone,
-    DataSinkComponent_Postgres,
-    DataSinkComponent_Qdrant,
     DataSinkCreate,
     DataSinkCreateComponent,
-    DataSinkCreateComponent_AzureAiSearch,
-    DataSinkCreateComponent_Milvus,
-    DataSinkCreateComponent_MongodbAtlas,
-    DataSinkCreateComponent_Pinecone,
-    DataSinkCreateComponent_Postgres,
-    DataSinkCreateComponent_Qdrant,
-    DataSinkDefinition,
     DataSource,
     DataSourceComponent,
-    DataSourceComponentOne,
-    DataSourceComponentOne_AzureStorageBlob,
-    DataSourceComponentOne_Box,
-    DataSourceComponentOne_Confluence,
-    DataSourceComponentOne_GoogleDrive,
-    DataSourceComponentOne_Jira,
-    DataSourceComponentOne_MicrosoftOnedrive,
-    DataSourceComponentOne_MicrosoftSharepoint,
-    DataSourceComponentOne_NotionPage,
-    DataSourceComponentOne_S3,
-    DataSourceComponentOne_Slack,
     DataSourceCreate,
     DataSourceCreateComponent,
-    DataSourceCreateComponentOne,
-    DataSourceCreateComponentOne_AzureStorageBlob,
-    DataSourceCreateComponentOne_Box,
-    DataSourceCreateComponentOne_Confluence,
-    DataSourceCreateComponentOne_GoogleDrive,
-    DataSourceCreateComponentOne_Jira,
-    DataSourceCreateComponentOne_MicrosoftOnedrive,
-    DataSourceCreateComponentOne_MicrosoftSharepoint,
-    DataSourceCreateComponentOne_NotionPage,
-    DataSourceCreateComponentOne_S3,
-    DataSourceCreateComponentOne_Slack,
     DataSourceCreateCustomMetadataValue,
     DataSourceCustomMetadataValue,
-    DataSourceDefinition,
     DataSourceUpdateDispatcherConfig,
     DeleteParams,
+    DocumentBlock,
+    DocumentChunkMode,
     DocumentIngestionJobParams,
     EditSuggestion,
     EditSuggestionBlocksItem,
@@ -142,6 +106,7 @@ from .types import (
     ExtractJobCreateDataSchemaOverride,
     ExtractJobCreateDataSchemaOverrideZeroValue,
     ExtractMode,
+    ExtractModels,
     ExtractResultset,
     ExtractResultsetData,
     ExtractResultsetDataItemValue,
@@ -195,10 +160,14 @@ from .types import (
     LlamaIndexCoreBaseLlmsTypesChatMessage,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Audio,
+    LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Document,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Image,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Text,
     LlamaParseParameters,
     LlamaParseSupportedFileExtensions,
+    LlmConfigResult,
+    LlmConfigResultLlmType,
+    LlmConfigsResponse,
     LlmModelData,
     LlmParameters,
     LoadFilesJobConfig,
@@ -256,17 +225,6 @@ from .types import (
     PipelineCreateTransformConfig,
     PipelineDataSource,
     PipelineDataSourceComponent,
-    PipelineDataSourceComponentOne,
-    PipelineDataSourceComponentOne_AzureStorageBlob,
-    PipelineDataSourceComponentOne_Box,
-    PipelineDataSourceComponentOne_Confluence,
-    PipelineDataSourceComponentOne_GoogleDrive,
-    PipelineDataSourceComponentOne_Jira,
-    PipelineDataSourceComponentOne_MicrosoftOnedrive,
-    PipelineDataSourceComponentOne_MicrosoftSharepoint,
-    PipelineDataSourceComponentOne_NotionPage,
-    PipelineDataSourceComponentOne_S3,
-    PipelineDataSourceComponentOne_Slack,
     PipelineDataSourceCreate,
     PipelineDataSourceCustomMetadataValue,
     PipelineDataSourceStatus,
@@ -291,6 +249,7 @@ from .types import (
     PipelineFileUpdaterConfig,
     PipelineManagedIngestionJobParams,
     PipelineMetadataConfig,
+    PipelineStatus,
     PipelineTransformConfig,
     PipelineTransformConfig_Advanced,
     PipelineTransformConfig_Auto,
@@ -348,7 +307,6 @@ from .types import (
     TextNodeRelationshipsValue,
     TextNodeWithScore,
     TokenChunkingConfig,
-    TransformationCategoryNames,
     UsageAndPlan,
     UsageMetricResponse,
     UsageResponse,
@@ -367,24 +325,7 @@ from .types import (
 from .errors import UnprocessableEntityError
 from .resources import (
     DataSinkUpdateComponent,
-    DataSinkUpdateComponent_AzureAiSearch,
-    DataSinkUpdateComponent_Milvus,
-    DataSinkUpdateComponent_MongodbAtlas,
-    DataSinkUpdateComponent_Pinecone,
-    DataSinkUpdateComponent_Postgres,
-    DataSinkUpdateComponent_Qdrant,
     DataSourceUpdateComponent,
-    DataSourceUpdateComponentOne,
-    DataSourceUpdateComponentOne_AzureStorageBlob,
-    DataSourceUpdateComponentOne_Box,
-    DataSourceUpdateComponentOne_Confluence,
-    DataSourceUpdateComponentOne_GoogleDrive,
-    DataSourceUpdateComponentOne_Jira,
-    DataSourceUpdateComponentOne_MicrosoftOnedrive,
-    DataSourceUpdateComponentOne_MicrosoftSharepoint,
-    DataSourceUpdateComponentOne_NotionPage,
-    DataSourceUpdateComponentOne_S3,
-    DataSourceUpdateComponentOne_Slack,
     DataSourceUpdateCustomMetadataValue,
     EmbeddingModelConfigCreateEmbeddingConfig,
     EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding,
@@ -416,9 +357,9 @@ from .resources import (
     PipelineUpdateEmbeddingConfig_VertexaiEmbedding,
     PipelineUpdateTransformConfig,
     UpdateReportPlanApiV1ReportsReportIdPlanPatchRequestAction,
+    admin,
     beta,
     chat_apps,
-    component_definitions,
     data_sinks,
     data_sources,
     embedding_model_configs,
@@ -495,77 +436,24 @@ __all__ = [
     "CompositeRetrievedTextNodeWithScore",
     "ConfigurableDataSinkNames",
     "ConfigurableDataSourceNames",
-    "ConfigurableTransformationDefinition",
-    "ConfigurableTransformationNames",
     "CreditType",
     "DataSink",
     "DataSinkComponent",
-    "DataSinkComponent_AzureAiSearch",
-    "DataSinkComponent_Milvus",
-    "DataSinkComponent_MongodbAtlas",
-    "DataSinkComponent_Pinecone",
-    "DataSinkComponent_Postgres",
-    "DataSinkComponent_Qdrant",
     "DataSinkCreate",
     "DataSinkCreateComponent",
-    "DataSinkCreateComponent_AzureAiSearch",
-    "DataSinkCreateComponent_Milvus",
-    "DataSinkCreateComponent_MongodbAtlas",
-    "DataSinkCreateComponent_Pinecone",
-    "DataSinkCreateComponent_Postgres",
-    "DataSinkCreateComponent_Qdrant",
-    "DataSinkDefinition",
     "DataSinkUpdateComponent",
-    "DataSinkUpdateComponent_AzureAiSearch",
-    "DataSinkUpdateComponent_Milvus",
-    "DataSinkUpdateComponent_MongodbAtlas",
-    "DataSinkUpdateComponent_Pinecone",
-    "DataSinkUpdateComponent_Postgres",
-    "DataSinkUpdateComponent_Qdrant",
     "DataSource",
     "DataSourceComponent",
-    "DataSourceComponentOne",
-    "DataSourceComponentOne_AzureStorageBlob",
-    "DataSourceComponentOne_Box",
-    "DataSourceComponentOne_Confluence",
-    "DataSourceComponentOne_GoogleDrive",
-    "DataSourceComponentOne_Jira",
-    "DataSourceComponentOne_MicrosoftOnedrive",
-    "DataSourceComponentOne_MicrosoftSharepoint",
-    "DataSourceComponentOne_NotionPage",
-    "DataSourceComponentOne_S3",
-    "DataSourceComponentOne_Slack",
     "DataSourceCreate",
     "DataSourceCreateComponent",
-    "DataSourceCreateComponentOne",
-    "DataSourceCreateComponentOne_AzureStorageBlob",
-    "DataSourceCreateComponentOne_Box",
-    "DataSourceCreateComponentOne_Confluence",
-    "DataSourceCreateComponentOne_GoogleDrive",
-    "DataSourceCreateComponentOne_Jira",
-    "DataSourceCreateComponentOne_MicrosoftOnedrive",
-    "DataSourceCreateComponentOne_MicrosoftSharepoint",
-    "DataSourceCreateComponentOne_NotionPage",
-    "DataSourceCreateComponentOne_S3",
-    "DataSourceCreateComponentOne_Slack",
     "DataSourceCreateCustomMetadataValue",
     "DataSourceCustomMetadataValue",
-    "DataSourceDefinition",
     "DataSourceUpdateComponent",
-    "DataSourceUpdateComponentOne",
-    "DataSourceUpdateComponentOne_AzureStorageBlob",
-    "DataSourceUpdateComponentOne_Box",
-    "DataSourceUpdateComponentOne_Confluence",
-    "DataSourceUpdateComponentOne_GoogleDrive",
-    "DataSourceUpdateComponentOne_Jira",
-    "DataSourceUpdateComponentOne_MicrosoftOnedrive",
-    "DataSourceUpdateComponentOne_MicrosoftSharepoint",
-    "DataSourceUpdateComponentOne_NotionPage",
-    "DataSourceUpdateComponentOne_S3",
-    "DataSourceUpdateComponentOne_Slack",
     "DataSourceUpdateCustomMetadataValue",
     "DataSourceUpdateDispatcherConfig",
     "DeleteParams",
+    "DocumentBlock",
+    "DocumentChunkMode",
     "DocumentIngestionJobParams",
     "EditSuggestion",
     "EditSuggestionBlocksItem",
@@ -611,6 +499,7 @@ __all__ = [
     "ExtractJobCreateDataSchemaOverride",
     "ExtractJobCreateDataSchemaOverrideZeroValue",
     "ExtractMode",
+    "ExtractModels",
     "ExtractResultset",
     "ExtractResultsetData",
     "ExtractResultsetDataItemValue",
@@ -670,10 +559,14 @@ __all__ = [
     "LlamaIndexCoreBaseLlmsTypesChatMessage",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Audio",
+    "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Document",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Image",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Text",
     "LlamaParseParameters",
     "LlamaParseSupportedFileExtensions",
+    "LlmConfigResult",
+    "LlmConfigResultLlmType",
+    "LlmConfigsResponse",
     "LlmModelData",
     "LlmParameters",
     "LoadFilesJobConfig",
@@ -731,17 +624,6 @@ __all__ = [
     "PipelineCreateTransformConfig",
     "PipelineDataSource",
     "PipelineDataSourceComponent",
-    "PipelineDataSourceComponentOne",
-    "PipelineDataSourceComponentOne_AzureStorageBlob",
-    "PipelineDataSourceComponentOne_Box",
-    "PipelineDataSourceComponentOne_Confluence",
-    "PipelineDataSourceComponentOne_GoogleDrive",
-    "PipelineDataSourceComponentOne_Jira",
-    "PipelineDataSourceComponentOne_MicrosoftOnedrive",
-    "PipelineDataSourceComponentOne_MicrosoftSharepoint",
-    "PipelineDataSourceComponentOne_NotionPage",
-    "PipelineDataSourceComponentOne_S3",
-    "PipelineDataSourceComponentOne_Slack",
     "PipelineDataSourceCreate",
     "PipelineDataSourceCustomMetadataValue",
     "PipelineDataSourceStatus",
@@ -767,6 +649,7 @@ __all__ = [
     "PipelineFileUpdaterConfig",
     "PipelineManagedIngestionJobParams",
     "PipelineMetadataConfig",
+    "PipelineStatus",
     "PipelineTransformConfig",
     "PipelineTransformConfig_Advanced",
     "PipelineTransformConfig_Auto",
@@ -833,7 +716,6 @@ __all__ = [
     "TextNodeRelationshipsValue",
     "TextNodeWithScore",
     "TokenChunkingConfig",
-    "TransformationCategoryNames",
     "UnprocessableEntityError",
     "UpdateReportPlanApiV1ReportsReportIdPlanPatchRequestAction",
     "UsageAndPlan",
@@ -850,9 +732,9 @@ __all__ = [
     "VertexAiEmbeddingConfig",
     "VertexEmbeddingMode",
     "VertexTextEmbedding",
+    "admin",
     "beta",
     "chat_apps",
-    "component_definitions",
     "data_sinks",
     "data_sources",
     "embedding_model_configs",

llama_cloud/client.py

@@ -6,9 +6,9 @@ import httpx
 
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import LlamaCloudEnvironment
+from .resources.admin.client import AdminClient, AsyncAdminClient
 from .resources.beta.client import AsyncBetaClient, BetaClient
 from .resources.chat_apps.client import AsyncChatAppsClient, ChatAppsClient
-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
@@ -50,8 +50,8 @@ class LlamaCloud:
         self.jobs = JobsClient(client_wrapper=self._client_wrapper)
         self.evals = EvalsClient(client_wrapper=self._client_wrapper)
         self.parsing = ParsingClient(client_wrapper=self._client_wrapper)
-        self.component_definitions = ComponentDefinitionsClient(client_wrapper=self._client_wrapper)
         self.chat_apps = ChatAppsClient(client_wrapper=self._client_wrapper)
+        self.admin = AdminClient(client_wrapper=self._client_wrapper)
         self.llama_extract = LlamaExtractClient(client_wrapper=self._client_wrapper)
         self.reports = ReportsClient(client_wrapper=self._client_wrapper)
         self.beta = BetaClient(client_wrapper=self._client_wrapper)
@@ -83,8 +83,8 @@ class AsyncLlamaCloud:
         self.jobs = AsyncJobsClient(client_wrapper=self._client_wrapper)
         self.evals = AsyncEvalsClient(client_wrapper=self._client_wrapper)
         self.parsing = AsyncParsingClient(client_wrapper=self._client_wrapper)
-        self.component_definitions = AsyncComponentDefinitionsClient(client_wrapper=self._client_wrapper)
         self.chat_apps = AsyncChatAppsClient(client_wrapper=self._client_wrapper)
+        self.admin = AsyncAdminClient(client_wrapper=self._client_wrapper)
         self.llama_extract = AsyncLlamaExtractClient(client_wrapper=self._client_wrapper)
         self.reports = AsyncReportsClient(client_wrapper=self._client_wrapper)
         self.beta = AsyncBetaClient(client_wrapper=self._client_wrapper)

llama_cloud/resources/__init__.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from . import (
+    admin,
     beta,
     chat_apps,
-    component_definitions,
     data_sinks,
     data_sources,
     embedding_model_configs,
@@ -18,30 +18,8 @@ from . import (
     reports,
     retrievers,
 )
-from .data_sinks import (
-    DataSinkUpdateComponent,
-    DataSinkUpdateComponent_AzureAiSearch,
-    DataSinkUpdateComponent_Milvus,
-    DataSinkUpdateComponent_MongodbAtlas,
-    DataSinkUpdateComponent_Pinecone,
-    DataSinkUpdateComponent_Postgres,
-    DataSinkUpdateComponent_Qdrant,
-)
-from .data_sources import (
-    DataSourceUpdateComponent,
-    DataSourceUpdateComponentOne,
-    DataSourceUpdateComponentOne_AzureStorageBlob,
-    DataSourceUpdateComponentOne_Box,
-    DataSourceUpdateComponentOne_Confluence,
-    DataSourceUpdateComponentOne_GoogleDrive,
-    DataSourceUpdateComponentOne_Jira,
-    DataSourceUpdateComponentOne_MicrosoftOnedrive,
-    DataSourceUpdateComponentOne_MicrosoftSharepoint,
-    DataSourceUpdateComponentOne_NotionPage,
-    DataSourceUpdateComponentOne_S3,
-    DataSourceUpdateComponentOne_Slack,
-    DataSourceUpdateCustomMetadataValue,
-)
+from .data_sinks import DataSinkUpdateComponent
+from .data_sources import DataSourceUpdateComponent, DataSourceUpdateCustomMetadataValue
 from .embedding_model_configs import (
     EmbeddingModelConfigCreateEmbeddingConfig,
     EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding,
@@ -79,24 +57,7 @@ from .reports import UpdateReportPlanApiV1ReportsReportIdPlanPatchRequestAction
 
 __all__ = [
     "DataSinkUpdateComponent",
-    "DataSinkUpdateComponent_AzureAiSearch",
-    "DataSinkUpdateComponent_Milvus",
-    "DataSinkUpdateComponent_MongodbAtlas",
-    "DataSinkUpdateComponent_Pinecone",
-    "DataSinkUpdateComponent_Postgres",
-    "DataSinkUpdateComponent_Qdrant",
     "DataSourceUpdateComponent",
-    "DataSourceUpdateComponentOne",
-    "DataSourceUpdateComponentOne_AzureStorageBlob",
-    "DataSourceUpdateComponentOne_Box",
-    "DataSourceUpdateComponentOne_Confluence",
-    "DataSourceUpdateComponentOne_GoogleDrive",
-    "DataSourceUpdateComponentOne_Jira",
-    "DataSourceUpdateComponentOne_MicrosoftOnedrive",
-    "DataSourceUpdateComponentOne_MicrosoftSharepoint",
-    "DataSourceUpdateComponentOne_NotionPage",
-    "DataSourceUpdateComponentOne_S3",
-    "DataSourceUpdateComponentOne_Slack",
     "DataSourceUpdateCustomMetadataValue",
     "EmbeddingModelConfigCreateEmbeddingConfig",
     "EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding",
@@ -128,9 +89,9 @@ __all__ = [
     "PipelineUpdateEmbeddingConfig_VertexaiEmbedding",
     "PipelineUpdateTransformConfig",
     "UpdateReportPlanApiV1ReportsReportIdPlanPatchRequestAction",
+    "admin",
     "beta",
     "chat_apps",
-    "component_definitions",
     "data_sinks",
     "data_sources",
     "embedding_model_configs",

llama_cloud/resources/admin/client.py

@@ -0,0 +1,78 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import urllib.parse
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.http_validation_error import HttpValidationError
+from ...types.llm_configs_response import LlmConfigsResponse
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class AdminClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def get_llm_configs(self) -> LlmConfigsResponse:
+        """
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.admin.get_llm_configs()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/admin/llm-configs"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(LlmConfigsResponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncAdminClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def get_llm_configs(self) -> LlmConfigsResponse:
+        """
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.admin.get_llm_configs()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/admin/llm-configs"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(LlmConfigsResponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

llama_cloud/resources/data_sinks/__init__.py

@@ -1,21 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import (
-    DataSinkUpdateComponent,
-    DataSinkUpdateComponent_AzureAiSearch,
-    DataSinkUpdateComponent_Milvus,
-    DataSinkUpdateComponent_MongodbAtlas,
-    DataSinkUpdateComponent_Pinecone,
-    DataSinkUpdateComponent_Postgres,
-    DataSinkUpdateComponent_Qdrant,
-)
+from .types import DataSinkUpdateComponent
 
-__all__ = [
-    "DataSinkUpdateComponent",
-    "DataSinkUpdateComponent_AzureAiSearch",
-    "DataSinkUpdateComponent_Milvus",
-    "DataSinkUpdateComponent_MongodbAtlas",
-    "DataSinkUpdateComponent_Pinecone",
-    "DataSinkUpdateComponent_Postgres",
-    "DataSinkUpdateComponent_Qdrant",
-]
+__all__ = ["DataSinkUpdateComponent"]

llama_cloud/resources/data_sinks/client.py

@@ -82,6 +82,19 @@ class DataSinksClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSinkCreate.
+        ---
+        from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.data_sinks.create_data_sink(
+            request=DataSinkCreate(
+                name="string",
+                sink_type=ConfigurableDataSinkNames.PINECONE,
+            ),
+        )
         """
         _response = self._client_wrapper.httpx_client.request(
             "POST",
@@ -118,6 +131,19 @@ class DataSinksClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSinkCreate.
+        ---
+        from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.data_sinks.upsert_data_sink(
+            request=DataSinkCreate(
+                name="string",
+                sink_type=ConfigurableDataSinkNames.PINECONE,
+            ),
+        )
         """
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
@@ -143,6 +169,15 @@ class DataSinksClient:
 
         Parameters:
             - data_sink_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.data_sinks.get_data_sink(
+            data_sink_id="string",
+        )
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
@@ -178,7 +213,18 @@ class DataSinksClient:
 
             - sink_type: ConfigurableDataSinkNames.
 
-            - component: typing.Optional[DataSinkUpdateComponent].
+            - component: typing.Optional[DataSinkUpdateComponent]. Component that implements the data sink
+        ---
+        from llama_cloud import ConfigurableDataSinkNames
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.data_sinks.update_data_sink(
+            data_sink_id="string",
+            sink_type=ConfigurableDataSinkNames.PINECONE,
+        )
         """
         _request: typing.Dict[str, typing.Any] = {"sink_type": sink_type}
         if name is not OMIT:
@@ -290,6 +336,19 @@ class AsyncDataSinksClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSinkCreate.
+        ---
+        from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.data_sinks.create_data_sink(
+            request=DataSinkCreate(
+                name="string",
+                sink_type=ConfigurableDataSinkNames.PINECONE,
+            ),
+        )
         """
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
@@ -326,6 +385,19 @@ class AsyncDataSinksClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSinkCreate.
+        ---
+        from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.data_sinks.upsert_data_sink(
+            request=DataSinkCreate(
+                name="string",
+                sink_type=ConfigurableDataSinkNames.PINECONE,
+            ),
+        )
         """
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
@@ -351,6 +423,15 @@ class AsyncDataSinksClient:
 
         Parameters:
             - data_sink_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.data_sinks.get_data_sink(
+            data_sink_id="string",
+        )
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
@@ -386,7 +467,18 @@ class AsyncDataSinksClient:
 
             - sink_type: ConfigurableDataSinkNames.
 
-            - component: typing.Optional[DataSinkUpdateComponent].
+            - component: typing.Optional[DataSinkUpdateComponent]. Component that implements the data sink
+        ---
+        from llama_cloud import ConfigurableDataSinkNames
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.data_sinks.update_data_sink(
+            data_sink_id="string",
+            sink_type=ConfigurableDataSinkNames.PINECONE,
+        )
         """
         _request: typing.Dict[str, typing.Any] = {"sink_type": sink_type}
         if name is not OMIT:

llama_cloud/resources/data_sinks/types/__init__.py

@@ -1,21 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .data_sink_update_component import (
-    DataSinkUpdateComponent,
-    DataSinkUpdateComponent_AzureAiSearch,
-    DataSinkUpdateComponent_Milvus,
-    DataSinkUpdateComponent_MongodbAtlas,
-    DataSinkUpdateComponent_Pinecone,
-    DataSinkUpdateComponent_Postgres,
-    DataSinkUpdateComponent_Qdrant,
-)
+from .data_sink_update_component import DataSinkUpdateComponent
 
-__all__ = [
-    "DataSinkUpdateComponent",
-    "DataSinkUpdateComponent_AzureAiSearch",
-    "DataSinkUpdateComponent_Milvus",
-    "DataSinkUpdateComponent_MongodbAtlas",
-    "DataSinkUpdateComponent_Pinecone",
-    "DataSinkUpdateComponent_Postgres",
-    "DataSinkUpdateComponent_Qdrant",
-]
+__all__ = ["DataSinkUpdateComponent"]