droidrun 0.3.8__py3-none-any.whl → 0.3.10.dev2__py3-none-any.whl

droidrun/agent/context/task_manager.py

@@ -1,7 +1,8 @@
+import copy
 import os
-from typing import List, Dict, Optional
 from dataclasses import dataclass
-import copy
+from typing import Dict, List, Optional
+
 
 @dataclass
 class Task:
@@ -14,13 +15,13 @@ class Task:
     # Optional fields to carry success/failure context back to the planner
     message: Optional[str] = None
     failure_reason: Optional[str] = None
-    
+
 
 class TaskManager:
     """
     Manages a list of tasks for an agent, each with a status and assigned specialized agent.
     """
-    STATUS_PENDING = "pending"      
+    STATUS_PENDING = "pending"
     STATUS_COMPLETED = "completed"
     STATUS_FAILED = "failed"
 
@@ -32,14 +33,14 @@ class TaskManager:
     def __init__(self):
         """Initializes an empty task list."""
         self.tasks: List[Task] = []
-        self.goal_completed = False 
+        self.goal_completed = False
         self.message = None
-        self.task_history = [] 
+        self.task_history = []
         self.file_path = os.path.join(os.path.dirname(__file__), "todo.txt")
 
     def get_all_tasks(self) -> List[Task]:
         return self.tasks
-        
+
     def get_task_history(self):
         return self.task_history
 
@@ -67,9 +68,6 @@ 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):
@@ -89,12 +87,12 @@ class TaskManager:
     def set_tasks_with_agents(self, task_assignments: List[Dict[str, str]]):
         """
         Clears the current task list and sets new tasks with their assigned agents.
-        
+
         Args:
             task_assignments: A list of dictionaries, each containing:
                             - 'task': The task description string
                             - 'agent': The agent type
-                            
+
         Example:
             task_manager.set_tasks_with_agents([
                 {'task': 'Open Gmail app', 'agent': 'AppStarterExpert'},
@@ -106,21 +104,21 @@ class TaskManager:
             for i, assignment in enumerate(task_assignments):
                 if not isinstance(assignment, dict) or 'task' not in assignment:
                     raise ValueError(f"Each task assignment must be a dictionary with 'task' key at index {i}.")
-                
+
                 task_description = assignment['task']
                 if not isinstance(task_description, str) or not task_description.strip():
                     raise ValueError(f"Task description must be a non-empty string at index {i}.")
-                
+
                 agent_type = assignment.get('agent', 'Default')
-                
+
                 task_obj = Task(
                     description=task_description.strip(),
                     status=self.STATUS_PENDING,
                     agent_type=agent_type
                 )
-                
+
                 self.tasks.append(task_obj)
-            
+
             print(f"Tasks set with agents: {len(self.tasks)} tasks added.")
             self.save_to_file()
         except Exception as e:

droidrun/agent/droid/__init__.py

@@ -10,4 +10,4 @@ from droidrun.agent.droid.droid_agent import DroidAgent
 __all__ = [
     "CodeActAgent",
     "DroidAgent"
-] 
+]