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

letta_client/resources/groups/messages.py

@@ -17,11 +17,12 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import make_request_options
+from ...pagination import SyncArrayPage, AsyncArrayPage
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.groups import message_list_params, message_send_params, message_stream_params, message_update_params
 from ...types.agents.message_type import MessageType
 from ...types.agents.letta_response import LettaResponse
-from ...types.groups.message_list_response import MessageListResponse
+from ...types.agents.letta_message_union import LettaMessageUnion
 from ...types.groups.message_update_response import MessageUpdateResponse
 from ...types.agents.letta_user_message_content_union_param import LettaUserMessageContentUnionParam
 from ...types.agents.letta_assistant_message_content_union_param import LettaAssistantMessageContentUnionParam
@@ -253,7 +254,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageListResponse:
+    ) -> SyncArrayPage[LettaMessageUnion]:
         """
         Retrieve message history for an agent.
 
@@ -289,8 +290,9 @@ class MessagesResource(SyncAPIResource):
         """
         if not group_id:
             raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
-        return self._get(
+        return self._get_api_list(
             f"/v1/groups/{group_id}/messages",
+            page=SyncArrayPage[LettaMessageUnion],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -310,7 +312,7 @@ class MessagesResource(SyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            cast_to=MessageListResponse,
+            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
         )
 
     def reset(
@@ -722,7 +724,7 @@ class AsyncMessagesResource(AsyncAPIResource):
             ),
         )
 
-    async def list(
+    def list(
         self,
         group_id: str,
         *,
@@ -740,7 +742,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageListResponse:
+    ) -> AsyncPaginator[LettaMessageUnion, AsyncArrayPage[LettaMessageUnion]]:
         """
         Retrieve message history for an agent.
 
@@ -776,14 +778,15 @@ class AsyncMessagesResource(AsyncAPIResource):
         """
         if not group_id:
             raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
-        return await self._get(
+        return self._get_api_list(
             f"/v1/groups/{group_id}/messages",
+            page=AsyncArrayPage[LettaMessageUnion],
             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,
                         "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
@@ -797,7 +800,7 @@ class AsyncMessagesResource(AsyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            cast_to=MessageListResponse,
+            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
         )
 
     async def reset(

letta_client/resources/identities/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, async_maybe_transform
+from ..._utils import maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -17,9 +17,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import make_request_options
+from ...pagination import SyncArrayPage, AsyncArrayPage
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.identities import agent_list_params
-from ...types.identities.agent_list_response import AgentListResponse
+from ...types.agent_state import AgentState
 
 __all__ = ["AgentsResource", "AsyncAgentsResource"]
 
@@ -71,7 +72,7 @@ class AgentsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentListResponse:
+    ) -> SyncArrayPage[AgentState]:
         """
         Get all agents associated with the specified identity.
 
@@ -104,8 +105,9 @@ class AgentsResource(SyncAPIResource):
         """
         if not identity_id:
             raise ValueError(f"Expected a non-empty value for `identity_id` but received {identity_id!r}")
-        return self._get(
+        return self._get_api_list(
             f"/v1/identities/{identity_id}/agents",
+            page=SyncArrayPage[AgentState],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -123,7 +125,7 @@ class AgentsResource(SyncAPIResource):
                     agent_list_params.AgentListParams,
                 ),
             ),
-            cast_to=AgentListResponse,
+            model=AgentState,
         )
 
 
@@ -147,7 +149,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         """
         return AsyncAgentsResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         identity_id: str,
         *,
@@ -174,7 +176,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentListResponse:
+    ) -> AsyncPaginator[AgentState, AsyncArrayPage[AgentState]]:
         """
         Get all agents associated with the specified identity.
 
@@ -207,14 +209,15 @@ class AsyncAgentsResource(AsyncAPIResource):
         """
         if not identity_id:
             raise ValueError(f"Expected a non-empty value for `identity_id` but received {identity_id!r}")
-        return await self._get(
+        return self._get_api_list(
             f"/v1/identities/{identity_id}/agents",
+            page=AsyncArrayPage[AgentState],
             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,
@@ -226,7 +229,7 @@ class AsyncAgentsResource(AsyncAPIResource):
                     agent_list_params.AgentListParams,
                 ),
             ),
-            cast_to=AgentListResponse,
+            model=AgentState,
         )
 
 

letta_client/resources/identities/blocks.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, async_maybe_transform
+from ..._utils import maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -17,9 +17,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import make_request_options
+from ...pagination import SyncArrayPage, AsyncArrayPage
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.identities import block_list_params
-from ...types.identities.block_list_response import BlockListResponse
+from ...types.agents.block import Block
 
 __all__ = ["BlocksResource", "AsyncBlocksResource"]
 
@@ -59,7 +60,7 @@ class BlocksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockListResponse:
+    ) -> SyncArrayPage[Block]:
         """
         Get all blocks associated with the specified identity.
 
@@ -89,8 +90,9 @@ class BlocksResource(SyncAPIResource):
         """
         if not identity_id:
             raise ValueError(f"Expected a non-empty value for `identity_id` but received {identity_id!r}")
-        return self._get(
+        return self._get_api_list(
             f"/v1/identities/{identity_id}/blocks",
+            page=SyncArrayPage[Block],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -107,7 +109,7 @@ class BlocksResource(SyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            cast_to=BlockListResponse,
+            model=Block,
         )
 
 
@@ -131,7 +133,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         """
         return AsyncBlocksResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         identity_id: str,
         *,
@@ -146,7 +148,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockListResponse:
+    ) -> AsyncPaginator[Block, AsyncArrayPage[Block]]:
         """
         Get all blocks associated with the specified identity.
 
@@ -176,14 +178,15 @@ class AsyncBlocksResource(AsyncAPIResource):
         """
         if not identity_id:
             raise ValueError(f"Expected a non-empty value for `identity_id` but received {identity_id!r}")
-        return await self._get(
+        return self._get_api_list(
             f"/v1/identities/{identity_id}/blocks",
+            page=AsyncArrayPage[Block],
             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,
@@ -194,7 +197,7 @@ class AsyncBlocksResource(AsyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            cast_to=BlockListResponse,
+            model=Block,
         )
 
 

letta_client/resources/identities/identities.py

@@ -48,10 +48,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import make_request_options
+from ...pagination import SyncArrayPage, AsyncArrayPage
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.identity import Identity
 from ...types.identity_type import IdentityType
-from ...types.identity_list_response import IdentityListResponse
 from ...types.identity_count_response import IdentityCountResponse
 from ...types.identity_property_param import IdentityPropertyParam
 
@@ -274,7 +274,7 @@ class IdentitiesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> IdentityListResponse:
+    ) -> SyncArrayPage[Identity]:
         """
         Get a list of all identities in the database
 
@@ -302,8 +302,9 @@ class IdentitiesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get(
+        return self._get_api_list(
             "/v1/identities/",
+            page=SyncArrayPage[Identity],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -324,7 +325,7 @@ class IdentitiesResource(SyncAPIResource):
                     identity_list_params.IdentityListParams,
                 ),
             ),
-            cast_to=IdentityListResponse,
+            model=Identity,
         )
 
     def delete(
@@ -647,7 +648,7 @@ class AsyncIdentitiesResource(AsyncAPIResource):
             cast_to=Identity,
         )
 
-    async def list(
+    def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -665,7 +666,7 @@ class AsyncIdentitiesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> IdentityListResponse:
+    ) -> AsyncPaginator[Identity, AsyncArrayPage[Identity]]:
         """
         Get a list of all identities in the database
 
@@ -693,14 +694,15 @@ class AsyncIdentitiesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._get(
+        return self._get_api_list(
             "/v1/identities/",
+            page=AsyncArrayPage[Identity],
             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,
@@ -715,7 +717,7 @@ class AsyncIdentitiesResource(AsyncAPIResource):
                     identity_list_params.IdentityListParams,
                 ),
             ),
-            cast_to=IdentityListResponse,
+            model=Identity,
         )
 
     async def delete(

letta_client/resources/runs/messages.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing import Any, Optional, cast
 from typing_extensions import Literal
 
 import httpx
@@ -17,9 +17,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from ...pagination import SyncArrayPage, AsyncArrayPage
 from ...types.runs import message_list_params, message_stream_params
-from ..._base_client import make_request_options
-from ...types.runs.message_list_response import MessageListResponse
+from ..._base_client import AsyncPaginator, make_request_options
+from ...types.agents.letta_message_union import LettaMessageUnion
 
 __all__ = ["MessagesResource", "AsyncMessagesResource"]
 
@@ -59,7 +60,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageListResponse:
+    ) -> SyncArrayPage[LettaMessageUnion]:
         """
         Get response messages associated with a run.
 
@@ -87,8 +88,9 @@ class MessagesResource(SyncAPIResource):
         """
         if not run_id:
             raise ValueError(f"Expected a non-empty value for `run_id` but received {run_id!r}")
-        return self._get(
+        return self._get_api_list(
             f"/v1/runs/{run_id}/messages",
+            page=SyncArrayPage[LettaMessageUnion],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -105,7 +107,7 @@ class MessagesResource(SyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            cast_to=MessageListResponse,
+            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
         )
 
     def stream(
@@ -185,7 +187,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         """
         return AsyncMessagesResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         run_id: str,
         *,
@@ -200,7 +202,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageListResponse:
+    ) -> AsyncPaginator[LettaMessageUnion, AsyncArrayPage[LettaMessageUnion]]:
         """
         Get response messages associated with a run.
 
@@ -228,14 +230,15 @@ class AsyncMessagesResource(AsyncAPIResource):
         """
         if not run_id:
             raise ValueError(f"Expected a non-empty value for `run_id` but received {run_id!r}")
-        return await self._get(
+        return self._get_api_list(
             f"/v1/runs/{run_id}/messages",
+            page=AsyncArrayPage[LettaMessageUnion],
             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,
@@ -246,7 +249,7 @@ class AsyncMessagesResource(AsyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            cast_to=MessageListResponse,
+            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
         )
 
     async def stream(

letta_client/resources/runs/runs.py

@@ -25,7 +25,7 @@ from .usage import (
 )
 from ...types import StopReasonType, run_list_params
 from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
-from ..._utils import maybe_transform, async_maybe_transform
+from ..._utils import maybe_transform
 from .messages import (
     MessagesResource,
     AsyncMessagesResource,
@@ -42,10 +42,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import make_request_options
+from ...pagination import SyncArrayPage, AsyncArrayPage
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.agents.run import Run
 from ...types.stop_reason_type import StopReasonType
-from ...types.run_list_response import RunListResponse
 
 __all__ = ["RunsResource", "AsyncRunsResource"]
 
@@ -136,7 +136,7 @@ class RunsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> RunListResponse:
+    ) -> SyncArrayPage[Run]:
         """
         List all runs.
 
@@ -178,8 +178,9 @@ class RunsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get(
+        return self._get_api_list(
             "/v1/runs/",
+            page=SyncArrayPage[Run],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -203,7 +204,7 @@ class RunsResource(SyncAPIResource):
                     run_list_params.RunListParams,
                 ),
             ),
-            cast_to=RunListResponse,
+            model=Run,
         )
 
 
@@ -272,7 +273,7 @@ class AsyncRunsResource(AsyncAPIResource):
             cast_to=Run,
         )
 
-    async def list(
+    def list(
         self,
         *,
         active: bool | Omit = omit,
@@ -293,7 +294,7 @@ class AsyncRunsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> RunListResponse:
+    ) -> AsyncPaginator[Run, AsyncArrayPage[Run]]:
         """
         List all runs.
 
@@ -335,14 +336,15 @@ class AsyncRunsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._get(
+        return self._get_api_list(
             "/v1/runs/",
+            page=AsyncArrayPage[Run],
             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(
                     {
                         "active": active,
                         "after": after,
@@ -360,7 +362,7 @@ class AsyncRunsResource(AsyncAPIResource):
                     run_list_params.RunListParams,
                 ),
             ),
-            cast_to=RunListResponse,
+            model=Run,
         )
 
 

letta_client/resources/runs/steps.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, async_maybe_transform
+from ..._utils import maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -17,9 +17,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from ...pagination import SyncArrayPage, AsyncArrayPage
 from ...types.runs import step_list_params
-from ..._base_client import make_request_options
-from ...types.runs.step_list_response import StepListResponse
+from ...types.step import Step
+from ..._base_client import AsyncPaginator, make_request_options
 
 __all__ = ["StepsResource", "AsyncStepsResource"]
 
@@ -59,7 +60,7 @@ class StepsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> StepListResponse:
+    ) -> SyncArrayPage[Step]:
         """
         Get steps associated with a run with filtering options.
 
@@ -85,8 +86,9 @@ class StepsResource(SyncAPIResource):
         """
         if not run_id:
             raise ValueError(f"Expected a non-empty value for `run_id` but received {run_id!r}")
-        return self._get(
+        return self._get_api_list(
             f"/v1/runs/{run_id}/steps",
+            page=SyncArrayPage[Step],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -103,7 +105,7 @@ class StepsResource(SyncAPIResource):
                     step_list_params.StepListParams,
                 ),
             ),
-            cast_to=StepListResponse,
+            model=Step,
         )
 
 
@@ -127,7 +129,7 @@ class AsyncStepsResource(AsyncAPIResource):
         """
         return AsyncStepsResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         run_id: str,
         *,
@@ -142,7 +144,7 @@ class AsyncStepsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> StepListResponse:
+    ) -> AsyncPaginator[Step, AsyncArrayPage[Step]]:
         """
         Get steps associated with a run with filtering options.
 
@@ -168,14 +170,15 @@ class AsyncStepsResource(AsyncAPIResource):
         """
         if not run_id:
             raise ValueError(f"Expected a non-empty value for `run_id` but received {run_id!r}")
-        return await self._get(
+        return self._get_api_list(
             f"/v1/runs/{run_id}/steps",
+            page=AsyncArrayPage[Step],
             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,
@@ -186,7 +189,7 @@ class AsyncStepsResource(AsyncAPIResource):
                     step_list_params.StepListParams,
                 ),
             ),
-            cast_to=StepListResponse,
+            model=Step,
         )