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

letta_client/agents/messages/client.py

@@ -3,7 +3,7 @@
 import typing
 from ...core.client_wrapper import SyncClientWrapper
 from ...core.request_options import RequestOptions
-from .types.messages_list_response import MessagesListResponse
+from ...types.letta_message_union import LettaMessageUnion
 from ...core.jsonable_encoder import jsonable_encoder
 from ...core.unchecked_base_model import construct_type
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
@@ -11,12 +11,11 @@ from ...types.http_validation_error import HttpValidationError
 from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
 from ...types.message_create import MessageCreate
-from ...types.letta_request_config import LettaRequestConfig
 from ...types.letta_response import LettaResponse
 from ...core.serialization import convert_and_respect_annotation_metadata
 from ...types.message_role import MessageRole
 from .types.message_update_content import MessageUpdateContent
-from ...types.chat_completion_message_tool_call import ChatCompletionMessageToolCall
+from ...types.chat_completion_message_tool_call_input import ChatCompletionMessageToolCallInput
 from ...types.message import Message
 from .types.letta_streaming_response import LettaStreamingResponse
 import httpx_sse
@@ -36,13 +35,14 @@ class MessagesClient:
         self,
         agent_id: str,
         *,
+        after: typing.Optional[str] = None,
         before: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
-        msg_object: typing.Optional[bool] = None,
+        use_assistant_message: typing.Optional[bool] = None,
         assistant_message_tool_name: typing.Optional[str] = None,
         assistant_message_tool_kwarg: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> MessagesListResponse:
+    ) -> typing.List[LettaMessageUnion]:
         """
         Retrieve message history for an agent.
 
@@ -50,27 +50,30 @@ class MessagesClient:
         ----------
         agent_id : str
 
+        after : typing.Optional[str]
+            Message after which to retrieve the returned messages.
+
         before : typing.Optional[str]
             Message before which to retrieve the returned messages.
 
         limit : typing.Optional[int]
             Maximum number of messages to retrieve.
 
-        msg_object : typing.Optional[bool]
-            If true, returns Message objects. If false, return LettaMessage objects.
+        use_assistant_message : typing.Optional[bool]
+            Whether to use assistant messages
 
         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.
+            The name of the message argument.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        MessagesListResponse
+        typing.List[LettaMessageUnion]
             Successful Response
 
         Examples
@@ -88,9 +91,10 @@ class MessagesClient:
             f"v1/agents/{jsonable_encoder(agent_id)}/messages",
             method="GET",
             params={
+                "after": after,
                 "before": before,
                 "limit": limit,
-                "msg_object": msg_object,
+                "use_assistant_message": use_assistant_message,
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
             },
@@ -99,9 +103,9 @@ class MessagesClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    MessagesListResponse,
+                    typing.List[LettaMessageUnion],
                     construct_type(
-                        type_=MessagesListResponse,  # type: ignore
+                        type_=typing.List[LettaMessageUnion],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -125,7 +129,9 @@ class MessagesClient:
         agent_id: str,
         *,
         messages: typing.Sequence[MessageCreate],
-        config: typing.Optional[LettaRequestConfig] = OMIT,
+        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,
     ) -> LettaResponse:
         """
@@ -139,8 +145,14 @@ class MessagesClient:
         messages : typing.Sequence[MessageCreate]
             The messages to be sent to the agent.
 
-        config : typing.Optional[LettaRequestConfig]
-            Configuration options for the LettaRequest.
+        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.
@@ -174,9 +186,9 @@ class MessagesClient:
                 "messages": convert_and_respect_annotation_metadata(
                     object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
                 ),
-                "config": convert_and_respect_annotation_metadata(
-                    object_=config, annotation=LettaRequestConfig, 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,
@@ -213,7 +225,7 @@ class MessagesClient:
         role: typing.Optional[MessageRole] = OMIT,
         content: typing.Optional[MessageUpdateContent] = OMIT,
         name: typing.Optional[str] = OMIT,
-        tool_calls: typing.Optional[typing.Sequence[ChatCompletionMessageToolCall]] = OMIT,
+        tool_calls: typing.Optional[typing.Sequence[ChatCompletionMessageToolCallInput]] = OMIT,
         tool_call_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Message:
@@ -235,7 +247,7 @@ class MessagesClient:
         name : typing.Optional[str]
             The name of the participant.
 
-        tool_calls : typing.Optional[typing.Sequence[ChatCompletionMessageToolCall]]
+        tool_calls : typing.Optional[typing.Sequence[ChatCompletionMessageToolCallInput]]
             The list of tool calls requested.
 
         tool_call_id : typing.Optional[str]
@@ -271,7 +283,9 @@ class MessagesClient:
                 ),
                 "name": name,
                 "tool_calls": convert_and_respect_annotation_metadata(
-                    object_=tool_calls, annotation=typing.Sequence[ChatCompletionMessageToolCall], direction="write"
+                    object_=tool_calls,
+                    annotation=typing.Sequence[ChatCompletionMessageToolCallInput],
+                    direction="write",
                 ),
                 "tool_call_id": tool_call_id,
             },
@@ -310,7 +324,9 @@ class MessagesClient:
         agent_id: str,
         *,
         messages: typing.Sequence[MessageCreate],
-        config: typing.Optional[LettaRequestConfig] = OMIT,
+        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,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Iterator[LettaStreamingResponse]:
@@ -326,8 +342,14 @@ class MessagesClient:
         messages : typing.Sequence[MessageCreate]
             The messages to be sent to the agent.
 
-        config : typing.Optional[LettaRequestConfig]
-            Configuration options for the LettaRequest.
+        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.
 
         stream_tokens : typing.Optional[bool]
             Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
@@ -366,9 +388,9 @@ class MessagesClient:
                 "messages": convert_and_respect_annotation_metadata(
                     object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
                 ),
-                "config": convert_and_respect_annotation_metadata(
-                    object_=config, annotation=LettaRequestConfig, 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,
             },
             headers={
@@ -413,7 +435,9 @@ class MessagesClient:
         agent_id: str,
         *,
         messages: typing.Sequence[MessageCreate],
-        config: typing.Optional[LettaRequestConfig] = OMIT,
+        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:
         """
@@ -427,8 +451,14 @@ class MessagesClient:
         messages : typing.Sequence[MessageCreate]
             The messages to be sent to the agent.
 
-        config : typing.Optional[LettaRequestConfig]
-            Configuration options for the LettaRequest.
+        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.
@@ -462,9 +492,9 @@ class MessagesClient:
                 "messages": convert_and_respect_annotation_metadata(
                     object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
                 ),
-                "config": convert_and_respect_annotation_metadata(
-                    object_=config, annotation=LettaRequestConfig, 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,
@@ -502,13 +532,14 @@ class AsyncMessagesClient:
         self,
         agent_id: str,
         *,
+        after: typing.Optional[str] = None,
         before: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
-        msg_object: typing.Optional[bool] = None,
+        use_assistant_message: typing.Optional[bool] = None,
         assistant_message_tool_name: typing.Optional[str] = None,
         assistant_message_tool_kwarg: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> MessagesListResponse:
+    ) -> typing.List[LettaMessageUnion]:
         """
         Retrieve message history for an agent.
 
@@ -516,27 +547,30 @@ class AsyncMessagesClient:
         ----------
         agent_id : str
 
+        after : typing.Optional[str]
+            Message after which to retrieve the returned messages.
+
         before : typing.Optional[str]
             Message before which to retrieve the returned messages.
 
         limit : typing.Optional[int]
             Maximum number of messages to retrieve.
 
-        msg_object : typing.Optional[bool]
-            If true, returns Message objects. If false, return LettaMessage objects.
+        use_assistant_message : typing.Optional[bool]
+            Whether to use assistant messages
 
         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.
+            The name of the message argument.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        MessagesListResponse
+        typing.List[LettaMessageUnion]
             Successful Response
 
         Examples
@@ -562,9 +596,10 @@ class AsyncMessagesClient:
             f"v1/agents/{jsonable_encoder(agent_id)}/messages",
             method="GET",
             params={
+                "after": after,
                 "before": before,
                 "limit": limit,
-                "msg_object": msg_object,
+                "use_assistant_message": use_assistant_message,
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
             },
@@ -573,9 +608,9 @@ class AsyncMessagesClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    MessagesListResponse,
+                    typing.List[LettaMessageUnion],
                     construct_type(
-                        type_=MessagesListResponse,  # type: ignore
+                        type_=typing.List[LettaMessageUnion],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -599,7 +634,9 @@ class AsyncMessagesClient:
         agent_id: str,
         *,
         messages: typing.Sequence[MessageCreate],
-        config: typing.Optional[LettaRequestConfig] = OMIT,
+        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,
     ) -> LettaResponse:
         """
@@ -613,8 +650,14 @@ class AsyncMessagesClient:
         messages : typing.Sequence[MessageCreate]
             The messages to be sent to the agent.
 
-        config : typing.Optional[LettaRequestConfig]
-            Configuration options for the LettaRequest.
+        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.
@@ -656,9 +699,9 @@ class AsyncMessagesClient:
                 "messages": convert_and_respect_annotation_metadata(
                     object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
                 ),
-                "config": convert_and_respect_annotation_metadata(
-                    object_=config, annotation=LettaRequestConfig, 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,
@@ -695,7 +738,7 @@ class AsyncMessagesClient:
         role: typing.Optional[MessageRole] = OMIT,
         content: typing.Optional[MessageUpdateContent] = OMIT,
         name: typing.Optional[str] = OMIT,
-        tool_calls: typing.Optional[typing.Sequence[ChatCompletionMessageToolCall]] = OMIT,
+        tool_calls: typing.Optional[typing.Sequence[ChatCompletionMessageToolCallInput]] = OMIT,
         tool_call_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Message:
@@ -717,7 +760,7 @@ class AsyncMessagesClient:
         name : typing.Optional[str]
             The name of the participant.
 
-        tool_calls : typing.Optional[typing.Sequence[ChatCompletionMessageToolCall]]
+        tool_calls : typing.Optional[typing.Sequence[ChatCompletionMessageToolCallInput]]
             The list of tool calls requested.
 
         tool_call_id : typing.Optional[str]
@@ -761,7 +804,9 @@ class AsyncMessagesClient:
                 ),
                 "name": name,
                 "tool_calls": convert_and_respect_annotation_metadata(
-                    object_=tool_calls, annotation=typing.Sequence[ChatCompletionMessageToolCall], direction="write"
+                    object_=tool_calls,
+                    annotation=typing.Sequence[ChatCompletionMessageToolCallInput],
+                    direction="write",
                 ),
                 "tool_call_id": tool_call_id,
             },
@@ -800,7 +845,9 @@ class AsyncMessagesClient:
         agent_id: str,
         *,
         messages: typing.Sequence[MessageCreate],
-        config: typing.Optional[LettaRequestConfig] = OMIT,
+        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,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.AsyncIterator[LettaStreamingResponse]:
@@ -816,8 +863,14 @@ class AsyncMessagesClient:
         messages : typing.Sequence[MessageCreate]
             The messages to be sent to the agent.
 
-        config : typing.Optional[LettaRequestConfig]
-            Configuration options for the LettaRequest.
+        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.
 
         stream_tokens : typing.Optional[bool]
             Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
@@ -864,9 +917,9 @@ class AsyncMessagesClient:
                 "messages": convert_and_respect_annotation_metadata(
                     object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
                 ),
-                "config": convert_and_respect_annotation_metadata(
-                    object_=config, annotation=LettaRequestConfig, 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,
             },
             headers={
@@ -911,7 +964,9 @@ class AsyncMessagesClient:
         agent_id: str,
         *,
         messages: typing.Sequence[MessageCreate],
-        config: typing.Optional[LettaRequestConfig] = OMIT,
+        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:
         """
@@ -925,8 +980,14 @@ class AsyncMessagesClient:
         messages : typing.Sequence[MessageCreate]
             The messages to be sent to the agent.
 
-        config : typing.Optional[LettaRequestConfig]
-            Configuration options for the LettaRequest.
+        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.
@@ -968,9 +1029,9 @@ class AsyncMessagesClient:
                 "messages": convert_and_respect_annotation_metadata(
                     object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
                 ),
-                "config": convert_and_respect_annotation_metadata(
-                    object_=config, annotation=LettaRequestConfig, 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,

letta_client/agents/messages/types/__init__.py

@@ -2,6 +2,5 @@
 
 from .letta_streaming_response import LettaStreamingResponse
 from .message_update_content import MessageUpdateContent
-from .messages_list_response import MessagesListResponse
 
-__all__ = ["LettaStreamingResponse", "MessageUpdateContent", "MessagesListResponse"]
+__all__ = ["LettaStreamingResponse", "MessageUpdateContent"]

letta_client/base_client.py

@@ -14,6 +14,7 @@ from .health.client import HealthClient
 from .providers.client import ProvidersClient
 from .runs.client import RunsClient
 from .tag.client import TagClient
+from .templates.client import TemplatesClient
 from .core.client_wrapper import AsyncClientWrapper
 from .tools.client import AsyncToolsClient
 from .sources.client import AsyncSourcesClient
@@ -25,6 +26,7 @@ from .health.client import AsyncHealthClient
 from .providers.client import AsyncProvidersClient
 from .runs.client import AsyncRunsClient
 from .tag.client import AsyncTagClient
+from .templates.client import AsyncTemplatesClient
 
 
 class LettaBase:
@@ -95,6 +97,7 @@ class LettaBase:
         self.providers = ProvidersClient(client_wrapper=self._client_wrapper)
         self.runs = RunsClient(client_wrapper=self._client_wrapper)
         self.tag = TagClient(client_wrapper=self._client_wrapper)
+        self.templates = TemplatesClient(client_wrapper=self._client_wrapper)
 
 
 class AsyncLettaBase:
@@ -165,6 +168,7 @@ class AsyncLettaBase:
         self.providers = AsyncProvidersClient(client_wrapper=self._client_wrapper)
         self.runs = AsyncRunsClient(client_wrapper=self._client_wrapper)
         self.tag = AsyncTagClient(client_wrapper=self._client_wrapper)
+        self.templates = AsyncTemplatesClient(client_wrapper=self._client_wrapper)
 
 
 def _get_base_url(*, base_url: typing.Optional[str] = None, environment: LettaEnvironment) -> str:

letta_client/client.py

@@ -31,7 +31,6 @@ class ToolsClient(ToolsClientBase):
         self,
         *,
         func: typing.Callable,
-        name: typing.Optional[str] = OMIT,
         description: typing.Optional[str] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
         source_type: typing.Optional[str] = OMIT,
@@ -44,7 +43,6 @@ class ToolsClient(ToolsClientBase):
         source_code = dedent(inspect.getsource(func))
         return self.create(
             source_code=source_code,
-            name=name,
             description=description,
             tags=tags,
             source_type=source_type,
@@ -57,7 +55,6 @@ class ToolsClient(ToolsClientBase):
         self,
         *,
         func: typing.Callable,
-        name: typing.Optional[str] = OMIT,
         description: typing.Optional[str] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
         source_type: typing.Optional[str] = OMIT,
@@ -70,7 +67,6 @@ class ToolsClient(ToolsClientBase):
         source_code = dedent(inspect.getsource(func))
         return self.upsert(
             source_code=source_code,
-            name=name,
             description=description,
             tags=tags,
             source_type=source_type,

letta_client/core/client_wrapper.py

@@ -16,7 +16,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "letta-client",
-            "X-Fern-SDK-Version": "0.1.21",
+            "X-Fern-SDK-Version": "0.1.23",
         }
         if self.token is not None:
             headers["Authorization"] = f"Bearer {self.token}"