letta-client 0.1.318__py3-none-any.whl → 0.1.320__py3-none-any.whl

letta_client/archives/raw_client.py

@@ -0,0 +1,448 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.http_response import AsyncHttpResponse, HttpResponse
+from ..core.jsonable_encoder import jsonable_encoder
+from ..core.request_options import RequestOptions
+from ..core.unchecked_base_model import construct_type
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.archive import Archive
+from ..types.http_validation_error import HttpValidationError
+from .types.list_archives_request_order import ListArchivesRequestOrder
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawArchivesClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list_archives(
+        self,
+        *,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        order: typing.Optional[ListArchivesRequestOrder] = None,
+        name: typing.Optional[str] = None,
+        agent_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[typing.List[Archive]]:
+        """
+        Get a list of all archives for the current organization with optional filters and pagination.
+
+        Parameters
+        ----------
+        before : typing.Optional[str]
+            Archive ID cursor for pagination. Returns archives that come before this archive ID in the specified sort order
+
+        after : typing.Optional[str]
+            Archive ID cursor for pagination. Returns archives that come after this archive ID in the specified sort order
+
+        limit : typing.Optional[int]
+            Maximum number of archives to return
+
+        order : typing.Optional[ListArchivesRequestOrder]
+            Sort order for archives by creation time. 'asc' for oldest first, 'desc' for newest first
+
+        name : typing.Optional[str]
+            Filter by archive name (exact match)
+
+        agent_id : typing.Optional[str]
+            Only archives attached to this agent ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[typing.List[Archive]]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/archives/",
+            method="GET",
+            params={
+                "before": before,
+                "after": after,
+                "limit": limit,
+                "order": order,
+                "name": name,
+                "agent_id": agent_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    typing.List[Archive],
+                    construct_type(
+                        type_=typing.List[Archive],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=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, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def create_archive(
+        self,
+        *,
+        name: str,
+        description: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[Archive]:
+        """
+        Create a new archive.
+
+        Parameters
+        ----------
+        name : str
+
+        description : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[Archive]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/archives/",
+            method="POST",
+            json={
+                "name": name,
+                "description": description,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    Archive,
+                    construct_type(
+                        type_=Archive,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=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, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def modify_archive(
+        self,
+        archive_id: str,
+        *,
+        name: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[Archive]:
+        """
+        Update an existing archive's name and/or description.
+
+        Parameters
+        ----------
+        archive_id : str
+
+        name : typing.Optional[str]
+
+        description : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[Archive]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/archives/{jsonable_encoder(archive_id)}",
+            method="PATCH",
+            json={
+                "name": name,
+                "description": description,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    Archive,
+                    construct_type(
+                        type_=Archive,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=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, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawArchivesClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list_archives(
+        self,
+        *,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        order: typing.Optional[ListArchivesRequestOrder] = None,
+        name: typing.Optional[str] = None,
+        agent_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[typing.List[Archive]]:
+        """
+        Get a list of all archives for the current organization with optional filters and pagination.
+
+        Parameters
+        ----------
+        before : typing.Optional[str]
+            Archive ID cursor for pagination. Returns archives that come before this archive ID in the specified sort order
+
+        after : typing.Optional[str]
+            Archive ID cursor for pagination. Returns archives that come after this archive ID in the specified sort order
+
+        limit : typing.Optional[int]
+            Maximum number of archives to return
+
+        order : typing.Optional[ListArchivesRequestOrder]
+            Sort order for archives by creation time. 'asc' for oldest first, 'desc' for newest first
+
+        name : typing.Optional[str]
+            Filter by archive name (exact match)
+
+        agent_id : typing.Optional[str]
+            Only archives attached to this agent ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[typing.List[Archive]]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/archives/",
+            method="GET",
+            params={
+                "before": before,
+                "after": after,
+                "limit": limit,
+                "order": order,
+                "name": name,
+                "agent_id": agent_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    typing.List[Archive],
+                    construct_type(
+                        type_=typing.List[Archive],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=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, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def create_archive(
+        self,
+        *,
+        name: str,
+        description: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[Archive]:
+        """
+        Create a new archive.
+
+        Parameters
+        ----------
+        name : str
+
+        description : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[Archive]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/archives/",
+            method="POST",
+            json={
+                "name": name,
+                "description": description,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    Archive,
+                    construct_type(
+                        type_=Archive,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=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, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def modify_archive(
+        self,
+        archive_id: str,
+        *,
+        name: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[Archive]:
+        """
+        Update an existing archive's name and/or description.
+
+        Parameters
+        ----------
+        archive_id : str
+
+        name : typing.Optional[str]
+
+        description : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[Archive]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/archives/{jsonable_encoder(archive_id)}",
+            method="PATCH",
+            json={
+                "name": name,
+                "description": description,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    Archive,
+                    construct_type(
+                        type_=Archive,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=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, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

letta_client/archives/types/__init__.py

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

letta_client/archives/types/list_archives_request_order.py

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

letta_client/base_client.py

@@ -4,18 +4,17 @@ import typing
 
 import httpx
 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 .client_side_access_tokens.client import AsyncClientSideAccessTokensClient, ClientSideAccessTokensClient
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from .embedding_models.client import AsyncEmbeddingModelsClient, EmbeddingModelsClient
 from .environment import LettaEnvironment
 from .folders.client import AsyncFoldersClient, FoldersClient
 from .groups.client import AsyncGroupsClient, GroupsClient
 from .health.client import AsyncHealthClient, HealthClient
 from .identities.client import AsyncIdentitiesClient, IdentitiesClient
 from .jobs.client import AsyncJobsClient, JobsClient
-from .messages.client import AsyncMessagesClient, MessagesClient
 from .models.client import AsyncModelsClient, ModelsClient
 from .projects.client import AsyncProjectsClient, ProjectsClient
 from .providers.client import AsyncProvidersClient, ProvidersClient
@@ -98,6 +97,7 @@ class LettaBase:
             else httpx.Client(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
+        self.archives = ArchivesClient(client_wrapper=self._client_wrapper)
         self.tools = ToolsClient(client_wrapper=self._client_wrapper)
         self.sources = SourcesClient(client_wrapper=self._client_wrapper)
         self.folders = FoldersClient(client_wrapper=self._client_wrapper)
@@ -105,7 +105,6 @@ class LettaBase:
         self.groups = GroupsClient(client_wrapper=self._client_wrapper)
         self.identities = IdentitiesClient(client_wrapper=self._client_wrapper)
         self.models = ModelsClient(client_wrapper=self._client_wrapper)
-        self.embedding_models = EmbeddingModelsClient(client_wrapper=self._client_wrapper)
         self.blocks = BlocksClient(client_wrapper=self._client_wrapper)
         self.jobs = JobsClient(client_wrapper=self._client_wrapper)
         self.health = HealthClient(client_wrapper=self._client_wrapper)
@@ -115,7 +114,6 @@ class LettaBase:
         self.tags = TagsClient(client_wrapper=self._client_wrapper)
         self.telemetry = TelemetryClient(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)
@@ -191,6 +189,7 @@ class AsyncLettaBase:
             else httpx.AsyncClient(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
+        self.archives = AsyncArchivesClient(client_wrapper=self._client_wrapper)
         self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
         self.sources = AsyncSourcesClient(client_wrapper=self._client_wrapper)
         self.folders = AsyncFoldersClient(client_wrapper=self._client_wrapper)
@@ -198,7 +197,6 @@ class AsyncLettaBase:
         self.groups = AsyncGroupsClient(client_wrapper=self._client_wrapper)
         self.identities = AsyncIdentitiesClient(client_wrapper=self._client_wrapper)
         self.models = AsyncModelsClient(client_wrapper=self._client_wrapper)
-        self.embedding_models = AsyncEmbeddingModelsClient(client_wrapper=self._client_wrapper)
         self.blocks = AsyncBlocksClient(client_wrapper=self._client_wrapper)
         self.jobs = AsyncJobsClient(client_wrapper=self._client_wrapper)
         self.health = AsyncHealthClient(client_wrapper=self._client_wrapper)
@@ -208,7 +206,6 @@ class AsyncLettaBase:
         self.tags = AsyncTagsClient(client_wrapper=self._client_wrapper)
         self.telemetry = AsyncTelemetryClient(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/batches/__init__.py

@@ -2,3 +2,8 @@
 
 # isort: skip_file
 
+from .types import BatchesListRequestOrder
+from . import messages
+from .messages import MessagesListRequestOrder
+
+__all__ = ["BatchesListRequestOrder", "MessagesListRequestOrder", "messages"]