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

letta_client/resources/agents/messages.py

@@ -17,6 +17,7 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from ..._streaming import Stream, AsyncStream
 from ...pagination import SyncArrayPage, AsyncArrayPage
 from ..._base_client import AsyncPaginator, make_request_options
 from ...types.agents import (
@@ -35,6 +36,7 @@ from ...types.agents.letta_response import LettaResponse
 from ...types.agents.letta_message_union import LettaMessageUnion
 from ...types.agents.message_cancel_response import MessageCancelResponse
 from ...types.agents.message_modify_response import MessageModifyResponse
+from ...types.agents.letta_streaming_response import LettaStreamingResponse
 from ...types.agents.letta_user_message_content_union_param import LettaUserMessageContentUnionParam
 from ...types.agents.letta_assistant_message_content_union_param import LettaAssistantMessageContentUnionParam
 
@@ -597,7 +599,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> object:
+    ) -> Stream[LettaStreamingResponse]:
         """Process a user message and return the agent's response.
 
         This endpoint accepts a
@@ -664,7 +666,11 @@ class MessagesResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=cast(
+                Any, LettaStreamingResponse
+            ),  # Union types cannot be passed in as arguments in the type system
+            stream=True,
+            stream_cls=Stream[LettaStreamingResponse],
         )
 
     def summarize(
@@ -1260,7 +1266,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> object:
+    ) -> AsyncStream[LettaStreamingResponse]:
         """Process a user message and return the agent's response.
 
         This endpoint accepts a
@@ -1327,7 +1333,11 @@ class AsyncMessagesResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=cast(
+                Any, LettaStreamingResponse
+            ),  # Union types cannot be passed in as arguments in the type system
+            stream=True,
+            stream_cls=AsyncStream[LettaStreamingResponse],
         )
 
     async def summarize(

letta_client/resources/agents/tools.py

@@ -17,11 +17,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...pagination import SyncArrayPage, AsyncArrayPage
-from ...types.tool import Tool
-from ..._base_client import AsyncPaginator, make_request_options
+from ..._base_client import make_request_options
 from ...types.agents import tool_list_params, tool_update_approval_params
 from ...types.agent_state import AgentState
+from ...types.agents.tool_list_response import ToolListResponse
 
 __all__ = ["ToolsResource", "AsyncToolsResource"]
 
@@ -61,7 +60,7 @@ class ToolsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[Tool]:
+    ) -> ToolListResponse:
         """
         Get tools from an existing agent.
 
@@ -91,9 +90,8 @@ class ToolsResource(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}/tools",
-            page=SyncArrayPage[Tool],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -110,7 +108,7 @@ class ToolsResource(SyncAPIResource):
                     tool_list_params.ToolListParams,
                 ),
             ),
-            model=Tool,
+            cast_to=ToolListResponse,
         )
 
     def attach(
@@ -271,7 +269,7 @@ class AsyncToolsResource(AsyncAPIResource):
         """
         return AsyncToolsResourceWithStreamingResponse(self)
 
-    def list(
+    async def list(
         self,
         agent_id: str,
         *,
@@ -286,7 +284,7 @@ class AsyncToolsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[Tool, AsyncArrayPage[Tool]]:
+    ) -> ToolListResponse:
         """
         Get tools from an existing agent.
 
@@ -316,15 +314,14 @@ class AsyncToolsResource(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}/tools",
-            page=AsyncArrayPage[Tool],
             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,
@@ -335,7 +332,7 @@ class AsyncToolsResource(AsyncAPIResource):
                     tool_list_params.ToolListParams,
                 ),
             ),
-            model=Tool,
+            cast_to=ToolListResponse,
         )
 
     async def attach(

letta_client/resources/archives.py

@@ -18,9 +18,9 @@ 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.archive import Archive
+from ..types.archive_list_response import ArchiveListResponse
 from ..types.embedding_config_param import EmbeddingConfigParam
 
 __all__ = ["ArchivesResource", "AsyncArchivesResource"]
@@ -140,7 +140,7 @@ class ArchivesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[Archive]:
+    ) -> ArchiveListResponse:
         """
         Get a list of all archives for the current organization with optional filters
         and pagination.
@@ -171,9 +171,8 @@ class ArchivesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._get(
             "/v1/archives/",
-            page=SyncArrayPage[Archive],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -192,7 +191,7 @@ class ArchivesResource(SyncAPIResource):
                     archive_list_params.ArchiveListParams,
                 ),
             ),
-            model=Archive,
+            cast_to=ArchiveListResponse,
         )
 
     def delete(
@@ -373,7 +372,7 @@ class AsyncArchivesResource(AsyncAPIResource):
             cast_to=Archive,
         )
 
-    def list(
+    async def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -389,7 +388,7 @@ class AsyncArchivesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[Archive, AsyncArrayPage[Archive]]:
+    ) -> ArchiveListResponse:
         """
         Get a list of all archives for the current organization with optional filters
         and pagination.
@@ -420,15 +419,14 @@ class AsyncArchivesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._get(
             "/v1/archives/",
-            page=AsyncArrayPage[Archive],
             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,
                         "agent_id": agent_id,
@@ -441,7 +439,7 @@ class AsyncArchivesResource(AsyncAPIResource):
                     archive_list_params.ArchiveListParams,
                 ),
             ),
-            model=Archive,
+            cast_to=ArchiveListResponse,
         )
 
     async def delete(

letta_client/resources/batches/batches.py

@@ -26,9 +26,9 @@ 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.batch_job import BatchJob
+from ...types.batch_list_response import BatchListResponse
 
 __all__ = ["BatchesResource", "AsyncBatchesResource"]
 
@@ -155,7 +155,7 @@ class BatchesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[BatchJob]:
+    ) -> BatchListResponse:
         """List all batch runs.
 
         Args:
@@ -182,9 +182,8 @@ class BatchesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._get(
             "/v1/messages/batches",
-            page=SyncArrayPage[BatchJob],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -201,7 +200,7 @@ class BatchesResource(SyncAPIResource):
                     batch_list_params.BatchListParams,
                 ),
             ),
-            model=BatchJob,
+            cast_to=BatchListResponse,
         )
 
     def cancel(
@@ -346,7 +345,7 @@ class AsyncBatchesResource(AsyncAPIResource):
             cast_to=BatchJob,
         )
 
-    def list(
+    async def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -360,7 +359,7 @@ class AsyncBatchesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[BatchJob, AsyncArrayPage[BatchJob]]:
+    ) -> BatchListResponse:
         """List all batch runs.
 
         Args:
@@ -387,15 +386,14 @@ class AsyncBatchesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._get(
             "/v1/messages/batches",
-            page=AsyncArrayPage[BatchJob],
             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,
@@ -406,7 +404,7 @@ class AsyncBatchesResource(AsyncAPIResource):
                     batch_list_params.BatchListParams,
                 ),
             ),
-            model=BatchJob,
+            cast_to=BatchListResponse,
         )
 
     async def cancel(

letta_client/resources/batches/messages.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,10 +17,9 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...pagination import SyncObjectPage, AsyncObjectPage
-from ..._base_client import AsyncPaginator, make_request_options
+from ..._base_client import make_request_options
 from ...types.batches import message_list_params
-from ...types.agents.message import Message
+from ...types.batches.message_list_response import MessageListResponse
 
 __all__ = ["MessagesResource", "AsyncMessagesResource"]
 
@@ -61,7 +60,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncObjectPage[Message]:
+    ) -> MessageListResponse:
         """
         Get response messages for a specific batch job.
 
@@ -91,9 +90,8 @@ class MessagesResource(SyncAPIResource):
         """
         if not batch_id:
             raise ValueError(f"Expected a non-empty value for `batch_id` but received {batch_id!r}")
-        return self._get_api_list(
+        return self._get(
             f"/v1/messages/batches/{batch_id}/messages",
-            page=SyncObjectPage[Message],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -111,7 +109,7 @@ class MessagesResource(SyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            model=Message,
+            cast_to=MessageListResponse,
         )
 
 
@@ -135,7 +133,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         """
         return AsyncMessagesResourceWithStreamingResponse(self)
 
-    def list(
+    async def list(
         self,
         batch_id: str,
         *,
@@ -151,7 +149,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[Message, AsyncObjectPage[Message]]:
+    ) -> MessageListResponse:
         """
         Get response messages for a specific batch job.
 
@@ -181,15 +179,14 @@ class AsyncMessagesResource(AsyncAPIResource):
         """
         if not batch_id:
             raise ValueError(f"Expected a non-empty value for `batch_id` but received {batch_id!r}")
-        return self._get_api_list(
+        return await self._get(
             f"/v1/messages/batches/{batch_id}/messages",
-            page=AsyncObjectPage[Message],
             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,
                         "agent_id": agent_id,
@@ -201,7 +198,7 @@ class AsyncMessagesResource(AsyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            model=Message,
+            cast_to=MessageListResponse,
         )
 
 

letta_client/resources/blocks/blocks.py

@@ -28,11 +28,8 @@ from ..._response import (
 )
 from ...pagination import SyncArrayPage, AsyncArrayPage
 from ..._base_client import AsyncPaginator, make_request_options
-from ...types.block_list_response import BlockListResponse
+from ...types.block_response import BlockResponse
 from ...types.block_count_response import BlockCountResponse
-from ...types.block_create_response import BlockCreateResponse
-from ...types.block_modify_response import BlockModifyResponse
-from ...types.block_retrieve_response import BlockRetrieveResponse
 
 __all__ = ["BlocksResource", "AsyncBlocksResource"]
 
@@ -85,7 +82,7 @@ class BlocksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockCreateResponse:
+    ) -> BlockResponse:
         """
         Create Block
 
@@ -151,7 +148,7 @@ class BlocksResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=BlockCreateResponse,
+            cast_to=BlockResponse,
         )
 
     def retrieve(
@@ -164,7 +161,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 Block
 
@@ -186,7 +183,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(
@@ -215,7 +212,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]:
         """List Blocks
 
         Args:
@@ -273,7 +270,7 @@ class BlocksResource(SyncAPIResource):
         """
         return self._get_api_list(
             "/v1/blocks/",
-            page=SyncArrayPage[BlockListResponse],
+            page=SyncArrayPage[BlockResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -302,7 +299,7 @@ class BlocksResource(SyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            model=BlockListResponse,
+            model=BlockResponse,
         )
 
     def delete(
@@ -384,7 +381,7 @@ class BlocksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockModifyResponse:
+    ) -> BlockResponse:
         """
         Modify Block
 
@@ -456,7 +453,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,
         )
 
 
@@ -508,7 +505,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockCreateResponse:
+    ) -> BlockResponse:
         """
         Create Block
 
@@ -574,7 +571,7 @@ class AsyncBlocksResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=BlockCreateResponse,
+            cast_to=BlockResponse,
         )
 
     async def retrieve(
@@ -587,7 +584,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 Block
 
@@ -609,7 +606,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(
@@ -638,7 +635,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]]:
         """List Blocks
 
         Args:
@@ -696,7 +693,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         """
         return self._get_api_list(
             "/v1/blocks/",
-            page=AsyncArrayPage[BlockListResponse],
+            page=AsyncArrayPage[BlockResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -725,7 +722,7 @@ class AsyncBlocksResource(AsyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            model=BlockListResponse,
+            model=BlockResponse,
         )
 
     async def delete(
@@ -807,7 +804,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockModifyResponse:
+    ) -> BlockResponse:
         """
         Modify Block
 
@@ -879,7 +876,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/folders/agents.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.folders import agent_list_params
 from ...types.folders.agent_list_response import AgentListResponse
 
@@ -60,7 +59,7 @@ class AgentsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[AgentListResponse]:
+    ) -> AgentListResponse:
         """
         Get all agent IDs that have the specified folder attached.
 
@@ -90,9 +89,8 @@ class AgentsResource(SyncAPIResource):
         """
         if not folder_id:
             raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
-        return self._get_api_list(
+        return self._get(
             f"/v1/folders/{folder_id}/agents",
-            page=SyncArrayPage[AgentListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -109,7 +107,7 @@ class AgentsResource(SyncAPIResource):
                     agent_list_params.AgentListParams,
                 ),
             ),
-            model=str,
+            cast_to=AgentListResponse,
         )
 
 
@@ -133,7 +131,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         """
         return AsyncAgentsResourceWithStreamingResponse(self)
 
-    def list(
+    async def list(
         self,
         folder_id: str,
         *,
@@ -148,7 +146,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[AgentListResponse, AsyncArrayPage[AgentListResponse]]:
+    ) -> AgentListResponse:
         """
         Get all agent IDs that have the specified folder attached.
 
@@ -178,15 +176,14 @@ class AsyncAgentsResource(AsyncAPIResource):
         """
         if not folder_id:
             raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
-        return self._get_api_list(
+        return await self._get(
             f"/v1/folders/{folder_id}/agents",
-            page=AsyncArrayPage[AgentListResponse],
             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,
@@ -197,7 +194,7 @@ class AsyncAgentsResource(AsyncAPIResource):
                     agent_list_params.AgentListParams,
                 ),
             ),
-            model=str,
+            cast_to=AgentListResponse,
         )