deepagents 0.2.6__py3-none-any.whl → 0.2.7__py3-none-any.whl

deepagents/middleware/__init__.py

@@ -1,13 +1,11 @@
 """Middleware for the DeepAgent."""
 
 from deepagents.middleware.filesystem import FilesystemMiddleware
-from deepagents.middleware.resumable_shell import ResumableShellToolMiddleware
 from deepagents.middleware.subagents import CompiledSubAgent, SubAgent, SubAgentMiddleware
 
 __all__ = [
     "CompiledSubAgent",
     "FilesystemMiddleware",
-    "ResumableShellToolMiddleware",
     "SubAgent",
     "SubAgentMiddleware",
 ]

{deepagents-0.2.6.dist-info → deepagents-0.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deepagents
-Version: 0.2.6
+Version: 0.2.7
 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

{deepagents-0.2.6.dist-info → deepagents-0.2.7.dist-info}/RECORD

@@ -8,12 +8,11 @@ deepagents/backends/sandbox.py,sha256=JueMe_2cZcA359JIqIi8kDUkmdtevC4VbKHw-fBPOW
 deepagents/backends/state.py,sha256=ST_tUExPxArJaA3U8vc1dyzxuYl2BQH-HU7P0eu_Ty8,6518
 deepagents/backends/store.py,sha256=f2LVSl65Dg-BZl-cY3pl3RqrUJCBMBm2kuAzZEODwsE,13098
 deepagents/backends/utils.py,sha256=Iyk2jW-gfoLvMnz-W_2FRCoJW_j3r1zoumU9iww-jd0,13973
-deepagents/middleware/__init__.py,sha256=x7UHqGcrKlhzORNdChPvnUwa_PIJCbFUHY6zTKVfloI,418
+deepagents/middleware/__init__.py,sha256=_OGIHcHZ2pRD0gzUBS7R48agwI6P7-FCBKBgjyaWlsg,303
 deepagents/middleware/filesystem.py,sha256=3PAetXqWy0i9bE6moM0FDZAEmjMm_M48B4AWWYl4Luk,37271
 deepagents/middleware/patch_tool_calls.py,sha256=PdNhxPaQqwnFkhEAZEE2kEzadTNAOO3_iJRA30WqpGE,1981
-deepagents/middleware/resumable_shell.py,sha256=KjhafjKu28Nf_8pDmSk_aWRK7pgkXZoubvWQljIEv3w,3382
 deepagents/middleware/subagents.py,sha256=RbNpWLXC0Bhr0nUIs40whybNnzNkhxG9Fie7QKsICRk,23748
-deepagents-0.2.6.dist-info/METADATA,sha256=gcNhcchWORoY_wyqYv8xU1lvvYccmOhuVDZIYTubNYI,18887
-deepagents-0.2.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-deepagents-0.2.6.dist-info/top_level.txt,sha256=drAzchOzPNePwpb3_pbPuvLuayXkN7SNqeIKMBWJoAo,11
-deepagents-0.2.6.dist-info/RECORD,,
+deepagents-0.2.7.dist-info/METADATA,sha256=H5FgpjWX5qzV5uHcpm0ba4F1Jwtw_Kvj79U9mdUR3Os,18887
+deepagents-0.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+deepagents-0.2.7.dist-info/top_level.txt,sha256=drAzchOzPNePwpb3_pbPuvLuayXkN7SNqeIKMBWJoAo,11
+deepagents-0.2.7.dist-info/RECORD,,

deepagents/middleware/resumable_shell.py

@@ -1,86 +0,0 @@
-"""Shell tool middleware that survives HITL pauses."""
-
-from __future__ import annotations
-
-from collections.abc import Awaitable, Callable
-from typing import Any, cast
-
-from langchain.agents.middleware.shell_tool import (
-    ShellToolMiddleware,
-    ShellToolState,
-    _PersistentShellTool,
-    _SessionResources,
-)
-from langchain.agents.middleware.types import AgentState
-from langchain.tools.tool_node import ToolCallRequest
-from langchain_core.messages import ToolMessage
-from langgraph.types import Command
-
-
-class ResumableShellToolMiddleware(ShellToolMiddleware):
-    """Shell middleware that recreates session resources after human interrupts.
-
-    ``ShellToolMiddleware`` stores its session handle in middleware state using an
-    ``UntrackedValue``. When a run pauses for human approval, that attribute is not
-    checkpointed. Upon resuming, LangGraph restores the state without the shell
-    resources, so the next tool execution fails with
-    ``Shell session resources are unavailable``.
-
-    This subclass lazily recreates the shell session the first time a resumed run
-    touches the shell tool again and only performs shutdown when a session is
-    actually active. This keeps behaviour identical for uninterrupted runs while
-    allowing HITL pauses to succeed.
-    """
-
-    def wrap_tool_call(
-        self,
-        request: ToolCallRequest,
-        handler: Callable[[ToolCallRequest], ToolMessage | Command],
-    ) -> ToolMessage | Command:
-        if isinstance(request.tool, _PersistentShellTool):
-            resources = self._get_or_create_resources(request.state)
-            return self._run_shell_tool(
-                resources,
-                request.tool_call["args"],
-                tool_call_id=request.tool_call.get("id"),
-            )
-        return super().wrap_tool_call(request, handler)
-
-    async def awrap_tool_call(
-        self,
-        request: ToolCallRequest,
-        handler: Callable[[ToolCallRequest], Awaitable[ToolMessage | Command]],
-    ) -> ToolMessage | Command:
-        if isinstance(request.tool, _PersistentShellTool):
-            resources = self._get_or_create_resources(request.state)
-            return self._run_shell_tool(
-                resources,
-                request.tool_call["args"],
-                tool_call_id=request.tool_call.get("id"),
-            )
-        return await super().awrap_tool_call(request, handler)
-
-    def after_agent(self, state: ShellToolState, runtime) -> None:  # type: ignore[override]
-        if self._has_resources(state):
-            super().after_agent(state, runtime)
-
-    async def aafter_agent(self, state: ShellToolState, runtime) -> None:  # type: ignore[override]
-        if self._has_resources(state):
-            await super().aafter_agent(state, runtime)
-
-    @staticmethod
-    def _has_resources(state: AgentState) -> bool:
-        resources = state.get("shell_session_resources")
-        return isinstance(resources, _SessionResources)
-
-    def _get_or_create_resources(self, state: AgentState) -> _SessionResources:
-        resources = state.get("shell_session_resources")
-        if isinstance(resources, _SessionResources):
-            return resources
-
-        new_resources = self._create_resources()
-        cast("dict[str, Any]", state)["shell_session_resources"] = new_resources
-        return new_resources
-
-
-__all__ = ["ResumableShellToolMiddleware"]