droidrun 0.5.0.dev1__tar.gz → 0.5.0.dev2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: droidrun
-Version: 0.5.0.dev1
+Version: 0.5.0.dev2
 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.dev1 → droidrun-0.5.0.dev2}/docs/v5/features/custom-tools.mdx

@@ -109,7 +109,7 @@ async def screenshot_and_count(*, ctx, **kwargs) -> str:
     screenshot = await ctx.driver.screenshot()
 
     # Get UI state via the state provider
-    ui_state = await ctx.state_provider.get_state(ctx.driver)
+    ui_state = await ctx.state_provider.get_state()
     element_count = len(ui_state.elements) if ui_state else 0
 
     return f"Screenshot taken. Found {element_count} UI elements"

{droidrun-0.5.0.dev1 → droidrun-0.5.0.dev2}/docs/v5/sdk/base-tools.mdx

@@ -99,7 +99,8 @@ All DeviceDriver implementations may provide these methods (check `supported` se
 
 ```python
 class StateProvider:
-    async def get_state(self, driver: DeviceDriver) -> UIState: ...
+    def __init__(self, driver: DeviceDriver): ...
+    async def get_state(self) -> UIState: ...
 ```
 
 Base class for state providers. Subclass to support different platforms.
@@ -298,8 +299,8 @@ async def my_action(param: str, *, ctx: ActionContext) -> ActionResult:
 from droidrun.tools import AndroidStateProvider, AndroidDriver
 
 # StateProvider handles fetching + parsing + retries
-provider = AndroidStateProvider(tree_filter=my_filter, tree_formatter=my_formatter)
-ui_state = await provider.get_state(driver)
+provider = AndroidStateProvider(driver, tree_filter=my_filter, tree_formatter=my_formatter)
+ui_state = await provider.get_state()
 
 # UIState provides element lookup
 element = ui_state.get_element(5)

{droidrun-0.5.0.dev1 → droidrun-0.5.0.dev2}/droidrun/agent/codeact/codeact_agent.py

@@ -290,7 +290,7 @@ class CodeActAgent(Workflow):
 
         # Get device state
         try:
-            ui_state = await self.state_provider.get_state(self.action_ctx.driver)
+            ui_state = await self.state_provider.get_state()
             self.action_ctx.ui = ui_state
 
             # Update shared state

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

@@ -240,7 +240,7 @@ class FastAgent(Workflow):
 
         # Get device state
         try:
-            ui_state = await self.state_provider.get_state(self.action_ctx.driver)
+            ui_state = await self.state_provider.get_state()
             self.action_ctx.ui = ui_state
 
             # Update shared state

{droidrun-0.5.0.dev1 → droidrun-0.5.0.dev2}/droidrun/agent/droid/droid_agent.py

@@ -430,12 +430,14 @@ class DroidAgent(Workflow):
             self.state_provider = self._injected_state_provider
         elif is_ios:
             self.state_provider = IOSStateProvider(
+                driver,
                 use_normalized=self.config.agent.use_normalized_coordinates,
             )
         else:
             tree_filter = ConciseFilter() if vision_enabled else DetailedFilter()
             tree_formatter = IndexedFormatter()
             self.state_provider = AndroidStateProvider(
+                driver,
                 tree_filter=tree_filter,
                 tree_formatter=tree_formatter,
                 use_normalized=self.config.agent.use_normalized_coordinates,

{droidrun-0.5.0.dev1 → droidrun-0.5.0.dev2}/droidrun/agent/manager/manager_agent.py

@@ -377,7 +377,7 @@ class ManagerAgent(Workflow):
         logger.debug("💬 Preparing manager context...")
 
         # Get and format device state
-        ui_state = await self.state_provider.get_state(self.action_ctx.driver)
+        ui_state = await self.state_provider.get_state()
         self.action_ctx.ui = ui_state
 
         # Update shared state (previous ← current, current ← new)

{droidrun-0.5.0.dev1 → droidrun-0.5.0.dev2}/droidrun/agent/manager/stateless_manager_agent.py

@@ -176,7 +176,7 @@ class StatelessManagerAgent(Workflow):
     async def prepare_context(
         self, ctx: Context, ev: StartEvent
     ) -> ManagerContextEvent:
-        ui_state = await self.state_provider.get_state(self.action_ctx.driver)
+        ui_state = await self.state_provider.get_state()
         self.action_ctx.ui = ui_state
 
         self.shared_state.previous_formatted_device_state = (

{droidrun-0.5.0.dev1 → droidrun-0.5.0.dev2}/droidrun/tools/ui/ios_provider.py

@@ -48,12 +48,13 @@ class IOSStateProvider(StateProvider):
 
     supported = {"element_index", "convert_point"}
 
-    def __init__(self, use_normalized: bool = False) -> None:
+    def __init__(self, driver: DeviceDriver, use_normalized: bool = False) -> None:
+        super().__init__(driver)
         # TODO: normalized coordinates untested on iOS
         self.use_normalized = use_normalized
 
-    async def get_state(self, driver: DeviceDriver) -> UIState:
-        raw = await driver.get_ui_tree()
+    async def get_state(self) -> UIState:
+        raw = await self.driver.get_ui_tree()
         a11y_text = raw.get("a11y_raw", "")
         phone_state = raw.get("phone_state", {})
 

{droidrun-0.5.0.dev1 → droidrun-0.5.0.dev2}/droidrun/tools/ui/provider.py

@@ -26,7 +26,10 @@ class StateProvider:
 
     supported: set[str] = set()
 
-    async def get_state(self, driver: "DeviceDriver") -> UIState:
+    def __init__(self, driver: "DeviceDriver") -> None:
+        self.driver = driver
+
+    async def get_state(self) -> UIState:
         raise NotImplementedError
 
 
@@ -41,17 +44,19 @@ class AndroidStateProvider(StateProvider):
 
     def __init__(
         self,
+        driver: "DeviceDriver",
         tree_filter: "TreeFilter",
         tree_formatter: "TreeFormatter",
         use_normalized: bool = False,
         stealth: bool = False,
     ) -> None:
+        super().__init__(driver)
         self.tree_filter = tree_filter
         self.tree_formatter = tree_formatter
         self.use_normalized = use_normalized
         self._ui_cls = StealthUIState if stealth else UIState
 
-    async def get_state(self, driver: "DeviceDriver") -> UIState:
+    async def get_state(self) -> UIState:
         max_retries = 3
         last_error = None
 
@@ -59,7 +64,7 @@ class AndroidStateProvider(StateProvider):
             try:
                 logger.debug(f"Getting state (attempt {attempt + 1}/{max_retries})")
 
-                combined_data = await driver.get_ui_tree()
+                combined_data = await self.driver.get_ui_tree()
 
                 if "error" in combined_data:
                     raise Exception(

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

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

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

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