feat: bump langgraph dependency to >=v1.0.2 with fixed ExecutorToolNode

Author: KYN4

minitap/mobile_use/agents/executor/tool_node.py

@@ -5,7 +5,8 @@
 from langchain_core.messages import AnyMessage, ToolCall, ToolMessage
 from langchain_core.runnables import RunnableConfig
 from langgraph.prebuilt import ToolNode
-from langgraph.store.base import BaseStore
+from langgraph.prebuilt.tool_node import ToolRuntime
+from langgraph.runtime import Runtime
 from langgraph.types import Command
 from pydantic import BaseModel
 
@@ -30,46 +31,61 @@ async def _afunc(
         self,
         input: list[AnyMessage] | dict[str, Any] | BaseModel,
         config: RunnableConfig,
-        *,
-        store: BaseStore | None,
+        runtime: Runtime,
     ):
-        return await self.__func(is_async=True, input=input, config=config, store=store)
+        return await self.__func(is_async=True, input=input, config=config, runtime=runtime)
 
     @override
     def _func(
         self,
         input: list[AnyMessage] | dict[str, Any] | BaseModel,
         config: RunnableConfig,
-        *,
-        store: BaseStore | None,
+        runtime: Runtime,
     ) -> Any:
         loop = asyncio.get_event_loop()
         return loop.run_until_complete(
-            self.__func(is_async=False, input=input, config=config, store=store)
+            self.__func(is_async=False, input=input, config=config, runtime=runtime)
+        )
+
+    def _build_tool_runtime(
+        self,
+        call: ToolCall,
+        input: list[AnyMessage] | dict[str, Any] | BaseModel,
+        config: RunnableConfig,
+        runtime: Runtime,
+    ) -> Any:
+        state = self._extract_state(input)
+        return ToolRuntime(
+            state=state,
+            tool_call_id=call["id"],
+            config=config,
+            context=runtime.context,
+            store=runtime.store,
+            stream_writer=runtime.stream_writer,
         )
 
     async def __func(
         self,
         is_async: bool,
         input: list[AnyMessage] | dict[str, Any] | BaseModel,
         config: RunnableConfig,
-        *,
-        store: BaseStore | None,
+        runtime: Runtime,
     ) -> Any:
-        tool_calls, input_type = self._parse_input(input, store)
+        tool_calls, input_type = self._parse_input(input)
         outputs: list[Command | ToolMessage] = []
         failed = False
         for call in tool_calls:
+            tool_runtime = self._build_tool_runtime(call, input, config, runtime)
             if failed:
                 output = self._get_erroneous_command(
                     call=call,
                     message="Aborted: a previous tool call failed!",
                 )
             else:
                 if is_async:
-                    output = await self._arun_one(call, input_type, config)
+                    output = await self._arun_one(call, input_type, tool_runtime)
                 else:
-                    output = self._run_one(call, input_type, config)
+                    output = self._run_one(call, input_type, tool_runtime)
                 failed = self._has_tool_call_failed(call, output)
                 if failed is None:
                     output = self._get_erroneous_command(
@@ -133,11 +149,11 @@ def _get_erroneous_command(self, call: ToolCall, message: str) -> Command:
         tool_message = ToolMessage(
             name=call["name"], tool_call_id=call["id"], content=message, status="error"
         )
-        return Command(update={self.messages_key: [tool_message]})
+        return Command(update={self._messages_key: [tool_message]})
 
     def _get_tool_message(self, cmd: Command) -> ToolMessage:
         if isinstance(cmd.update, dict):
-            msg = cmd.update.get(self.messages_key)
+            msg = cmd.update.get(self._messages_key)
             if isinstance(msg, list):
                 if len(msg) == 0:
                     raise ValueError("No messages found in command update")
@@ -147,6 +163,6 @@ def _get_tool_message(self, cmd: Command) -> ToolMessage:
             elif isinstance(msg, ToolMessage):
                 return msg
             elif msg is None:
-                raise ValueError(f"Missing '{self.messages_key}' in command update")
+                raise ValueError(f"Missing '{self._messages_key}' in command update")
             raise ValueError(f"Unexpected message type in command update: {type(msg)}")
         raise ValueError("Command update is not a dict")

minitap/mobile_use/main.py

@@ -6,7 +6,7 @@
 
 import typer
 from adbutils import AdbClient
-from langchain.callbacks.base import Callbacks
+from langchain_core.callbacks.base import Callbacks
 from rich.console import Console
 
 from minitap.mobile_use.clients.ios_client_config import (

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "minitap-mobile-use"
-version = "3.5.7"
+version = "3.6.0"
 description = "AI-powered multi-agent system that automates real Android and iOS devices through low-level control using LangGraph."
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -14,25 +14,25 @@ authors = [
 requires-python = ">=3.12"
 
 dependencies = [
-    "langgraph>=1.0.0",
-    "adbutils>=2.9.3",
-    "langchain-google-genai>=2.1.10",
-    "langchain>=0.3.27",
-    "langchain-core>=0.3.75",
-    "jinja2>=3.1.6",
-    "python-dotenv>=1.1.1",
-    "pydantic-settings>=2.10.1",
-    "langchain-mcp-adapters>=0.1.7",
-    "langchain-openai>=0.3.27",
-    "typer>=0.16.0",
-    "langchain-cerebras>=0.5.0",
+    "langgraph>=1.0.2,<2.0.0",
+    "adbutils==2.9.3",
+    "langchain-google-genai>=4.0.0",
+    "langchain>=1.0.0",
+    "langchain-core>=1.0.0",
+    "jinja2==3.1.6",
+    "python-dotenv==1.1.1",
+    "pydantic-settings==2.10.1",
+    "langchain-mcp-adapters>=0.2.0",
+    "langchain-openai>=1.0.0",
+    "typer==0.16.0",
+    "langchain-cerebras>=0.8.0",
     "inquirer>=3.4.0",
     "sseclient-py>=1.8.0",
     "fastapi>=0.111.0",
     "uvicorn[standard]>=0.30.1",
     "colorama>=0.4.6",
     "psutil>=5.9.0",
-    "langchain-google-vertexai>=2.0.28",
+    "langchain-google-vertexai>=3.0.0",
     "httpx>=0.28.1",
     "uiautomator2>=3.5.0",
     "fb-idb>=1.1.7",