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

letta_client/agents/core_memory/client.py

@@ -3,14 +3,13 @@
 import typing
 from ...core.client_wrapper import SyncClientWrapper
 from ...core.request_options import RequestOptions
-from ...types.letta_schemas_message_message import LettaSchemasMessageMessage
+from ...types.memory import Memory
 from ...core.jsonable_encoder import jsonable_encoder
 from ...core.unchecked_base_model import construct_type
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.http_validation_error import HttpValidationError
 from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
-from ...types.memory import Memory
 from ...types.block import Block
 from ...core.client_wrapper import AsyncClientWrapper
 
@@ -22,64 +21,6 @@ class CoreMemoryClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def list_in_context(
-        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[LettaSchemasMessageMessage]:
-        """
-        Retrieve the messages in the context of a specific agent.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[LettaSchemasMessageMessage]
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            token="YOUR_TOKEN",
-        )
-        client.agents.core_memory.list_in_context(
-            agent_id="agent_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/messages",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[LettaSchemasMessageMessage],
-                    construct_type(
-                        type_=typing.List[LettaSchemasMessageMessage],  # 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 get(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Memory:
         """
         Retrieve the memory state of a specific agent.
@@ -109,7 +50,7 @@ class CoreMemoryClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory",
             method="GET",
             request_options=request_options,
         )
@@ -170,7 +111,7 @@ class CoreMemoryClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/block/{jsonable_encoder(block_label)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
             method="GET",
             request_options=request_options,
         )
@@ -231,7 +172,7 @@ class CoreMemoryClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/block/{jsonable_encoder(block_label)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
             method="DELETE",
             request_options=request_options,
         )
@@ -324,7 +265,7 @@ class CoreMemoryClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/block/{jsonable_encoder(block_label)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
             method="PATCH",
             json={
                 "value": value,
@@ -362,9 +303,7 @@ class CoreMemoryClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_blocks(
-        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[Block]:
+    def list(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[Block]:
         """
         Retrieve the memory blocks of a specific agent.
 
@@ -387,12 +326,12 @@ class CoreMemoryClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.core_memory.get_blocks(
+        client.agents.core_memory.list(
             agent_id="agent_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/block",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks",
             method="GET",
             request_options=request_options,
         )
@@ -482,7 +421,7 @@ class CoreMemoryClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/block",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks",
             method="POST",
             json={
                 "value": value,
@@ -525,72 +464,6 @@ class AsyncCoreMemoryClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def list_in_context(
-        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[LettaSchemasMessageMessage]:
-        """
-        Retrieve the messages in the context of a specific agent.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[LettaSchemasMessageMessage]
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.core_memory.list_in_context(
-                agent_id="agent_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/messages",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[LettaSchemasMessageMessage],
-                    construct_type(
-                        type_=typing.List[LettaSchemasMessageMessage],  # 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 get(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Memory:
         """
         Retrieve the memory state of a specific agent.
@@ -628,7 +501,7 @@ class AsyncCoreMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory",
             method="GET",
             request_options=request_options,
         )
@@ -697,7 +570,7 @@ class AsyncCoreMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/block/{jsonable_encoder(block_label)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
             method="GET",
             request_options=request_options,
         )
@@ -766,7 +639,7 @@ class AsyncCoreMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/block/{jsonable_encoder(block_label)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
             method="DELETE",
             request_options=request_options,
         )
@@ -867,7 +740,7 @@ class AsyncCoreMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/block/{jsonable_encoder(block_label)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
             method="PATCH",
             json={
                 "value": value,
@@ -905,7 +778,7 @@ class AsyncCoreMemoryClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_blocks(
+    async def list(
         self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[Block]:
         """
@@ -935,7 +808,7 @@ class AsyncCoreMemoryClient:
 
 
         async def main() -> None:
-            await client.agents.core_memory.get_blocks(
+            await client.agents.core_memory.list(
                 agent_id="agent_id",
             )
 
@@ -943,7 +816,7 @@ class AsyncCoreMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/block",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks",
             method="GET",
             request_options=request_options,
         )
@@ -1041,7 +914,7 @@ class AsyncCoreMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory/block",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks",
             method="POST",
             json={
                 "value": value,

letta_client/agents/memory_variables/client.py

@@ -48,7 +48,7 @@ class MemoryVariablesClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/variables",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/variables",
             method="GET",
             request_options=request_options,
         )
@@ -121,7 +121,7 @@ class AsyncMemoryVariablesClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/variables",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/variables",
             method="GET",
             request_options=request_options,
         )

letta_client/agents/messages/__init__.py

@@ -10,13 +10,6 @@ from .types import (
     LettaStreamingResponse_UsageStatistics,
     LettaStreamingResponse_UserMessage,
     MessagesListResponse,
-    MessagesListResponseItem,
-    MessagesListResponseItem_AssistantMessage,
-    MessagesListResponseItem_ReasoningMessage,
-    MessagesListResponseItem_SystemMessage,
-    MessagesListResponseItem_ToolCallMessage,
-    MessagesListResponseItem_ToolReturnMessage,
-    MessagesListResponseItem_UserMessage,
 )
 
 __all__ = [
@@ -29,11 +22,4 @@ __all__ = [
     "LettaStreamingResponse_UsageStatistics",
     "LettaStreamingResponse_UserMessage",
     "MessagesListResponse",
-    "MessagesListResponseItem",
-    "MessagesListResponseItem_AssistantMessage",
-    "MessagesListResponseItem_ReasoningMessage",
-    "MessagesListResponseItem_SystemMessage",
-    "MessagesListResponseItem_ToolCallMessage",
-    "MessagesListResponseItem_ToolReturnMessage",
-    "MessagesListResponseItem_UserMessage",
 ]

letta_client/agents/messages/client.py

@@ -306,7 +306,7 @@ class MessagesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def stream(
+    def create_stream(
         self,
         agent_id: str,
         *,
@@ -348,7 +348,7 @@ class MessagesClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        response = client.agents.messages.stream(
+        response = client.agents.messages.create_stream(
             agent_id="agent_id",
             messages=[
                 MessageCreate(
@@ -796,7 +796,7 @@ class AsyncMessagesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def stream(
+    async def create_stream(
         self,
         agent_id: str,
         *,
@@ -843,7 +843,7 @@ class AsyncMessagesClient:
 
 
         async def main() -> None:
-            response = await client.agents.messages.stream(
+            response = await client.agents.messages.create_stream(
                 agent_id="agent_id",
                 messages=[
                     MessageCreate(

letta_client/agents/messages/types/__init__.py

@@ -11,15 +11,6 @@ from .letta_streaming_response import (
     LettaStreamingResponse_UserMessage,
 )
 from .messages_list_response import MessagesListResponse
-from .messages_list_response_item import (
-    MessagesListResponseItem,
-    MessagesListResponseItem_AssistantMessage,
-    MessagesListResponseItem_ReasoningMessage,
-    MessagesListResponseItem_SystemMessage,
-    MessagesListResponseItem_ToolCallMessage,
-    MessagesListResponseItem_ToolReturnMessage,
-    MessagesListResponseItem_UserMessage,
-)
 
 __all__ = [
     "LettaStreamingResponse",
@@ -31,11 +22,4 @@ __all__ = [
     "LettaStreamingResponse_UsageStatistics",
     "LettaStreamingResponse_UserMessage",
     "MessagesListResponse",
-    "MessagesListResponseItem",
-    "MessagesListResponseItem_AssistantMessage",
-    "MessagesListResponseItem_ReasoningMessage",
-    "MessagesListResponseItem_SystemMessage",
-    "MessagesListResponseItem_ToolCallMessage",
-    "MessagesListResponseItem_ToolReturnMessage",
-    "MessagesListResponseItem_UserMessage",
 ]

letta_client/agents/messages/types/messages_list_response.py

@@ -2,6 +2,6 @@
 
 import typing
 from ....types.letta_schemas_message_message import LettaSchemasMessageMessage
-from .messages_list_response_item import MessagesListResponseItem
+from ....types.letta_message_union import LettaMessageUnion
 
-MessagesListResponse = typing.Union[typing.List[LettaSchemasMessageMessage], typing.List[MessagesListResponseItem]]
+MessagesListResponse = typing.Union[typing.List[LettaSchemasMessageMessage], typing.List[LettaMessageUnion]]

letta_client/core/client_wrapper.py

@@ -16,7 +16,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "letta-client",
-            "X-Fern-SDK-Version": "0.1.15",
+            "X-Fern-SDK-Version": "0.1.17",
         }
         if self.token is not None:
             headers["Authorization"] = f"Bearer {self.token}"