letta-client 0.1.112__tar.gz → 0.1.113__tar.gz

{letta_client-0.1.112 → letta_client-0.1.113}/PKG-INFO

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

{letta_client-0.1.112 → letta_client-0.1.113}/pyproject.toml

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

{letta_client-0.1.112 → letta_client-0.1.113}/src/letta_client/__init__.py

@@ -113,6 +113,8 @@ from .types import (
     JobStatus,
     JobType,
     JsonSchema,
+    LettaBatchRequest,
+    LettaBatchResponse,
     LettaMessageContentUnion,
     LettaMessageUnion,
     LettaRequest,
@@ -391,6 +393,8 @@ __all__ = [
     "JobType",
     "JsonSchema",
     "Letta",
+    "LettaBatchRequest",
+    "LettaBatchResponse",
     "LettaEnvironment",
     "LettaMessageContentUnion",
     "LettaMessageUnion",

{letta_client-0.1.112 → letta_client-0.1.113}/src/letta_client/agents/client.py

@@ -31,6 +31,8 @@ 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
@@ -1158,6 +1160,83 @@ 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 search(
         self,
         *,
@@ -2427,6 +2506,96 @@ 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 search(
         self,
         *,

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

{letta_client-0.1.112 → letta_client-0.1.113}/src/letta_client/types/__init__.py

@@ -112,6 +112,8 @@ from .job import Job
 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
@@ -334,6 +336,8 @@ __all__ = [
     "JobStatus",
     "JobType",
     "JsonSchema",
+    "LettaBatchRequest",
+    "LettaBatchResponse",
     "LettaMessageContentUnion",
     "LettaMessageUnion",
     "LettaRequest",

letta_client-0.1.113/src/letta_client/types/letta_batch_request.py

@@ -0,0 +1,43 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.unchecked_base_model import UncheckedBaseModel
+import typing
+from .message_create import MessageCreate
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class LettaBatchRequest(UncheckedBaseModel):
+    messages: typing.List[MessageCreate] = pydantic.Field()
+    """
+    The messages to be sent to the agent.
+    """
+
+    use_assistant_message: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Whether the server should parse specific tool call arguments (default `send_message`) as `AssistantMessage` objects.
+    """
+
+    assistant_message_tool_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The name of the designated message tool.
+    """
+
+    assistant_message_tool_kwarg: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The name of the message argument in the designated message tool.
+    """
+
+    agent_id: str = pydantic.Field()
+    """
+    The ID of the agent to send this batch request for
+    """
+
+    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.113/src/letta_client/types/letta_batch_response.py

@@ -0,0 +1,44 @@
+# 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