letta-client 1.0.0a10__py3-none-any.whl → 1.0.0a12__py3-none-any.whl

letta_client/_client.py

@@ -49,6 +49,7 @@ from .resources.batches import batches
 from .resources.folders import folders
 from .resources.templates import templates
 from .resources.identities import identities
+from .resources.mcp_servers import mcp_servers
 from .types.health_response import HealthResponse
 
 __all__ = [
@@ -64,7 +65,7 @@ __all__ = [
 ]
 
 ENVIRONMENTS: Dict[str, str] = {
-    "cloud": "https://app.letta.com",
+    "cloud": "https://api.letta.com",
     "local": "http://localhost:8283",
 }
 
@@ -83,11 +84,13 @@ class Letta(SyncAPIClient):
     tags: tags.TagsResource
     batches: batches.BatchesResource
     templates: templates.TemplatesResource
+    mcp_servers: mcp_servers.McpServersResource
     with_raw_response: LettaWithRawResponse
     with_streaming_response: LettaWithStreamedResponse
 
     # client options
     api_key: str
+    project_id: str | None
 
     _environment: Literal["cloud", "local"] | NotGiven
 
@@ -95,6 +98,7 @@ class Letta(SyncAPIClient):
         self,
         *,
         api_key: str | None = None,
+        project_id: str | None = None,
         environment: Literal["cloud", "local"] | NotGiven = not_given,
         base_url: str | httpx.URL | None | NotGiven = not_given,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -127,6 +131,8 @@ class Letta(SyncAPIClient):
             )
         self.api_key = api_key
 
+        self.project_id = project_id
+
         self._environment = environment
 
         base_url_env = os.environ.get("LETTA_BASE_URL")
@@ -164,6 +170,8 @@ class Letta(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._default_stream_cls = Stream
+
         self.archives = archives.ArchivesResource(self)
         self.tools = tools.ToolsResource(self)
         self.folders = folders.FoldersResource(self)
@@ -177,6 +185,7 @@ class Letta(SyncAPIClient):
         self.tags = tags.TagsResource(self)
         self.batches = batches.BatchesResource(self)
         self.templates = templates.TemplatesResource(self)
+        self.mcp_servers = mcp_servers.McpServersResource(self)
         self.with_raw_response = LettaWithRawResponse(self)
         self.with_streaming_response = LettaWithStreamedResponse(self)
 
@@ -197,6 +206,7 @@ class Letta(SyncAPIClient):
         return {
             **super().default_headers,
             "X-Stainless-Async": "false",
+            "X-Project": self.project_id if self.project_id is not None else Omit(),
             **self._custom_headers,
         }
 
@@ -204,6 +214,7 @@ class Letta(SyncAPIClient):
         self,
         *,
         api_key: str | None = None,
+        project_id: str | None = None,
         environment: Literal["cloud", "local"] | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -239,6 +250,7 @@ class Letta(SyncAPIClient):
         http_client = http_client or self._client
         return self.__class__(
             api_key=api_key or self.api_key,
+            project_id=project_id or self.project_id,
             base_url=base_url or self.base_url,
             environment=environment or self._environment,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
@@ -320,11 +332,13 @@ class AsyncLetta(AsyncAPIClient):
     tags: tags.AsyncTagsResource
     batches: batches.AsyncBatchesResource
     templates: templates.AsyncTemplatesResource
+    mcp_servers: mcp_servers.AsyncMcpServersResource
     with_raw_response: AsyncLettaWithRawResponse
     with_streaming_response: AsyncLettaWithStreamedResponse
 
     # client options
     api_key: str
+    project_id: str | None
 
     _environment: Literal["cloud", "local"] | NotGiven
 
@@ -332,6 +346,7 @@ class AsyncLetta(AsyncAPIClient):
         self,
         *,
         api_key: str | None = None,
+        project_id: str | None = None,
         environment: Literal["cloud", "local"] | NotGiven = not_given,
         base_url: str | httpx.URL | None | NotGiven = not_given,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -364,6 +379,8 @@ class AsyncLetta(AsyncAPIClient):
             )
         self.api_key = api_key
 
+        self.project_id = project_id
+
         self._environment = environment
 
         base_url_env = os.environ.get("LETTA_BASE_URL")
@@ -401,6 +418,8 @@ class AsyncLetta(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._default_stream_cls = AsyncStream
+
         self.archives = archives.AsyncArchivesResource(self)
         self.tools = tools.AsyncToolsResource(self)
         self.folders = folders.AsyncFoldersResource(self)
@@ -414,6 +433,7 @@ class AsyncLetta(AsyncAPIClient):
         self.tags = tags.AsyncTagsResource(self)
         self.batches = batches.AsyncBatchesResource(self)
         self.templates = templates.AsyncTemplatesResource(self)
+        self.mcp_servers = mcp_servers.AsyncMcpServersResource(self)
         self.with_raw_response = AsyncLettaWithRawResponse(self)
         self.with_streaming_response = AsyncLettaWithStreamedResponse(self)
 
@@ -434,6 +454,7 @@ class AsyncLetta(AsyncAPIClient):
         return {
             **super().default_headers,
             "X-Stainless-Async": f"async:{get_async_library()}",
+            "X-Project": self.project_id if self.project_id is not None else Omit(),
             **self._custom_headers,
         }
 
@@ -441,6 +462,7 @@ class AsyncLetta(AsyncAPIClient):
         self,
         *,
         api_key: str | None = None,
+        project_id: str | None = None,
         environment: Literal["cloud", "local"] | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -476,6 +498,7 @@ class AsyncLetta(AsyncAPIClient):
         http_client = http_client or self._client
         return self.__class__(
             api_key=api_key or self.api_key,
+            project_id=project_id or self.project_id,
             base_url=base_url or self.base_url,
             environment=environment or self._environment,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
@@ -558,6 +581,7 @@ class LettaWithRawResponse:
         self.tags = tags.TagsResourceWithRawResponse(client.tags)
         self.batches = batches.BatchesResourceWithRawResponse(client.batches)
         self.templates = templates.TemplatesResourceWithRawResponse(client.templates)
+        self.mcp_servers = mcp_servers.McpServersResourceWithRawResponse(client.mcp_servers)
 
         self.health = to_raw_response_wrapper(
             client.health,
@@ -579,6 +603,7 @@ class AsyncLettaWithRawResponse:
         self.tags = tags.AsyncTagsResourceWithRawResponse(client.tags)
         self.batches = batches.AsyncBatchesResourceWithRawResponse(client.batches)
         self.templates = templates.AsyncTemplatesResourceWithRawResponse(client.templates)
+        self.mcp_servers = mcp_servers.AsyncMcpServersResourceWithRawResponse(client.mcp_servers)
 
         self.health = async_to_raw_response_wrapper(
             client.health,
@@ -600,6 +625,7 @@ class LettaWithStreamedResponse:
         self.tags = tags.TagsResourceWithStreamingResponse(client.tags)
         self.batches = batches.BatchesResourceWithStreamingResponse(client.batches)
         self.templates = templates.TemplatesResourceWithStreamingResponse(client.templates)
+        self.mcp_servers = mcp_servers.McpServersResourceWithStreamingResponse(client.mcp_servers)
 
         self.health = to_streamed_response_wrapper(
             client.health,
@@ -621,6 +647,7 @@ class AsyncLettaWithStreamedResponse:
         self.tags = tags.AsyncTagsResourceWithStreamingResponse(client.tags)
         self.batches = batches.AsyncBatchesResourceWithStreamingResponse(client.batches)
         self.templates = templates.AsyncTemplatesResourceWithStreamingResponse(client.templates)
+        self.mcp_servers = mcp_servers.AsyncMcpServersResourceWithStreamingResponse(client.mcp_servers)
 
         self.health = async_to_streamed_response_wrapper(
             client.health,

letta_client/_streaming.py

@@ -55,11 +55,29 @@ class Stream(Generic[_T]):
         iterator = self._iter_events()
 
         for sse in iterator:
-            yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+            if sse.data.startswith("[DONE]"):
+                break
 
-        # Ensure the entire stream is consumed
-        for _sse in iterator:
-            ...
+            if sse.event == "error":
+                body = sse.data
+
+                try:
+                    body = sse.json()
+                    err_msg = f"{body}"
+                except Exception:
+                    err_msg = sse.data or f"Error code: {response.status_code}"
+
+                raise self._client._make_status_error(
+                    err_msg,
+                    body=body,
+                    response=self.response,
+                )
+
+            if sse.event is None:
+                yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+
+        # As we might not fully consume the response stream, we need to close it explicitly
+        response.close()
 
     def __enter__(self) -> Self:
         return self
@@ -119,11 +137,29 @@ class AsyncStream(Generic[_T]):
         iterator = self._iter_events()
 
         async for sse in iterator:
-            yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+            if sse.data.startswith("[DONE]"):
+                break
+
+            if sse.event == "error":
+                body = sse.data
+
+                try:
+                    body = sse.json()
+                    err_msg = f"{body}"
+                except Exception:
+                    err_msg = sse.data or f"Error code: {response.status_code}"
+
+                raise self._client._make_status_error(
+                    err_msg,
+                    body=body,
+                    response=self.response,
+                )
+
+            if sse.event is None:
+                yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 
-        # Ensure the entire stream is consumed
-        async for _sse in iterator:
-            ...
+        # As we might not fully consume the response stream, we need to close it explicitly
+        await response.aclose()
 
     async def __aenter__(self) -> Self:
         return self

letta_client/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "letta_client"
-__version__ = "1.0.0-alpha.10"  # x-release-please-version
+__version__ = "1.0.0-alpha.12"  # x-release-please-version

letta_client/resources/__init__.py

@@ -104,6 +104,14 @@ from .identities import (
     IdentitiesResourceWithStreamingResponse,
     AsyncIdentitiesResourceWithStreamingResponse,
 )
+from .mcp_servers import (
+    McpServersResource,
+    AsyncMcpServersResource,
+    McpServersResourceWithRawResponse,
+    AsyncMcpServersResourceWithRawResponse,
+    McpServersResourceWithStreamingResponse,
+    AsyncMcpServersResourceWithStreamingResponse,
+)
 
 __all__ = [
     "ArchivesResource",
@@ -184,4 +192,10 @@ __all__ = [
     "AsyncTemplatesResourceWithRawResponse",
     "TemplatesResourceWithStreamingResponse",
     "AsyncTemplatesResourceWithStreamingResponse",
+    "McpServersResource",
+    "AsyncMcpServersResource",
+    "McpServersResourceWithRawResponse",
+    "AsyncMcpServersResourceWithRawResponse",
+    "McpServersResourceWithStreamingResponse",
+    "AsyncMcpServersResourceWithStreamingResponse",
 ]

letta_client/resources/agents/agents.py

@@ -191,7 +191,6 @@ class AgentsResource(SyncAPIResource):
         tool_ids: Optional[SequenceNotStr[str]] | Omit = omit,
         tool_rules: Optional[Iterable[agent_create_params.ToolRule]] | Omit = omit,
         tools: Optional[SequenceNotStr[str]] | Omit = omit,
-        x_project: 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,
@@ -306,8 +305,6 @@ class AgentsResource(SyncAPIResource):
 
           tools: The tools used by the agent.
 
-          x_project: The project slug to associate with the agent (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -316,7 +313,6 @@ class AgentsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return self._post(
             "/v1/agents/",
             body=maybe_transform(
@@ -1026,7 +1022,6 @@ class AsyncAgentsResource(AsyncAPIResource):
         tool_ids: Optional[SequenceNotStr[str]] | Omit = omit,
         tool_rules: Optional[Iterable[agent_create_params.ToolRule]] | Omit = omit,
         tools: Optional[SequenceNotStr[str]] | Omit = omit,
-        x_project: 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,
@@ -1141,8 +1136,6 @@ class AsyncAgentsResource(AsyncAPIResource):
 
           tools: The tools used by the agent.
 
-          x_project: The project slug to associate with the agent (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -1151,7 +1144,6 @@ class AsyncAgentsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return await self._post(
             "/v1/agents/",
             body=await async_maybe_transform(

letta_client/resources/agents/blocks.py

@@ -21,9 +21,7 @@ from ...pagination import SyncArrayPage, AsyncArrayPage
 from ..._base_client import AsyncPaginator, make_request_options
 from ...types.agents import block_list_params, block_modify_params
 from ...types.agent_state import AgentState
-from ...types.agents.block_list_response import BlockListResponse
-from ...types.agents.block_modify_response import BlockModifyResponse
-from ...types.agents.block_retrieve_response import BlockRetrieveResponse
+from ...types.block_response import BlockResponse
 
 __all__ = ["BlocksResource", "AsyncBlocksResource"]
 
@@ -59,7 +57,7 @@ class BlocksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockRetrieveResponse:
+    ) -> BlockResponse:
         """
         Retrieve a core memory block from an agent.
 
@@ -83,7 +81,7 @@ class BlocksResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=BlockRetrieveResponse,
+            cast_to=BlockResponse,
         )
 
     def list(
@@ -101,7 +99,7 @@ class BlocksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[BlockListResponse]:
+    ) -> SyncArrayPage[BlockResponse]:
         """
         Retrieve the core memory blocks of a specific agent.
 
@@ -133,7 +131,7 @@ class BlocksResource(SyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
         return self._get_api_list(
             f"/v1/agents/{agent_id}/core-memory/blocks",
-            page=SyncArrayPage[BlockListResponse],
+            page=SyncArrayPage[BlockResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -150,7 +148,7 @@ class BlocksResource(SyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            model=BlockListResponse,
+            model=BlockResponse,
         )
 
     def attach(
@@ -259,7 +257,7 @@ class BlocksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockModifyResponse:
+    ) -> BlockResponse:
         """
         Updates a core memory block of an agent.
 
@@ -333,7 +331,7 @@ class BlocksResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=BlockModifyResponse,
+            cast_to=BlockResponse,
         )
 
 
@@ -368,7 +366,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockRetrieveResponse:
+    ) -> BlockResponse:
         """
         Retrieve a core memory block from an agent.
 
@@ -392,7 +390,7 @@ class AsyncBlocksResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=BlockRetrieveResponse,
+            cast_to=BlockResponse,
         )
 
     def list(
@@ -410,7 +408,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[BlockListResponse, AsyncArrayPage[BlockListResponse]]:
+    ) -> AsyncPaginator[BlockResponse, AsyncArrayPage[BlockResponse]]:
         """
         Retrieve the core memory blocks of a specific agent.
 
@@ -442,7 +440,7 @@ class AsyncBlocksResource(AsyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
         return self._get_api_list(
             f"/v1/agents/{agent_id}/core-memory/blocks",
-            page=AsyncArrayPage[BlockListResponse],
+            page=AsyncArrayPage[BlockResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -459,7 +457,7 @@ class AsyncBlocksResource(AsyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            model=BlockListResponse,
+            model=BlockResponse,
         )
 
     async def attach(
@@ -568,7 +566,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockModifyResponse:
+    ) -> BlockResponse:
         """
         Updates a core memory block of an agent.
 
@@ -642,7 +640,7 @@ class AsyncBlocksResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=BlockModifyResponse,
+            cast_to=BlockResponse,
         )
 
 

letta_client/resources/agents/files.py

@@ -8,7 +8,7 @@ from typing_extensions import Literal
 import httpx
 
 from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from ..._utils import maybe_transform
+from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -17,8 +17,7 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...pagination import SyncNextFilesPage, AsyncNextFilesPage
-from ..._base_client import AsyncPaginator, make_request_options
+from ..._base_client import make_request_options
 from ...types.agents import file_list_params
 from ...types.agents.file_list_response import FileListResponse
 from ...types.agents.file_open_response import FileOpenResponse
@@ -64,7 +63,7 @@ class FilesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncNextFilesPage[FileListResponse]:
+    ) -> FileListResponse:
         """
         Get the files attached to an agent with their open/closed status.
 
@@ -98,9 +97,8 @@ class FilesResource(SyncAPIResource):
         """
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._get_api_list(
+        return self._get(
             f"/v1/agents/{agent_id}/files",
-            page=SyncNextFilesPage[FileListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -119,7 +117,7 @@ class FilesResource(SyncAPIResource):
                     file_list_params.FileListParams,
                 ),
             ),
-            model=FileListResponse,
+            cast_to=FileListResponse,
         )
 
     def close(
@@ -268,7 +266,7 @@ class AsyncFilesResource(AsyncAPIResource):
         """
         return AsyncFilesResourceWithStreamingResponse(self)
 
-    def list(
+    async def list(
         self,
         agent_id: str,
         *,
@@ -285,7 +283,7 @@ class AsyncFilesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[FileListResponse, AsyncNextFilesPage[FileListResponse]]:
+    ) -> FileListResponse:
         """
         Get the files attached to an agent with their open/closed status.
 
@@ -319,15 +317,14 @@ class AsyncFilesResource(AsyncAPIResource):
         """
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._get_api_list(
+        return await self._get(
             f"/v1/agents/{agent_id}/files",
-            page=AsyncNextFilesPage[FileListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform(
+                query=await async_maybe_transform(
                     {
                         "after": after,
                         "before": before,
@@ -340,7 +337,7 @@ class AsyncFilesResource(AsyncAPIResource):
                     file_list_params.FileListParams,
                 ),
             ),
-            model=FileListResponse,
+            cast_to=FileListResponse,
         )
 
     async def close(

letta_client/resources/agents/folders.py

@@ -8,7 +8,7 @@ from typing_extensions import Literal
 import httpx
 
 from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from ..._utils import maybe_transform
+from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -17,8 +17,7 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...pagination import SyncArrayPage, AsyncArrayPage
-from ..._base_client import AsyncPaginator, make_request_options
+from ..._base_client import make_request_options
 from ...types.agents import folder_list_params
 from ...types.agent_state import AgentState
 from ...types.agents.folder_list_response import FolderListResponse
@@ -61,7 +60,7 @@ class FoldersResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[FolderListResponse]:
+    ) -> FolderListResponse:
         """
         Get the folders associated with an agent.
 
@@ -91,9 +90,8 @@ class FoldersResource(SyncAPIResource):
         """
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._get_api_list(
+        return self._get(
             f"/v1/agents/{agent_id}/folders",
-            page=SyncArrayPage[FolderListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -110,7 +108,7 @@ class FoldersResource(SyncAPIResource):
                     folder_list_params.FolderListParams,
                 ),
             ),
-            model=FolderListResponse,
+            cast_to=FolderListResponse,
         )
 
     def attach(
@@ -214,7 +212,7 @@ class AsyncFoldersResource(AsyncAPIResource):
         """
         return AsyncFoldersResourceWithStreamingResponse(self)
 
-    def list(
+    async def list(
         self,
         agent_id: str,
         *,
@@ -229,7 +227,7 @@ class AsyncFoldersResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[FolderListResponse, AsyncArrayPage[FolderListResponse]]:
+    ) -> FolderListResponse:
         """
         Get the folders associated with an agent.
 
@@ -259,15 +257,14 @@ class AsyncFoldersResource(AsyncAPIResource):
         """
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._get_api_list(
+        return await self._get(
             f"/v1/agents/{agent_id}/folders",
-            page=AsyncArrayPage[FolderListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform(
+                query=await async_maybe_transform(
                     {
                         "after": after,
                         "before": before,
@@ -278,7 +275,7 @@ class AsyncFoldersResource(AsyncAPIResource):
                     folder_list_params.FolderListParams,
                 ),
             ),
-            model=FolderListResponse,
+            cast_to=FolderListResponse,
         )
 
     async def attach(