letta-client 0.1.116__py3-none-any.whl → 0.1.117__py3-none-any.whl

letta_client/__init__.py

@@ -21,6 +21,7 @@ from .types import (
     AuthSchemeField,
     BadRequestErrorBody,
     BaseToolRuleSchema,
+    BatchJob,
     Block,
     BlockUpdate,
     ChatCompletionAssistantMessageParam,
@@ -115,7 +116,6 @@ from .types import (
     JobType,
     JsonSchema,
     LettaBatchRequest,
-    LettaBatchResponse,
     LettaMessageContentUnion,
     LettaMessageUnion,
     LettaRequest,
@@ -230,6 +230,7 @@ from . import (
     health,
     identities,
     jobs,
+    messages,
     models,
     projects,
     providers,
@@ -310,6 +311,7 @@ __all__ = [
     "BadRequestError",
     "BadRequestErrorBody",
     "BaseToolRuleSchema",
+    "BatchJob",
     "Block",
     "BlockUpdate",
     "ChatCompletionAssistantMessageParam",
@@ -418,7 +420,6 @@ __all__ = [
     "JsonSchema",
     "Letta",
     "LettaBatchRequest",
-    "LettaBatchResponse",
     "LettaEnvironment",
     "LettaMessageContentUnion",
     "LettaMessageUnion",
@@ -541,6 +542,7 @@ __all__ = [
     "health",
     "identities",
     "jobs",
+    "messages",
     "models",
     "projects",
     "providers",

letta_client/agents/client.py

@@ -31,8 +31,6 @@ from .types.update_agent_tool_rules_item import UpdateAgentToolRulesItem
 import datetime as dt
 from ..types.passage import Passage
 from ..types.group import Group
-from ..types.letta_batch_request import LettaBatchRequest
-from ..types.letta_batch_response import LettaBatchResponse
 from .types.agents_search_request_search_item import AgentsSearchRequestSearchItem
 from .types.agents_search_response import AgentsSearchResponse
 from ..core.client_wrapper import AsyncClientWrapper
@@ -1160,141 +1158,6 @@ class AgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def create_batch_message_request(
-        self, *, request: typing.Sequence[LettaBatchRequest], request_options: typing.Optional[RequestOptions] = None
-    ) -> LettaBatchResponse:
-        """
-        Submit a batch of agent messages for asynchronous processing.
-        Creates a job that will fan out messages to all listed agents and process them in parallel.
-
-        Parameters
-        ----------
-        request : typing.Sequence[LettaBatchRequest]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        LettaBatchResponse
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta, LettaBatchRequest, MessageCreate, TextContent
-
-        client = Letta(
-            token="YOUR_TOKEN",
-        )
-        client.agents.create_batch_message_request(
-            request=[
-                LettaBatchRequest(
-                    messages=[
-                        MessageCreate(
-                            role="user",
-                            content=[
-                                TextContent(
-                                    text="text",
-                                )
-                            ],
-                        )
-                    ],
-                    agent_id="agent_id",
-                )
-            ],
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/agents/messages/batches",
-            method="POST",
-            json=convert_and_respect_annotation_metadata(
-                object_=request, annotation=typing.Sequence[LettaBatchRequest], direction="write"
-            ),
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    LettaBatchResponse,
-                    construct_type(
-                        type_=LettaBatchResponse,  # 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_batch_message_request(
-        self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> LettaBatchResponse:
-        """
-        Retrieve the result or current status of a previously submitted batch message request.
-
-        Parameters
-        ----------
-        batch_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        LettaBatchResponse
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            token="YOUR_TOKEN",
-        )
-        client.agents.retrieve_batch_message_request(
-            batch_id="batch_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/messages/batches/{jsonable_encoder(batch_id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    LettaBatchResponse,
-                    construct_type(
-                        type_=LettaBatchResponse,  # 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 search(
         self,
         *,
@@ -2564,162 +2427,6 @@ class AsyncAgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def create_batch_message_request(
-        self, *, request: typing.Sequence[LettaBatchRequest], request_options: typing.Optional[RequestOptions] = None
-    ) -> LettaBatchResponse:
-        """
-        Submit a batch of agent messages for asynchronous processing.
-        Creates a job that will fan out messages to all listed agents and process them in parallel.
-
-        Parameters
-        ----------
-        request : typing.Sequence[LettaBatchRequest]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        LettaBatchResponse
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import (
-            AsyncLetta,
-            LettaBatchRequest,
-            MessageCreate,
-            TextContent,
-        )
-
-        client = AsyncLetta(
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.create_batch_message_request(
-                request=[
-                    LettaBatchRequest(
-                        messages=[
-                            MessageCreate(
-                                role="user",
-                                content=[
-                                    TextContent(
-                                        text="text",
-                                    )
-                                ],
-                            )
-                        ],
-                        agent_id="agent_id",
-                    )
-                ],
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/agents/messages/batches",
-            method="POST",
-            json=convert_and_respect_annotation_metadata(
-                object_=request, annotation=typing.Sequence[LettaBatchRequest], direction="write"
-            ),
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    LettaBatchResponse,
-                    construct_type(
-                        type_=LettaBatchResponse,  # 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_batch_message_request(
-        self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> LettaBatchResponse:
-        """
-        Retrieve the result or current status of a previously submitted batch message request.
-
-        Parameters
-        ----------
-        batch_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        LettaBatchResponse
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.retrieve_batch_message_request(
-                batch_id="batch_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/messages/batches/{jsonable_encoder(batch_id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    LettaBatchResponse,
-                    construct_type(
-                        type_=LettaBatchResponse,  # 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 search(
         self,
         *,

letta_client/base_client.py

@@ -17,6 +17,7 @@ from .providers.client import ProvidersClient
 from .runs.client import RunsClient
 from .steps.client import StepsClient
 from .tag.client import TagClient
+from .messages.client import MessagesClient
 from .voice.client import VoiceClient
 from .templates.client import TemplatesClient
 from .client_side_access_tokens.client import ClientSideAccessTokensClient
@@ -35,6 +36,7 @@ from .providers.client import AsyncProvidersClient
 from .runs.client import AsyncRunsClient
 from .steps.client import AsyncStepsClient
 from .tag.client import AsyncTagClient
+from .messages.client import AsyncMessagesClient
 from .voice.client import AsyncVoiceClient
 from .templates.client import AsyncTemplatesClient
 from .client_side_access_tokens.client import AsyncClientSideAccessTokensClient
@@ -112,6 +114,7 @@ class LettaBase:
         self.runs = RunsClient(client_wrapper=self._client_wrapper)
         self.steps = StepsClient(client_wrapper=self._client_wrapper)
         self.tag = TagClient(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)
@@ -189,6 +192,7 @@ class AsyncLettaBase:
         self.runs = AsyncRunsClient(client_wrapper=self._client_wrapper)
         self.steps = AsyncStepsClient(client_wrapper=self._client_wrapper)
         self.tag = AsyncTagClient(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/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.116",
+            "X-Fern-SDK-Version": "0.1.117",
         }
         if self.token is not None:
             headers["Authorization"] = f"Bearer {self.token}"

letta_client/messages/__init__.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from . import batches
+
+__all__ = ["batches"]

letta_client/messages/batches/__init__.py

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

letta_client/messages/batches/client.py

@@ -0,0 +1,521 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from ...core.client_wrapper import SyncClientWrapper
+from ...core.request_options import RequestOptions
+from ...types.batch_job import BatchJob
+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 ...types.letta_batch_request import LettaBatchRequest
+from ...core.serialization import convert_and_respect_annotation_metadata
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.client_wrapper import AsyncClientWrapper
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class BatchesClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[BatchJob]:
+        """
+        List all batch runs.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[BatchJob]
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.messages.batches.list()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/messages/batches",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[BatchJob],
+                    construct_type(
+                        type_=typing.List[BatchJob],  # 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 create(
+        self, *, requests: typing.Sequence[LettaBatchRequest], request_options: typing.Optional[RequestOptions] = None
+    ) -> BatchJob:
+        """
+        Submit a batch of agent messages for asynchronous processing.
+        Creates a job that will fan out messages to all listed agents and process them in parallel.
+
+        Parameters
+        ----------
+        requests : typing.Sequence[LettaBatchRequest]
+            List of requests to be processed in batch.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        BatchJob
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta, LettaBatchRequest, MessageCreate, TextContent
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.messages.batches.create(
+            requests=[
+                LettaBatchRequest(
+                    messages=[
+                        MessageCreate(
+                            role="user",
+                            content=[
+                                TextContent(
+                                    text="text",
+                                )
+                            ],
+                        )
+                    ],
+                    agent_id="agent_id",
+                )
+            ],
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/messages/batches",
+            method="POST",
+            json={
+                "requests": convert_and_respect_annotation_metadata(
+                    object_=requests, annotation=typing.Sequence[LettaBatchRequest], direction="write"
+                ),
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    BatchJob,
+                    construct_type(
+                        type_=BatchJob,  # 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, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> BatchJob:
+        """
+        Get the status of a batch run.
+
+        Parameters
+        ----------
+        batch_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        BatchJob
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.messages.batches.retrieve(
+            batch_id="batch_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/messages/batches/{jsonable_encoder(batch_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    BatchJob,
+                    construct_type(
+                        type_=BatchJob,  # 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 cancel(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+        """
+        Parameters
+        ----------
+        batch_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.messages.batches.cancel(
+            batch_id="batch_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/messages/batches/{jsonable_encoder(batch_id)}",
+            method="PATCH",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _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 AsyncBatchesClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[BatchJob]:
+        """
+        List all batch runs.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[BatchJob]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.messages.batches.list()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/messages/batches",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[BatchJob],
+                    construct_type(
+                        type_=typing.List[BatchJob],  # 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 create(
+        self, *, requests: typing.Sequence[LettaBatchRequest], request_options: typing.Optional[RequestOptions] = None
+    ) -> BatchJob:
+        """
+        Submit a batch of agent messages for asynchronous processing.
+        Creates a job that will fan out messages to all listed agents and process them in parallel.
+
+        Parameters
+        ----------
+        requests : typing.Sequence[LettaBatchRequest]
+            List of requests to be processed in batch.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        BatchJob
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import (
+            AsyncLetta,
+            LettaBatchRequest,
+            MessageCreate,
+            TextContent,
+        )
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.messages.batches.create(
+                requests=[
+                    LettaBatchRequest(
+                        messages=[
+                            MessageCreate(
+                                role="user",
+                                content=[
+                                    TextContent(
+                                        text="text",
+                                    )
+                                ],
+                            )
+                        ],
+                        agent_id="agent_id",
+                    )
+                ],
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/messages/batches",
+            method="POST",
+            json={
+                "requests": convert_and_respect_annotation_metadata(
+                    object_=requests, annotation=typing.Sequence[LettaBatchRequest], direction="write"
+                ),
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    BatchJob,
+                    construct_type(
+                        type_=BatchJob,  # 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, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> BatchJob:
+        """
+        Get the status of a batch run.
+
+        Parameters
+        ----------
+        batch_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        BatchJob
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.messages.batches.retrieve(
+                batch_id="batch_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/messages/batches/{jsonable_encoder(batch_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    BatchJob,
+                    construct_type(
+                        type_=BatchJob,  # 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 cancel(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+        """
+        Parameters
+        ----------
+        batch_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.messages.batches.cancel(
+                batch_id="batch_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/messages/batches/{jsonable_encoder(batch_id)}",
+            method="PATCH",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _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/messages/client.py

@@ -0,0 +1,150 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.client_wrapper import SyncClientWrapper
+from .batches.client import BatchesClient
+import typing
+from ..core.request_options import RequestOptions
+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
+from .batches.client import AsyncBatchesClient
+
+
+class MessagesClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+        self.batches = BatchesClient(client_wrapper=self._client_wrapper)
+
+    def cancel_batch_run(
+        self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Optional[typing.Any]:
+        """
+        Cancel a batch run.
+
+        Parameters
+        ----------
+        batch_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.messages.cancel_batch_run(
+            batch_id="batch_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/messages/batches/{jsonable_encoder(batch_id)}/cancel",
+            method="PATCH",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    construct_type(
+                        type_=typing.Optional[typing.Any],  # 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
+        self.batches = AsyncBatchesClient(client_wrapper=self._client_wrapper)
+
+    async def cancel_batch_run(
+        self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Optional[typing.Any]:
+        """
+        Cancel a batch run.
+
+        Parameters
+        ----------
+        batch_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.messages.cancel_batch_run(
+                batch_id="batch_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/messages/batches/{jsonable_encoder(batch_id)}/cancel",
+            method="PATCH",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    construct_type(
+                        type_=typing.Optional[typing.Any],  # 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/types/__init__.py

@@ -20,6 +20,7 @@ from .auth_response import AuthResponse
 from .auth_scheme_field import AuthSchemeField
 from .bad_request_error_body import BadRequestErrorBody
 from .base_tool_rule_schema import BaseToolRuleSchema
+from .batch_job import BatchJob
 from .block import Block
 from .block_update import BlockUpdate
 from .chat_completion_assistant_message_param import ChatCompletionAssistantMessageParam
@@ -114,7 +115,6 @@ from .job_status import JobStatus
 from .job_type import JobType
 from .json_schema import JsonSchema
 from .letta_batch_request import LettaBatchRequest
-from .letta_batch_response import LettaBatchResponse
 from .letta_message_content_union import LettaMessageContentUnion
 from .letta_message_union import LettaMessageUnion
 from .letta_request import LettaRequest
@@ -245,6 +245,7 @@ __all__ = [
     "AuthSchemeField",
     "BadRequestErrorBody",
     "BaseToolRuleSchema",
+    "BatchJob",
     "Block",
     "BlockUpdate",
     "ChatCompletionAssistantMessageParam",
@@ -339,7 +340,6 @@ __all__ = [
     "JobType",
     "JsonSchema",
     "LettaBatchRequest",
-    "LettaBatchResponse",
     "LettaMessageContentUnion",
     "LettaMessageUnion",
     "LettaRequest",

letta_client/types/batch_job.py

@@ -0,0 +1,61 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.unchecked_base_model import UncheckedBaseModel
+import typing
+import pydantic
+import datetime as dt
+from .job_status import JobStatus
+from .job_type import JobType
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class BatchJob(UncheckedBaseModel):
+    created_by_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The id of the user that made this object.
+    """
+
+    last_updated_by_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The id of the user that made this object.
+    """
+
+    created_at: typing.Optional[dt.datetime] = pydantic.Field(default=None)
+    """
+    The timestamp when the object was created.
+    """
+
+    updated_at: typing.Optional[dt.datetime] = pydantic.Field(default=None)
+    """
+    The timestamp when the object was last updated.
+    """
+
+    status: typing.Optional[JobStatus] = pydantic.Field(default=None)
+    """
+    The status of the job.
+    """
+
+    completed_at: typing.Optional[dt.datetime] = pydantic.Field(default=None)
+    """
+    The unix timestamp of when the job was completed.
+    """
+
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    The metadata of the job.
+    """
+
+    job_type: typing.Optional[JobType] = None
+    id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The human-friendly ID of the Job
+    """
+
+    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/types/job_type.py

@@ -2,4 +2,4 @@
 
 import typing
 
-JobType = typing.Union[typing.Literal["job", "run"], typing.Any]
+JobType = typing.Union[typing.Literal["job", "run", "batch"], typing.Any]

{letta_client-0.1.116.dist-info → letta_client-0.1.117.dist-info}/METADATA

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

{letta_client-0.1.116.dist-info → letta_client-0.1.117.dist-info}/RECORD

@@ -1,8 +1,8 @@
-letta_client/__init__.py,sha256=8odG3X3w-YCoHYdIinxLvVvTRJh6e2vrmeTjerbSnXk,16164
+letta_client/__init__.py,sha256=oU-1Vq1arydqXE9q_Rp4B7VshBLJO4C3ZIZ6Sg2Q70s,16174
 letta_client/agents/__init__.py,sha256=CveigJGrnkw3yZ8S9yZ2DpK1HV0v1fU-khsiLJJ0uaU,1452
 letta_client/agents/blocks/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 letta_client/agents/blocks/client.py,sha256=u5zvutxoH_DqfSLWhRtNSRBC9_ezQDx682cxkxDz3JA,23822
-letta_client/agents/client.py,sha256=VGKkSVTJ5j52fIh7PSNmMBD7gd0ZBu3dYcDPRoYU6as,103492
+letta_client/agents/client.py,sha256=ku9SdcBr0WmR5pjwD1kfq4EMuH-BM3WROO-uda2pul4,93545
 letta_client/agents/context/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 letta_client/agents/context/client.py,sha256=GKKvoG4N_K8Biz9yDjeIHpFG0C8Cwc7tHmEX3pTL_9U,4815
 letta_client/agents/core_memory/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
@@ -38,7 +38,7 @@ letta_client/agents/types/agents_search_request_search_item_zero.py,sha256=tGjwn
 letta_client/agents/types/agents_search_response.py,sha256=AQJVKps-bjCx2ujqESzW1Iy9ZYFS17hH_UFIeBeK4S8,815
 letta_client/agents/types/create_agent_request_tool_rules_item.py,sha256=L3FNsFTG9kVmuPbQhbCKNg3H2E5bB2Rgp92gWmGd-LM,689
 letta_client/agents/types/update_agent_tool_rules_item.py,sha256=k9MmcVPsK-EGl8XlT3JQwdlBNLgpGw528jmi8fCFS7g,682
-letta_client/base_client.py,sha256=G8Ag35gITyN71r_CDrb_hsmJ3PlHPgIK6cCEMU2fVw0,9253
+letta_client/base_client.py,sha256=37vSxqxIwuotfwpLaJWyqePXUUoNt7ZS6TgJcp4G77c,9503
 letta_client/blocks/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 letta_client/blocks/client.py,sha256=LE9dsHaBxFLC3G035f0VpNDG7XKWRK8y9OXpeFCMvUw,30082
 letta_client/client.py,sha256=k2mZqqEWciVmEQHgipjCK4kQILk74hpSqzcdNwdql9A,21212
@@ -53,7 +53,7 @@ letta_client/client_side_access_tokens/types/client_side_access_tokens_create_cl
 letta_client/client_side_access_tokens/types/client_side_access_tokens_create_client_side_access_token_response_policy_data_item_access_item.py,sha256=vefuUVB3pLCbCNBJACXMsEKZuwhMAGvLwSU3TU3Q4l4,275
 letta_client/core/__init__.py,sha256=OKbX2aCZXgHCDUsCouqv-OiX32xA6eFFCKIUH9M5Vzk,1591
 letta_client/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-letta_client/core/client_wrapper.py,sha256=60C6g_Z9X2ZqS7osrPIqdXkJqLaAQOwLKTN3nmztSM0,1998
+letta_client/core/client_wrapper.py,sha256=S7b3SY_tgDb58OaozkpNkuqGQ2s5L2LpUl83woui6nM,1998
 letta_client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 letta_client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 letta_client/core/http_client.py,sha256=Z77OIxIbL4OAB2IDqjRq_sYa5yNYAWfmdhdCSSvh6Y4,19552
@@ -88,6 +88,10 @@ letta_client/identities/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPA
 letta_client/identities/client.py,sha256=kFcShBwJF3RSmOcEVBYM1-IFTFWD3zvCD0JsHSWDeh4,40986
 letta_client/jobs/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 letta_client/jobs/client.py,sha256=z1Zq6dGs2xbf3EAFuD3-m-qbpbUeqpCBYqtIFKkGoMk,15622
+letta_client/messages/__init__.py,sha256=YH5-krRUPFJQGtFgUr_krZfjlEHcRUAIJZ5J1sIE9bM,110
+letta_client/messages/batches/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
+letta_client/messages/batches/client.py,sha256=cG5Cq7MwNCtYN3wnQlo4feur9T988XU6lhA7u35ai0E,16797
+letta_client/messages/client.py,sha256=MTBnxo2irqz8Y3i0iDEtW6SWYOIG4XjEqUZRur3edrQ,5016
 letta_client/models/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 letta_client/models/client.py,sha256=Rd9IHjSdXRzzZyabpq8pDTc9XDnwLPnmm5by335g1D0,6306
 letta_client/projects/__init__.py,sha256=CccxELBuPLqC6SfAJMNP3I4tEzYBNNA9CUrhuPk-TVI,243
@@ -123,7 +127,7 @@ letta_client/tools/types/add_mcp_server_request.py,sha256=EieZjfOT95sjkpxXdqy7gl
 letta_client/tools/types/add_mcp_server_response_item.py,sha256=TWdsKqGb1INhYtpGnAckz0Pw4nZShumSp4pfocRfxCA,270
 letta_client/tools/types/delete_mcp_server_response_item.py,sha256=MeZObU-7tMSCd-S5yuUjNDse6A1hUz1LLjbko0pXaro,273
 letta_client/tools/types/list_mcp_servers_response_value.py,sha256=AIoXu4bO8QNSU7zjL1jj0Rg4313wVtPaTt13W0aevLQ,273
-letta_client/types/__init__.py,sha256=WL1kUXfr7I6CDtXjtiw05YOpJ-7DYHFKslCKkoQgw8I,19858
+letta_client/types/__init__.py,sha256=X-4mwCyHwKdCmS6dQZeoaZT2fCR6ChlNl9mRH1IAdBI,19827
 letta_client/types/action_model.py,sha256=y1e2XMv3skFaNJIBdYoBKgiORzGh05aOVvu-qVR9uHg,1240
 letta_client/types/action_parameters_model.py,sha256=LgKf5aPZG3-OHGxFdXiSokIDgce8c02xPYIAY05VgW8,828
 letta_client/types/action_response_model.py,sha256=yq2Fd9UU8j7vvtE3VqXUoRRvDzWcfJPj_95ynGdeHCs,824
@@ -144,6 +148,7 @@ letta_client/types/auth_response.py,sha256=jtG9Nn0voJcOWkBtvnuGGwhpUhYz9A8O7soOJ
 letta_client/types/auth_scheme_field.py,sha256=W4-qgKtKUSpBHaSvjLyzLybOIsGo7Ggk4VECpsoPnqQ,881
 letta_client/types/bad_request_error_body.py,sha256=E4_eWEc9xeW9BkXGViBDrevV8Jf6PjgEweeGS3vJLD4,567
 letta_client/types/base_tool_rule_schema.py,sha256=FbnJy6gb8wY_DPiU3Gs-u1Ol_l4K7-nAmPTc1oR3kOo,582
+letta_client/types/batch_job.py,sha256=SyBdYiC0sI2FqtEjaSfecJsfrV-TJWpxzrv4uWIwbeI,1759
 letta_client/types/block.py,sha256=J8McqSpellhd-KsPYontU8DYg3YV41_fQW5rR-85qMk,2900
 letta_client/types/block_update.py,sha256=oIgxvSnav5vxztBdslRMiWOgRaAp3dh43pinZpoLzxk,1496
 letta_client/types/chat_completion_assistant_message_param.py,sha256=QwxAJ9RQqxtZKnt6g6RfDppuMIt-1RAIlpnfSrVdHgg,1219
@@ -235,10 +240,9 @@ letta_client/types/input_audio_format.py,sha256=QQFfndI9w66wIbGyHwfmJnk2bEJDPmEs
 letta_client/types/internal_server_error_body.py,sha256=xR9n1zptgmImbH6apQAuwBblYOWAYNLFzY8s0SUcEug,653
 letta_client/types/job.py,sha256=VJBdFIY0rwqh4hObTchlU2jrloTjZwUEA44pNtY_JBg,2321
 letta_client/types/job_status.py,sha256=lX5Q0QMQFnw-WiirqHD6kgBvGr6I7r8rKLnMJdqhlT8,239
-letta_client/types/job_type.py,sha256=Roa04Ry0I-8YMYcDHiHSQwqBavZyPonzkZtjf098e-Q,145
+letta_client/types/job_type.py,sha256=HXYrfzPwxI54PqV7OVcMhewSJ_pBNHc14s9LcExr7Ss,154
 letta_client/types/json_schema.py,sha256=EHcLKBSGRsSzCKTpujKFHylcLJG6ODQIBrjQkU4lWDQ,870
 letta_client/types/letta_batch_request.py,sha256=HdIuaaUaqINbau98jPqyIc7Ge84VlCf6VhAQpIhCFvQ,1354
-letta_client/types/letta_batch_response.py,sha256=i1KP1aZh710acu5bKNmcmdiheuPrKbkJr4_e2K2ki_k,1161
 letta_client/types/letta_message_content_union.py,sha256=YxzyXKxUMeqbqWOlDs9LC8HUiqEhgkNCV9a76GS3spg,486
 letta_client/types/letta_message_union.py,sha256=TTQwlur2CZNdZ466Nb_2TFcSFXrgoMliaNzD33t7Ktw,603
 letta_client/types/letta_request.py,sha256=bCPDRJhSJSo5eILJp0mTw_k26O3dZL1vChfAcaZ0rE8,1240
@@ -348,6 +352,6 @@ letta_client/voice/__init__.py,sha256=7hX85553PiRMtIMM12a0DSoFzsglNiUziYR2ekS84Q
 letta_client/voice/client.py,sha256=STjswa5oOLoP59QwTJvQwi73kgn0UzKOaXc2CsTRI4k,6912
 letta_client/voice/types/__init__.py,sha256=FRc3iKRTONE4N8Lf1IqvnqWZ2kXdrFFvkL7PxVcR8Ew,212
 letta_client/voice/types/create_voice_chat_completions_request_body.py,sha256=ZLfKgNK1T6IAwLEvaBVFfy7jEAoPUXP28n-nfmHkklc,391
-letta_client-0.1.116.dist-info/METADATA,sha256=gzBG41gPHxqDKIDINM2rLVUrGZdzl37tSYjutqsVhL4,5042
-letta_client-0.1.116.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-letta_client-0.1.116.dist-info/RECORD,,
+letta_client-0.1.117.dist-info/METADATA,sha256=-ZUQYGScs53WPyFaYLTG_5jN8S-dPf7W4RKvujbJHpI,5042
+letta_client-0.1.117.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+letta_client-0.1.117.dist-info/RECORD,,

letta_client/types/letta_batch_response.py

@@ -1,44 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from ..core.unchecked_base_model import UncheckedBaseModel
-import pydantic
-from .job_status import JobStatus
-import datetime as dt
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import typing
-
-
-class LettaBatchResponse(UncheckedBaseModel):
-    batch_id: str = pydantic.Field()
-    """
-    A unique identifier for this batch request.
-    """
-
-    status: JobStatus = pydantic.Field()
-    """
-    The current status of the batch request.
-    """
-
-    agent_count: int = pydantic.Field()
-    """
-    The number of agents in the batch request.
-    """
-
-    last_polled_at: dt.datetime = pydantic.Field()
-    """
-    The timestamp when the batch was last polled for updates.
-    """
-
-    created_at: dt.datetime = pydantic.Field()
-    """
-    The timestamp when the batch request was created.
-    """
-
-    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