letta-client 1.0.0a1__py3-none-any.whl → 1.0.0a3__py3-none-any.whl

letta_client/agents/tools/raw_client.py

@@ -13,6 +13,7 @@ from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.agent_state import AgentState
 from ...types.http_validation_error import HttpValidationError
 from ...types.tool import Tool
+from .types.tools_list_request_order import ToolsListRequestOrder
 
 
 class RawToolsClient:
@@ -20,7 +21,15 @@ class RawToolsClient:
         self._client_wrapper = client_wrapper
 
     def list(
-        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
+        self,
+        agent_id: str,
+        *,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        order: typing.Optional[ToolsListRequestOrder] = None,
+        order_by: typing.Optional[typing.Literal["created_at"]] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[typing.List[Tool]]:
         """
         Get tools from an existing agent
@@ -29,6 +38,21 @@ class RawToolsClient:
         ----------
         agent_id : str
 
+        before : typing.Optional[str]
+            Tool ID cursor for pagination. Returns tools that come before this tool ID in the specified sort order
+
+        after : typing.Optional[str]
+            Tool ID cursor for pagination. Returns tools that come after this tool ID in the specified sort order
+
+        limit : typing.Optional[int]
+            Maximum number of tools to return
+
+        order : typing.Optional[ToolsListRequestOrder]
+            Sort order for tools by creation time. 'asc' for oldest first, 'desc' for newest first
+
+        order_by : typing.Optional[typing.Literal["created_at"]]
+            Field to sort by
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -40,6 +64,13 @@ class RawToolsClient:
         _response = self._client_wrapper.httpx_client.request(
             f"v1/agents/{jsonable_encoder(agent_id)}/tools",
             method="GET",
+            params={
+                "before": before,
+                "after": after,
+                "limit": limit,
+                "order": order,
+                "order_by": order_by,
+            },
             request_options=request_options,
         )
         try:
@@ -237,7 +268,15 @@ class AsyncRawToolsClient:
         self._client_wrapper = client_wrapper
 
     async def list(
-        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
+        self,
+        agent_id: str,
+        *,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        order: typing.Optional[ToolsListRequestOrder] = None,
+        order_by: typing.Optional[typing.Literal["created_at"]] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[typing.List[Tool]]:
         """
         Get tools from an existing agent
@@ -246,6 +285,21 @@ class AsyncRawToolsClient:
         ----------
         agent_id : str
 
+        before : typing.Optional[str]
+            Tool ID cursor for pagination. Returns tools that come before this tool ID in the specified sort order
+
+        after : typing.Optional[str]
+            Tool ID cursor for pagination. Returns tools that come after this tool ID in the specified sort order
+
+        limit : typing.Optional[int]
+            Maximum number of tools to return
+
+        order : typing.Optional[ToolsListRequestOrder]
+            Sort order for tools by creation time. 'asc' for oldest first, 'desc' for newest first
+
+        order_by : typing.Optional[typing.Literal["created_at"]]
+            Field to sort by
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -257,6 +311,13 @@ class AsyncRawToolsClient:
         _response = await self._client_wrapper.httpx_client.request(
             f"v1/agents/{jsonable_encoder(agent_id)}/tools",
             method="GET",
+            params={
+                "before": before,
+                "after": after,
+                "limit": limit,
+                "order": order,
+                "order_by": order_by,
+            },
             request_options=request_options,
         )
         try:

letta_client/agents/tools/types/__init__.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .tools_list_request_order import ToolsListRequestOrder
+
+__all__ = ["ToolsListRequestOrder"]

letta_client/agents/tools/types/tools_list_request_order.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ToolsListRequestOrder = typing.Union[typing.Literal["asc", "desc"], typing.Any]

letta_client/archives/client.py

@@ -77,7 +77,14 @@ class ArchivesClient:
             project="YOUR_PROJECT",
             token="YOUR_TOKEN",
         )
-        client.archives.list_archives()
+        client.archives.list_archives(
+            before="before",
+            after="after",
+            limit=1,
+            order="asc",
+            name="name",
+            agent_id="agent_id",
+        )
         """
         _response = self._raw_client.list_archives(
             before=before,
@@ -244,7 +251,14 @@ class AsyncArchivesClient:
 
 
         async def main() -> None:
-            await client.archives.list_archives()
+            await client.archives.list_archives(
+                before="before",
+                after="after",
+                limit=1,
+                order="asc",
+                name="name",
+                agent_id="agent_id",
+            )
 
 
         asyncio.run(main())

letta_client/base_client.py

@@ -7,6 +7,7 @@ from .agents.client import AgentsClient, AsyncAgentsClient
 from .archives.client import ArchivesClient, AsyncArchivesClient
 from .batches.client import AsyncBatchesClient, BatchesClient
 from .blocks.client import AsyncBlocksClient, BlocksClient
+from .chat.client import AsyncChatClient, ChatClient
 from .client_side_access_tokens.client import AsyncClientSideAccessTokensClient, ClientSideAccessTokensClient
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import LettaEnvironment
@@ -102,6 +103,7 @@ class LettaBase:
         self.sources = SourcesClient(client_wrapper=self._client_wrapper)
         self.folders = FoldersClient(client_wrapper=self._client_wrapper)
         self.agents = AgentsClient(client_wrapper=self._client_wrapper)
+        self.chat = ChatClient(client_wrapper=self._client_wrapper)
         self.groups = GroupsClient(client_wrapper=self._client_wrapper)
         self.identities = IdentitiesClient(client_wrapper=self._client_wrapper)
         self.models = ModelsClient(client_wrapper=self._client_wrapper)
@@ -194,6 +196,7 @@ class AsyncLettaBase:
         self.sources = AsyncSourcesClient(client_wrapper=self._client_wrapper)
         self.folders = AsyncFoldersClient(client_wrapper=self._client_wrapper)
         self.agents = AsyncAgentsClient(client_wrapper=self._client_wrapper)
+        self.chat = AsyncChatClient(client_wrapper=self._client_wrapper)
         self.groups = AsyncGroupsClient(client_wrapper=self._client_wrapper)
         self.identities = AsyncIdentitiesClient(client_wrapper=self._client_wrapper)
         self.models = AsyncModelsClient(client_wrapper=self._client_wrapper)

letta_client/batches/client.py

@@ -76,7 +76,12 @@ class BatchesClient:
             project="YOUR_PROJECT",
             token="YOUR_TOKEN",
         )
-        client.batches.list()
+        client.batches.list(
+            before="before",
+            after="after",
+            limit=1,
+            order="asc",
+        )
         """
         _response = self._raw_client.list(
             before=before, after=after, limit=limit, order=order, order_by=order_by, request_options=request_options
@@ -275,7 +280,12 @@ class AsyncBatchesClient:
 
 
         async def main() -> None:
-            await client.batches.list()
+            await client.batches.list(
+                before="before",
+                after="after",
+                limit=1,
+                order="asc",
+            )
 
 
         asyncio.run(main())

letta_client/batches/messages/client.py

@@ -79,6 +79,11 @@ class MessagesClient:
         )
         client.batches.messages.list(
             batch_id="batch_id",
+            before="before",
+            after="after",
+            limit=1,
+            order="asc",
+            agent_id="agent_id",
         )
         """
         _response = self._raw_client.list(
@@ -169,6 +174,11 @@ class AsyncMessagesClient:
         async def main() -> None:
             await client.batches.messages.list(
                 batch_id="batch_id",
+                before="before",
+                after="after",
+                limit=1,
+                order="asc",
+                agent_id="agent_id",
             )
 
 

letta_client/blocks/agents/client.py

@@ -80,6 +80,10 @@ class AgentsClient:
         )
         client.blocks.agents.list(
             block_id="block_id",
+            before="before",
+            after="after",
+            limit=1,
+            order="asc",
         )
         """
         _response = self._raw_client.list(
@@ -171,6 +175,10 @@ class AsyncAgentsClient:
         async def main() -> None:
             await client.blocks.agents.list(
                 block_id="block_id",
+                before="before",
+                after="after",
+                limit=1,
+                order="asc",
             )
 
 

letta_client/blocks/client.py

@@ -121,7 +121,22 @@ class BlocksClient:
             project="YOUR_PROJECT",
             token="YOUR_TOKEN",
         )
-        client.blocks.list()
+        client.blocks.list(
+            label="label",
+            templates_only=True,
+            name="name",
+            identity_id="identity_id",
+            project_id="project_id",
+            limit=1,
+            before="before",
+            after="after",
+            order="asc",
+            label_search="label_search",
+            description_search="description_search",
+            value_search="value_search",
+            connected_to_agents_count_gt=1,
+            connected_to_agents_count_lt=1,
+        )
         """
         _response = self._raw_client.list(
             label=label,
@@ -556,7 +571,22 @@ class AsyncBlocksClient:
 
 
         async def main() -> None:
-            await client.blocks.list()
+            await client.blocks.list(
+                label="label",
+                templates_only=True,
+                name="name",
+                identity_id="identity_id",
+                project_id="project_id",
+                limit=1,
+                before="before",
+                after="after",
+                order="asc",
+                label_search="label_search",
+                description_search="description_search",
+                value_search="value_search",
+                connected_to_agents_count_gt=1,
+                connected_to_agents_count_lt=1,
+            )
 
 
         asyncio.run(main())

letta_client/chat/__init__.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .types import ChatCompletionRequestMessagesItem, ChatCompletionRequestStop
+
+__all__ = ["ChatCompletionRequestMessagesItem", "ChatCompletionRequestStop"]

letta_client/chat/client.py

@@ -0,0 +1,255 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from ..types.chat_completion import ChatCompletion
+from .raw_client import AsyncRawChatClient, RawChatClient
+from .types.chat_completion_request_messages_item import ChatCompletionRequestMessagesItem
+from .types.chat_completion_request_stop import ChatCompletionRequestStop
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class ChatClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawChatClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawChatClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawChatClient
+        """
+        return self._raw_client
+
+    def create_chat_completion(
+        self,
+        *,
+        model: str,
+        messages: typing.Sequence[ChatCompletionRequestMessagesItem],
+        temperature: typing.Optional[float] = OMIT,
+        top_p: typing.Optional[float] = OMIT,
+        n: typing.Optional[int] = OMIT,
+        stream: typing.Optional[bool] = OMIT,
+        stop: typing.Optional[ChatCompletionRequestStop] = OMIT,
+        max_tokens: typing.Optional[int] = OMIT,
+        presence_penalty: typing.Optional[float] = OMIT,
+        frequency_penalty: typing.Optional[float] = OMIT,
+        user: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ChatCompletion:
+        """
+        Create a chat completion using a Letta agent (OpenAI-compatible).
+
+        This endpoint provides full OpenAI API compatibility. The agent is selected based on:
+        - The 'model' parameter in the request (should contain an agent ID in format 'agent-...')
+
+        When streaming is enabled (stream=true), the response will be Server-Sent Events
+        with ChatCompletionChunk objects.
+
+        Parameters
+        ----------
+        model : str
+            ID of the model to use
+
+        messages : typing.Sequence[ChatCompletionRequestMessagesItem]
+            Messages comprising the conversation so far
+
+        temperature : typing.Optional[float]
+            Sampling temperature
+
+        top_p : typing.Optional[float]
+            Nucleus sampling parameter
+
+        n : typing.Optional[int]
+            Number of chat completion choices to generate
+
+        stream : typing.Optional[bool]
+            Whether to stream back partial progress
+
+        stop : typing.Optional[ChatCompletionRequestStop]
+            Sequences where the API will stop generating
+
+        max_tokens : typing.Optional[int]
+            Maximum number of tokens to generate
+
+        presence_penalty : typing.Optional[float]
+            Presence penalty
+
+        frequency_penalty : typing.Optional[float]
+            Frequency penalty
+
+        user : typing.Optional[str]
+            A unique identifier representing your end-user
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ChatCompletion
+            Successful response
+
+        Examples
+        --------
+        from letta_client import ChatCompletionDeveloperMessageParam, Letta
+
+        client = Letta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+        client.chat.create_chat_completion(
+            model="model",
+            messages=[
+                ChatCompletionDeveloperMessageParam(
+                    content="content",
+                )
+            ],
+        )
+        """
+        _response = self._raw_client.create_chat_completion(
+            model=model,
+            messages=messages,
+            temperature=temperature,
+            top_p=top_p,
+            n=n,
+            stream=stream,
+            stop=stop,
+            max_tokens=max_tokens,
+            presence_penalty=presence_penalty,
+            frequency_penalty=frequency_penalty,
+            user=user,
+            request_options=request_options,
+        )
+        return _response.data
+
+
+class AsyncChatClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawChatClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawChatClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawChatClient
+        """
+        return self._raw_client
+
+    async def create_chat_completion(
+        self,
+        *,
+        model: str,
+        messages: typing.Sequence[ChatCompletionRequestMessagesItem],
+        temperature: typing.Optional[float] = OMIT,
+        top_p: typing.Optional[float] = OMIT,
+        n: typing.Optional[int] = OMIT,
+        stream: typing.Optional[bool] = OMIT,
+        stop: typing.Optional[ChatCompletionRequestStop] = OMIT,
+        max_tokens: typing.Optional[int] = OMIT,
+        presence_penalty: typing.Optional[float] = OMIT,
+        frequency_penalty: typing.Optional[float] = OMIT,
+        user: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ChatCompletion:
+        """
+        Create a chat completion using a Letta agent (OpenAI-compatible).
+
+        This endpoint provides full OpenAI API compatibility. The agent is selected based on:
+        - The 'model' parameter in the request (should contain an agent ID in format 'agent-...')
+
+        When streaming is enabled (stream=true), the response will be Server-Sent Events
+        with ChatCompletionChunk objects.
+
+        Parameters
+        ----------
+        model : str
+            ID of the model to use
+
+        messages : typing.Sequence[ChatCompletionRequestMessagesItem]
+            Messages comprising the conversation so far
+
+        temperature : typing.Optional[float]
+            Sampling temperature
+
+        top_p : typing.Optional[float]
+            Nucleus sampling parameter
+
+        n : typing.Optional[int]
+            Number of chat completion choices to generate
+
+        stream : typing.Optional[bool]
+            Whether to stream back partial progress
+
+        stop : typing.Optional[ChatCompletionRequestStop]
+            Sequences where the API will stop generating
+
+        max_tokens : typing.Optional[int]
+            Maximum number of tokens to generate
+
+        presence_penalty : typing.Optional[float]
+            Presence penalty
+
+        frequency_penalty : typing.Optional[float]
+            Frequency penalty
+
+        user : typing.Optional[str]
+            A unique identifier representing your end-user
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ChatCompletion
+            Successful response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta, ChatCompletionDeveloperMessageParam
+
+        client = AsyncLetta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.chat.create_chat_completion(
+                model="model",
+                messages=[
+                    ChatCompletionDeveloperMessageParam(
+                        content="content",
+                    )
+                ],
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_chat_completion(
+            model=model,
+            messages=messages,
+            temperature=temperature,
+            top_p=top_p,
+            n=n,
+            stream=stream,
+            stop=stop,
+            max_tokens=max_tokens,
+            presence_penalty=presence_penalty,
+            frequency_penalty=frequency_penalty,
+            user=user,
+            request_options=request_options,
+        )
+        return _response.data