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

letta/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 → 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,8 +33,8 @@ class MemoryClient:
 
         Returns
         -------
-        Memory
-            Successful Response
+        MemoryVariablesGetResponse
+            200
 
         Examples
         --------
@@ -44,30 +43,30 @@ class MemoryClient:
         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,8 +98,8 @@ class AsyncMemoryClient:
 
         Returns
         -------
-        Memory
-            Successful Response
+        MemoryVariablesGetResponse
+            200
 
         Examples
         --------
@@ -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/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 → 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:

letta/agents/messages/client.py

@@ -5,7 +5,7 @@ from ...core.client_wrapper import SyncClientWrapper
 from ...core.request_options import RequestOptions
 from .types.messages_list_response import MessagesListResponse
 from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import parse_obj_as
+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
@@ -21,6 +21,7 @@ from ...types.letta_schemas_message_message import LettaSchemasMessageMessage
 from .types.letta_streaming_response import LettaStreamingResponse
 import httpx_sse
 import json
+from ...types.job import Job
 from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -99,7 +100,7 @@ class MessagesClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     MessagesListResponse,
-                    parse_obj_as(
+                    construct_type(
                         type_=MessagesListResponse,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -108,7 +109,7 @@ class MessagesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -124,6 +125,7 @@ class MessagesClient:
         agent_id: str,
         *,
         messages: typing.Sequence[MessageCreate],
+        use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -139,6 +141,9 @@ class MessagesClient:
         messages : typing.Sequence[MessageCreate]
             The messages to be sent to the agent.
 
+        use_assistant_message : typing.Optional[bool]
+            Whether the server should parse specific tool call arguments (default `send_message`) as `AssistantMessage` objects.
+
         assistant_message_tool_name : typing.Optional[str]
             The name of the designated message tool.
 
@@ -177,6 +182,7 @@ class MessagesClient:
                 "messages": convert_and_respect_annotation_metadata(
                     object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
                 ),
+                "use_assistant_message": use_assistant_message,
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
             },
@@ -187,7 +193,7 @@ class MessagesClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     LettaResponse,
-                    parse_obj_as(
+                    construct_type(
                         type_=LettaResponse,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -196,7 +202,7 @@ class MessagesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -287,7 +293,7 @@ class MessagesClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     LettaSchemasMessageMessage,
-                    parse_obj_as(
+                    construct_type(
                         type_=LettaSchemasMessageMessage,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -296,7 +302,7 @@ class MessagesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -312,6 +318,7 @@ class MessagesClient:
         agent_id: str,
         *,
         messages: typing.Sequence[MessageCreate],
+        use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         stream_tokens: typing.Optional[bool] = OMIT,
@@ -329,6 +336,9 @@ class MessagesClient:
         messages : typing.Sequence[MessageCreate]
             The messages to be sent to the agent.
 
+        use_assistant_message : typing.Optional[bool]
+            Whether the server should parse specific tool call arguments (default `send_message`) as `AssistantMessage` objects.
+
         assistant_message_tool_name : typing.Optional[str]
             The name of the designated message tool.
 
@@ -372,6 +382,7 @@ class MessagesClient:
                 "messages": convert_and_respect_annotation_metadata(
                     object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
                 ),
+                "use_assistant_message": use_assistant_message,
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "stream_tokens": stream_tokens,
@@ -389,7 +400,7 @@ class MessagesClient:
                         try:
                             yield typing.cast(
                                 LettaStreamingResponse,
-                                parse_obj_as(
+                                construct_type(
                                     type_=LettaStreamingResponse,  # type: ignore
                                     object_=json.loads(_sse.data),
                                 ),
@@ -402,7 +413,7 @@ class MessagesClient:
                     raise UnprocessableEntityError(
                         typing.cast(
                             HttpValidationError,
-                            parse_obj_as(
+                            construct_type(
                                 type_=HttpValidationError,  # type: ignore
                                 object_=_response.json(),
                             ),
@@ -413,6 +424,99 @@ class MessagesClient:
                 raise ApiError(status_code=_response.status_code, body=_response.text)
             raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def create_async(
+        self,
+        agent_id: str,
+        *,
+        messages: typing.Sequence[MessageCreate],
+        use_assistant_message: typing.Optional[bool] = OMIT,
+        assistant_message_tool_name: typing.Optional[str] = OMIT,
+        assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Job:
+        """
+        Asynchronously process a user message and return a job ID.
+        The actual processing happens in the background, and the status can be checked using the job ID.
+
+        Parameters
+        ----------
+        agent_id : str
+
+        messages : typing.Sequence[MessageCreate]
+            The messages to be sent to the agent.
+
+        use_assistant_message : typing.Optional[bool]
+            Whether the server should parse specific tool call arguments (default `send_message`) as `AssistantMessage` objects.
+
+        assistant_message_tool_name : typing.Optional[str]
+            The name of the designated message tool.
+
+        assistant_message_tool_kwarg : typing.Optional[str]
+            The name of the message argument in the designated message tool.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Job
+            Successful Response
+
+        Examples
+        --------
+        from letta import Letta, MessageCreate
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.agents.messages.create_async(
+            agent_id="agent_id",
+            messages=[
+                MessageCreate(
+                    role="user",
+                    text="text",
+                )
+            ],
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/agents/{jsonable_encoder(agent_id)}/messages/async",
+            method="POST",
+            json={
+                "messages": convert_and_respect_annotation_metadata(
+                    object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
+                ),
+                "use_assistant_message": use_assistant_message,
+                "assistant_message_tool_name": assistant_message_tool_name,
+                "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    Job,
+                    construct_type(
+                        type_=Job,  # 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 AsyncMessagesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -494,7 +598,7 @@ class AsyncMessagesClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     MessagesListResponse,
-                    parse_obj_as(
+                    construct_type(
                         type_=MessagesListResponse,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -503,7 +607,7 @@ class AsyncMessagesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -519,6 +623,7 @@ class AsyncMessagesClient:
         agent_id: str,
         *,
         messages: typing.Sequence[MessageCreate],
+        use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -534,6 +639,9 @@ class AsyncMessagesClient:
         messages : typing.Sequence[MessageCreate]
             The messages to be sent to the agent.
 
+        use_assistant_message : typing.Optional[bool]
+            Whether the server should parse specific tool call arguments (default `send_message`) as `AssistantMessage` objects.
+
         assistant_message_tool_name : typing.Optional[str]
             The name of the designated message tool.
 
@@ -580,6 +688,7 @@ class AsyncMessagesClient:
                 "messages": convert_and_respect_annotation_metadata(
                     object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
                 ),
+                "use_assistant_message": use_assistant_message,
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
             },
@@ -590,7 +699,7 @@ class AsyncMessagesClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     LettaResponse,
-                    parse_obj_as(
+                    construct_type(
                         type_=LettaResponse,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -599,7 +708,7 @@ class AsyncMessagesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -698,7 +807,7 @@ class AsyncMessagesClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     LettaSchemasMessageMessage,
-                    parse_obj_as(
+                    construct_type(
                         type_=LettaSchemasMessageMessage,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -707,7 +816,7 @@ class AsyncMessagesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -723,6 +832,7 @@ class AsyncMessagesClient:
         agent_id: str,
         *,
         messages: typing.Sequence[MessageCreate],
+        use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         stream_tokens: typing.Optional[bool] = OMIT,
@@ -740,6 +850,9 @@ class AsyncMessagesClient:
         messages : typing.Sequence[MessageCreate]
             The messages to be sent to the agent.
 
+        use_assistant_message : typing.Optional[bool]
+            Whether the server should parse specific tool call arguments (default `send_message`) as `AssistantMessage` objects.
+
         assistant_message_tool_name : typing.Optional[str]
             The name of the designated message tool.
 
@@ -791,6 +904,7 @@ class AsyncMessagesClient:
                 "messages": convert_and_respect_annotation_metadata(
                     object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
                 ),
+                "use_assistant_message": use_assistant_message,
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "stream_tokens": stream_tokens,
@@ -808,7 +922,7 @@ class AsyncMessagesClient:
                         try:
                             yield typing.cast(
                                 LettaStreamingResponse,
-                                parse_obj_as(
+                                construct_type(
                                     type_=LettaStreamingResponse,  # type: ignore
                                     object_=json.loads(_sse.data),
                                 ),
@@ -821,7 +935,7 @@ class AsyncMessagesClient:
                     raise UnprocessableEntityError(
                         typing.cast(
                             HttpValidationError,
-                            parse_obj_as(
+                            construct_type(
                                 type_=HttpValidationError,  # type: ignore
                                 object_=_response.json(),
                             ),
@@ -831,3 +945,104 @@ class AsyncMessagesClient:
             except JSONDecodeError:
                 raise ApiError(status_code=_response.status_code, body=_response.text)
             raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def create_async(
+        self,
+        agent_id: str,
+        *,
+        messages: typing.Sequence[MessageCreate],
+        use_assistant_message: typing.Optional[bool] = OMIT,
+        assistant_message_tool_name: typing.Optional[str] = OMIT,
+        assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Job:
+        """
+        Asynchronously process a user message and return a job ID.
+        The actual processing happens in the background, and the status can be checked using the job ID.
+
+        Parameters
+        ----------
+        agent_id : str
+
+        messages : typing.Sequence[MessageCreate]
+            The messages to be sent to the agent.
+
+        use_assistant_message : typing.Optional[bool]
+            Whether the server should parse specific tool call arguments (default `send_message`) as `AssistantMessage` objects.
+
+        assistant_message_tool_name : typing.Optional[str]
+            The name of the designated message tool.
+
+        assistant_message_tool_kwarg : typing.Optional[str]
+            The name of the message argument in the designated message tool.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Job
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta import AsyncLetta, MessageCreate
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.agents.messages.create_async(
+                agent_id="agent_id",
+                messages=[
+                    MessageCreate(
+                        role="user",
+                        text="text",
+                    )
+                ],
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/agents/{jsonable_encoder(agent_id)}/messages/async",
+            method="POST",
+            json={
+                "messages": convert_and_respect_annotation_metadata(
+                    object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
+                ),
+                "use_assistant_message": use_assistant_message,
+                "assistant_message_tool_name": assistant_message_tool_name,
+                "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    Job,
+                    construct_type(
+                        type_=Job,  # 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)