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

letta_client/agents/core_memory/client.py

@@ -11,6 +11,7 @@ from ...types.http_validation_error import HttpValidationError
 from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
 from ...types.block import Block
+from ...types.agent_state import AgentState
 from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -21,7 +22,7 @@ class CoreMemoryClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def get(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Memory:
+    def retrieve(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Memory:
         """
         Retrieve the memory state of a specific agent.
         This endpoint fetches the current memory state of the agent identified by the user ID and agent ID.
@@ -45,12 +46,12 @@ class CoreMemoryClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.core_memory.get(
+        client.agents.core_memory.retrieve(
             agent_id="agent_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory",
             method="GET",
             request_options=request_options,
         )
@@ -78,7 +79,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_block(
+    def retrieve_block(
         self, agent_id: str, block_label: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> Block:
         """
@@ -105,13 +106,13 @@ class CoreMemoryClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.core_memory.get_block(
+        client.agents.core_memory.retrieve_block(
             agent_id="agent_id",
             block_label="block_label",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/blocks/{jsonable_encoder(block_label)}",
             method="GET",
             request_options=request_options,
         )
@@ -139,68 +140,7 @@ class CoreMemoryClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def remove_block(
-        self, agent_id: str, block_label: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> Memory:
-        """
-        Removes a memory block from an agent by unlnking it. If the block is not linked to any other agent, it is deleted.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        block_label : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        Memory
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            token="YOUR_TOKEN",
-        )
-        client.agents.core_memory.remove_block(
-            agent_id="agent_id",
-            block_label="block_label",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
-            method="DELETE",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Memory,
-                    construct_type(
-                        type_=Memory,  # 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 update_block(
+    def modify_block(
         self,
         agent_id: str,
         block_label: str,
@@ -215,7 +155,7 @@ class CoreMemoryClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Block:
         """
-        Removes a memory block from an agent by unlnking it. If the block is not linked to any other agent, it is deleted.
+        Updates a memory block of an agent.
 
         Parameters
         ----------
@@ -259,13 +199,13 @@ class CoreMemoryClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.core_memory.update_block(
+        client.agents.core_memory.modify_block(
             agent_id="agent_id",
             block_label="block_label",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/blocks/{jsonable_encoder(block_label)}",
             method="PATCH",
             json={
                 "value": value,
@@ -274,7 +214,7 @@ class CoreMemoryClient:
                 "is_template": is_template,
                 "label": label,
                 "description": description,
-                "metadata_": metadata,
+                "metadata": metadata,
             },
             request_options=request_options,
             omit=OMIT,
@@ -303,7 +243,9 @@ class CoreMemoryClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def list(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[Block]:
+    def list_blocks(
+        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[Block]:
         """
         Retrieve the memory blocks of a specific agent.
 
@@ -326,12 +268,12 @@ class CoreMemoryClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.core_memory.list(
+        client.agents.core_memory.list_blocks(
             agent_id="agent_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/blocks",
             method="GET",
             request_options=request_options,
         )
@@ -359,52 +301,24 @@ class CoreMemoryClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def add_block(
-        self,
-        agent_id: str,
-        *,
-        value: str,
-        label: str,
-        limit: typing.Optional[int] = OMIT,
-        name: typing.Optional[str] = OMIT,
-        is_template: typing.Optional[bool] = OMIT,
-        description: typing.Optional[str] = OMIT,
-        metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> Memory:
+    def attach_block(
+        self, agent_id: str, block_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AgentState:
         """
-        Creates a memory block and links it to the agent.
+        Attach a block to an agent.
 
         Parameters
         ----------
         agent_id : str
 
-        value : str
-            Value of the block.
-
-        label : str
-            Label of the block.
-
-        limit : typing.Optional[int]
-            Character limit of the block.
-
-        name : typing.Optional[str]
-            Name of the block if it is a template.
-
-        is_template : typing.Optional[bool]
-
-        description : typing.Optional[str]
-            Description of the block.
-
-        metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-            Metadata of the block.
+        block_id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        Memory
+        AgentState
             Successful Response
 
         Examples
@@ -414,33 +328,22 @@ class CoreMemoryClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.core_memory.add_block(
+        client.agents.core_memory.attach_block(
             agent_id="agent_id",
-            value="value",
-            label="label",
+            block_id="block_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks",
-            method="POST",
-            json={
-                "value": value,
-                "limit": limit,
-                "name": name,
-                "is_template": is_template,
-                "label": label,
-                "description": description,
-                "metadata_": metadata,
-            },
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/blocks/attach/{jsonable_encoder(block_id)}",
+            method="PATCH",
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    Memory,
+                    AgentState,
                     construct_type(
-                        type_=Memory,  # type: ignore
+                        type_=AgentState,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -459,58 +362,49 @@ class CoreMemoryClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-
-class AsyncCoreMemoryClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def get(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Memory:
+    def detach_block(
+        self, agent_id: str, block_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AgentState:
         """
-        Retrieve the memory state of a specific agent.
-        This endpoint fetches the current memory state of the agent identified by the user ID and agent ID.
+        Detach a block from an agent.
 
         Parameters
         ----------
         agent_id : str
 
+        block_id : str
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        Memory
+        AgentState
             Successful Response
 
         Examples
         --------
-        import asyncio
-
-        from letta_client import AsyncLetta
+        from letta_client import Letta
 
-        client = AsyncLetta(
+        client = Letta(
             token="YOUR_TOKEN",
         )
-
-
-        async def main() -> None:
-            await client.agents.core_memory.get(
-                agent_id="agent_id",
-            )
-
-
-        asyncio.run(main())
+        client.agents.core_memory.detach_block(
+            agent_id="agent_id",
+            block_id="block_id",
+        )
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory",
-            method="GET",
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/blocks/detach/{jsonable_encoder(block_id)}",
+            method="PATCH",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    Memory,
+                    AgentState,
                     construct_type(
-                        type_=Memory,  # type: ignore
+                        type_=AgentState,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -529,24 +423,26 @@ 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_block(
-        self, agent_id: str, block_label: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> Block:
+
+class AsyncCoreMemoryClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def retrieve(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Memory:
         """
-        Retrieve a memory block from an agent.
+        Retrieve the memory state of a specific agent.
+        This endpoint fetches the current memory state of the agent identified by the user ID and agent ID.
 
         Parameters
         ----------
         agent_id : str
 
-        block_label : str
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        Block
+        Memory
             Successful Response
 
         Examples
@@ -561,25 +457,24 @@ class AsyncCoreMemoryClient:
 
 
         async def main() -> None:
-            await client.agents.core_memory.get_block(
+            await client.agents.core_memory.retrieve(
                 agent_id="agent_id",
-                block_label="block_label",
             )
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory",
             method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    Block,
+                    Memory,
                     construct_type(
-                        type_=Block,  # type: ignore
+                        type_=Memory,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -598,11 +493,11 @@ class AsyncCoreMemoryClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def remove_block(
+    async def retrieve_block(
         self, agent_id: str, block_label: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> Memory:
+    ) -> Block:
         """
-        Removes a memory block from an agent by unlnking it. If the block is not linked to any other agent, it is deleted.
+        Retrieve a memory block from an agent.
 
         Parameters
         ----------
@@ -615,7 +510,7 @@ class AsyncCoreMemoryClient:
 
         Returns
         -------
-        Memory
+        Block
             Successful Response
 
         Examples
@@ -630,7 +525,7 @@ class AsyncCoreMemoryClient:
 
 
         async def main() -> None:
-            await client.agents.core_memory.remove_block(
+            await client.agents.core_memory.retrieve_block(
                 agent_id="agent_id",
                 block_label="block_label",
             )
@@ -639,16 +534,16 @@ class AsyncCoreMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
-            method="DELETE",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/blocks/{jsonable_encoder(block_label)}",
+            method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    Memory,
+                    Block,
                     construct_type(
-                        type_=Memory,  # type: ignore
+                        type_=Block,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -667,7 +562,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 update_block(
+    async def modify_block(
         self,
         agent_id: str,
         block_label: str,
@@ -682,7 +577,7 @@ class AsyncCoreMemoryClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Block:
         """
-        Removes a memory block from an agent by unlnking it. If the block is not linked to any other agent, it is deleted.
+        Updates a memory block of an agent.
 
         Parameters
         ----------
@@ -731,7 +626,7 @@ class AsyncCoreMemoryClient:
 
 
         async def main() -> None:
-            await client.agents.core_memory.update_block(
+            await client.agents.core_memory.modify_block(
                 agent_id="agent_id",
                 block_label="block_label",
             )
@@ -740,7 +635,7 @@ class AsyncCoreMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks/{jsonable_encoder(block_label)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/blocks/{jsonable_encoder(block_label)}",
             method="PATCH",
             json={
                 "value": value,
@@ -749,7 +644,7 @@ class AsyncCoreMemoryClient:
                 "is_template": is_template,
                 "label": label,
                 "description": description,
-                "metadata_": metadata,
+                "metadata": metadata,
             },
             request_options=request_options,
             omit=OMIT,
@@ -778,7 +673,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 list(
+    async def list_blocks(
         self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[Block]:
         """
@@ -808,7 +703,7 @@ class AsyncCoreMemoryClient:
 
 
         async def main() -> None:
-            await client.agents.core_memory.list(
+            await client.agents.core_memory.list_blocks(
                 agent_id="agent_id",
             )
 
@@ -816,7 +711,7 @@ class AsyncCoreMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/blocks",
             method="GET",
             request_options=request_options,
         )
@@ -844,52 +739,93 @@ class AsyncCoreMemoryClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def add_block(
-        self,
-        agent_id: str,
-        *,
-        value: str,
-        label: str,
-        limit: typing.Optional[int] = OMIT,
-        name: typing.Optional[str] = OMIT,
-        is_template: typing.Optional[bool] = OMIT,
-        description: typing.Optional[str] = OMIT,
-        metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> Memory:
+    async def attach_block(
+        self, agent_id: str, block_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AgentState:
         """
-        Creates a memory block and links it to the agent.
+        Attach a block to an agent.
 
         Parameters
         ----------
         agent_id : str
 
-        value : str
-            Value of the block.
+        block_id : str
 
-        label : str
-            Label of the block.
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
 
-        limit : typing.Optional[int]
-            Character limit of the block.
+        Returns
+        -------
+        AgentState
+            Successful Response
 
-        name : typing.Optional[str]
-            Name of the block if it is a template.
+        Examples
+        --------
+        import asyncio
 
-        is_template : typing.Optional[bool]
+        from letta_client import AsyncLetta
 
-        description : typing.Optional[str]
-            Description of the block.
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
 
-        metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-            Metadata of the block.
+
+        async def main() -> None:
+            await client.agents.core_memory.attach_block(
+                agent_id="agent_id",
+                block_id="block_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/blocks/attach/{jsonable_encoder(block_id)}",
+            method="PATCH",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    AgentState,
+                    construct_type(
+                        type_=AgentState,  # 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_block(
+        self, agent_id: str, block_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AgentState:
+        """
+        Detach a block from an agent.
+
+        Parameters
+        ----------
+        agent_id : str
+
+        block_id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        Memory
+        AgentState
             Successful Response
 
         Examples
@@ -904,36 +840,25 @@ class AsyncCoreMemoryClient:
 
 
         async def main() -> None:
-            await client.agents.core_memory.add_block(
+            await client.agents.core_memory.detach_block(
                 agent_id="agent_id",
-                value="value",
-                label="label",
+                block_id="block_id",
             )
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/blocks",
-            method="POST",
-            json={
-                "value": value,
-                "limit": limit,
-                "name": name,
-                "is_template": is_template,
-                "label": label,
-                "description": description,
-                "metadata_": metadata,
-            },
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/blocks/detach/{jsonable_encoder(block_id)}",
+            method="PATCH",
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    Memory,
+                    AgentState,
                     construct_type(
-                        type_=Memory,  # type: ignore
+                        type_=AgentState,  # type: ignore
                         object_=_response.json(),
                     ),
                 )