letta-client 0.1.6__py3-none-any.whl → 0.1.10__py3-none-any.whl

letta_client/agents/memory_variables/__init__.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .types import MemoryVariablesGetResponse
+
+__all__ = ["MemoryVariablesGetResponse"]

{letta/agents/memory → letta_client/agents/memory_variables}/client.py

@@ -1,29 +1,28 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ...core.client_wrapper import SyncClientWrapper
-from .messages.client import MessagesClient
 import typing
 from ...core.request_options import RequestOptions
-from ...types.memory import Memory
+from .types.memory_variables_get_response import MemoryVariablesGetResponse
 from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import parse_obj_as
-from ...errors.unprocessable_entity_error import UnprocessableEntityError
-from ...types.http_validation_error import HttpValidationError
+from ...core.unchecked_base_model import construct_type
+from ...errors.not_found_error import NotFoundError
 from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper
-from .messages.client import AsyncMessagesClient
 
 
-class MemoryClient:
+class MemoryVariablesClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
-        self.messages = MessagesClient(client_wrapper=self._client_wrapper)
 
-    def get(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Memory:
+    def get(
+        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> MemoryVariablesGetResponse:
         """
-        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.
+        <Note>This endpoint is only available on Letta Cloud.</Note>
+
+        Returns the memory variables associated with an agent.
 
         Parameters
         ----------
@@ -34,40 +33,40 @@ class MemoryClient:
 
         Returns
         -------
-        Memory
-            Successful Response
+        MemoryVariablesGetResponse
+            200
 
         Examples
         --------
-        from letta import Letta
+        from letta_client import Letta
 
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.memory.get(
+        client.agents.memory_variables.get(
             agent_id="agent_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/memory",
+            f"v1/agents/{jsonable_encoder(agent_id)}/variables",
             method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    Memory,
-                    parse_obj_as(
-                        type_=Memory,  # type: ignore
+                    MemoryVariablesGetResponse,
+                    construct_type(
+                        type_=MemoryVariablesGetResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
+            if _response.status_code == 404:
+                raise NotFoundError(
                     typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
+                        typing.Optional[typing.Any],
+                        construct_type(
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     )
@@ -78,15 +77,17 @@ class MemoryClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
 
-class AsyncMemoryClient:
+class AsyncMemoryVariablesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
-        self.messages = AsyncMessagesClient(client_wrapper=self._client_wrapper)
 
-    async def get(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Memory:
+    async def get(
+        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> MemoryVariablesGetResponse:
         """
-        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.
+        <Note>This endpoint is only available on Letta Cloud.</Note>
+
+        Returns the memory variables associated with an agent.
 
         Parameters
         ----------
@@ -97,14 +98,14 @@ class AsyncMemoryClient:
 
         Returns
         -------
-        Memory
-            Successful Response
+        MemoryVariablesGetResponse
+            200
 
         Examples
         --------
         import asyncio
 
-        from letta import AsyncLetta
+        from letta_client import AsyncLetta
 
         client = AsyncLetta(
             token="YOUR_TOKEN",
@@ -112,7 +113,7 @@ class AsyncMemoryClient:
 
 
         async def main() -> None:
-            await client.agents.memory.get(
+            await client.agents.memory_variables.get(
                 agent_id="agent_id",
             )
 
@@ -120,25 +121,25 @@ class AsyncMemoryClient:
         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)}/variables",
             method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    Memory,
-                    parse_obj_as(
-                        type_=Memory,  # type: ignore
+                    MemoryVariablesGetResponse,
+                    construct_type(
+                        type_=MemoryVariablesGetResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
+            if _response.status_code == 404:
+                raise NotFoundError(
                     typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
+                        typing.Optional[typing.Any],
+                        construct_type(
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     )

letta_client/agents/memory_variables/types/__init__.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .memory_variables_get_response import MemoryVariablesGetResponse
+
+__all__ = ["MemoryVariablesGetResponse"]

letta/agents/types/agents_get_agent_variables_response.py → letta_client/agents/memory_variables/types/memory_variables_get_response.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ...core.pydantic_utilities import UniversalBaseModel
+from ....core.unchecked_base_model import UncheckedBaseModel
 import typing
-from ...core.pydantic_utilities import IS_PYDANTIC_V2
+from ....core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class AgentsGetAgentVariablesResponse(UniversalBaseModel):
+class MemoryVariablesGetResponse(UncheckedBaseModel):
     variables: typing.Dict[str, str]
 
     if IS_PYDANTIC_V2: