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

deepagents/middleware.py

@@ -7,8 +7,8 @@ from langchain_core.tools import BaseTool, tool, InjectedToolCallId
 from langchain_core.messages import ToolMessage
 from langchain.chat_models import init_chat_model
 from langgraph.types import Command
-from langchain.agents.tool_node import InjectedState
-from typing import Annotated, Optional
+from langchain.tools.tool_node import InjectedState
+from typing import Annotated
 from deepagents.state import PlanningState, FilesystemState
 from deepagents.tools import write_todos, ls, read_file, write_file, edit_file
 from deepagents.prompts import WRITE_TODOS_SYSTEM_PROMPT, TASK_SYSTEM_PROMPT, FILESYSTEM_SYSTEM_PROMPT, TASK_TOOL_DESCRIPTION, BASE_AGENT_PROMPT

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,7 +1,8 @@
+from re import L
 from langchain_core.tools import tool, InjectedToolCallId
 from langchain_core.messages import ToolMessage
 from langgraph.types import Command
-from langchain.agents.tool_node import InjectedState
+from langchain.tools.tool_node import InjectedState
 from typing import Annotated, Union
 from deepagents.state import Todo, FilesystemState
 from deepagents.prompts import (
@@ -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.9.dist-info → deepagents-0.0.11rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deepagents
-Version: 0.0.9
+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
@@ -8,8 +8,9 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: langgraph>=1.0.0a3
 Requires-Dist: langchain-anthropic>=0.1.23
-Requires-Dist: langchain>=1.0.0a9
+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"
@@ -163,10 +164,10 @@ agent = create_deep_agent(
 For more complex use cases, you can provide your own pre-built LangGraph graph as a subagent:
 
 ```python
-from langgraph.prebuilt import create_react_agent
+from langchain.agents import create_agent
 
 # Create a custom agent graph
-custom_graph = create_react_agent(
+custom_graph = create_agent(
     model=your_model,
     tools=specialized_tools,
     prompt="You are a specialized agent for data analysis..."

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

@@ -1,17 +1,17 @@
 deepagents/__init__.py,sha256=fA_91ByxPb3e8aPfci43zOXrWz8ylh_CFQALo7EUKi8,312
 deepagents/graph.py,sha256=sLN0gWJma9dGgJI_RHDK-I1cEZMCo8nfxuF4a8vN7Po,6288
-deepagents/middleware.py,sha256=UysnjeHQoALtbays0AKFQ7-BQ3Ivlci_9-6mxzTNzNM,7265
+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=NpjO2ZEAnL8ZKsYBfxK36AYaayOxC7hWYOvEBCL_5tA,4777
+deepagents/tools.py,sha256=kTS-O8jWJsUnX0tBl19Iyg8fSR2jhD8c7jDz_xNrw0I,7138
 deepagents/types.py,sha256=5KBSUPlWOnv9It3SnJCMHrOtp9Y4_NQGtGCp69JsEjE,694
-deepagents-0.0.9.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.9.dist-info/METADATA,sha256=QKPY9e5P9Y3X7TR5l4pZiKHqaaW_iHCUedgukldhytw,17304
-deepagents-0.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-deepagents-0.0.9.dist-info/top_level.txt,sha256=_w9VMQtG4YDNg5A5eAeUre7dF7x7hk9zRpT9zsFaukY,17
-deepagents-0.0.9.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,,