letta-client 0.1.217__py3-none-any.whl → 0.1.219__py3-none-any.whl

letta_client/agents/__init__.py

@@ -16,7 +16,20 @@ from .types import (
     UpdateAgentResponseFormat,
     UpdateAgentToolRulesItem,
 )
-from . import blocks, context, core_memory, groups, memory_variables, messages, passages, sources, templates, tools
+from . import (
+    blocks,
+    context,
+    core_memory,
+    files,
+    folders,
+    groups,
+    memory_variables,
+    messages,
+    passages,
+    sources,
+    templates,
+    tools,
+)
 from .memory_variables import MemoryVariablesListResponse
 from .messages import (
     LettaStreamingResponse,
@@ -52,6 +65,8 @@ __all__ = [
     "blocks",
     "context",
     "core_memory",
+    "files",
+    "folders",
     "groups",
     "memory_variables",
     "messages",

letta_client/agents/client.py

@@ -5,6 +5,8 @@ from ..core.client_wrapper import SyncClientWrapper
 from .context.client import ContextClient
 from .tools.client import ToolsClient
 from .sources.client import SourcesClient
+from .folders.client import FoldersClient
+from .files.client import FilesClient
 from .core_memory.client import CoreMemoryClient
 from .blocks.client import BlocksClient
 from .passages.client import PassagesClient
@@ -39,6 +41,8 @@ from ..core.client_wrapper import AsyncClientWrapper
 from .context.client import AsyncContextClient
 from .tools.client import AsyncToolsClient
 from .sources.client import AsyncSourcesClient
+from .folders.client import AsyncFoldersClient
+from .files.client import AsyncFilesClient
 from .core_memory.client import AsyncCoreMemoryClient
 from .blocks.client import AsyncBlocksClient
 from .passages.client import AsyncPassagesClient
@@ -57,6 +61,8 @@ class AgentsClient:
         self.context = ContextClient(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)
+        self.files = FilesClient(client_wrapper=self._client_wrapper)
         self.core_memory = CoreMemoryClient(client_wrapper=self._client_wrapper)
         self.blocks = BlocksClient(client_wrapper=self._client_wrapper)
         self.passages = PassagesClient(client_wrapper=self._client_wrapper)
@@ -1019,199 +1025,6 @@ class AgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def close_all_open_files(
-        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[str]:
-        """
-        Closes all currently open files for a given agent.
-
-        This endpoint updates the file state for the agent so that no files are marked as open.
-        Typically used to reset the working memory view for the agent.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[str]
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-        client.agents.close_all_open_files(
-            agent_id="agent_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/files/close-all",
-            method="PATCH",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[str],
-                    construct_type(
-                        type_=typing.List[str],  # 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 open_file(
-        self, agent_id: str, file_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[str]:
-        """
-        Opens a specific file for a given agent.
-
-        This endpoint marks a specific file as open in the agent's file state.
-        The file will be included in the agent's working memory view.
-        Returns a list of file names that were closed due to LRU eviction.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        file_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[str]
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-        client.agents.open_file(
-            agent_id="agent_id",
-            file_id="file_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/files/{jsonable_encoder(file_id)}/open",
-            method="PATCH",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[str],
-                    construct_type(
-                        type_=typing.List[str],  # 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 close_file(
-        self, agent_id: str, file_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.Optional[typing.Any]:
-        """
-        Closes a specific file for a given agent.
-
-        This endpoint marks a specific file as closed in the agent's file state.
-        The file will be removed from the agent's working memory view.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        file_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(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-        client.agents.close_file(
-            agent_id="agent_id",
-            file_id="file_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/files/{jsonable_encoder(file_id)}/close",
-            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)
-
     def summarize_agent_conversation(
         self, agent_id: str, *, max_message_length: int, request_options: typing.Optional[RequestOptions] = None
     ) -> AgentState:
@@ -1373,6 +1186,8 @@ class AsyncAgentsClient:
         self.context = AsyncContextClient(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)
+        self.files = AsyncFilesClient(client_wrapper=self._client_wrapper)
         self.core_memory = AsyncCoreMemoryClient(client_wrapper=self._client_wrapper)
         self.blocks = AsyncBlocksClient(client_wrapper=self._client_wrapper)
         self.passages = AsyncPassagesClient(client_wrapper=self._client_wrapper)
@@ -2399,223 +2214,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 close_all_open_files(
-        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[str]:
-        """
-        Closes all currently open files for a given agent.
-
-        This endpoint updates the file state for the agent so that no files are marked as open.
-        Typically used to reset the working memory view for the agent.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[str]
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.close_all_open_files(
-                agent_id="agent_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/files/close-all",
-            method="PATCH",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[str],
-                    construct_type(
-                        type_=typing.List[str],  # 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 open_file(
-        self, agent_id: str, file_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[str]:
-        """
-        Opens a specific file for a given agent.
-
-        This endpoint marks a specific file as open in the agent's file state.
-        The file will be included in the agent's working memory view.
-        Returns a list of file names that were closed due to LRU eviction.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        file_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[str]
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.open_file(
-                agent_id="agent_id",
-                file_id="file_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/files/{jsonable_encoder(file_id)}/open",
-            method="PATCH",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[str],
-                    construct_type(
-                        type_=typing.List[str],  # 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 close_file(
-        self, agent_id: str, file_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.Optional[typing.Any]:
-        """
-        Closes a specific file for a given agent.
-
-        This endpoint marks a specific file as closed in the agent's file state.
-        The file will be removed from the agent's working memory view.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        file_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(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.close_file(
-                agent_id="agent_id",
-                file_id="file_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/files/{jsonable_encoder(file_id)}/close",
-            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)
-
     async def summarize_agent_conversation(
         self, agent_id: str, *, max_message_length: int, request_options: typing.Optional[RequestOptions] = None
     ) -> AgentState:

letta_client/agents/files/__init__.py

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