letta-client 0.1.136__tar.gz → 0.1.138__tar.gz

{letta_client-0.1.136 → letta_client-0.1.138}/PKG-INFO

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

{letta_client-0.1.136 → letta_client-0.1.138}/pyproject.toml

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

{letta_client-0.1.136 → letta_client-0.1.138}/src/letta_client/__init__.py

@@ -118,6 +118,7 @@ from .types import (
     JsonObjectResponseFormat,
     JsonSchema,
     JsonSchemaResponseFormat,
+    LettaBatchMessages,
     LettaBatchRequest,
     LettaMessageContentUnion,
     LettaMessageUnion,
@@ -241,6 +242,7 @@ from . import (
     health,
     identities,
     jobs,
+    messages,
     models,
     projects,
     providers,
@@ -431,6 +433,7 @@ __all__ = [
     "JsonSchema",
     "JsonSchemaResponseFormat",
     "Letta",
+    "LettaBatchMessages",
     "LettaBatchRequest",
     "LettaEnvironment",
     "LettaMessageContentUnion",
@@ -562,6 +565,7 @@ __all__ = [
     "health",
     "identities",
     "jobs",
+    "messages",
     "models",
     "projects",
     "providers",

{letta_client-0.1.136 → letta_client-0.1.138}/src/letta_client/base_client.py

@@ -19,6 +19,7 @@ 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 .templates.client import TemplatesClient
 from .client_side_access_tokens.client import ClientSideAccessTokensClient
@@ -39,6 +40,7 @@ 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 .templates.client import AsyncTemplatesClient
 from .client_side_access_tokens.client import AsyncClientSideAccessTokensClient
@@ -118,6 +120,7 @@ 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.templates = TemplatesClient(client_wrapper=self._client_wrapper)
         self.client_side_access_tokens = ClientSideAccessTokensClient(client_wrapper=self._client_wrapper)
@@ -197,6 +200,7 @@ 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.templates = AsyncTemplatesClient(client_wrapper=self._client_wrapper)
         self.client_side_access_tokens = AsyncClientSideAccessTokensClient(client_wrapper=self._client_wrapper)

{letta_client-0.1.136 → letta_client-0.1.138}/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.136 → letta_client-0.1.138}/src/letta_client/blocks/client.py

@@ -198,6 +198,58 @@ class BlocksClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
+        """
+        Count all blocks created by a user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.blocks.count()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/blocks/count",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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)
+
     def retrieve(self, block_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Block:
         """
         Parameters
@@ -600,6 +652,66 @@ class AsyncBlocksClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
+        """
+        Count all blocks created by a user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.blocks.count()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/blocks/count",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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)
+
     async def retrieve(self, block_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Block:
         """
         Parameters

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

{letta_client-0.1.136 → letta_client-0.1.138}/src/letta_client/groups/client.py

@@ -201,6 +201,58 @@ class GroupsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
+        """
+        Get the count of all groups associated with a given user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.groups.count()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/groups/count",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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)
+
     def retrieve(self, group_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Group:
         """
         Retrieve the group by id.
@@ -604,6 +656,66 @@ class AsyncGroupsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
+        """
+        Get the count of all groups associated with a given user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.groups.count()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/groups/count",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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)
+
     async def retrieve(self, group_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Group:
         """
         Retrieve the group by id.

letta_client-0.1.138/src/letta_client/messages/client.py

@@ -0,0 +1,211 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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
+from json.decoder import JSONDecodeError
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper
+
+
+class MessagesClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    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 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
+        -------
+        LettaBatchMessages
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.messages.list_batch_messages(
+            batch_id="batch_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/messages/batches/{jsonable_encoder(batch_id)}/messages",
+            method="GET",
+            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(
+                    LettaBatchMessages,
+                    construct_type(
+                        type_=LettaBatchMessages,  # 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):
+        self._client_wrapper = client_wrapper
+
+    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 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
+        -------
+        LettaBatchMessages
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.messages.list_batch_messages(
+                batch_id="batch_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/messages/batches/{jsonable_encoder(batch_id)}/messages",
+            method="GET",
+            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(
+                    LettaBatchMessages,
+                    construct_type(
+                        type_=LettaBatchMessages,  # 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_client-0.1.138/src/letta_client/tags/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

{letta_client-0.1.136 → letta_client-0.1.138}/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.138/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.136 → letta_client-0.1.138}/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.136 → letta_client-0.1.138}/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