droidrun 0.5.0.dev2__tar.gz → 0.5.0.dev3__tar.gz

{droidrun-0.5.0.dev2 → droidrun-0.5.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: droidrun
-Version: 0.5.0.dev2
+Version: 0.5.0.dev3
 Summary: A framework for controlling Android devices through LLM agents
 Project-URL: Homepage, https://github.com/droidrun/droidrun
 Project-URL: Bug Tracker, https://github.com/droidrun/droidrun/issues

{droidrun-0.5.0.dev2 → droidrun-0.5.0.dev3}/droidrun/agent/codeact/tools_agent.py

@@ -408,7 +408,7 @@ class FastAgent(Workflow):
             else:
                 # Dispatch via registry
                 action_result = await self.registry.execute(
-                    call.name, call.parameters, self.action_ctx
+                    call.name, call.parameters, self.action_ctx, workflow_ctx=ctx
                 )
             results.append(
                 ToolResult(

{droidrun-0.5.0.dev2 → droidrun-0.5.0.dev3}/droidrun/agent/common/events.py

@@ -9,3 +9,12 @@ class ScreenshotEvent(Event):
 
 class RecordUIStateEvent(Event):
     ui_state: list[Dict[str, Any]]
+
+
+class ToolExecutionEvent(Event):
+    """Emitted after every tool call dispatched through ToolRegistry."""
+
+    tool_name: str
+    tool_args: Dict[str, Any]
+    success: bool
+    summary: str

{droidrun-0.5.0.dev2 → droidrun-0.5.0.dev3}/droidrun/agent/executor/executor_agent.py

@@ -241,7 +241,9 @@ class ExecutorAgent(Workflow):
         action_type = action_dict.get("action", "unknown")
         action_args = {k: v for k, v in action_dict.items() if k != "action"}
 
-        result = await self.registry.execute(action_type, action_args, self.action_ctx)
+        result = await self.registry.execute(
+            action_type, action_args, self.action_ctx, workflow_ctx=ctx
+        )
 
         await asyncio.sleep(self.agent_config.after_sleep_action)
 

{droidrun-0.5.0.dev2 → droidrun-0.5.0.dev3}/droidrun/agent/tool_registry.py

@@ -15,6 +15,8 @@ from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Set
 from droidrun.agent.action_result import ActionResult
 
 if TYPE_CHECKING:
+    from llama_index.core.workflow import Context as WorkflowContext
+
     from droidrun.agent.action_context import ActionContext
 
 logger = logging.getLogger("droidrun")
@@ -105,11 +107,16 @@ class ToolRegistry:
         name: str,
         args: Dict[str, Any],
         ctx: "ActionContext",
+        workflow_ctx: "Optional[WorkflowContext]" = None,
     ) -> ActionResult:
         """Dispatch action by name.
 
         All actions receive ``ctx=ctx`` as a keyword argument.
 
+        Args:
+            workflow_ctx: Optional llama-index workflow Context. When provided,
+                a ``ToolExecutionEvent`` is streamed after execution.
+
         Handles:
         - Unknown tool names
         - Sync vs async functions
@@ -117,12 +124,14 @@ class ToolRegistry:
         - General exceptions
         """
         if name not in self.tools:
-            return ActionResult(
+            result = ActionResult(
                 success=False,
                 summary=(
                     f"Unknown tool: {name}. " f"Available: {list(self.tools.keys())}"
                 ),
             )
+            self._emit_event(workflow_ctx, name, args, result)
+            return result
 
         entry = self.tools[name]
         try:
@@ -131,25 +140,55 @@ class ToolRegistry:
             else:
                 result = entry.fn(**args, ctx=ctx)
         except TypeError as e:
-            return ActionResult(
+            result = ActionResult(
                 success=False,
                 summary=f"Invalid arguments for {name}: {e}",
             )
+            self._emit_event(workflow_ctx, name, args, result)
+            return result
         except Exception as e:
-            return ActionResult(
+            result = ActionResult(
                 success=False,
                 summary=f"Failed to execute {name}: {e}",
             )
+            self._emit_event(workflow_ctx, name, args, result)
+            return result
 
         # Normalise the return value into ActionResult
         if isinstance(result, ActionResult):
-            return result
-        if isinstance(result, tuple):
-            return ActionResult(success=result[0], summary=str(result[1]))
-        if isinstance(result, str):
+            action_result = result
+        elif isinstance(result, tuple):
+            action_result = ActionResult(success=result[0], summary=str(result[1]))
+        elif isinstance(result, str):
             success = not result.startswith("Failed")
-            return ActionResult(success=success, summary=result)
-        return ActionResult(success=True, summary=str(result) if result else "Done")
+            action_result = ActionResult(success=success, summary=result)
+        else:
+            action_result = ActionResult(
+                success=True, summary=str(result) if result else "Done"
+            )
+
+        self._emit_event(workflow_ctx, name, args, action_result)
+        return action_result
+
+    @staticmethod
+    def _emit_event(
+        workflow_ctx: "Optional[WorkflowContext]",
+        name: str,
+        args: Dict[str, Any],
+        result: ActionResult,
+    ) -> None:
+        if workflow_ctx is None:
+            return
+        from droidrun.agent.common.events import ToolExecutionEvent
+
+        workflow_ctx.write_event_to_stream(
+            ToolExecutionEvent(
+                tool_name=name,
+                tool_args=args,
+                success=result.success,
+                summary=result.summary,
+            )
+        )
 
     # -- prompt helpers ------------------------------------------------------
 

{droidrun-0.5.0.dev2 → droidrun-0.5.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "droidrun"
-version = "0.5.0.dev2"
+version = "0.5.0.dev3"
 description = "A framework for controlling Android devices through LLM agents"
 authors = [{ name = "Niels Schmidt", email = "niels.schmidt@droidrun.ai" }]
 dependencies = [

{droidrun-0.5.0.dev2 → droidrun-0.5.0.dev3}/uv.lock

@@ -760,7 +760,7 @@ wheels = [
 
 [[package]]
 name = "droidrun"
-version = "0.5.0.dev2"
+version = "0.5.0.dev3"
 source = { editable = "." }
 dependencies = [
     { name = "aiofiles" },