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

{letta → letta_client}/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.run import Run
 from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -74,7 +75,7 @@ class MessagesClient:
 
         Examples
         --------
-        from letta import Letta
+        from letta_client import Letta
 
         client = Letta(
             token="YOUR_TOKEN",
@@ -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.
 
@@ -155,7 +160,7 @@ class MessagesClient:
 
         Examples
         --------
-        from letta import Letta, MessageCreate
+        from letta_client import Letta, MessageCreate
 
         client = Letta(
             token="YOUR_TOKEN",
@@ -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(),
                         ),
@@ -253,7 +259,7 @@ class MessagesClient:
 
         Examples
         --------
-        from letta import Letta
+        from letta_client import Letta
 
         client = Letta(
             token="YOUR_TOKEN",
@@ -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.
 
@@ -348,7 +358,7 @@ class MessagesClient:
 
         Examples
         --------
-        from letta import Letta, MessageCreate
+        from letta_client import Letta, MessageCreate
 
         client = Letta(
             token="YOUR_TOKEN",
@@ -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,
+    ) -> Run:
+        """
+        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
+        -------
+        Run
+            Successful Response
+
+        Examples
+        --------
+        from letta_client 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(
+                    Run,
+                    construct_type(
+                        type_=Run,  # 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):
@@ -463,7 +567,7 @@ class AsyncMessagesClient:
         --------
         import asyncio
 
-        from letta import AsyncLetta
+        from letta_client import AsyncLetta
 
         client = AsyncLetta(
             token="YOUR_TOKEN",
@@ -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.
 
@@ -552,7 +660,7 @@ class AsyncMessagesClient:
         --------
         import asyncio
 
-        from letta import AsyncLetta, MessageCreate
+        from letta_client import AsyncLetta, MessageCreate
 
         client = AsyncLetta(
             token="YOUR_TOKEN",
@@ -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(),
                         ),
@@ -658,7 +767,7 @@ class AsyncMessagesClient:
         --------
         import asyncio
 
-        from letta import AsyncLetta
+        from letta_client import AsyncLetta
 
         client = AsyncLetta(
             token="YOUR_TOKEN",
@@ -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.
 
@@ -761,7 +874,7 @@ class AsyncMessagesClient:
         --------
         import asyncio
 
-        from letta import AsyncLetta, MessageCreate
+        from letta_client import AsyncLetta, MessageCreate
 
         client = AsyncLetta(
             token="YOUR_TOKEN",
@@ -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,
+    ) -> Run:
+        """
+        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
+        -------
+        Run
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client 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(
+                    Run,
+                    construct_type(
+                        type_=Run,  # 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 → letta_client}/agents/messages/types/letta_streaming_response.py

@@ -1,16 +1,18 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ....core.pydantic_utilities import UniversalBaseModel
+from ....core.unchecked_base_model import UncheckedBaseModel
 import typing
 import datetime as dt
 from ....core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 from ....types.tool_call_message_tool_call import ToolCallMessageToolCall
 from ....types.tool_return_message_status import ToolReturnMessageStatus
+import typing_extensions
+from ....core.unchecked_base_model import UnionMetadata
 
 
-class LettaStreamingResponse_SystemMessage(UniversalBaseModel):
+class LettaStreamingResponse_SystemMessage(UncheckedBaseModel):
     message_type: typing.Literal["system_message"] = "system_message"
     id: str
     date: dt.datetime
@@ -26,7 +28,7 @@ class LettaStreamingResponse_SystemMessage(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-class LettaStreamingResponse_UserMessage(UniversalBaseModel):
+class LettaStreamingResponse_UserMessage(UncheckedBaseModel):
     message_type: typing.Literal["user_message"] = "user_message"
     id: str
     date: dt.datetime
@@ -42,7 +44,7 @@ class LettaStreamingResponse_UserMessage(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-class LettaStreamingResponse_ReasoningMessage(UniversalBaseModel):
+class LettaStreamingResponse_ReasoningMessage(UncheckedBaseModel):
     message_type: typing.Literal["reasoning_message"] = "reasoning_message"
     id: str
     date: dt.datetime
@@ -58,7 +60,7 @@ class LettaStreamingResponse_ReasoningMessage(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-class LettaStreamingResponse_ToolCallMessage(UniversalBaseModel):
+class LettaStreamingResponse_ToolCallMessage(UncheckedBaseModel):
     message_type: typing.Literal["tool_call_message"] = "tool_call_message"
     id: str
     date: dt.datetime
@@ -74,7 +76,7 @@ class LettaStreamingResponse_ToolCallMessage(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-class LettaStreamingResponse_ToolReturnMessage(UniversalBaseModel):
+class LettaStreamingResponse_ToolReturnMessage(UncheckedBaseModel):
     message_type: typing.Literal["tool_return_message"] = "tool_return_message"
     id: str
     date: dt.datetime
@@ -94,7 +96,7 @@ class LettaStreamingResponse_ToolReturnMessage(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-class LettaStreamingResponse_AssistantMessage(UniversalBaseModel):
+class LettaStreamingResponse_AssistantMessage(UncheckedBaseModel):
     message_type: typing.Literal["assistant_message"] = "assistant_message"
     id: str
     date: dt.datetime
@@ -110,7 +112,7 @@ class LettaStreamingResponse_AssistantMessage(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-class LettaStreamingResponse_UsageStatistics(UniversalBaseModel):
+class LettaStreamingResponse_UsageStatistics(UncheckedBaseModel):
     message_type: typing.Literal["usage_statistics"] = "usage_statistics"
     completion_tokens: typing.Optional[int] = None
     prompt_tokens: typing.Optional[int] = None
@@ -127,12 +129,15 @@ class LettaStreamingResponse_UsageStatistics(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-LettaStreamingResponse = typing.Union[
-    LettaStreamingResponse_SystemMessage,
-    LettaStreamingResponse_UserMessage,
-    LettaStreamingResponse_ReasoningMessage,
-    LettaStreamingResponse_ToolCallMessage,
-    LettaStreamingResponse_ToolReturnMessage,
-    LettaStreamingResponse_AssistantMessage,
-    LettaStreamingResponse_UsageStatistics,
+LettaStreamingResponse = typing_extensions.Annotated[
+    typing.Union[
+        LettaStreamingResponse_SystemMessage,
+        LettaStreamingResponse_UserMessage,
+        LettaStreamingResponse_ReasoningMessage,
+        LettaStreamingResponse_ToolCallMessage,
+        LettaStreamingResponse_ToolReturnMessage,
+        LettaStreamingResponse_AssistantMessage,
+        LettaStreamingResponse_UsageStatistics,
+    ],
+    UnionMetadata(discriminant="message_type"),
 ]