deepagents 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

deepagents/backends/utils.py

@@ -6,10 +6,11 @@ enable composition without fragile string parsing.
 """
 
 import re
-import wcmatch.glob as wcglob
 from datetime import UTC, datetime
 from pathlib import Path
-from typing import Any, Literal, TypedDict, List, Dict
+from typing import Any, Literal, TypedDict
+
+import wcmatch.glob as wcglob
 
 EMPTY_CONTENT_WARNING = "System reminder: File exists but has empty contents"
 MAX_LINE_LENGTH = 10000
@@ -24,6 +25,7 @@ class FileInfo(TypedDict, total=False):
     Minimal contract used across backends. Only "path" is required.
     Other fields are best-effort and may be absent depending on backend.
     """
+
     path: str
     is_dir: bool
     size: int  # bytes (approx)
@@ -32,14 +34,15 @@ class FileInfo(TypedDict, total=False):
 
 class GrepMatch(TypedDict):
     """Structured grep match entry."""
+
     path: str
     line: int
     text: str
 
 
 def sanitize_tool_call_id(tool_call_id: str) -> str:
-    """Sanitize tool_call_id to prevent path traversal and separator issues. 
-    
+    r"""Sanitize tool_call_id to prevent path traversal and separator issues.
+
     Replaces dangerous characters (., /, \) with underscores.
     """
     sanitized = tool_call_id.replace(".", "_").replace("/", "_").replace("\\", "_")
@@ -94,10 +97,10 @@ def format_content_with_line_numbers(
 
 def check_empty_content(content: str) -> str | None:
     """Check if content is empty and return warning message.
-    
+
     Args:
         content: Content to check
-    
+
     Returns:
         Warning message if empty, None otherwise
     """
@@ -108,10 +111,10 @@ def check_empty_content(content: str) -> str | None:
 
 def file_data_to_string(file_data: dict[str, Any]) -> str:
     """Convert FileData to plain string content.
-    
+
     Args:
         file_data: FileData dict with 'content' key
-    
+
     Returns:
         Content as string with lines joined by newlines
     """
@@ -164,12 +167,12 @@ def format_read_response(
     limit: int,
 ) -> str:
     """Format file data for read response with line numbers.
-    
+
     Args:
         file_data: FileData dict
         offset: Line offset (0-indexed)
         limit: Maximum number of lines
-    
+
     Returns:
         Formatted content or error message
     """
@@ -177,14 +180,14 @@ def format_read_response(
     empty_msg = check_empty_content(content)
     if empty_msg:
         return empty_msg
-    
+
     lines = content.splitlines()
     start_idx = offset
     end_idx = min(start_idx + limit, len(lines))
-    
+
     if start_idx >= len(lines):
         return f"Error: Line offset {offset} exceeds file length ({len(lines)} lines)"
-    
+
     selected_lines = lines[start_idx:end_idx]
     return format_content_with_line_numbers(selected_lines, start_line=start_idx + 1)
 
@@ -196,24 +199,24 @@ def perform_string_replacement(
     replace_all: bool,
 ) -> tuple[str, int] | str:
     """Perform string replacement with occurrence validation.
-    
+
     Args:
         content: Original content
         old_string: String to replace
         new_string: Replacement string
         replace_all: Whether to replace all occurrences
-    
+
     Returns:
         Tuple of (new_content, occurrences) on success, or error message string
     """
     occurrences = content.count(old_string)
-    
+
     if occurrences == 0:
         return f"Error: String not found in file: '{old_string}'"
-    
+
     if occurrences > 1 and not replace_all:
         return f"Error: String '{old_string}' appears {occurrences} times in file. Use replace_all=True to replace all instances, or provide a more specific string with surrounding context."
-    
+
     new_content = content.replace(old_string, new_string)
     return new_content, occurrences
 
@@ -225,33 +228,33 @@ def truncate_if_too_long(result: list[str] | str) -> list[str] | str:
         if total_chars > TOOL_RESULT_TOKEN_LIMIT * 4:
             return result[: len(result) * TOOL_RESULT_TOKEN_LIMIT * 4 // total_chars] + [TRUNCATION_GUIDANCE]
         return result
-    else:  # string
-        if len(result) > TOOL_RESULT_TOKEN_LIMIT * 4:
-            return result[: TOOL_RESULT_TOKEN_LIMIT * 4] + "\n" + TRUNCATION_GUIDANCE
-        return result
+    # string
+    if len(result) > TOOL_RESULT_TOKEN_LIMIT * 4:
+        return result[: TOOL_RESULT_TOKEN_LIMIT * 4] + "\n" + TRUNCATION_GUIDANCE
+    return result
 
 
 def _validate_path(path: str | None) -> str:
     """Validate and normalize a path.
-    
+
     Args:
         path: Path to validate
-    
+
     Returns:
         Normalized path starting with /
-    
+
     Raises:
         ValueError: If path is invalid
     """
     path = path or "/"
     if not path or path.strip() == "":
         raise ValueError("Path cannot be empty")
-    
+
     normalized = path if path.startswith("/") else "/" + path
-    
+
     if not normalized.endswith("/"):
         normalized += "/"
-    
+
     return normalized
 
 
@@ -261,16 +264,16 @@ def _glob_search_files(
     path: str = "/",
 ) -> str:
     """Search files dict for paths matching glob pattern.
-    
+
     Args:
         files: Dictionary of file paths to FileData.
         pattern: Glob pattern (e.g., "*.py", "**/*.ts").
         path: Base path to search from.
-    
+
     Returns:
         Newline-separated file paths, sorted by modification time (most recent first).
         Returns "No files found" if no matches.
-    
+
     Example:
         ```python
         files = {"/src/main.py": FileData(...), "/test.py": FileData(...)}
@@ -313,29 +316,28 @@ def _format_grep_results(
     output_mode: Literal["files_with_matches", "content", "count"],
 ) -> str:
     """Format grep search results based on output mode.
-    
+
     Args:
         results: Dictionary mapping file paths to list of (line_num, line_content) tuples
         output_mode: Output format - "files_with_matches", "content", or "count"
-    
+
     Returns:
         Formatted string output
     """
     if output_mode == "files_with_matches":
         return "\n".join(sorted(results.keys()))
-    elif output_mode == "count":
+    if output_mode == "count":
         lines = []
         for file_path in sorted(results.keys()):
             count = len(results[file_path])
             lines.append(f"{file_path}: {count}")
         return "\n".join(lines)
-    else:
-        lines = []
-        for file_path in sorted(results.keys()):
-            lines.append(f"{file_path}:")
-            for line_num, line in results[file_path]:
-                lines.append(f"  {line_num}: {line}")
-        return "\n".join(lines)
+    lines = []
+    for file_path in sorted(results.keys()):
+        lines.append(f"{file_path}:")
+        for line_num, line in results[file_path]:
+            lines.append(f"  {line_num}: {line}")
+    return "\n".join(lines)
 
 
 def _grep_search_files(
@@ -346,17 +348,17 @@ def _grep_search_files(
     output_mode: Literal["files_with_matches", "content", "count"] = "files_with_matches",
 ) -> str:
     """Search file contents for regex pattern.
-    
+
     Args:
         files: Dictionary of file paths to FileData.
         pattern: Regex pattern to search for.
         path: Base path to search from.
         glob: Optional glob pattern to filter files (e.g., "*.py").
         output_mode: Output format - "files_with_matches", "content", or "count".
-    
+
     Returns:
         Formatted search results. Returns "No matches found" if no results.
-    
+
     Example:
         ```python
         files = {"/file.py": FileData(content=["import os", "print('hi')"], ...)}
@@ -394,6 +396,7 @@ def _grep_search_files(
 
 # -------- Structured helpers for composition --------
 
+
 def grep_matches_from_files(
     files: dict[str, Any],
     pattern: str,
@@ -419,11 +422,7 @@ def grep_matches_from_files(
     filtered = {fp: fd for fp, fd in files.items() if fp.startswith(normalized_path)}
 
     if glob:
-        filtered = {
-            fp: fd
-            for fp, fd in filtered.items()
-            if wcglob.globmatch(Path(fp).name, glob, flags=wcglob.BRACE)
-        }
+        filtered = {fp: fd for fp, fd in filtered.items() if wcglob.globmatch(Path(fp).name, glob, flags=wcglob.BRACE)}
 
     matches: list[GrepMatch] = []
     for file_path, file_data in filtered.items():
@@ -433,16 +432,16 @@ def grep_matches_from_files(
     return matches
 
 
-def build_grep_results_dict(matches: List[GrepMatch]) -> Dict[str, list[tuple[int, str]]]:
+def build_grep_results_dict(matches: list[GrepMatch]) -> dict[str, list[tuple[int, str]]]:
     """Group structured matches into the legacy dict form used by formatters."""
-    grouped: Dict[str, list[tuple[int, str]]] = {}
+    grouped: dict[str, list[tuple[int, str]]] = {}
     for m in matches:
         grouped.setdefault(m["path"], []).append((m["line"], m["text"]))
     return grouped
 
 
 def format_grep_matches(
-    matches: List[GrepMatch],
+    matches: list[GrepMatch],
     output_mode: Literal["files_with_matches", "content", "count"],
 ) -> str:
     """Format structured grep matches using existing formatting logic."""

{deepagents-0.2.2.dist-info → deepagents-0.2.4.dist-info}/METADATA

@@ -1,8 +1,14 @@
 Metadata-Version: 2.4
 Name: deepagents
-Version: 0.2.2
+Version: 0.2.4
 Summary: General purpose 'deep agent' with sub-agent spawning, todo list capabilities, and mock file system. Built on LangGraph.
 License: MIT
+Project-URL: Homepage, https://docs.langchain.com/oss/python/deepagents/overview
+Project-URL: Documentation, https://reference.langchain.com/python/deepagents/
+Project-URL: Source, https://github.com/langchain-ai/deepagents
+Project-URL: Twitter, https://x.com/LangChainAI
+Project-URL: Slack, https://www.langchain.com/join-community
+Project-URL: Reddit, https://www.reddit.com/r/LangChain/
 Requires-Python: <4.0,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE

{deepagents-0.2.2.dist-info → deepagents-0.2.4.dist-info}/RECORD

@@ -6,15 +6,14 @@ deepagents/backends/filesystem.py,sha256=U9Tmf8BDTqKbw-gQjJkJOYG1nXQo9FhxyfzN97W
 deepagents/backends/protocol.py,sha256=fwqJa_Ec6F4BoNYz0bcPHL_fiKksxw2RoyA6x5wr7dc,4181
 deepagents/backends/state.py,sha256=BxMNm1kDpxtgzIzpuF78h1NuYh9VIpXqnUbbETGe4Y4,6584
 deepagents/backends/store.py,sha256=VsPSj6ayABPjkKiN6CcvOGm7YCWKuWP_ltJWvFJ1nF0,13358
-deepagents/backends/utils.py,sha256=vQDMFMjf7pmfKqprpTlF7851FWmswZnMdLj-cezTsBk,14432
+deepagents/backends/utils.py,sha256=_0mkTs3fD4eaChlOjOivYtF9i_AuJHdV64jRlGRF5LE,14207
 deepagents/middleware/__init__.py,sha256=x7UHqGcrKlhzORNdChPvnUwa_PIJCbFUHY6zTKVfloI,418
-deepagents/middleware/agent_memory.py,sha256=BRP8Dyuzl1ms4Eja-3nRHI3g2vNWfK8tUW6zBr2JJOc,9196
 deepagents/middleware/filesystem.py,sha256=Zwpt6ILniHbNzfLWXrSGLbd__ZFkkO1xv0mGiRsNB7s,28144
 deepagents/middleware/patch_tool_calls.py,sha256=PdNhxPaQqwnFkhEAZEE2kEzadTNAOO3_iJRA30WqpGE,1981
 deepagents/middleware/resumable_shell.py,sha256=WbtjW81DItZgFiy5k4cSLAu5NTBHdOk2UEnFecuqlJU,3353
 deepagents/middleware/subagents.py,sha256=JxXwZvi41pBKKMguKlyVqwjCoydnZboWEgJGkWOCIY8,23503
-deepagents-0.2.2.dist-info/licenses/LICENSE,sha256=c__BaxUCK69leo2yEKynf8lWndu8iwYwge1CbyqAe-E,1071
-deepagents-0.2.2.dist-info/METADATA,sha256=qCF7ZAsyD2cW8nqskeDWJ3DZxsp-V4ZF2RiVZ0vHtcY,18660
-deepagents-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-deepagents-0.2.2.dist-info/top_level.txt,sha256=drAzchOzPNePwpb3_pbPuvLuayXkN7SNqeIKMBWJoAo,11
-deepagents-0.2.2.dist-info/RECORD,,
+deepagents-0.2.4.dist-info/licenses/LICENSE,sha256=c__BaxUCK69leo2yEKynf8lWndu8iwYwge1CbyqAe-E,1071
+deepagents-0.2.4.dist-info/METADATA,sha256=byw286YaV5O5vsd-USesxgs7coiW63tg5GE1wqe-eI0,19050
+deepagents-0.2.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+deepagents-0.2.4.dist-info/top_level.txt,sha256=drAzchOzPNePwpb3_pbPuvLuayXkN7SNqeIKMBWJoAo,11
+deepagents-0.2.4.dist-info/RECORD,,

deepagents/middleware/agent_memory.py

@@ -1,222 +0,0 @@
-"""Middleware for loading agent-specific long-term memory into the system prompt."""
-
-from collections.abc import Awaitable, Callable
-from typing import TYPE_CHECKING, Any
-
-if TYPE_CHECKING:
-    from langgraph.runtime import Runtime
-
-from langchain.agents.middleware.types import (
-    AgentMiddleware,
-    AgentState,
-    ModelRequest,
-    ModelResponse,
-)
-from typing_extensions import NotRequired, TypedDict
-
-from deepagents.backends.protocol import BackendProtocol
-
-
-class AgentMemoryState(AgentState):
-    """State for the agent memory middleware."""
-
-    agent_memory: NotRequired[str | None]
-    """Long-term memory content for the agent."""
-
-
-AGENT_MEMORY_FILE_PATH = "/agent.md"
-
-# Long-term Memory Documentation
-LONGTERM_MEMORY_SYSTEM_PROMPT = """
-
-## Long-term Memory
-
-You have access to a long-term memory system using the {memory_path} path prefix.
-Files stored in {memory_path} persist across sessions and conversations.
-
-Your system prompt is loaded from {memory_path}agent.md at startup. You can update your own instructions by editing this file.
-
-**When to CHECK/READ memories (CRITICAL - do this FIRST):**
-- **At the start of ANY new session**: Run `ls {memory_path}` to see what you know
-- **BEFORE answering questions**: If asked "what do you know about X?" or "how do I do Y?", check `ls {memory_path}` for relevant files FIRST
-- **When user asks you to do something**: Check if you have guides, examples, or patterns in {memory_path} before proceeding
-- **When user references past work or conversations**: Search {memory_path} for related content
-- **If you're unsure**: Check your memories rather than guessing or using only general knowledge
-
-**Memory-first response pattern:**
-1. User asks a question → Run `ls {memory_path}` to check for relevant files
-2. If relevant files exist → Read them with `read_file {memory_path}[filename]`
-3. Base your answer on saved knowledge (from memories) supplemented by general knowledge
-4. If no relevant memories exist → Use general knowledge, then consider if this is worth saving
-
-**When to update memories:**
-- **IMMEDIATELY when the user describes your role or how you should behave** (e.g., "you are a web researcher", "you are an expert in X")
-- **IMMEDIATELY when the user gives feedback on your work** - Before continuing, update memories to capture what was wrong and how to do it better
-- When the user explicitly asks you to remember something
-- When patterns or preferences emerge (coding styles, conventions, workflows)
-- After significant work where context would help in future sessions
-
-**Learning from feedback:**
-- When user says something is better/worse, capture WHY and encode it as a pattern
-- Each correction is a chance to improve permanently - don't just fix the immediate issue, update your instructions
-- When user says "you should remember X" or "be careful about Y", treat this as HIGH PRIORITY - update memories IMMEDIATELY
-- Look for the underlying principle behind corrections, not just the specific mistake
-- If it's something you "should have remembered", identify where that instruction should live permanently
-
-**What to store where:**
-- **{memory_path}agent.md**: Update this to modify your core instructions and behavioral patterns
-- **Other {memory_path} files**: Use for project-specific context, reference information, or structured notes
-  - If you create additional memory files, add references to them in {memory_path}agent.md so you remember to consult them
-
-The portion of your system prompt that comes from {memory_path}agent.md is marked with `<agent_memory>` tags so you can identify what instructions come from your persistent memory.
-
-Example: `ls {memory_path}` to see what memories you have
-Example: `read_file '{memory_path}deep-agents-guide.md'` to recall saved knowledge
-Example: `edit_file('{memory_path}agent.md', ...)` to update your instructions
-Example: `write_file('{memory_path}project_context.md', ...)` for project-specific notes, then reference it in agent.md
-
-Remember: To interact with the longterm filesystem, you must prefix the filename with the {memory_path} path."""
-
-
-DEFAULT_MEMORY_SNIPPET = """<agent_memory>
-{agent_memory}
-</agent_memory>
-"""
-
-class AgentMemoryMiddleware(AgentMiddleware):
-    """Middleware for loading agent-specific long-term memory.
-
-    This middleware loads the agent's long-term memory from a file (agent.md)
-    and injects it into the system prompt. The memory is loaded once at the
-    start of the conversation and stored in state.
-
-    Args:
-        backend: Backend to use for loading the agent memory file.
-        system_prompt_template: Optional custom template for how to inject
-            the agent memory into the system prompt. Use {agent_memory} as
-            a placeholder. Defaults to a simple section header.
-
-    Example:
-        ```python
-        from deepagents.middleware.agent_memory import AgentMemoryMiddleware
-        from deepagents.memory.backends import FilesystemBackend
-        from pathlib import Path
-
-        # Set up backend pointing to agent's directory
-        agent_dir = Path.home() / ".deepagents" / "my-agent"
-        backend = FilesystemBackend(root_dir=agent_dir)
-        
-        # Create middleware
-        middleware = AgentMemoryMiddleware(backend=backend)
-        ```
-    """
-
-    state_schema = AgentMemoryState
-
-    def __init__(
-        self,
-        *,
-        backend: BackendProtocol,
-        memory_path: str,
-        system_prompt_template: str | None = None,
-    ) -> None:
-        """Initialize the agent memory middleware.
-
-        Args:
-            backend: Backend to use for loading the agent memory file.
-            system_prompt_template: Optional custom template for injecting
-                agent memory into system prompt.
-        """
-        self.backend = backend
-        self.memory_path = memory_path
-        self.system_prompt_template = system_prompt_template or DEFAULT_MEMORY_SNIPPET
-
-    def before_agent(
-        self,
-        state: AgentMemoryState,
-        runtime,
-    ) -> AgentMemoryState:
-        """Load agent memory from file before agent execution.
-
-        Args:
-            state: Current agent state.
-            handler: Handler function to call after loading memory.
-
-        Returns:
-            Updated state with agent_memory populated.
-        """
-        # Only load memory if it hasn't been loaded yet
-        if "agent_memory" not in state or state.get("agent_memory") is None:
-            file_data = self.backend.read(AGENT_MEMORY_FILE_PATH)
-            return {"agent_memory": file_data}
-
-    async def abefore_agent(
-        self,
-        state: AgentMemoryState,
-        runtime,
-    ) -> AgentMemoryState:
-        """(async) Load agent memory from file before agent execution.
-
-        Args:
-            state: Current agent state.
-            handler: Handler function to call after loading memory.
-
-        Returns:
-            Updated state with agent_memory populated.
-        """
-        # Only load memory if it hasn't been loaded yet
-        if "agent_memory" not in state or state.get("agent_memory") is None:
-            file_data = self.backend.read(AGENT_MEMORY_FILE_PATH)
-            return {"agent_memory": file_data}
-
-    def wrap_model_call(
-        self,
-        request: ModelRequest,
-        handler: Callable[[ModelRequest], ModelResponse],
-    ) -> ModelResponse:
-        """Inject agent memory into the system prompt.
-
-        Args:
-            request: The model request being processed.
-            handler: The handler function to call with the modified request.
-
-        Returns:
-            The model response from the handler.
-        """
-        # Get agent memory from state
-        agent_memory = request.state.get("agent_memory", "")
-        
-        memory_section = self.system_prompt_template.format(agent_memory=agent_memory)
-        if request.system_prompt:
-            request.system_prompt = memory_section + "\n\n" + request.system_prompt
-        else:
-            request.system_prompt = memory_section
-        request.system_prompt = request.system_prompt + "\n\n" + LONGTERM_MEMORY_SYSTEM_PROMPT.format(memory_path=self.memory_path)
-
-        return handler(request)
-
-    async def awrap_model_call(
-        self,
-        request: ModelRequest,
-        handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
-    ) -> ModelResponse:
-        """(async) Inject agent memory into the system prompt.
-
-        Args:
-            request: The model request being processed.
-            handler: The handler function to call with the modified request.
-
-        Returns:
-            The model response from the handler.
-        """
-        # Get agent memory from state
-        agent_memory = request.state.get("agent_memory", "")
-        
-        memory_section = self.system_prompt_template.format(agent_memory=agent_memory)
-        if request.system_prompt:
-            request.system_prompt = memory_section + "\n\n" + request.system_prompt
-        else:
-            request.system_prompt = memory_section
-        request.system_prompt = request.system_prompt + "\n\n" + LONGTERM_MEMORY_SYSTEM_PROMPT.format(memory_path=self.memory_path)
-
-        return await handler(request)