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

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.
 
@@ -345,6 +355,25 @@ class MessagesClient:
         ------
         typing.Iterator[LettaStreamingResponse]
             Successful response
+
+        Examples
+        --------
+        from letta import Letta, MessageCreate
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        response = client.agents.messages.stream(
+            agent_id="agent_id",
+            messages=[
+                MessageCreate(
+                    role="user",
+                    text="text",
+                )
+            ],
+        )
+        for chunk in response:
+            yield chunk
         """
         with self._client_wrapper.httpx_client.stream(
             f"v1/agents/{jsonable_encoder(agent_id)}/messages/stream",
@@ -353,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,
@@ -370,7 +400,7 @@ class MessagesClient:
                         try:
                             yield typing.cast(
                                 LettaStreamingResponse,
-                                parse_obj_as(
+                                construct_type(
                                     type_=LettaStreamingResponse,  # type: ignore
                                     object_=json.loads(_sse.data),
                                 ),
@@ -383,7 +413,7 @@ class MessagesClient:
                     raise UnprocessableEntityError(
                         typing.cast(
                             HttpValidationError,
-                            parse_obj_as(
+                            construct_type(
                                 type_=HttpValidationError,  # type: ignore
                                 object_=_response.json(),
                             ),
@@ -394,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):
@@ -475,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(),
                     ),
@@ -484,7 +607,7 @@ class AsyncMessagesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -500,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,
@@ -515,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.
 
@@ -561,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,
             },
@@ -571,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(),
                     ),
@@ -580,7 +708,7 @@ class AsyncMessagesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -679,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(),
                     ),
@@ -688,7 +816,7 @@ class AsyncMessagesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -704,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,
@@ -721,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.
 
@@ -737,6 +869,33 @@ class AsyncMessagesClient:
         ------
         typing.AsyncIterator[LettaStreamingResponse]
             Successful response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta import AsyncLetta, MessageCreate
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            response = await client.agents.messages.stream(
+                agent_id="agent_id",
+                messages=[
+                    MessageCreate(
+                        role="user",
+                        text="text",
+                    )
+                ],
+            )
+            async for chunk in response:
+                yield chunk
+
+
+        asyncio.run(main())
         """
         async with self._client_wrapper.httpx_client.stream(
             f"v1/agents/{jsonable_encoder(agent_id)}/messages/stream",
@@ -745,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,
@@ -762,7 +922,7 @@ class AsyncMessagesClient:
                         try:
                             yield typing.cast(
                                 LettaStreamingResponse,
-                                parse_obj_as(
+                                construct_type(
                                     type_=LettaStreamingResponse,  # type: ignore
                                     object_=json.loads(_sse.data),
                                 ),
@@ -775,7 +935,7 @@ class AsyncMessagesClient:
                     raise UnprocessableEntityError(
                         typing.cast(
                             HttpValidationError,
-                            parse_obj_as(
+                            construct_type(
                                 type_=HttpValidationError,  # type: ignore
                                 object_=_response.json(),
                             ),
@@ -785,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)

letta/agents/messages/types/__init__.py

@@ -1,7 +1,41 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .letta_streaming_response import LettaStreamingResponse
+from .letta_streaming_response import (
+    LettaStreamingResponse,
+    LettaStreamingResponse_AssistantMessage,
+    LettaStreamingResponse_ReasoningMessage,
+    LettaStreamingResponse_SystemMessage,
+    LettaStreamingResponse_ToolCallMessage,
+    LettaStreamingResponse_ToolReturnMessage,
+    LettaStreamingResponse_UsageStatistics,
+    LettaStreamingResponse_UserMessage,
+)
 from .messages_list_response import MessagesListResponse
-from .messages_list_response_item import MessagesListResponseItem
+from .messages_list_response_item import (
+    MessagesListResponseItem,
+    MessagesListResponseItem_AssistantMessage,
+    MessagesListResponseItem_ReasoningMessage,
+    MessagesListResponseItem_SystemMessage,
+    MessagesListResponseItem_ToolCallMessage,
+    MessagesListResponseItem_ToolReturnMessage,
+    MessagesListResponseItem_UserMessage,
+)
 
-__all__ = ["LettaStreamingResponse", "MessagesListResponse", "MessagesListResponseItem"]
+__all__ = [
+    "LettaStreamingResponse",
+    "LettaStreamingResponse_AssistantMessage",
+    "LettaStreamingResponse_ReasoningMessage",
+    "LettaStreamingResponse_SystemMessage",
+    "LettaStreamingResponse_ToolCallMessage",
+    "LettaStreamingResponse_ToolReturnMessage",
+    "LettaStreamingResponse_UsageStatistics",
+    "LettaStreamingResponse_UserMessage",
+    "MessagesListResponse",
+    "MessagesListResponseItem",
+    "MessagesListResponseItem_AssistantMessage",
+    "MessagesListResponseItem_ReasoningMessage",
+    "MessagesListResponseItem_SystemMessage",
+    "MessagesListResponseItem_ToolCallMessage",
+    "MessagesListResponseItem_ToolReturnMessage",
+    "MessagesListResponseItem_UserMessage",
+]