letta-client 0.1.135__tar.gz → 0.1.137__tar.gz

{letta_client-0.1.135 → letta_client-0.1.137}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: letta-client
-Version: 0.1.135
+Version: 0.1.137
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers

{letta_client-0.1.135 → letta_client-0.1.137}/pyproject.toml

@@ -3,7 +3,7 @@ name = "letta-client"
 
 [tool.poetry]
 name = "letta-client"
-version = "0.1.135"
+version = "0.1.137"
 description = ""
 readme = "README.md"
 authors = []

{letta_client-0.1.135 → letta_client-0.1.137}/src/letta_client/__init__.py

@@ -118,6 +118,7 @@ from .types import (
     JsonObjectResponseFormat,
     JsonSchema,
     JsonSchemaResponseFormat,
+    LettaBatchMessages,
     LettaBatchRequest,
     LettaMessageContentUnion,
     LettaMessageUnion,
@@ -237,11 +238,11 @@ from . import (
     blocks,
     client_side_access_tokens,
     embedding_models,
-    embeddings,
     groups,
     health,
     identities,
     jobs,
+    messages,
     models,
     projects,
     providers,
@@ -432,6 +433,7 @@ __all__ = [
     "JsonSchema",
     "JsonSchemaResponseFormat",
     "Letta",
+    "LettaBatchMessages",
     "LettaBatchRequest",
     "LettaEnvironment",
     "LettaMessageContentUnion",
@@ -559,11 +561,11 @@ __all__ = [
     "blocks",
     "client_side_access_tokens",
     "embedding_models",
-    "embeddings",
     "groups",
     "health",
     "identities",
     "jobs",
+    "messages",
     "models",
     "projects",
     "providers",

{letta_client-0.1.135 → letta_client-0.1.137}/src/letta_client/base_client.py

@@ -19,8 +19,8 @@ from .runs.client import RunsClient
 from .steps.client import StepsClient
 from .tags.client import TagsClient
 from .batches.client import BatchesClient
+from .messages.client import MessagesClient
 from .voice.client import VoiceClient
-from .embeddings.client import EmbeddingsClient
 from .templates.client import TemplatesClient
 from .client_side_access_tokens.client import ClientSideAccessTokensClient
 from .projects.client import ProjectsClient
@@ -40,8 +40,8 @@ from .runs.client import AsyncRunsClient
 from .steps.client import AsyncStepsClient
 from .tags.client import AsyncTagsClient
 from .batches.client import AsyncBatchesClient
+from .messages.client import AsyncMessagesClient
 from .voice.client import AsyncVoiceClient
-from .embeddings.client import AsyncEmbeddingsClient
 from .templates.client import AsyncTemplatesClient
 from .client_side_access_tokens.client import AsyncClientSideAccessTokensClient
 from .projects.client import AsyncProjectsClient
@@ -120,8 +120,8 @@ class LettaBase:
         self.steps = StepsClient(client_wrapper=self._client_wrapper)
         self.tags = TagsClient(client_wrapper=self._client_wrapper)
         self.batches = BatchesClient(client_wrapper=self._client_wrapper)
+        self.messages = MessagesClient(client_wrapper=self._client_wrapper)
         self.voice = VoiceClient(client_wrapper=self._client_wrapper)
-        self.embeddings = EmbeddingsClient(client_wrapper=self._client_wrapper)
         self.templates = TemplatesClient(client_wrapper=self._client_wrapper)
         self.client_side_access_tokens = ClientSideAccessTokensClient(client_wrapper=self._client_wrapper)
         self.projects = ProjectsClient(client_wrapper=self._client_wrapper)
@@ -200,8 +200,8 @@ class AsyncLettaBase:
         self.steps = AsyncStepsClient(client_wrapper=self._client_wrapper)
         self.tags = AsyncTagsClient(client_wrapper=self._client_wrapper)
         self.batches = AsyncBatchesClient(client_wrapper=self._client_wrapper)
+        self.messages = AsyncMessagesClient(client_wrapper=self._client_wrapper)
         self.voice = AsyncVoiceClient(client_wrapper=self._client_wrapper)
-        self.embeddings = AsyncEmbeddingsClient(client_wrapper=self._client_wrapper)
         self.templates = AsyncTemplatesClient(client_wrapper=self._client_wrapper)
         self.client_side_access_tokens = AsyncClientSideAccessTokensClient(client_wrapper=self._client_wrapper)
         self.projects = AsyncProjectsClient(client_wrapper=self._client_wrapper)

{letta_client-0.1.135 → letta_client-0.1.137}/src/letta_client/batches/client.py

@@ -91,7 +91,7 @@ class BatchesClient:
             List of requests to be processed in batch.
 
         callback_url : typing.Optional[str]
-            Optional URL to call via POST when the batch completes.
+            Optional URL to call via POST when the batch completes. The callback payload will be a JSON object with the following fields: {'job_id': string, 'status': string, 'completed_at': string}. Where 'job_id' is the unique batch job identifier, 'status' is the final batch status (e.g., 'completed', 'failed'), and 'completed_at' is an ISO 8601 timestamp indicating when the batch job completed.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -361,7 +361,7 @@ class AsyncBatchesClient:
             List of requests to be processed in batch.
 
         callback_url : typing.Optional[str]
-            Optional URL to call via POST when the batch completes.
+            Optional URL to call via POST when the batch completes. The callback payload will be a JSON object with the following fields: {'job_id': string, 'status': string, 'completed_at': string}. Where 'job_id' is the unique batch job identifier, 'status' is the final batch status (e.g., 'completed', 'failed'), and 'completed_at' is an ISO 8601 timestamp indicating when the batch job completed.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.

{letta_client-0.1.135 → letta_client-0.1.137}/src/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.135",
+            "X-Fern-SDK-Version": "0.1.137",
         }
         if self.token is not None:
             headers["Authorization"] = f"Bearer {self.token}"

{letta_client-0.1.135/src/letta_client/embeddings → letta_client-0.1.137/src/letta_client/messages}/client.py

@@ -3,6 +3,8 @@
 from ..core.client_wrapper import SyncClientWrapper
 import typing
 from ..core.request_options import RequestOptions
+from ..types.letta_batch_messages import LettaBatchMessages
+from ..core.jsonable_encoder import jsonable_encoder
 from ..core.unchecked_base_model import construct_type
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.http_validation_error import HttpValidationError
@@ -11,22 +13,52 @@ from ..core.api_error import ApiError
 from ..core.client_wrapper import AsyncClientWrapper
 
 
-class EmbeddingsClient:
+class MessagesClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def get_embeddings_total_storage_size(self, *, request_options: typing.Optional[RequestOptions] = None) -> float:
+    def list_batch_messages(
+        self,
+        batch_id: str,
+        *,
+        limit: typing.Optional[int] = None,
+        cursor: typing.Optional[str] = None,
+        agent_id: typing.Optional[str] = None,
+        sort_descending: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> LettaBatchMessages:
         """
-        Get the total size of all embeddings in the database for a user in the storage unit given.
+        Get messages for a specific batch job.
+
+        Returns messages associated with the batch in chronological order.
+
+        Pagination:
+        - For the first page, omit the cursor parameter
+        - For subsequent pages, use the ID of the last message from the previous response as the cursor
+        - Results will include messages before/after the cursor based on sort_descending
 
         Parameters
         ----------
+        batch_id : str
+
+        limit : typing.Optional[int]
+            Maximum number of messages to return
+
+        cursor : typing.Optional[str]
+            Message ID to use as pagination cursor (get messages before/after this ID) depending on sort_descending.
+
+        agent_id : typing.Optional[str]
+            Filter messages by agent ID
+
+        sort_descending : typing.Optional[bool]
+            Sort messages by creation time (true=newest first)
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        float
+        LettaBatchMessages
             Successful Response
 
         Examples
@@ -36,22 +68,27 @@ class EmbeddingsClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.embeddings.get_embeddings_total_storage_size()
+        client.messages.list_batch_messages(
+            batch_id="batch_id",
+        )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "v1/embeddings/total_storage_size",
+            f"v1/messages/batches/{jsonable_encoder(batch_id)}/messages",
             method="GET",
-            headers={
-                "storage-unit": "GB",
+            params={
+                "limit": limit,
+                "cursor": cursor,
+                "agent_id": agent_id,
+                "sort_descending": sort_descending,
             },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    float,
+                    LettaBatchMessages,
                     construct_type(
-                        type_=float,  # type: ignore
+                        type_=LettaBatchMessages,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -71,24 +108,52 @@ class EmbeddingsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
 
-class AsyncEmbeddingsClient:
+class AsyncMessagesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def get_embeddings_total_storage_size(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> float:
+    async def list_batch_messages(
+        self,
+        batch_id: str,
+        *,
+        limit: typing.Optional[int] = None,
+        cursor: typing.Optional[str] = None,
+        agent_id: typing.Optional[str] = None,
+        sort_descending: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> LettaBatchMessages:
         """
-        Get the total size of all embeddings in the database for a user in the storage unit given.
+        Get messages for a specific batch job.
+
+        Returns messages associated with the batch in chronological order.
+
+        Pagination:
+        - For the first page, omit the cursor parameter
+        - For subsequent pages, use the ID of the last message from the previous response as the cursor
+        - Results will include messages before/after the cursor based on sort_descending
 
         Parameters
         ----------
+        batch_id : str
+
+        limit : typing.Optional[int]
+            Maximum number of messages to return
+
+        cursor : typing.Optional[str]
+            Message ID to use as pagination cursor (get messages before/after this ID) depending on sort_descending.
+
+        agent_id : typing.Optional[str]
+            Filter messages by agent ID
+
+        sort_descending : typing.Optional[bool]
+            Sort messages by creation time (true=newest first)
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        float
+        LettaBatchMessages
             Successful Response
 
         Examples
@@ -103,25 +168,30 @@ class AsyncEmbeddingsClient:
 
 
         async def main() -> None:
-            await client.embeddings.get_embeddings_total_storage_size()
+            await client.messages.list_batch_messages(
+                batch_id="batch_id",
+            )
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "v1/embeddings/total_storage_size",
+            f"v1/messages/batches/{jsonable_encoder(batch_id)}/messages",
             method="GET",
-            headers={
-                "storage-unit": "GB",
+            params={
+                "limit": limit,
+                "cursor": cursor,
+                "agent_id": agent_id,
+                "sort_descending": sort_descending,
             },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    float,
+                    LettaBatchMessages,
                     construct_type(
-                        type_=float,  # type: ignore
+                        type_=LettaBatchMessages,  # type: ignore
                         object_=_response.json(),
                     ),
                 )

{letta_client-0.1.135 → letta_client-0.1.137}/src/letta_client/types/__init__.py

@@ -117,6 +117,7 @@ from .job_type import JobType
 from .json_object_response_format import JsonObjectResponseFormat
 from .json_schema import JsonSchema
 from .json_schema_response_format import JsonSchemaResponseFormat
+from .letta_batch_messages import LettaBatchMessages
 from .letta_batch_request import LettaBatchRequest
 from .letta_message_content_union import LettaMessageContentUnion
 from .letta_message_union import LettaMessageUnion
@@ -351,6 +352,7 @@ __all__ = [
     "JsonObjectResponseFormat",
     "JsonSchema",
     "JsonSchemaResponseFormat",
+    "LettaBatchMessages",
     "LettaBatchRequest",
     "LettaMessageContentUnion",
     "LettaMessageUnion",

letta_client-0.1.137/src/letta_client/types/letta_batch_messages.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.unchecked_base_model import UncheckedBaseModel
+import typing
+from .message import Message
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class LettaBatchMessages(UncheckedBaseModel):
+    messages: typing.List[Message]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

{letta_client-0.1.135 → letta_client-0.1.137}/src/letta_client/types/message.py

@@ -118,6 +118,11 @@ class Message(UncheckedBaseModel):
     The id of the sender of the message, can be an identity id or agent id
     """
 
+    batch_item_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The id of the LLMBatchItem that this message is associated with
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

{letta_client-0.1.135 → letta_client-0.1.137}/src/letta_client/types/message_create.py

@@ -38,6 +38,11 @@ class MessageCreate(UncheckedBaseModel):
     The id of the sender of the message, can be an identity id or agent id
     """
 
+    batch_item_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The id of the LLMBatchItem that this message is associated with
+    """
+
     group_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     The multi-agent group that the message was sent in