deepagents 0.0.10__py3-none-any.whl → 0.0.11rc1__py3-none-any.whl

deepagents/prompts.py

@@ -406,10 +406,17 @@ When NOT to use the task tool:
 FILESYSTEM_SYSTEM_PROMPT = """## Filesystem Tools `ls`, `read_file`, `write_file`, `edit_file`
 
 You have access to a local, private filesystem which you can interact with using these tools.
-- ls: list all files in the local filesystem
-- read_file: read a file from the local filesystem
-- write_file: write to a file in the local filesystem
-- edit_file: edit a file in the local filesystem"""
+
+You also have access to a longterm filesystem in which you can store files that you want to keep around for longer than the current conversation.
+In order to interact with the longterm filesystem, you can use those same tools, but filenames must be prefixed with the memories/ path.
+
+- ls: list all files in the filesystem (local and longterm)
+- read_file: read a file from the filesystem (local and longterm)
+- write_file: write to a file in the filesystem (local and longterm)
+- edit_file: edit a file in the filesystem (local and longterm)
+
+Remember, to interact with the longterm filesystem, you must prefix the filename with the memories/ path.
+"""
 
 BASE_AGENT_PROMPT = """
 In order to complete the objective that the user asks of you, you have access to a number of standard tools.

deepagents/tools.py

@@ -1,3 +1,4 @@
+from re import L
 from langchain_core.tools import tool, InjectedToolCallId
 from langchain_core.messages import ToolMessage
 from langgraph.types import Command
@@ -11,7 +12,17 @@ from deepagents.prompts import (
     WRITE_FILE_TOOL_DESCRIPTION,
     EDIT_FILE_TOOL_DESCRIPTION,
 )
+from ai_filesystem import FilesystemClient
+import os
 
+def has_memories_prefix(file_path: str) -> bool:
+    return file_path.startswith("memories/")
+
+def append_memories_prefix(file_path: str) -> str:
+    return f"memories/{file_path}"
+
+def strip_memories_prefix(file_path: str) -> str:
+    return file_path.replace("memories/", "")
 
 @tool(description=WRITE_TODOS_TOOL_DESCRIPTION)
 def write_todos(
@@ -30,7 +41,17 @@ def write_todos(
 @tool(description=LIST_FILES_TOOL_DESCRIPTION)
 def ls(state: Annotated[FilesystemState, InjectedState]) -> list[str]:
     """List all files"""
-    return list(state.get("files", {}).keys())
+    files = []
+    files.extend(list(state.get("files", {}).keys()))
+    # Special handling for longterm filesystem
+    if os.getenv("LONGTERM_FILESYSTEM_NAME") and os.getenv("AGENT_FS_API_KEY"):
+        filesystem_client = FilesystemClient(
+            filesystem=os.getenv("LONGTERM_FILESYSTEM_NAME")
+        )
+        file_data_list = filesystem_client._list_files()
+        memories_files = [f"memories/{f.path}" for f in file_data_list]
+        files.extend(memories_files)
+    return files
 
 
 @tool(description=READ_FILE_TOOL_DESCRIPTION)
@@ -40,6 +61,15 @@ def read_file(
     offset: int = 0,
     limit: int = 2000,
 ) -> str:
+    # Special handling for longterm filesystem
+    if os.getenv("LONGTERM_FILESYSTEM_NAME") and os.getenv("AGENT_FS_API_KEY") and has_memories_prefix(file_path):
+        filesystem_client = FilesystemClient(
+            filesystem=os.getenv("LONGTERM_FILESYSTEM_NAME")
+        )
+        file_path = strip_memories_prefix(file_path)
+        content = filesystem_client.read_file(file_path)
+        return content
+
     mock_filesystem = state.get("files", {})
     if file_path not in mock_filesystem:
         return f"Error: File '{file_path}' not found"
@@ -85,14 +115,24 @@ def write_file(
     state: Annotated[FilesystemState, InjectedState],
     tool_call_id: Annotated[str, InjectedToolCallId],
 ) -> Command:
+    # Special handling for longterm filesystem
+    if os.getenv("LONGTERM_FILESYSTEM_NAME") and os.getenv("AGENT_FS_API_KEY") and has_memories_prefix(file_path):
+        filesystem_client = FilesystemClient(
+            filesystem=os.getenv("LONGTERM_FILESYSTEM_NAME")
+        )
+        short_file_path = strip_memories_prefix(file_path)
+        filesystem_client.create_file(short_file_path, content)
+        return Command(
+            update={
+                "messages": [ToolMessage(f"Updated longterm memories file {file_path}", tool_call_id=tool_call_id)]
+            }
+        )
+
     files = state.get("files", {})
-    files[file_path] = content
     return Command(
         update={
             "files": files,
-            "messages": [
-                ToolMessage(f"Updated file {file_path}", tool_call_id=tool_call_id)
-            ],
+            "messages": [ToolMessage(f"Updated file {file_path}", tool_call_id=tool_call_id)]
         }
     )
 
@@ -107,6 +147,19 @@ def edit_file(
     replace_all: bool = False,
 ) -> Union[Command, str]:
     """Write to a file."""
+    # Special handling for longterm filesystem
+    if os.getenv("LONGTERM_FILESYSTEM_NAME") and os.getenv("AGENT_FS_API_KEY") and has_memories_prefix(file_path):
+        filesystem_client = FilesystemClient(
+            filesystem=os.getenv("LONGTERM_FILESYSTEM_NAME")
+        )
+        short_file_path = strip_memories_prefix(file_path)
+        filesystem_client.edit_file(short_file_path, old_string, new_string, replace_all)
+        return Command(
+            update={
+                "messages": [ToolMessage(f"Successfully edited longterm memories file {file_path}", tool_call_id=tool_call_id)]
+            }
+        )
+
     mock_filesystem = state.get("files", {})
     # Check if file exists in mock filesystem
     if file_path not in mock_filesystem:

{deepagents-0.0.10.dist-info → deepagents-0.0.11rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deepagents
-Version: 0.0.10
+Version: 0.0.11rc1
 Summary: General purpose 'deep agent' with sub-agent spawning, todo list capabilities, and mock file system. Built on LangGraph.
 License: MIT
 Requires-Python: <4.0,>=3.11
@@ -10,6 +10,7 @@ Requires-Dist: langgraph>=1.0.0a3
 Requires-Dist: langchain-anthropic>=0.1.23
 Requires-Dist: langchain>=1.0.0a10
 Requires-Dist: langgraph-prebuilt>=0.7.0a2
+Requires-Dist: ai-filesystem>=0.1.4
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"

{deepagents-0.0.10.dist-info → deepagents-0.0.11rc1.dist-info}/RECORD

@@ -2,16 +2,16 @@ deepagents/__init__.py,sha256=fA_91ByxPb3e8aPfci43zOXrWz8ylh_CFQALo7EUKi8,312
 deepagents/graph.py,sha256=sLN0gWJma9dGgJI_RHDK-I1cEZMCo8nfxuF4a8vN7Po,6288
 deepagents/middleware.py,sha256=P2y1crsedTH1uZlKS3fpc7pd4jwz8dR8jnw7repUjLk,7254
 deepagents/model.py,sha256=VyRIkdeXJH8HqLrudTKucHpBTtrwMFTQGRlXBj0kUyo,155
-deepagents/prompts.py,sha256=mCnNGTRljfDUMXEfVRiBNGvhoc_wQjt0epiGMJkktgo,25150
+deepagents/prompts.py,sha256=hiyiUriv-B3IQ0lwRqiJsvtgvwSS4wdeMNdXojHUisc,25603
 deepagents/state.py,sha256=8so3MgL-zRPYP8Ci_OuVg4wHrs5uAXCErKF1AjjCSt8,726
-deepagents/tools.py,sha256=KJw1lr6uO0Bw7mUNFmZ6lSuavJA_qx-3hJa3fjK5G6I,4776
+deepagents/tools.py,sha256=kTS-O8jWJsUnX0tBl19Iyg8fSR2jhD8c7jDz_xNrw0I,7138
 deepagents/types.py,sha256=5KBSUPlWOnv9It3SnJCMHrOtp9Y4_NQGtGCp69JsEjE,694
-deepagents-0.0.10.dist-info/licenses/LICENSE,sha256=c__BaxUCK69leo2yEKynf8lWndu8iwYwge1CbyqAe-E,1071
+deepagents-0.0.11rc1.dist-info/licenses/LICENSE,sha256=c__BaxUCK69leo2yEKynf8lWndu8iwYwge1CbyqAe-E,1071
 tests/test_deepagents.py,sha256=SwtOiJF4c1O3r_Q3AiM7XZu6tVq4uMIcZlnsfRjx8Ig,7648
 tests/test_hitl.py,sha256=B16ZFiyaVSOcDLz7mh1RTaQZ93EMTKOPUY-IEslkcfM,2460
 tests/test_middleware.py,sha256=3HYmTx0Jw4XTNJjqLYeyGS_QZzcqkFuKfShtajIDhF4,2146
 tests/utils.py,sha256=Ln_DYaMkwAVBo4XQ-QKwlCWP8zZYMenWTcFhsneoL0g,2913
-deepagents-0.0.10.dist-info/METADATA,sha256=Fc_-F5LiDzCdN-MdHbd-oo1Gyftb_Fq6xKsryZeqh5o,17292
-deepagents-0.0.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-deepagents-0.0.10.dist-info/top_level.txt,sha256=_w9VMQtG4YDNg5A5eAeUre7dF7x7hk9zRpT9zsFaukY,17
-deepagents-0.0.10.dist-info/RECORD,,
+deepagents-0.0.11rc1.dist-info/METADATA,sha256=nWOCcZljXdWY9tm3Gc_0Wyt7ZK0WScU0MhZY-jlJKuY,17331
+deepagents-0.0.11rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+deepagents-0.0.11rc1.dist-info/top_level.txt,sha256=_w9VMQtG4YDNg5A5eAeUre7dF7x7hk9zRpT9zsFaukY,17
+deepagents-0.0.11rc1.dist-info/RECORD,,