letta-client 0.1.121__py3-none-any.whl → 0.1.122__py3-none-any.whl

letta_client/runs/client.py

@@ -1,6 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ..core.client_wrapper import SyncClientWrapper
+from .messages.client import MessagesClient
+from .usage.client import UsageClient
+from .steps.client import StepsClient
 import typing
 from ..core.request_options import RequestOptions
 from ..types.run import Run
@@ -10,16 +13,18 @@ from ..types.http_validation_error import HttpValidationError
 from json.decoder import JSONDecodeError
 from ..core.api_error import ApiError
 from ..core.jsonable_encoder import jsonable_encoder
-from ..types.message_role import MessageRole
-from ..types.letta_message_union import LettaMessageUnion
-from ..types.usage_statistics import UsageStatistics
-from ..types.step import Step
 from ..core.client_wrapper import AsyncClientWrapper
+from .messages.client import AsyncMessagesClient
+from .usage.client import AsyncUsageClient
+from .steps.client import AsyncStepsClient
 
 
 class RunsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
+        self.messages = MessagesClient(client_wrapper=self._client_wrapper)
+        self.usage = UsageClient(client_wrapper=self._client_wrapper)
+        self.steps = StepsClient(client_wrapper=self._client_wrapper)
 
     def list_runs(
         self,
@@ -84,7 +89,7 @@ class RunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def list_active_runs(
+    def list_active(
         self,
         *,
         agent_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
@@ -113,7 +118,7 @@ class RunsClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.runs.list_active_runs()
+        client.runs.list_active()
         """
         _response = self._client_wrapper.httpx_client.request(
             "v1/runs/active",
@@ -147,7 +152,7 @@ class RunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def retrieve_run(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
+    def retrieve(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
         """
         Get the status of a run.
 
@@ -170,7 +175,7 @@ class RunsClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.runs.retrieve_run(
+        client.runs.retrieve(
             run_id="run_id",
         )
         """
@@ -203,7 +208,7 @@ class RunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def delete_run(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
+    def delete(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
         """
         Delete a run by its run_id.
 
@@ -226,7 +231,7 @@ class RunsClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.runs.delete_run(
+        client.runs.delete(
             run_id="run_id",
         )
         """
@@ -259,211 +264,16 @@ class RunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def list_run_messages(
-        self,
-        run_id: str,
-        *,
-        before: typing.Optional[str] = None,
-        after: typing.Optional[str] = None,
-        limit: typing.Optional[int] = None,
-        order: typing.Optional[str] = None,
-        role: typing.Optional[MessageRole] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[LettaMessageUnion]:
-        """
-        Get messages associated with a run with filtering options.
-
-        Args:
-            run_id: ID of the run
-            before: A cursor for use in pagination. `before` is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, starting with obj_foo, your subsequent call can include before=obj_foo in order to fetch the previous page of the list.
-            after: A cursor for use in pagination. `after` is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, ending with obj_foo, your subsequent call can include after=obj_foo in order to fetch the next page of the list.
-            limit: Maximum number of messages to return
-            order: Sort order by the created_at timestamp of the objects. asc for ascending order and desc for descending order.
-            role: Filter by role (user/assistant/system/tool)
-            return_message_object: Whether to return Message objects or LettaMessage objects
-            user_id: ID of the user making the request
-
-        Returns:
-            A list of messages associated with the run. Default is List[LettaMessage].
-
-        Parameters
-        ----------
-        run_id : str
-
-        before : typing.Optional[str]
-            Cursor for pagination
-
-        after : typing.Optional[str]
-            Cursor for pagination
-
-        limit : typing.Optional[int]
-            Maximum number of messages to return
-
-        order : typing.Optional[str]
-            Sort order by the created_at timestamp of the objects. asc for ascending order and desc for descending order.
-
-        role : typing.Optional[MessageRole]
-            Filter by role
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[LettaMessageUnion]
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            token="YOUR_TOKEN",
-        )
-        client.runs.list_run_messages(
-            run_id="run_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/runs/{jsonable_encoder(run_id)}/messages",
-            method="GET",
-            params={
-                "before": before,
-                "after": after,
-                "limit": limit,
-                "order": order,
-                "role": role,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[LettaMessageUnion],
-                    construct_type(
-                        type_=typing.List[LettaMessageUnion],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    def retrieve_run_usage(
-        self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> UsageStatistics:
-        """
-        Get usage statistics for a run.
-
-        Parameters
-        ----------
-        run_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        UsageStatistics
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            token="YOUR_TOKEN",
-        )
-        client.runs.retrieve_run_usage(
-            run_id="run_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/runs/{jsonable_encoder(run_id)}/usage",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    UsageStatistics,
-                    construct_type(
-                        type_=UsageStatistics,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    def list_run_steps(
-        self,
-        run_id: str,
-        *,
-        before: typing.Optional[str] = None,
-        after: typing.Optional[str] = None,
-        limit: typing.Optional[int] = None,
-        order: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[Step]:
+    def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
-        Get messages associated with a run with filtering options.
-
-        Args:
-            run_id: ID of the run
-            before: A cursor for use in pagination. `before` is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, starting with obj_foo, your subsequent call can include before=obj_foo in order to fetch the previous page of the list.
-            after: A cursor for use in pagination. `after` is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, ending with obj_foo, your subsequent call can include after=obj_foo in order to fetch the next page of the list.
-            limit: Maximum number of steps to return
-            order: Sort order by the created_at timestamp of the objects. asc for ascending order and desc for descending order.
-
-        Returns:
-            A list of steps associated with the run.
-
         Parameters
         ----------
-        run_id : str
-
-        before : typing.Optional[str]
-            Cursor for pagination
-
-        after : typing.Optional[str]
-            Cursor for pagination
-
-        limit : typing.Optional[int]
-            Maximum number of messages to return
-
-        order : typing.Optional[str]
-            Sort order by the created_at timestamp of the objects. asc for ascending order and desc for descending order.
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[Step]
-            Successful Response
+        None
 
         Examples
         --------
@@ -472,40 +282,16 @@ class RunsClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.runs.list_run_steps(
-            run_id="run_id",
-        )
+        client.runs.list()
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/runs/{jsonable_encoder(run_id)}/steps",
+            "v1/runs",
             method="GET",
-            params={
-                "before": before,
-                "after": after,
-                "limit": limit,
-                "order": order,
-            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Step],
-                    construct_type(
-                        type_=typing.List[Step],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -515,6 +301,9 @@ class RunsClient:
 class AsyncRunsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
+        self.messages = AsyncMessagesClient(client_wrapper=self._client_wrapper)
+        self.usage = AsyncUsageClient(client_wrapper=self._client_wrapper)
+        self.steps = AsyncStepsClient(client_wrapper=self._client_wrapper)
 
     async def list_runs(
         self,
@@ -587,7 +376,7 @@ class AsyncRunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def list_active_runs(
+    async def list_active(
         self,
         *,
         agent_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
@@ -621,7 +410,7 @@ class AsyncRunsClient:
 
 
         async def main() -> None:
-            await client.runs.list_active_runs()
+            await client.runs.list_active()
 
 
         asyncio.run(main())
@@ -658,7 +447,7 @@ class AsyncRunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def retrieve_run(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
+    async def retrieve(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
         """
         Get the status of a run.
 
@@ -686,7 +475,7 @@ class AsyncRunsClient:
 
 
         async def main() -> None:
-            await client.runs.retrieve_run(
+            await client.runs.retrieve(
                 run_id="run_id",
             )
 
@@ -722,7 +511,7 @@ class AsyncRunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def delete_run(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
+    async def delete(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
         """
         Delete a run by its run_id.
 
@@ -750,7 +539,7 @@ class AsyncRunsClient:
 
 
         async def main() -> None:
-            await client.runs.delete_run(
+            await client.runs.delete(
                 run_id="run_id",
             )
 
@@ -786,227 +575,16 @@ class AsyncRunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def list_run_messages(
-        self,
-        run_id: str,
-        *,
-        before: typing.Optional[str] = None,
-        after: typing.Optional[str] = None,
-        limit: typing.Optional[int] = None,
-        order: typing.Optional[str] = None,
-        role: typing.Optional[MessageRole] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[LettaMessageUnion]:
-        """
-        Get messages associated with a run with filtering options.
-
-        Args:
-            run_id: ID of the run
-            before: A cursor for use in pagination. `before` is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, starting with obj_foo, your subsequent call can include before=obj_foo in order to fetch the previous page of the list.
-            after: A cursor for use in pagination. `after` is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, ending with obj_foo, your subsequent call can include after=obj_foo in order to fetch the next page of the list.
-            limit: Maximum number of messages to return
-            order: Sort order by the created_at timestamp of the objects. asc for ascending order and desc for descending order.
-            role: Filter by role (user/assistant/system/tool)
-            return_message_object: Whether to return Message objects or LettaMessage objects
-            user_id: ID of the user making the request
-
-        Returns:
-            A list of messages associated with the run. Default is List[LettaMessage].
-
-        Parameters
-        ----------
-        run_id : str
-
-        before : typing.Optional[str]
-            Cursor for pagination
-
-        after : typing.Optional[str]
-            Cursor for pagination
-
-        limit : typing.Optional[int]
-            Maximum number of messages to return
-
-        order : typing.Optional[str]
-            Sort order by the created_at timestamp of the objects. asc for ascending order and desc for descending order.
-
-        role : typing.Optional[MessageRole]
-            Filter by role
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[LettaMessageUnion]
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.runs.list_run_messages(
-                run_id="run_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/runs/{jsonable_encoder(run_id)}/messages",
-            method="GET",
-            params={
-                "before": before,
-                "after": after,
-                "limit": limit,
-                "order": order,
-                "role": role,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[LettaMessageUnion],
-                    construct_type(
-                        type_=typing.List[LettaMessageUnion],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    async def retrieve_run_usage(
-        self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> UsageStatistics:
-        """
-        Get usage statistics for a run.
-
-        Parameters
-        ----------
-        run_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        UsageStatistics
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.runs.retrieve_run_usage(
-                run_id="run_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/runs/{jsonable_encoder(run_id)}/usage",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    UsageStatistics,
-                    construct_type(
-                        type_=UsageStatistics,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    async def list_run_steps(
-        self,
-        run_id: str,
-        *,
-        before: typing.Optional[str] = None,
-        after: typing.Optional[str] = None,
-        limit: typing.Optional[int] = None,
-        order: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[Step]:
+    async def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
-        Get messages associated with a run with filtering options.
-
-        Args:
-            run_id: ID of the run
-            before: A cursor for use in pagination. `before` is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, starting with obj_foo, your subsequent call can include before=obj_foo in order to fetch the previous page of the list.
-            after: A cursor for use in pagination. `after` is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, ending with obj_foo, your subsequent call can include after=obj_foo in order to fetch the next page of the list.
-            limit: Maximum number of steps to return
-            order: Sort order by the created_at timestamp of the objects. asc for ascending order and desc for descending order.
-
-        Returns:
-            A list of steps associated with the run.
-
         Parameters
         ----------
-        run_id : str
-
-        before : typing.Optional[str]
-            Cursor for pagination
-
-        after : typing.Optional[str]
-            Cursor for pagination
-
-        limit : typing.Optional[int]
-            Maximum number of messages to return
-
-        order : typing.Optional[str]
-            Sort order by the created_at timestamp of the objects. asc for ascending order and desc for descending order.
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[Step]
-            Successful Response
+        None
 
         Examples
         --------
@@ -1020,43 +598,19 @@ class AsyncRunsClient:
 
 
         async def main() -> None:
-            await client.runs.list_run_steps(
-                run_id="run_id",
-            )
+            await client.runs.list()
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/runs/{jsonable_encoder(run_id)}/steps",
+            "v1/runs",
             method="GET",
-            params={
-                "before": before,
-                "after": after,
-                "limit": limit,
-                "order": order,
-            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Step],
-                    construct_type(
-                        type_=typing.List[Step],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)

letta_client/runs/messages/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+