letta-client 0.1.318__py3-none-any.whl → 0.1.320__py3-none-any.whl

letta_client/agents/files/raw_client.py

@@ -11,6 +11,7 @@ from ...core.request_options import RequestOptions
 from ...core.unchecked_base_model import construct_type
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.http_validation_error import HttpValidationError
+from ...types.paginated_agent_files import PaginatedAgentFiles
 
 
 class RawFilesClient:
@@ -180,27 +181,70 @@ class RawFilesClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-    def list(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> HttpResponse[None]:
+    def list(
+        self,
+        agent_id: str,
+        *,
+        cursor: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        is_open: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[PaginatedAgentFiles]:
         """
+        Get the files attached to an agent with their open/closed status (paginated).
+
         Parameters
         ----------
         agent_id : str
 
+        cursor : typing.Optional[str]
+            Pagination cursor from previous response
+
+        limit : typing.Optional[int]
+            Number of items to return (1-100)
+
+        is_open : typing.Optional[bool]
+            Filter by open status (true for open files, false for closed files)
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        HttpResponse[None]
+        HttpResponse[PaginatedAgentFiles]
+            Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
             f"v1/agents/{jsonable_encoder(agent_id)}/files",
-            method="PATCH",
+            method="GET",
+            params={
+                "cursor": cursor,
+                "limit": limit,
+                "is_open": is_open,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return HttpResponse(response=_response, data=None)
+                _data = typing.cast(
+                    PaginatedAgentFiles,
+                    construct_type(
+                        type_=PaginatedAgentFiles,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=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, headers=dict(_response.headers), body=_response.text)
@@ -375,28 +419,69 @@ class AsyncRawFilesClient:
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     async def list(
-        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> AsyncHttpResponse[None]:
+        self,
+        agent_id: str,
+        *,
+        cursor: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        is_open: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[PaginatedAgentFiles]:
         """
+        Get the files attached to an agent with their open/closed status (paginated).
+
         Parameters
         ----------
         agent_id : str
 
+        cursor : typing.Optional[str]
+            Pagination cursor from previous response
+
+        limit : typing.Optional[int]
+            Number of items to return (1-100)
+
+        is_open : typing.Optional[bool]
+            Filter by open status (true for open files, false for closed files)
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AsyncHttpResponse[None]
+        AsyncHttpResponse[PaginatedAgentFiles]
+            Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"v1/agents/{jsonable_encoder(agent_id)}/files",
-            method="PATCH",
+            method="GET",
+            params={
+                "cursor": cursor,
+                "limit": limit,
+                "is_open": is_open,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return AsyncHttpResponse(response=_response, data=None)
+                _data = typing.cast(
+                    PaginatedAgentFiles,
+                    construct_type(
+                        type_=PaginatedAgentFiles,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=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, headers=dict(_response.headers), body=_response.text)

letta_client/agents/messages/__init__.py

@@ -8,7 +8,7 @@ from .types import (
     MessageSearchRequestSearchMode,
     MessagesModifyRequest,
     MessagesModifyResponse,
-    MessagesPreviewRawPayloadRequest,
+    MessagesPreviewRequest,
 )
 
 __all__ = [
@@ -17,5 +17,5 @@ __all__ = [
     "MessageSearchRequestSearchMode",
     "MessagesModifyRequest",
     "MessagesModifyResponse",
-    "MessagesPreviewRawPayloadRequest",
+    "MessagesPreviewRequest",
 ]

letta_client/agents/messages/client.py

@@ -20,7 +20,7 @@ from .types.letta_streaming_response import LettaStreamingResponse
 from .types.message_search_request_search_mode import MessageSearchRequestSearchMode
 from .types.messages_modify_request import MessagesModifyRequest
 from .types.messages_modify_response import MessagesModifyResponse
-from .types.messages_preview_raw_payload_request import MessagesPreviewRawPayloadRequest
+from .types.messages_preview_request import MessagesPreviewRequest
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -185,7 +185,7 @@ class MessagesClient:
                     role="user",
                     content=[
                         TextContent(
-                            text="text",
+                            text="The sky above the port was the color of television, tuned to a dead channel.",
                         )
                     ],
                 )
@@ -329,7 +329,7 @@ class MessagesClient:
                     role="user",
                     content=[
                         TextContent(
-                            text="text",
+                            text="The sky above the port was the color of television, tuned to a dead channel.",
                         )
                     ],
                 )
@@ -403,13 +403,14 @@ class MessagesClient:
         search_mode: typing.Optional[MessageSearchRequestSearchMode] = OMIT,
         roles: typing.Optional[typing.Sequence[MessageRole]] = OMIT,
         project_id: typing.Optional[str] = OMIT,
+        template_id: typing.Optional[str] = OMIT,
         limit: typing.Optional[int] = OMIT,
         start_date: typing.Optional[dt.datetime] = OMIT,
         end_date: typing.Optional[dt.datetime] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[MessageSearchResult]:
         """
-        Search messages across the entire organization with optional project filtering. Returns messages with FTS/vector ranks and total RRF score.
+        Search messages across the entire organization with optional project and template filtering. Returns messages with FTS/vector ranks and total RRF score.
 
         This is a cloud-only feature.
 
@@ -427,6 +428,9 @@ class MessagesClient:
         project_id : typing.Optional[str]
             Filter messages by project ID
 
+        template_id : typing.Optional[str]
+            Filter messages by template ID
+
         limit : typing.Optional[int]
             Maximum number of results to return
 
@@ -459,6 +463,7 @@ class MessagesClient:
             search_mode=search_mode,
             roles=roles,
             project_id=project_id,
+            template_id=template_id,
             limit=limit,
             start_date=start_date,
             end_date=end_date,
@@ -601,12 +606,8 @@ class MessagesClient:
         )
         return _response.data
 
-    def preview_raw_payload(
-        self,
-        agent_id: str,
-        *,
-        request: MessagesPreviewRawPayloadRequest,
-        request_options: typing.Optional[RequestOptions] = None,
+    def preview(
+        self, agent_id: str, *, request: MessagesPreviewRequest, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.Dict[str, typing.Optional[typing.Any]]:
         """
         Inspect the raw LLM request payload without sending it.
@@ -619,7 +620,7 @@ class MessagesClient:
         ----------
         agent_id : str
 
-        request : MessagesPreviewRawPayloadRequest
+        request : MessagesPreviewRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -637,7 +638,7 @@ class MessagesClient:
             project="YOUR_PROJECT",
             token="YOUR_TOKEN",
         )
-        client.agents.messages.preview_raw_payload(
+        client.agents.messages.preview(
             agent_id="agent_id",
             request=LettaRequest(
                 messages=[
@@ -653,7 +654,48 @@ class MessagesClient:
             ),
         )
         """
-        _response = self._raw_client.preview_raw_payload(agent_id, request=request, request_options=request_options)
+        _response = self._raw_client.preview(agent_id, request=request, request_options=request_options)
+        return _response.data
+
+    def summarize(
+        self, agent_id: str, *, max_message_length: int, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Summarize an agent's conversation history to a target message length.
+
+        This endpoint summarizes the current message history for a given agent,
+        truncating and compressing it down to the specified `max_message_length`.
+
+        Parameters
+        ----------
+        agent_id : str
+
+        max_message_length : int
+            Maximum number of messages to retain after summarization.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+        client.agents.messages.summarize(
+            agent_id="agent_id",
+            max_message_length=1,
+        )
+        """
+        _response = self._raw_client.summarize(
+            agent_id, max_message_length=max_message_length, request_options=request_options
+        )
         return _response.data
 
 
@@ -829,7 +871,7 @@ class AsyncMessagesClient:
                         role="user",
                         content=[
                             TextContent(
-                                text="text",
+                                text="The sky above the port was the color of television, tuned to a dead channel.",
                             )
                         ],
                     )
@@ -991,7 +1033,7 @@ class AsyncMessagesClient:
                         role="user",
                         content=[
                             TextContent(
-                                text="text",
+                                text="The sky above the port was the color of television, tuned to a dead channel.",
                             )
                         ],
                     )
@@ -1077,13 +1119,14 @@ class AsyncMessagesClient:
         search_mode: typing.Optional[MessageSearchRequestSearchMode] = OMIT,
         roles: typing.Optional[typing.Sequence[MessageRole]] = OMIT,
         project_id: typing.Optional[str] = OMIT,
+        template_id: typing.Optional[str] = OMIT,
         limit: typing.Optional[int] = OMIT,
         start_date: typing.Optional[dt.datetime] = OMIT,
         end_date: typing.Optional[dt.datetime] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[MessageSearchResult]:
         """
-        Search messages across the entire organization with optional project filtering. Returns messages with FTS/vector ranks and total RRF score.
+        Search messages across the entire organization with optional project and template filtering. Returns messages with FTS/vector ranks and total RRF score.
 
         This is a cloud-only feature.
 
@@ -1101,6 +1144,9 @@ class AsyncMessagesClient:
         project_id : typing.Optional[str]
             Filter messages by project ID
 
+        template_id : typing.Optional[str]
+            Filter messages by template ID
+
         limit : typing.Optional[int]
             Maximum number of results to return
 
@@ -1141,6 +1187,7 @@ class AsyncMessagesClient:
             search_mode=search_mode,
             roles=roles,
             project_id=project_id,
+            template_id=template_id,
             limit=limit,
             start_date=start_date,
             end_date=end_date,
@@ -1299,12 +1346,8 @@ class AsyncMessagesClient:
         )
         return _response.data
 
-    async def preview_raw_payload(
-        self,
-        agent_id: str,
-        *,
-        request: MessagesPreviewRawPayloadRequest,
-        request_options: typing.Optional[RequestOptions] = None,
+    async def preview(
+        self, agent_id: str, *, request: MessagesPreviewRequest, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.Dict[str, typing.Optional[typing.Any]]:
         """
         Inspect the raw LLM request payload without sending it.
@@ -1317,7 +1360,7 @@ class AsyncMessagesClient:
         ----------
         agent_id : str
 
-        request : MessagesPreviewRawPayloadRequest
+        request : MessagesPreviewRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1340,7 +1383,7 @@ class AsyncMessagesClient:
 
 
         async def main() -> None:
-            await client.agents.messages.preview_raw_payload(
+            await client.agents.messages.preview(
                 agent_id="agent_id",
                 request=LettaRequest(
                     messages=[
@@ -1359,7 +1402,54 @@ class AsyncMessagesClient:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.preview_raw_payload(
-            agent_id, request=request, request_options=request_options
+        _response = await self._raw_client.preview(agent_id, request=request, request_options=request_options)
+        return _response.data
+
+    async def summarize(
+        self, agent_id: str, *, max_message_length: int, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Summarize an agent's conversation history to a target message length.
+
+        This endpoint summarizes the current message history for a given agent,
+        truncating and compressing it down to the specified `max_message_length`.
+
+        Parameters
+        ----------
+        agent_id : str
+
+        max_message_length : int
+            Maximum number of messages to retain after summarization.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.agents.messages.summarize(
+                agent_id="agent_id",
+                max_message_length=1,
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.summarize(
+            agent_id, max_message_length=max_message_length, request_options=request_options
         )
         return _response.data