deepagents-cli 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

deepagents/middleware/patch_tool_calls.py

@@ -0,0 +1,44 @@
+"""Middleware to patch dangling tool calls in the messages history."""
+
+from typing import Any
+
+from langchain.agents.middleware import AgentMiddleware, AgentState
+from langchain_core.messages import RemoveMessage, ToolMessage
+from langgraph.graph.message import REMOVE_ALL_MESSAGES
+from langgraph.runtime import Runtime
+
+
+class PatchToolCallsMiddleware(AgentMiddleware):
+    """Middleware to patch dangling tool calls in the messages history."""
+
+    def before_agent(self, state: AgentState, runtime: Runtime[Any]) -> dict[str, Any] | None:  # noqa: ARG002
+        """Before the agent runs, handle dangling tool calls from any AIMessage."""
+        messages = state["messages"]
+        if not messages or len(messages) == 0:
+            return None
+
+        patched_messages = []
+        # Iterate over the messages and add any dangling tool calls
+        for i, msg in enumerate(messages):
+            patched_messages.append(msg)
+            if msg.type == "ai" and msg.tool_calls:
+                for tool_call in msg.tool_calls:
+                    corresponding_tool_msg = next(
+                        (msg for msg in messages[i:] if msg.type == "tool" and msg.tool_call_id == tool_call["id"]),
+                        None,
+                    )
+                    if corresponding_tool_msg is None:
+                        # We have a dangling tool call which needs a ToolMessage
+                        tool_msg = (
+                            f"Tool call {tool_call['name']} with id {tool_call['id']} was "
+                            "cancelled - another message came in before it could be completed."
+                        )
+                        patched_messages.append(
+                            ToolMessage(
+                                content=tool_msg,
+                                name=tool_call["name"],
+                                tool_call_id=tool_call["id"],
+                            )
+                        )
+
+        return {"messages": [RemoveMessage(id=REMOVE_ALL_MESSAGES), *patched_messages]}

deepagents/middleware/subagents.py

@@ -14,11 +14,6 @@ from langchain_core.runnables import Runnable
 from langchain_core.tools import StructuredTool
 from langgraph.types import Command
 
-try:
-    from langchain_anthropic.middleware.prompt_caching import AnthropicPromptCachingMiddleware
-except ImportError:
-    AnthropicPromptCachingMiddleware = None
-
 
 class SubAgent(TypedDict):
     """Specification for an agent.
@@ -267,7 +262,7 @@ def _get_subagents(
 
         subagent_model = agent_.get("model", default_model)
 
-        _middleware = [*default_subagent_middleware, *agent_["middleware"]] if "middleware" in agent_ else default_subagent_middleware
+        _middleware = [*default_subagent_middleware, *agent_["middleware"]] if "middleware" in agent_ else [*default_subagent_middleware]
 
         interrupt_on = agent_.get("interrupt_on", default_interrupt_on)
         if interrupt_on:
@@ -353,6 +348,9 @@ def _create_task_tool(
     ) -> str | Command:
         subagent, subagent_state = _validate_and_prepare_state(subagent_type, description, runtime)
         result = subagent.invoke(subagent_state)
+        if not runtime.tool_call_id:
+            value_error_msg = "Tool call ID is required for subagent invocation"
+            raise ValueError(value_error_msg)
         return _return_command_with_state_update(result, runtime.tool_call_id)
 
     async def atask(
@@ -362,6 +360,9 @@ def _create_task_tool(
     ) -> str | Command:
         subagent, subagent_state = _validate_and_prepare_state(subagent_type, description, runtime)
         result = await subagent.ainvoke(subagent_state)
+        if not runtime.tool_call_id:
+            value_error_msg = "Tool call ID is required for subagent invocation"
+            raise ValueError(value_error_msg)
         return _return_command_with_state_update(result, runtime.tool_call_id)
 
     return StructuredTool.from_function(

deepagents/pretty_cli.py

@@ -0,0 +1,289 @@
+#!/usr/bin/env python3
+"""Minimalist prompt-toolkit TUI for DeepAgents sandbox chat.
+
+Beautiful emerald green theme with streaming responses.
+
+Controls:
+- Enter: Submit message
+- Alt-Enter (or Esc then Enter): New line for multiline input
+- Ctrl+C or Ctrl+D: Quit
+"""
+
+import asyncio
+
+from prompt_toolkit import PromptSession
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.styles import Style
+from rich.console import Console
+from rich.spinner import Spinner
+
+from chat_agent import create_sandbox_chat_agent
+
+
+# ============================================================================
+# CONFIGURATION
+# ============================================================================
+
+COLORS = {
+    "primary": "#10b981",    # Emerald 500
+    "dim": "#6b7280",        # Gray 500
+    "user": "#ffffff",       # White
+    "agent": "#10b981",      # Emerald 500
+    "thinking": "#34d399",   # Emerald 400
+    "tool": "#fbbf24",       # Amber 400
+}
+
+DEEP_AGENTS_ASCII = """
+ ██████╗  ███████╗ ███████╗ ██████╗
+ ██╔══██╗ ██╔════╝ ██╔════╝ ██╔══██╗
+ ██║  ██║ █████╗   █████╗   ██████╔╝
+ ██║  ██║ ██╔══╝   ██╔══╝   ██╔═══╝
+ ██████╔╝ ███████╗ ███████╗ ██║
+ ╚═════╝  ╚══════╝ ╚══════╝ ╚═╝
+
+  █████╗   ██████╗  ███████╗ ███╗   ██╗ ████████╗ ███████╗
+ ██╔══██╗ ██╔════╝  ██╔════╝ ████╗  ██║ ╚══██╔══╝ ██╔════╝
+ ███████║ ██║  ███╗ █████╗   ██╔██╗ ██║    ██║    ███████╗
+ ██╔══██║ ██║   ██║ ██╔══╝   ██║╚██╗██║    ██║    ╚════██║
+ ██║  ██║ ╚██████╔╝ ███████╗ ██║ ╚████║    ██║    ███████║
+ ╚═╝  ╚═╝  ╚═════╝  ╚══════╝ ╚═╝  ╚═══╝    ╚═╝    ╚══════╝
+"""
+
+# ============================================================================
+# GLOBALS
+# ============================================================================
+
+console = Console()
+agent = None
+config = {"configurable": {"thread_id": "sandbox-chat"}}
+
+# ============================================================================
+# UTILITIES
+# ============================================================================
+
+MAX_ARG_LENGTH = 150
+
+TOOL_ICONS = {
+    "shell": "⚡",
+    "write_file": "✏️",
+    "read_file": "📖",
+    "edit_file": "✂️",
+    "ls": "📁",
+    "glob": "🔍",
+    "grep": "🔎",
+}
+
+
+def truncate(text: str, max_len: int) -> str:
+    """Truncate text with ellipsis."""
+    return text[:max_len] + "..." if len(text) > max_len else text
+
+
+# ============================================================================
+# DISPLAY FUNCTIONS
+# ============================================================================
+
+
+def show_welcome():
+    """Display welcome screen and wait for Enter."""
+    console.clear()
+    console.print(DEEP_AGENTS_ASCII, style=f"bold {COLORS['primary']}")
+    console.print("\n")
+    console.print("Press Enter to start", style=COLORS["dim"])
+    input()
+    # Don't clear - keep ASCII art visible!
+
+
+# ============================================================================
+# AGENT INTERACTION
+# ============================================================================
+
+
+async def stream_agent_response(user_input: str):
+    """Stream agent response using async iteration."""
+    global agent
+
+    has_responded = False
+    current_text = ""
+
+    # Start spinner manually so we can stop it when we have content
+    status = console.status("[bold #34d399]Agent is thinking...", spinner="dots")
+    status.start()
+    spinner_active = True
+
+    async for _, chunk in agent.astream(
+        {"messages": [{"role": "user", "content": user_input}]},
+        stream_mode="updates",
+        subgraphs=True,
+        config=config,
+        durability="exit",
+    ):
+        chunk_data = list(chunk.values())[0]
+        if not chunk_data or "messages" not in chunk_data:
+            continue
+
+        last_message = chunk_data["messages"][-1]
+        message_role = getattr(last_message, "type", None)
+        message_content = getattr(last_message, "content", None)
+
+        # Handle tool calls from AI messages (LangChain tool_calls attribute)
+        tool_calls = getattr(last_message, "tool_calls", None)
+        if tool_calls and message_role == "ai":
+            for tool_call in tool_calls:
+                tool_name = tool_call.get("name", "unknown")
+                tool_args = tool_call.get("args", {})
+
+                icon = TOOL_ICONS.get(tool_name, "🔧")
+                args_str = ", ".join(
+                    f"{k}={truncate(str(v), 50)}" for k, v in tool_args.items()
+                )
+
+                # Stop spinner temporarily to print tool call
+                if spinner_active:
+                    status.stop()
+                console.print(f"  {icon} {tool_name}({args_str})", style=f"dim {COLORS['tool']}")
+                # Restart spinner for next tool/processing
+                if spinner_active:
+                    status.start()
+
+        # Skip tool results - they're verbose and the agent will summarize
+        if message_role == "tool":
+            continue
+
+        if not message_content:
+            continue
+
+        # Handle tool calls from content blocks (alternative format)
+        if message_role == "ai" and isinstance(message_content, list):
+            for block in message_content:
+                if isinstance(block, dict) and block.get("type") == "tool_use":
+                    tool_name = block.get("name", "unknown")
+                    tool_input = block.get("input", {})
+
+                    icon = TOOL_ICONS.get(tool_name, "🔧")
+                    args = ", ".join(
+                        f"{k}={truncate(str(v), 50)}" for k, v in tool_input.items()
+                    )
+
+                    # Stop spinner temporarily to print tool call
+                    if spinner_active:
+                        status.stop()
+                    console.print(f"  {icon} {tool_name}({args})", style=f"dim {COLORS['tool']}")
+                    # Restart spinner for next tool/processing
+                    if spinner_active:
+                        status.start()
+
+        # Handle agent text responses
+        if message_role == "ai":
+            text_content = ""
+
+            if isinstance(message_content, str):
+                text_content = message_content
+            elif isinstance(message_content, list):
+                for block in message_content:
+                    if isinstance(block, dict) and block.get("type") == "text":
+                        text_content = block.get("text", "")
+                        break
+
+            if text_content.strip():
+                # Stop spinner when we have actual text to display
+                if spinner_active:
+                    status.stop()
+                    spinner_active = False
+
+                # Print prefix on first response
+                if not has_responded:
+                    console.print("... ", style=COLORS["agent"], end="")
+                    has_responded = True
+
+                # Stream new content
+                if text_content != current_text:
+                    new_text = text_content[len(current_text) :]
+                    console.print(new_text, style=COLORS["agent"], end="")
+                    current_text = text_content
+
+    # Make sure spinner is stopped (in case loop ended without content)
+    if spinner_active:
+        status.stop()
+
+    if has_responded:
+        console.print()  # Newline
+        console.print()  # Blank line
+
+
+# ============================================================================
+# MAIN
+# ============================================================================
+
+
+async def main():
+    """Main entry point."""
+    global agent
+
+    # Show welcome
+    show_welcome()
+
+    # Initialize agent
+    console.print("\nInitializing agent...", style=COLORS["dim"])
+    agent = create_sandbox_chat_agent()
+
+    # Show ready message
+    console.print("\n... Ready to code! What would you like to build?", style=COLORS["agent"])
+    console.print()
+
+    # One-time hint for multiline input
+    console.print("  Tip: Alt-Enter for newline, Enter to submit", style=f"dim {COLORS['dim']}")
+
+    # Setup key bindings for multiline input
+    kb = KeyBindings()
+
+    @kb.add('enter')
+    def _(event):
+        buffer = event.current_buffer
+        if buffer.text.strip():  # Only submit if buffer has content
+            buffer.validate_and_handle()
+        # else: do nothing - no newline, no submission
+
+    @kb.add('escape', 'enter')  # Alt-Enter (or Esc then Enter) for newline
+    def _(event):
+        event.current_buffer.insert_text('\n')
+
+    # Setup prompt session with multiline support
+    style = Style.from_dict({"prompt": COLORS["user"]})
+    session = PromptSession(
+        message="> ",
+        style=style,
+        multiline=True,
+        prompt_continuation=lambda width, line_number, is_soft_wrap: "  ",
+        key_bindings=kb,
+    )
+
+    # Main chat loop
+    while True:
+        try:
+            # Get user input
+            user_input = await session.prompt_async()
+
+            # Handle quit commands
+            if user_input.strip().lower() in ["quit", "exit", "q"]:
+                break
+
+            # Skip empty input
+            if not user_input.strip():
+                continue
+
+            # Add spacing
+            console.print()
+
+            # Stream agent response
+            await stream_agent_response(user_input)
+
+        except (KeyboardInterrupt, EOFError):
+            break
+
+    # Goodbye message
+    console.print("\nGoodbye!", style=COLORS["primary"])
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

{deepagents_cli-0.0.1.dist-info → deepagents_cli-0.0.3.dist-info}/METADATA

@@ -1,20 +1,22 @@
 Metadata-Version: 2.4
 Name: deepagents-cli
-Version: 0.0.1
+Version: 0.0.3
 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
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: langchain-anthropic==1.0.0
-Requires-Dist: langchain==1.0.0
-Requires-Dist: langchain-core==1.0.0
 Requires-Dist: tavily-python
 Requires-Dist: python-dotenv
 Requires-Dist: requests
 Requires-Dist: rich>=13.0.0
+Requires-Dist: langchain-anthropic<2.0.0,>=1.0.0
+Requires-Dist: langchain<2.0.0,>=1.0.0
+Requires-Dist: langchain-core<2.0.0,>=1.0.0
+Requires-Dist: wcmatch
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-recording; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: build; extra == "dev"
 Requires-Dist: twine; extra == "dev"
@@ -319,40 +321,30 @@ agent = create_deep_agent(
 )
 ```
 
-### `use_longterm_memory`
-Deep agents come with a local filesystem to offload memory to. This filesystem is stored in state, and is therefore transient to a single thread.
+### `memory_backend`
+Deep agents come with a local filesystem to offload memory to. By default, this filesystem is stored in state (ephemeral, transient to a single thread).
 
-You can extend deep agents with long-term memory by providing a Store and setting use_longterm_memory=True.
+You can configure persistent long-term memory using a CompositeBackend with StoreBackend:
 
 ```python
 from deepagents import create_deep_agent
+from deepagents.memory.backends import StateBackend, StoreBackend, CompositeBackend
 from langgraph.store.memory import InMemoryStore
 
 store = InMemoryStore()  # Or any other Store object
-agent = create_deep_agent(
-    store=store,
-    use_longterm_memory=True
-)
-```
 
-### `use_local_filesystem`
-If you prefer the agent to operate on your machine's filesystem (read/write actual files), enable local mode:
-
-```python
-from deepagents import create_deep_agent
+# Create a hybrid backend: ephemeral files in / and persistent files in /memories/
+backend = CompositeBackend(
+    default=StateBackend(),
+    routes={"/memories/": StoreBackend()}
+)
 
 agent = create_deep_agent(
-    use_local_filesystem=True,   # Tools operate on disk
-    # use_longterm_memory=False  # Must be False when using local filesystem
+    memory_backend=backend,
+    store=store
 )
 ```
 
-Notes:
-- Local filesystem mode injects `ls`, `read_file`, `write_file`, `edit_file`, plus `glob` and `grep` (ripgrep-powered) for discovery/search.
-- Long-term memory is not supported in local mode; attempting to set both `use_local_filesystem=True` and `use_longterm_memory=True` raises a ValueError.
-- Large tool outputs are automatically evicted to the in-memory filesystem state as files under `/large_tool_results/{tool_call_id}`; use `read_file` with `offset`/`limit` to page through results.
-- Be careful when enabling local mode in sensitive environments — tools can read and write files on disk.
-
 ### `interrupt_on`
 A common reality for agents is that some tool operations may be sensitive and require human approval before execution. Deep Agents supports human-in-the-loop workflows through LangGraph’s interrupt capabilities. You can configure which tools require approval using a checkpointer.
 
@@ -416,17 +408,16 @@ agent = create_agent(
 ### FilesystemMiddleware
 
 Context engineering is one of the main challenges in building effective agents. This can be particularly hard when using tools that can return variable length results (ex. web_search, rag), as long ToolResults can quickly fill up your context window.
-**FilesystemMiddleware** provides four tools to your agent to interact with short-term memory (in agent state) and, optionally, long-term memory (via a Store).
+**FilesystemMiddleware** provides four tools to your agent to interact with both short-term and long-term memory.
 - **ls**: List the files in your filesystem
 - **read_file**: Read an entire file, or a certain number of lines from a file
 - **write_file**: Write a new file to your filesystem
 - **edit_file**: Edit an existing file in your filesystem
 
-If you want these tools to operate on your host filesystem instead of the agent state, use `create_deep_agent(..., use_local_filesystem=True)` or attach `LocalFilesystemMiddleware` directly. Local mode also provides `glob` and `grep` helpers for file discovery and content search.
-
 ```python
 from langchain.agents import create_agent
 from deepagents.middleware.filesystem import FilesystemMiddleware
+from deepagents.memory.backends import StateBackend, StoreBackend, CompositeBackend
 
 # FilesystemMiddleware is included by default in create_deep_agent
 # You can customize it if building a custom agent
@@ -434,8 +425,13 @@ agent = create_agent(
     model="anthropic:claude-sonnet-4-20250514",
     middleware=[
         FilesystemMiddleware(
-            long_term_memory=False,  # Enables access to long-term memory, defaults to False. You must attach a store to use long-term memory.
-            system_prompt="Write to the filesystem when...",  # Optional custom addition to the system prompt
+            memory_backend=StateBackend(),  # Optional: customize storage backend (defaults to StateBackend)
+            # For persistent memory, use CompositeBackend:
+            # memory_backend=CompositeBackend(
+            #     default=StateBackend(),
+            #     routes={"/memories/": StoreBackend()}
+            # )
+            system_prompt="Write to the filesystem when...",  # Optional custom system prompt override
             custom_tool_descriptions={
                 "ls": "Use the ls tool when...",
                 "read_file": "Use the read_file tool to..."

deepagents_cli-0.0.3.dist-info/RECORD

@@ -0,0 +1,24 @@
+deepagents/__init__.py,sha256=9BVNn4lfF5N8l2KY8Ttxi82zO609I-fGqoSIF7DAxiU,342
+deepagents/cli.py,sha256=3UA2SQvAGmUv_elLhi03LjT5U1uyPJ4_XC6wg8Qpl4E,21381
+deepagents/default_agent_prompt.md,sha256=4UbINqwOHbkolh3TTeV3FZWadrVdViHbvIuewnf7wxQ,2925
+deepagents/graph.py,sha256=238tz_eU8HUXFz2PE6uqbW64cPbTKxT8VNfzmNEoLBM,6245
+deepagents/pretty_cli.py,sha256=h43O2j-NA3d6_blvB9_fikSstSY6Nske5K4PjYaOajc,10326
+deepagents/memory/__init__.py,sha256=AnII-JcpwkJlBNzReZy2mgWbZQ7nrc-QN9kNxPoK4fg,353
+deepagents/memory/protocol.py,sha256=Rb_I6Fs27xL-iwHIiD60JluWcho_B_LilamAVlRs8RE,6100
+deepagents/memory/backends/__init__.py,sha256=PdBH3KkEuVAp5SHmGlitM80OV5JXzhfjfehTMOe5Zg4,472
+deepagents/memory/backends/composite.py,sha256=Ar53CGX8yzJrjOqV1dy3i7iGvzEkTfJgwIUYK9ffV14,10261
+deepagents/memory/backends/filesystem.py,sha256=-i_OG2K-dyPROXlKbKgIBvHdj0MWQUWzy7xm603kjWY,11684
+deepagents/memory/backends/state.py,sha256=iP46YlS1lR65WE5lGj2J1cH09UGjntoESVQ_v42Co9Y,6754
+deepagents/memory/backends/store.py,sha256=aWhpiR43-32HwO2VlL238_OqhU7qVabUL6i49UINaH8,11822
+deepagents/memory/backends/utils.py,sha256=n7z73Jes7qZUN8-Tjbz6fFAoZ3XHqZpm5rD9i-0Uvbk,9605
+deepagents/middleware/__init__.py,sha256=Uf4L69XweeHTcQiFz-IEd27wM5L8Mrq5u8OpJ3nwvQY,400
+deepagents/middleware/agent_memory.py,sha256=4bMs-EFbp0-pqDDfOjh_kQkP49M_p5KvUL7JafC4rag,8061
+deepagents/middleware/filesystem.py,sha256=oPTlcW35ijcB643pX4PldF3Qxpj-o2wEdQ9HILx7SLk,25334
+deepagents/middleware/patch_tool_calls.py,sha256=Cu8rUpt1GjrYgfMvZG6wOowvnmFeYTCauOJhlltNPmo,2045
+deepagents/middleware/subagents.py,sha256=NH7QEShEPAosc3HLbSFwlgtlZv3SZpkPEAqHRbuqE_c,23538
+deepagents_cli-0.0.3.dist-info/licenses/LICENSE,sha256=c__BaxUCK69leo2yEKynf8lWndu8iwYwge1CbyqAe-E,1071
+deepagents_cli-0.0.3.dist-info/METADATA,sha256=r7jWMUaQJIf22JG6x9_bOrqoFVE1gqItgV3oI21B-M4,19792
+deepagents_cli-0.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+deepagents_cli-0.0.3.dist-info/entry_points.txt,sha256=5qO2sNhal5xQqcexm2VtT981A29FBKt-75aE4gatH8Q,55
+deepagents_cli-0.0.3.dist-info/top_level.txt,sha256=drAzchOzPNePwpb3_pbPuvLuayXkN7SNqeIKMBWJoAo,11
+deepagents_cli-0.0.3.dist-info/RECORD,,

deepagents/middleware/common.py

@@ -1,16 +0,0 @@
-"""Common middleware constants shared across filesystem middlewares."""
-
-# Message used when evicting a large tool result to the in-memory filesystem
-TOO_LARGE_TOOL_MSG = (
-    "Tool result too large, the result of this tool call {tool_call_id} was saved in the filesystem at this path: {file_path}\n"
-    "You can read the result from the filesystem by using the read_file tool, but make sure to only read part of the result at a time.\n"
-    "You can do this by specifying an offset and limit in the read_file tool call.\n"
-    "For example, to read the first 100 lines, you can use the read_file tool with offset=0 and limit=100.\n\n"
-    "Here are the first 10 lines of the result:\n{content_sample}\n"
-)
-
-# Supplement lines for tools that aren't listed in the base filesystem system prompt
-FILESYSTEM_SYSTEM_PROMPT_GLOB_GREP_SUPPLEMENT = (
-    "\n- glob: find files/directories by pattern\n"
-    "- grep: search file contents using ripgrep (rg)"
-)