letta-client 0.1.17__py3-none-any.whl → 0.1.21__py3-none-any.whl

letta_client/runs/client.py

@@ -124,7 +124,7 @@ class RunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_run(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
+    def retrieve_run(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
         """
         Get the status of a run.
 
@@ -147,7 +147,7 @@ class RunsClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.runs.get_run(
+        client.runs.retrieve_run(
             run_id="run_id",
         )
         """
@@ -236,13 +236,14 @@ class RunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_run_messages(
+    def list_run_messages(
         self,
         run_id: str,
         *,
-        cursor: typing.Optional[str] = None,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
-        ascending: typing.Optional[bool] = None,
+        order: typing.Optional[str] = None,
         role: typing.Optional[MessageRole] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[LettaMessageUnion]:
@@ -251,9 +252,10 @@ class RunsClient:
 
         Args:
             run_id: ID of the run
-            cursor: Cursor for pagination
+            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
-            ascending: Sort order by creation time
+            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
@@ -265,14 +267,17 @@ class RunsClient:
         ----------
         run_id : str
 
-        cursor : typing.Optional[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
 
-        ascending : typing.Optional[bool]
-            Sort order by creation time
+        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
@@ -292,7 +297,7 @@ class RunsClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.runs.get_run_messages(
+        client.runs.list_run_messages(
             run_id="run_id",
         )
         """
@@ -300,9 +305,10 @@ class RunsClient:
             f"v1/runs/{jsonable_encoder(run_id)}/messages",
             method="GET",
             params={
-                "cursor": cursor,
+                "before": before,
+                "after": after,
                 "limit": limit,
-                "ascending": ascending,
+                "order": order,
                 "role": role,
             },
             request_options=request_options,
@@ -331,7 +337,9 @@ class RunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_run_usage(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> UsageStatistics:
+    def retrieve_run_usage(
+        self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> UsageStatistics:
         """
         Get usage statistics for a run.
 
@@ -354,7 +362,7 @@ class RunsClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.runs.get_run_usage(
+        client.runs.retrieve_run_usage(
             run_id="run_id",
         )
         """
@@ -512,7 +520,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 get_run(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
+    async def retrieve_run(self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Run:
         """
         Get the status of a run.
 
@@ -540,7 +548,7 @@ class AsyncRunsClient:
 
 
         async def main() -> None:
-            await client.runs.get_run(
+            await client.runs.retrieve_run(
                 run_id="run_id",
             )
 
@@ -640,13 +648,14 @@ class AsyncRunsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_run_messages(
+    async def list_run_messages(
         self,
         run_id: str,
         *,
-        cursor: typing.Optional[str] = None,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
-        ascending: typing.Optional[bool] = None,
+        order: typing.Optional[str] = None,
         role: typing.Optional[MessageRole] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[LettaMessageUnion]:
@@ -655,9 +664,10 @@ class AsyncRunsClient:
 
         Args:
             run_id: ID of the run
-            cursor: Cursor for pagination
+            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
-            ascending: Sort order by creation time
+            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
@@ -669,14 +679,17 @@ class AsyncRunsClient:
         ----------
         run_id : str
 
-        cursor : typing.Optional[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
 
-        ascending : typing.Optional[bool]
-            Sort order by creation time
+        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
@@ -701,7 +714,7 @@ class AsyncRunsClient:
 
 
         async def main() -> None:
-            await client.runs.get_run_messages(
+            await client.runs.list_run_messages(
                 run_id="run_id",
             )
 
@@ -712,9 +725,10 @@ class AsyncRunsClient:
             f"v1/runs/{jsonable_encoder(run_id)}/messages",
             method="GET",
             params={
-                "cursor": cursor,
+                "before": before,
+                "after": after,
                 "limit": limit,
-                "ascending": ascending,
+                "order": order,
                 "role": role,
             },
             request_options=request_options,
@@ -743,7 +757,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 get_run_usage(
+    async def retrieve_run_usage(
         self, run_id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> UsageStatistics:
         """
@@ -773,7 +787,7 @@ class AsyncRunsClient:
 
 
         async def main() -> None:
-            await client.runs.get_run_usage(
+            await client.runs.retrieve_run_usage(
                 run_id="run_id",
             )
 

letta_client/sources/client.py

@@ -28,7 +28,7 @@ class SourcesClient:
         self.files = FilesClient(client_wrapper=self._client_wrapper)
         self.passages = PassagesClient(client_wrapper=self._client_wrapper)
 
-    def get(self, source_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Source:
+    def retrieve(self, source_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Source:
         """
         Get all sources
 
@@ -51,7 +51,7 @@ class SourcesClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.sources.get(
+        client.sources.retrieve(
             source_id="source_id",
         )
         """
@@ -142,7 +142,7 @@ class SourcesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def update(
+    def modify(
         self,
         source_id: str,
         *,
@@ -186,7 +186,7 @@ class SourcesClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.sources.update(
+        client.sources.modify(
             source_id="source_id",
         )
         """
@@ -196,7 +196,7 @@ class SourcesClient:
             json={
                 "name": name,
                 "description": description,
-                "metadata_": metadata,
+                "metadata": metadata,
                 "embedding_config": convert_and_respect_annotation_metadata(
                     object_=embedding_config, annotation=EmbeddingConfig, direction="write"
                 ),
@@ -393,7 +393,7 @@ class SourcesClient:
                     object_=embedding_config, annotation=EmbeddingConfig, direction="write"
                 ),
                 "description": description,
-                "metadata_": metadata,
+                "metadata": metadata,
             },
             headers={
                 "content-type": "application/json",
@@ -425,136 +425,6 @@ class SourcesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def attach(
-        self, source_id: str, *, agent_id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> Source:
-        """
-        Attach a data source to an existing agent.
-
-        Parameters
-        ----------
-        source_id : str
-
-        agent_id : str
-            The unique identifier of the agent to attach the source to.
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        Source
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            token="YOUR_TOKEN",
-        )
-        client.sources.attach(
-            source_id="source_id",
-            agent_id="agent_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/sources/{jsonable_encoder(source_id)}/attach",
-            method="POST",
-            params={
-                "agent_id": agent_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Source,
-                    construct_type(
-                        type_=Source,  # 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 detach(
-        self, source_id: str, *, agent_id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> Source:
-        """
-        Detach a data source from an existing agent.
-
-        Parameters
-        ----------
-        source_id : str
-
-        agent_id : str
-            The unique identifier of the agent to detach the source from.
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        Source
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            token="YOUR_TOKEN",
-        )
-        client.sources.detach(
-            source_id="source_id",
-            agent_id="agent_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/sources/{jsonable_encoder(source_id)}/detach",
-            method="POST",
-            params={
-                "agent_id": agent_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Source,
-                    construct_type(
-                        type_=Source,  # 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)
-
 
 class AsyncSourcesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -562,7 +432,7 @@ class AsyncSourcesClient:
         self.files = AsyncFilesClient(client_wrapper=self._client_wrapper)
         self.passages = AsyncPassagesClient(client_wrapper=self._client_wrapper)
 
-    async def get(self, source_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Source:
+    async def retrieve(self, source_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Source:
         """
         Get all sources
 
@@ -590,7 +460,7 @@ class AsyncSourcesClient:
 
 
         async def main() -> None:
-            await client.sources.get(
+            await client.sources.retrieve(
                 source_id="source_id",
             )
 
@@ -692,7 +562,7 @@ class AsyncSourcesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def update(
+    async def modify(
         self,
         source_id: str,
         *,
@@ -741,7 +611,7 @@ class AsyncSourcesClient:
 
 
         async def main() -> None:
-            await client.sources.update(
+            await client.sources.modify(
                 source_id="source_id",
             )
 
@@ -754,7 +624,7 @@ class AsyncSourcesClient:
             json={
                 "name": name,
                 "description": description,
-                "metadata_": metadata,
+                "metadata": metadata,
                 "embedding_config": convert_and_respect_annotation_metadata(
                     object_=embedding_config, annotation=EmbeddingConfig, direction="write"
                 ),
@@ -975,7 +845,7 @@ class AsyncSourcesClient:
                     object_=embedding_config, annotation=EmbeddingConfig, direction="write"
                 ),
                 "description": description,
-                "metadata_": metadata,
+                "metadata": metadata,
             },
             headers={
                 "content-type": "application/json",
@@ -1006,149 +876,3 @@ class AsyncSourcesClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    async def attach(
-        self, source_id: str, *, agent_id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> Source:
-        """
-        Attach a data source to an existing agent.
-
-        Parameters
-        ----------
-        source_id : str
-
-        agent_id : str
-            The unique identifier of the agent to attach the source to.
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        Source
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.sources.attach(
-                source_id="source_id",
-                agent_id="agent_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/sources/{jsonable_encoder(source_id)}/attach",
-            method="POST",
-            params={
-                "agent_id": agent_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Source,
-                    construct_type(
-                        type_=Source,  # 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 detach(
-        self, source_id: str, *, agent_id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> Source:
-        """
-        Detach a data source from an existing agent.
-
-        Parameters
-        ----------
-        source_id : str
-
-        agent_id : str
-            The unique identifier of the agent to detach the source from.
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        Source
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.sources.detach(
-                source_id="source_id",
-                agent_id="agent_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/sources/{jsonable_encoder(source_id)}/detach",
-            method="POST",
-            params={
-                "agent_id": agent_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Source,
-                    construct_type(
-                        type_=Source,  # 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)

letta_client/sources/files/client.py

@@ -93,7 +93,7 @@ class FilesClient:
         source_id: str,
         *,
         limit: typing.Optional[int] = None,
-        cursor: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[FileMetadata]:
         """
@@ -106,7 +106,7 @@ class FilesClient:
         limit : typing.Optional[int]
             Number of files to return
 
-        cursor : typing.Optional[str]
+        after : typing.Optional[str]
             Pagination cursor to fetch the next set of results
 
         request_options : typing.Optional[RequestOptions]
@@ -133,7 +133,7 @@ class FilesClient:
             method="GET",
             params={
                 "limit": limit,
-                "cursor": cursor,
+                "after": after,
             },
             request_options=request_options,
         )
@@ -297,7 +297,7 @@ class AsyncFilesClient:
         source_id: str,
         *,
         limit: typing.Optional[int] = None,
-        cursor: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[FileMetadata]:
         """
@@ -310,7 +310,7 @@ class AsyncFilesClient:
         limit : typing.Optional[int]
             Number of files to return
 
-        cursor : typing.Optional[str]
+        after : typing.Optional[str]
             Pagination cursor to fetch the next set of results
 
         request_options : typing.Optional[RequestOptions]
@@ -345,7 +345,7 @@ class AsyncFilesClient:
             method="GET",
             params={
                 "limit": limit,
-                "cursor": cursor,
+                "after": after,
             },
             request_options=request_options,
         )