droidrun 0.3.4__tar.gz → 0.3.6__tar.gz

{droidrun-0.3.4 → droidrun-0.3.6}/.github/workflows/bounty.yml

@@ -64,7 +64,7 @@ jobs:
             eligible)      OPT_NAME="📝 Eligible"      ;;
             bounty-official) OPT_NAME="💰 Bounty" ;;
             claimed)       OPT_NAME="🛠️ In Progress"   ;;
-            *) echo "No matching status for $LABEL"; exit 0 ;;
+            *) echo "No matching status for $LABEL"; exit 78 ;;
           esac
 
           # extract from JSON map
@@ -131,4 +131,4 @@ jobs:
           -f proj="${PROJECT_ID}" \
           -f item="${ITEM_ID}" \
           -f field="${STATUS_FIELD_ID}" \
-          -f opt="${OPTION_ID}"
+          -f opt="${OPTION_ID}"

{droidrun-0.3.4 → droidrun-0.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: droidrun
-Version: 0.3.4
+Version: 0.3.6
 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
@@ -26,32 +26,26 @@ Classifier: Topic :: Software Development :: Testing :: Acceptance
 Classifier: Topic :: System :: Emulators
 Classifier: Topic :: Utilities
 Requires-Python: >=3.11
-Requires-Dist: adbutils==2.10.0
-Requires-Dist: aiofiles>=23.0.0
-Requires-Dist: anthropic>=0.7.0
+Requires-Dist: adbutils>=2.10.2
+Requires-Dist: anthropic>=0.67.0
 Requires-Dist: apkutils==2.0.0
-Requires-Dist: arize-phoenix
-Requires-Dist: click>=8.1.0
-Requires-Dist: llama-index
-Requires-Dist: llama-index-callbacks-arize-phoenix
-Requires-Dist: llama-index-llms-anthropic
-Requires-Dist: llama-index-llms-deepseek
-Requires-Dist: llama-index-llms-google-genai
-Requires-Dist: llama-index-llms-ollama
-Requires-Dist: llama-index-llms-openai
-Requires-Dist: llama-index-llms-openai-like
-Requires-Dist: openai>=1.0.0
-Requires-Dist: pillow>=10.0.0
-Requires-Dist: posthog==6.0.2
-Requires-Dist: pydantic>=2.0.0
-Requires-Dist: python-dotenv>=1.0.0
-Requires-Dist: rich>=13.0.0
+Requires-Dist: llama-index-llms-anthropic>=0.8.6
+Requires-Dist: llama-index-llms-deepseek>=0.2.1
+Requires-Dist: llama-index-llms-google-genai>=0.3.1
+Requires-Dist: llama-index-llms-ollama>=0.7.2
+Requires-Dist: llama-index-llms-openai-like>=0.5.1
+Requires-Dist: llama-index-llms-openai>=0.5.6
+Requires-Dist: llama-index>=0.14.0
+Requires-Dist: openai>=1.107.1
+Requires-Dist: posthog>=6.7.4
+Requires-Dist: pydantic>=2.11.7
+Requires-Dist: rich>=14.1.0
 Provides-Extra: dev
-Requires-Dist: bandit>=1.7.0; extra == 'dev'
+Requires-Dist: bandit>=1.8.6; extra == 'dev'
 Requires-Dist: black>=23.0.0; extra == 'dev'
 Requires-Dist: mypy>=1.0.0; extra == 'dev'
-Requires-Dist: ruff>=0.1.0; extra == 'dev'
-Requires-Dist: safety>=2.0.0; extra == 'dev'
+Requires-Dist: ruff>=0.13.0; extra == 'dev'
+Requires-Dist: safety>=3.2.11; extra == 'dev'
 Description-Content-Type: text/markdown
 
 <picture>

{droidrun-0.3.4 → droidrun-0.3.6}/docs/docs.json

@@ -14,7 +14,7 @@
         "tab": "Framework",
         "versions": [
           {
-            "version": "0.3.3",
+            "version": "0.3.6",
             "groups": [
               {
                 "group": "Introduction",

{droidrun-0.3.4 → droidrun-0.3.6}/docs/v3/quickstart.mdx

@@ -26,8 +26,9 @@ Before installing DroidRun, ensure you have:
    - Connected via USB or on the same network (for wireless debugging)
 
 ### Install from PyPI
+Use [`uv`](https://docs.astral.sh/uv/getting-started/installation/)
 ```bash
-pip install droidrun
+uv pip install droidrun
 ```
 
 ### Setup the Portal APK
@@ -91,4 +92,4 @@ Now that you've got DroidRun running, you can:
 - Learn about the [Agent](/v3/concepts/agent)
 - See supported [LLM providers](/v3/concepts/models)
 - Discover [Android interactions](/v3/concepts/android-control)
-- Learn about the [Portal App](/v3/concepts/portal-app) 
+- Learn about the [Portal App](/v3/concepts/portal-app) 

{droidrun-0.3.4 → droidrun-0.3.6}/docs/v3/sdk/droid-agent.mdx

@@ -33,7 +33,7 @@ def __init__(
     reflection: bool = False,
     enable_tracing: bool = False,
     debug: bool = False,
-    save_trajectories: str = "none",
+    save_trajectories: str = "none",  
     excluded_tools: List[str] = None,
     *args,
     **kwargs

{droidrun-0.3.4 → droidrun-0.3.6}/droidrun/agent/codeact/codeact_agent.py

@@ -18,7 +18,8 @@ from droidrun.agent.codeact.events import (
     TaskThinkingEvent,
     EpisodicMemoryEvent,
 )
-from droidrun.agent.common.events import ScreenshotEvent
+from droidrun.agent.common.events import ScreenshotEvent, RecordUIStateEvent
+from droidrun.agent.usage import get_usage_from_response
 from droidrun.agent.utils import chat_utils
 from droidrun.agent.utils.executer import SimpleCodeExecutor
 from droidrun.agent.codeact.prompts import (
@@ -182,6 +183,7 @@ class CodeActAgent(Workflow):
                 try:
                     state = self.tools.get_state()
                     await ctx.set("ui_state", state["a11y_tree"])
+                    ctx.write_event_to_stream(RecordUIStateEvent(ui_state=state["a11y_tree"]))
                     chat_history = await chat_utils.add_ui_text_block(
                         state["a11y_tree"], chat_history
                     )
@@ -202,11 +204,17 @@ class CodeActAgent(Workflow):
                 success=False, reason="LLM response is None. This is a critical error."
             )
 
+        try:
+            usage = get_usage_from_response(self.llm.class_name(), response)
+        except Exception as e:
+            logger.warning(f"Could not get llm usage from response: {e}")
+            usage = None
+
         await self.chat_memory.aput(response.message)
 
         code, thoughts = chat_utils.extract_code_and_thought(response.message.content)
 
-        event = TaskThinkingEvent(thoughts=thoughts, code=code)
+        event = TaskThinkingEvent(thoughts=thoughts, code=code, usage=usage)
         ctx.write_event_to_stream(event)
         return event
 
@@ -255,6 +263,10 @@ class CodeActAgent(Workflow):
             for screenshot in screenshots[:-1]: # the last screenshot will be captured by next step
                 ctx.write_event_to_stream(ScreenshotEvent(screenshot=screenshot))
 
+            ui_states = result['ui_states']
+            for ui_state in ui_states[:-1]:
+                ctx.write_event_to_stream(RecordUIStateEvent(ui_state=ui_state['a11y_tree']))
+
             if self.tools.finished == True:
                 logger.debug("  - Task completed.")
                 event = TaskEndEvent(
@@ -311,7 +323,8 @@ class CodeActAgent(Workflow):
         await ctx.set("chat_memory", self.chat_memory)
         
         # Add final state observation to episodic memory
-        await self._add_final_state_observation(ctx)
+        if self.vision:
+            await self._add_final_state_observation(ctx)
         
         result = {}
         result.update(

{droidrun-0.3.4 → droidrun-0.3.6}/droidrun/agent/codeact/events.py

@@ -1,6 +1,8 @@
 from llama_index.core.llms import ChatMessage
 from llama_index.core.workflow import Event
 from typing import Optional
+
+from droidrun.agent.usage import UsageResult
 from ..context.episodic_memory import EpisodicMemory
 
 class TaskInputEvent(Event):
@@ -11,6 +13,7 @@ class TaskInputEvent(Event):
 class TaskThinkingEvent(Event):
     thoughts: Optional[str] = None
     code: Optional[str] = None  
+    usage: Optional[UsageResult] = None
 
 class TaskExecutionEvent(Event):
     code: str

{droidrun-0.3.4 → droidrun-0.3.6}/droidrun/agent/common/events.py

@@ -1,4 +1,5 @@
 from llama_index.core.workflow import Event
+from typing import Dict, Any
 
 class ScreenshotEvent(Event):
     screenshot: bytes
@@ -44,4 +45,7 @@ class KeyPressActionEvent(MacroEvent):
 class StartAppEvent(MacroEvent):
     """"Event for starting an app"""
     package: str
-    activity: str = None
+    activity: str = None
+
+class RecordUIStateEvent(Event):
+    ui_state: list[Dict[str, Any]]

{droidrun-0.3.4 → droidrun-0.3.6}/droidrun/agent/context/task_manager.py

@@ -43,6 +43,13 @@ class TaskManager:
     def get_task_history(self):
         return self.task_history
 
+    def get_current_task(self) -> Optional[Task]:
+        """Return the first task with status "pending" from the task list."""
+        for task in self.tasks:
+            if task.status == self.STATUS_PENDING:
+                return task
+        return None
+
     def complete_task(self, task: Task, message: Optional[str] = None):
         task = copy.deepcopy(task)
         task.status = self.STATUS_COMPLETED
@@ -60,6 +67,9 @@ class TaskManager:
 
     def get_failed_tasks(self) -> list[dict]:
         return [task for task in self.task_history if task.status == self.STATUS_FAILED]
+    
+    def get_task_history(self) -> list[dict]:
+        return self.task_history
 
 
     def save_to_file(self):