letta-client 1.0.0a6__py3-none-any.whl → 1.0.0a8__py3-none-any.whl

letta_sdk/resources/providers.py → letta_client/resources/folders/folders.py

@@ -2,90 +2,109 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing import Dict, Optional
 from typing_extensions import Literal
 
 import httpx
 
-from ..types import (
-    ProviderType,
-    provider_list_params,
-    provider_check_params,
-    provider_create_params,
-    provider_update_params,
+from .files import (
+    FilesResource,
+    AsyncFilesResource,
+    FilesResourceWithRawResponse,
+    AsyncFilesResourceWithRawResponse,
+    FilesResourceWithStreamingResponse,
+    AsyncFilesResourceWithStreamingResponse,
 )
-from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from .._utils import maybe_transform, async_maybe_transform
-from .._compat import cached_property
-from .._resource import SyncAPIResource, AsyncAPIResource
-from .._response import (
+from .agents import (
+    AgentsResource,
+    AsyncAgentsResource,
+    AgentsResourceWithRawResponse,
+    AsyncAgentsResourceWithRawResponse,
+    AgentsResourceWithStreamingResponse,
+    AsyncAgentsResourceWithStreamingResponse,
+)
+from ...types import folder_list_params, folder_create_params, folder_update_params
+from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
+from ..._utils import maybe_transform, async_maybe_transform
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+from ..._response import (
     to_raw_response_wrapper,
     to_streamed_response_wrapper,
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from .._base_client import make_request_options
-from ..types.provider import Provider
-from ..types.provider_type import ProviderType
-from ..types.provider_list_response import ProviderListResponse
+from ...pagination import SyncArrayPage, AsyncArrayPage
+from ..._base_client import AsyncPaginator, make_request_options
+from ...types.folder import Folder
+from ...types.folder_count_response import FolderCountResponse
+from ...types.embedding_config_param import EmbeddingConfigParam
+
+__all__ = ["FoldersResource", "AsyncFoldersResource"]
+
 
-__all__ = ["ProvidersResource", "AsyncProvidersResource"]
+class FoldersResource(SyncAPIResource):
+    @cached_property
+    def files(self) -> FilesResource:
+        return FilesResource(self._client)
 
+    @cached_property
+    def agents(self) -> AgentsResource:
+        return AgentsResource(self._client)
 
-class ProvidersResource(SyncAPIResource):
     @cached_property
-    def with_raw_response(self) -> ProvidersResourceWithRawResponse:
+    def with_raw_response(self) -> FoldersResourceWithRawResponse:
         """
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
         For more information, see https://www.github.com/letta-ai/letta-python#accessing-raw-response-data-eg-headers
         """
-        return ProvidersResourceWithRawResponse(self)
+        return FoldersResourceWithRawResponse(self)
 
     @cached_property
-    def with_streaming_response(self) -> ProvidersResourceWithStreamingResponse:
+    def with_streaming_response(self) -> FoldersResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
         For more information, see https://www.github.com/letta-ai/letta-python#with_streaming_response
         """
-        return ProvidersResourceWithStreamingResponse(self)
+        return FoldersResourceWithStreamingResponse(self)
 
     def create(
         self,
         *,
-        api_key: str,
         name: str,
-        provider_type: ProviderType,
-        access_key: Optional[str] | Omit = omit,
-        api_version: Optional[str] | Omit = omit,
-        base_url: Optional[str] | Omit = omit,
-        region: Optional[str] | Omit = omit,
+        description: Optional[str] | Omit = omit,
+        embedding: Optional[str] | Omit = omit,
+        embedding_chunk_size: Optional[int] | Omit = omit,
+        embedding_config: Optional[EmbeddingConfigParam] | Omit = omit,
+        instructions: Optional[str] | Omit = omit,
+        metadata: Optional[Dict[str, object]] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Provider:
+    ) -> Folder:
         """
-        Create a new custom provider.
+        Create a new data folder.
 
         Args:
-          api_key: API key or secret key used for requests to the provider.
+          name: The name of the source.
 
-          name: The name of the provider.
+          description: The description of the source.
 
-          provider_type: The type of the provider.
+          embedding: The handle for the embedding config used by the source.
 
-          access_key: Access key used for requests to the provider.
+          embedding_chunk_size: The chunk size of the embedding.
 
-          api_version: API version used for requests to the provider.
+          embedding_config: Configuration for embedding model connection and processing parameters.
 
-          base_url: Base URL used for requests to the provider.
+          instructions: Instructions for how to use the source.
 
-          region: Region used for requests to the provider.
+          metadata: Metadata associated with the source.
 
           extra_headers: Send extra headers
 
@@ -96,28 +115,28 @@ class ProvidersResource(SyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return self._post(
-            "/v1/providers/",
+            "/v1/folders/",
             body=maybe_transform(
                 {
-                    "api_key": api_key,
                     "name": name,
-                    "provider_type": provider_type,
-                    "access_key": access_key,
-                    "api_version": api_version,
-                    "base_url": base_url,
-                    "region": region,
+                    "description": description,
+                    "embedding": embedding,
+                    "embedding_chunk_size": embedding_chunk_size,
+                    "embedding_config": embedding_config,
+                    "instructions": instructions,
+                    "metadata": metadata,
                 },
-                provider_create_params.ProviderCreateParams,
+                folder_create_params.FolderCreateParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=Provider,
+            cast_to=Folder,
         )
 
     def retrieve(
         self,
-        provider_id: str,
+        folder_id: str,
         *,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -125,11 +144,13 @@ class ProvidersResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Provider:
+    ) -> Folder:
         """
-        Get a provider by ID.
+        Get a folder by ID
 
         Args:
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -138,45 +159,47 @@ class ProvidersResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        if not provider_id:
-            raise ValueError(f"Expected a non-empty value for `provider_id` but received {provider_id!r}")
+        if not folder_id:
+            raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
         return self._get(
-            f"/v1/providers/{provider_id}",
+            f"/v1/folders/{folder_id}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=Provider,
+            cast_to=Folder,
         )
 
     def update(
         self,
-        provider_id: str,
+        folder_id: str,
         *,
-        api_key: str,
-        access_key: Optional[str] | Omit = omit,
-        api_version: Optional[str] | Omit = omit,
-        base_url: Optional[str] | Omit = omit,
-        region: Optional[str] | Omit = omit,
+        description: Optional[str] | Omit = omit,
+        embedding_config: Optional[EmbeddingConfigParam] | Omit = omit,
+        instructions: Optional[str] | Omit = omit,
+        metadata: Optional[Dict[str, object]] | Omit = omit,
+        name: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Provider:
+    ) -> Folder:
         """
-        Update an existing custom provider.
+        Update the name or documentation of an existing data folder.
 
         Args:
-          api_key: API key or secret key used for requests to the provider.
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
+          description: The description of the source.
 
-          access_key: Access key used for requests to the provider.
+          embedding_config: Configuration for embedding model connection and processing parameters.
 
-          api_version: API version used for requests to the provider.
+          instructions: Instructions for how to use the source.
 
-          base_url: Base URL used for requests to the provider.
+          metadata: Metadata associated with the source.
 
-          region: Region used for requests to the provider.
+          name: The name of the source.
 
           extra_headers: Send extra headers
 
@@ -186,24 +209,24 @@ class ProvidersResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        if not provider_id:
-            raise ValueError(f"Expected a non-empty value for `provider_id` but received {provider_id!r}")
+        if not folder_id:
+            raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
         return self._patch(
-            f"/v1/providers/{provider_id}",
+            f"/v1/folders/{folder_id}",
             body=maybe_transform(
                 {
-                    "api_key": api_key,
-                    "access_key": access_key,
-                    "api_version": api_version,
-                    "base_url": base_url,
-                    "region": region,
+                    "description": description,
+                    "embedding_config": embedding_config,
+                    "instructions": instructions,
+                    "metadata": metadata,
+                    "name": name,
                 },
-                provider_update_params.ProviderUpdateParams,
+                folder_update_params.FolderUpdateParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=Provider,
+            cast_to=Folder,
         )
 
     def list(
@@ -215,35 +238,32 @@ class ProvidersResource(SyncAPIResource):
         name: Optional[str] | Omit = omit,
         order: Literal["asc", "desc"] | Omit = omit,
         order_by: Literal["created_at"] | Omit = omit,
-        provider_type: Optional[ProviderType] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> ProviderListResponse:
+    ) -> SyncArrayPage[Folder]:
         """
-        Get a list of all custom providers.
+        List all data folders created by a user.
 
         Args:
-          after: Provider ID cursor for pagination. Returns providers that come after this
-              provider ID in the specified sort order
+          after: Folder ID cursor for pagination. Returns folders that come after this folder ID
+              in the specified sort order
 
-          before: Provider ID cursor for pagination. Returns providers that come before this
-              provider ID in the specified sort order
+          before: Folder ID cursor for pagination. Returns folders that come before this folder ID
+              in the specified sort order
 
-          limit: Maximum number of providers to return
+          limit: Maximum number of folders to return
 
-          name: Filter providers by name
+          name: Folder name to filter by
 
-          order: Sort order for providers by creation time. 'asc' for oldest first, 'desc' for
+          order: Sort order for folders by creation time. 'asc' for oldest first, 'desc' for
               newest first
 
           order_by: Field to sort by
 
-          provider_type: Filter providers by type
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -252,8 +272,9 @@ class ProvidersResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get(
-            "/v1/providers/",
+        return self._get_api_list(
+            "/v1/folders/",
+            page=SyncArrayPage[Folder],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -267,17 +288,16 @@ class ProvidersResource(SyncAPIResource):
                         "name": name,
                         "order": order,
                         "order_by": order_by,
-                        "provider_type": provider_type,
                     },
-                    provider_list_params.ProviderListParams,
+                    folder_list_params.FolderListParams,
                 ),
             ),
-            cast_to=ProviderListResponse,
+            model=Folder,
         )
 
     def delete(
         self,
-        provider_id: str,
+        folder_id: str,
         *,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -287,9 +307,11 @@ class ProvidersResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> object:
         """
-        Delete an existing custom provider.
+        Delete a data folder.
 
         Args:
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -298,130 +320,98 @@ class ProvidersResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        if not provider_id:
-            raise ValueError(f"Expected a non-empty value for `provider_id` but received {provider_id!r}")
+        if not folder_id:
+            raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
         return self._delete(
-            f"/v1/providers/{provider_id}",
+            f"/v1/folders/{folder_id}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
             cast_to=object,
         )
 
-    def check(
+    def count(
         self,
         *,
-        api_key: str,
-        provider_type: ProviderType,
-        access_key: Optional[str] | Omit = omit,
-        api_version: Optional[str] | Omit = omit,
-        base_url: Optional[str] | Omit = omit,
-        region: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> object:
-        """
-        Verify the API key and additional parameters for a provider.
-
-        Args:
-          api_key: API key or secret key used for requests to the provider.
-
-          provider_type: The type of the provider.
-
-          access_key: Access key used for requests to the provider.
-
-          api_version: API version used for requests to the provider.
-
-          base_url: Base URL used for requests to the provider.
-
-          region: Region used for requests to the provider.
-
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        return self._post(
-            "/v1/providers/check",
-            body=maybe_transform(
-                {
-                    "api_key": api_key,
-                    "provider_type": provider_type,
-                    "access_key": access_key,
-                    "api_version": api_version,
-                    "base_url": base_url,
-                    "region": region,
-                },
-                provider_check_params.ProviderCheckParams,
-            ),
+    ) -> FolderCountResponse:
+        """Count all data folders created by a user."""
+        return self._get(
+            "/v1/folders/count",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=int,
         )
 
 
-class AsyncProvidersResource(AsyncAPIResource):
+class AsyncFoldersResource(AsyncAPIResource):
+    @cached_property
+    def files(self) -> AsyncFilesResource:
+        return AsyncFilesResource(self._client)
+
+    @cached_property
+    def agents(self) -> AsyncAgentsResource:
+        return AsyncAgentsResource(self._client)
+
     @cached_property
-    def with_raw_response(self) -> AsyncProvidersResourceWithRawResponse:
+    def with_raw_response(self) -> AsyncFoldersResourceWithRawResponse:
         """
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
         For more information, see https://www.github.com/letta-ai/letta-python#accessing-raw-response-data-eg-headers
         """
-        return AsyncProvidersResourceWithRawResponse(self)
+        return AsyncFoldersResourceWithRawResponse(self)
 
     @cached_property
-    def with_streaming_response(self) -> AsyncProvidersResourceWithStreamingResponse:
+    def with_streaming_response(self) -> AsyncFoldersResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
         For more information, see https://www.github.com/letta-ai/letta-python#with_streaming_response
         """
-        return AsyncProvidersResourceWithStreamingResponse(self)
+        return AsyncFoldersResourceWithStreamingResponse(self)
 
     async def create(
         self,
         *,
-        api_key: str,
         name: str,
-        provider_type: ProviderType,
-        access_key: Optional[str] | Omit = omit,
-        api_version: Optional[str] | Omit = omit,
-        base_url: Optional[str] | Omit = omit,
-        region: Optional[str] | Omit = omit,
+        description: Optional[str] | Omit = omit,
+        embedding: Optional[str] | Omit = omit,
+        embedding_chunk_size: Optional[int] | Omit = omit,
+        embedding_config: Optional[EmbeddingConfigParam] | Omit = omit,
+        instructions: Optional[str] | Omit = omit,
+        metadata: Optional[Dict[str, object]] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Provider:
+    ) -> Folder:
         """
-        Create a new custom provider.
+        Create a new data folder.
 
         Args:
-          api_key: API key or secret key used for requests to the provider.
+          name: The name of the source.
 
-          name: The name of the provider.
+          description: The description of the source.
 
-          provider_type: The type of the provider.
+          embedding: The handle for the embedding config used by the source.
 
-          access_key: Access key used for requests to the provider.
+          embedding_chunk_size: The chunk size of the embedding.
 
-          api_version: API version used for requests to the provider.
+          embedding_config: Configuration for embedding model connection and processing parameters.
 
-          base_url: Base URL used for requests to the provider.
+          instructions: Instructions for how to use the source.
 
-          region: Region used for requests to the provider.
+          metadata: Metadata associated with the source.
 
           extra_headers: Send extra headers
 
@@ -432,28 +422,28 @@ class AsyncProvidersResource(AsyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return await self._post(
-            "/v1/providers/",
+            "/v1/folders/",
             body=await async_maybe_transform(
                 {
-                    "api_key": api_key,
                     "name": name,
-                    "provider_type": provider_type,
-                    "access_key": access_key,
-                    "api_version": api_version,
-                    "base_url": base_url,
-                    "region": region,
+                    "description": description,
+                    "embedding": embedding,
+                    "embedding_chunk_size": embedding_chunk_size,
+                    "embedding_config": embedding_config,
+                    "instructions": instructions,
+                    "metadata": metadata,
                 },
-                provider_create_params.ProviderCreateParams,
+                folder_create_params.FolderCreateParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=Provider,
+            cast_to=Folder,
         )
 
     async def retrieve(
         self,
-        provider_id: str,
+        folder_id: str,
         *,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -461,11 +451,13 @@ class AsyncProvidersResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Provider:
+    ) -> Folder:
         """
-        Get a provider by ID.
+        Get a folder by ID
 
         Args:
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -474,45 +466,47 @@ class AsyncProvidersResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        if not provider_id:
-            raise ValueError(f"Expected a non-empty value for `provider_id` but received {provider_id!r}")
+        if not folder_id:
+            raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
         return await self._get(
-            f"/v1/providers/{provider_id}",
+            f"/v1/folders/{folder_id}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=Provider,
+            cast_to=Folder,
         )
 
     async def update(
         self,
-        provider_id: str,
+        folder_id: str,
         *,
-        api_key: str,
-        access_key: Optional[str] | Omit = omit,
-        api_version: Optional[str] | Omit = omit,
-        base_url: Optional[str] | Omit = omit,
-        region: Optional[str] | Omit = omit,
+        description: Optional[str] | Omit = omit,
+        embedding_config: Optional[EmbeddingConfigParam] | Omit = omit,
+        instructions: Optional[str] | Omit = omit,
+        metadata: Optional[Dict[str, object]] | Omit = omit,
+        name: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Provider:
+    ) -> Folder:
         """
-        Update an existing custom provider.
+        Update the name or documentation of an existing data folder.
 
         Args:
-          api_key: API key or secret key used for requests to the provider.
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
+          description: The description of the source.
 
-          access_key: Access key used for requests to the provider.
+          embedding_config: Configuration for embedding model connection and processing parameters.
 
-          api_version: API version used for requests to the provider.
+          instructions: Instructions for how to use the source.
 
-          base_url: Base URL used for requests to the provider.
+          metadata: Metadata associated with the source.
 
-          region: Region used for requests to the provider.
+          name: The name of the source.
 
           extra_headers: Send extra headers
 
@@ -522,27 +516,27 @@ class AsyncProvidersResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        if not provider_id:
-            raise ValueError(f"Expected a non-empty value for `provider_id` but received {provider_id!r}")
+        if not folder_id:
+            raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
         return await self._patch(
-            f"/v1/providers/{provider_id}",
+            f"/v1/folders/{folder_id}",
             body=await async_maybe_transform(
                 {
-                    "api_key": api_key,
-                    "access_key": access_key,
-                    "api_version": api_version,
-                    "base_url": base_url,
-                    "region": region,
+                    "description": description,
+                    "embedding_config": embedding_config,
+                    "instructions": instructions,
+                    "metadata": metadata,
+                    "name": name,
                 },
-                provider_update_params.ProviderUpdateParams,
+                folder_update_params.FolderUpdateParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=Provider,
+            cast_to=Folder,
         )
 
-    async def list(
+    def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -551,35 +545,32 @@ class AsyncProvidersResource(AsyncAPIResource):
         name: Optional[str] | Omit = omit,
         order: Literal["asc", "desc"] | Omit = omit,
         order_by: Literal["created_at"] | Omit = omit,
-        provider_type: Optional[ProviderType] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> ProviderListResponse:
+    ) -> AsyncPaginator[Folder, AsyncArrayPage[Folder]]:
         """
-        Get a list of all custom providers.
+        List all data folders created by a user.
 
         Args:
-          after: Provider ID cursor for pagination. Returns providers that come after this
-              provider ID in the specified sort order
+          after: Folder ID cursor for pagination. Returns folders that come after this folder ID
+              in the specified sort order
 
-          before: Provider ID cursor for pagination. Returns providers that come before this
-              provider ID in the specified sort order
+          before: Folder ID cursor for pagination. Returns folders that come before this folder ID
+              in the specified sort order
 
-          limit: Maximum number of providers to return
+          limit: Maximum number of folders to return
 
-          name: Filter providers by name
+          name: Folder name to filter by
 
-          order: Sort order for providers by creation time. 'asc' for oldest first, 'desc' for
+          order: Sort order for folders by creation time. 'asc' for oldest first, 'desc' for
               newest first
 
           order_by: Field to sort by
 
-          provider_type: Filter providers by type
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -588,14 +579,15 @@ class AsyncProvidersResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._get(
-            "/v1/providers/",
+        return self._get_api_list(
+            "/v1/folders/",
+            page=AsyncArrayPage[Folder],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=await async_maybe_transform(
+                query=maybe_transform(
                     {
                         "after": after,
                         "before": before,
@@ -603,17 +595,16 @@ class AsyncProvidersResource(AsyncAPIResource):
                         "name": name,
                         "order": order,
                         "order_by": order_by,
-                        "provider_type": provider_type,
                     },
-                    provider_list_params.ProviderListParams,
+                    folder_list_params.FolderListParams,
                 ),
             ),
-            cast_to=ProviderListResponse,
+            model=Folder,
         )
 
     async def delete(
         self,
-        provider_id: str,
+        folder_id: str,
         *,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -623,9 +614,11 @@ class AsyncProvidersResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> object:
         """
-        Delete an existing custom provider.
+        Delete a data folder.
 
         Args:
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -634,167 +627,159 @@ class AsyncProvidersResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        if not provider_id:
-            raise ValueError(f"Expected a non-empty value for `provider_id` but received {provider_id!r}")
+        if not folder_id:
+            raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
         return await self._delete(
-            f"/v1/providers/{provider_id}",
+            f"/v1/folders/{folder_id}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
             cast_to=object,
         )
 
-    async def check(
+    async def count(
         self,
         *,
-        api_key: str,
-        provider_type: ProviderType,
-        access_key: Optional[str] | Omit = omit,
-        api_version: Optional[str] | Omit = omit,
-        base_url: Optional[str] | Omit = omit,
-        region: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> object:
-        """
-        Verify the API key and additional parameters for a provider.
-
-        Args:
-          api_key: API key or secret key used for requests to the provider.
-
-          provider_type: The type of the provider.
-
-          access_key: Access key used for requests to the provider.
-
-          api_version: API version used for requests to the provider.
-
-          base_url: Base URL used for requests to the provider.
-
-          region: Region used for requests to the provider.
-
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        return await self._post(
-            "/v1/providers/check",
-            body=await async_maybe_transform(
-                {
-                    "api_key": api_key,
-                    "provider_type": provider_type,
-                    "access_key": access_key,
-                    "api_version": api_version,
-                    "base_url": base_url,
-                    "region": region,
-                },
-                provider_check_params.ProviderCheckParams,
-            ),
+    ) -> FolderCountResponse:
+        """Count all data folders created by a user."""
+        return await self._get(
+            "/v1/folders/count",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=int,
         )
 
 
-class ProvidersResourceWithRawResponse:
-    def __init__(self, providers: ProvidersResource) -> None:
-        self._providers = providers
+class FoldersResourceWithRawResponse:
+    def __init__(self, folders: FoldersResource) -> None:
+        self._folders = folders
 
         self.create = to_raw_response_wrapper(
-            providers.create,
+            folders.create,
         )
         self.retrieve = to_raw_response_wrapper(
-            providers.retrieve,
+            folders.retrieve,
         )
         self.update = to_raw_response_wrapper(
-            providers.update,
+            folders.update,
         )
         self.list = to_raw_response_wrapper(
-            providers.list,
+            folders.list,
         )
         self.delete = to_raw_response_wrapper(
-            providers.delete,
+            folders.delete,
         )
-        self.check = to_raw_response_wrapper(
-            providers.check,
+        self.count = to_raw_response_wrapper(
+            folders.count,
         )
 
+    @cached_property
+    def files(self) -> FilesResourceWithRawResponse:
+        return FilesResourceWithRawResponse(self._folders.files)
+
+    @cached_property
+    def agents(self) -> AgentsResourceWithRawResponse:
+        return AgentsResourceWithRawResponse(self._folders.agents)
 
-class AsyncProvidersResourceWithRawResponse:
-    def __init__(self, providers: AsyncProvidersResource) -> None:
-        self._providers = providers
+
+class AsyncFoldersResourceWithRawResponse:
+    def __init__(self, folders: AsyncFoldersResource) -> None:
+        self._folders = folders
 
         self.create = async_to_raw_response_wrapper(
-            providers.create,
+            folders.create,
         )
         self.retrieve = async_to_raw_response_wrapper(
-            providers.retrieve,
+            folders.retrieve,
         )
         self.update = async_to_raw_response_wrapper(
-            providers.update,
+            folders.update,
         )
         self.list = async_to_raw_response_wrapper(
-            providers.list,
+            folders.list,
         )
         self.delete = async_to_raw_response_wrapper(
-            providers.delete,
+            folders.delete,
         )
-        self.check = async_to_raw_response_wrapper(
-            providers.check,
+        self.count = async_to_raw_response_wrapper(
+            folders.count,
         )
 
+    @cached_property
+    def files(self) -> AsyncFilesResourceWithRawResponse:
+        return AsyncFilesResourceWithRawResponse(self._folders.files)
+
+    @cached_property
+    def agents(self) -> AsyncAgentsResourceWithRawResponse:
+        return AsyncAgentsResourceWithRawResponse(self._folders.agents)
+
 
-class ProvidersResourceWithStreamingResponse:
-    def __init__(self, providers: ProvidersResource) -> None:
-        self._providers = providers
+class FoldersResourceWithStreamingResponse:
+    def __init__(self, folders: FoldersResource) -> None:
+        self._folders = folders
 
         self.create = to_streamed_response_wrapper(
-            providers.create,
+            folders.create,
         )
         self.retrieve = to_streamed_response_wrapper(
-            providers.retrieve,
+            folders.retrieve,
         )
         self.update = to_streamed_response_wrapper(
-            providers.update,
+            folders.update,
         )
         self.list = to_streamed_response_wrapper(
-            providers.list,
+            folders.list,
         )
         self.delete = to_streamed_response_wrapper(
-            providers.delete,
+            folders.delete,
         )
-        self.check = to_streamed_response_wrapper(
-            providers.check,
+        self.count = to_streamed_response_wrapper(
+            folders.count,
         )
 
+    @cached_property
+    def files(self) -> FilesResourceWithStreamingResponse:
+        return FilesResourceWithStreamingResponse(self._folders.files)
+
+    @cached_property
+    def agents(self) -> AgentsResourceWithStreamingResponse:
+        return AgentsResourceWithStreamingResponse(self._folders.agents)
 
-class AsyncProvidersResourceWithStreamingResponse:
-    def __init__(self, providers: AsyncProvidersResource) -> None:
-        self._providers = providers
+
+class AsyncFoldersResourceWithStreamingResponse:
+    def __init__(self, folders: AsyncFoldersResource) -> None:
+        self._folders = folders
 
         self.create = async_to_streamed_response_wrapper(
-            providers.create,
+            folders.create,
         )
         self.retrieve = async_to_streamed_response_wrapper(
-            providers.retrieve,
+            folders.retrieve,
         )
         self.update = async_to_streamed_response_wrapper(
-            providers.update,
+            folders.update,
         )
         self.list = async_to_streamed_response_wrapper(
-            providers.list,
+            folders.list,
         )
         self.delete = async_to_streamed_response_wrapper(
-            providers.delete,
+            folders.delete,
         )
-        self.check = async_to_streamed_response_wrapper(
-            providers.check,
+        self.count = async_to_streamed_response_wrapper(
+            folders.count,
         )
+
+    @cached_property
+    def files(self) -> AsyncFilesResourceWithStreamingResponse:
+        return AsyncFilesResourceWithStreamingResponse(self._folders.files)
+
+    @cached_property
+    def agents(self) -> AsyncAgentsResourceWithStreamingResponse:
+        return AsyncAgentsResourceWithStreamingResponse(self._folders.agents)