lybic-guiagents 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

gui_agents/__init__.py

@@ -37,7 +37,7 @@ from .agents.hardware_interface import HardwareInterface
 from .store.registry import Registry
 from .agents.global_state import GlobalState
 
-__version__ = "0.3.0"
+__version__ = "0.4.0"
 
 # Primary exports (what users should typically use)
 __all__ = [

gui_agents/agents/agent_s.py

@@ -172,8 +172,6 @@ class AgentS2(UIAgent):
         else:
             print(f"Found local knowledge base path: {kb_platform_path}")
 
-        self.reset()
-
     def reset(self) -> None:
         """
         Reinitialize core components and reset the agent's runtime state.
@@ -219,12 +217,18 @@ class AgentS2(UIAgent):
         self.subtasks: List[Node] = []
         self.search_query: str = ""
         self.subtask_status: str = "Start"
-        self.global_state: GlobalState = Registry.get("GlobalStateStore") # type: ignore
+        # Use task-specific registry if task_id is available, otherwise fall back to global registry
+        if self.task_id:
+            self.global_state: GlobalState = Registry.get_from_context("GlobalStateStore", self.task_id) # type: ignore
+        else:
+            self.global_state: GlobalState = Registry.get("GlobalStateStore") # type: ignore
 
         # Pass task_id to components
-        if self.task_id:
-            self.manager.task_id = self.task_id
-            self.worker.task_id = self.task_id
+        self.manager.set_task_id(self.task_id)
+        self.worker.set_task_id(self.task_id)
+        # Grounding doesn't have task_id in normal mode, but we set it if available
+        if hasattr(self, 'grounding') and hasattr(self.grounding, 'set_task_id'):
+            self.grounding.set_task_id(self.task_id)
 
     def set_task_id(self, task_id: str) -> None:
         """
@@ -236,9 +240,9 @@ class AgentS2(UIAgent):
         self.task_id = task_id
         # Also set task_id for components if they exist
         if hasattr(self, 'manager') and self.manager:
-            self.manager.task_id = task_id
+            self.manager.set_task_id(task_id)
         if hasattr(self, 'worker') and self.worker:
-            self.worker.task_id = task_id
+            self.worker.set_task_id(task_id)
 
     def reset_executor_state(self) -> None:
         """Reset executor and step counter"""
@@ -733,8 +737,6 @@ class AgentSFast(UIAgent):
         else:
             print(f"Found local knowledge base path: {kb_platform_path}")
 
-        self.reset()
-
     def reset(self) -> None:
         """
         Reinitialize the fast-agent components and reset internal runtime state.
@@ -816,19 +818,25 @@ class AgentSFast(UIAgent):
         # Reset state variables
         self.step_count: int = 0
         self.turn_count: int = 0
-        self.global_state: GlobalState = Registry.get("GlobalStateStore") # type: ignore
+        # Use task-specific registry if task_id is available, otherwise fall back to global registry
+        if self.task_id:
+            self.global_state: GlobalState = Registry.get_from_context("GlobalStateStore", self.task_id) # type: ignore
+        else:
+            self.global_state: GlobalState = Registry.get("GlobalStateStore") # type: ignore
         self.latest_action = None
 
-        # Pass task_id to tools if available
-        if self.task_id:
-            self.fast_action_generator.task_id = self.task_id
-            if self.enable_reflection and hasattr(self, 'reflection_agent'):
-                self.reflection_agent.task_id = self.task_id
+        # Pass task_id to tools and components if available
+        self.fast_action_generator.task_id = self.task_id
+        if self.enable_reflection and hasattr(self, 'reflection_agent'):
+            self.reflection_agent.task_id = self.task_id
+        # Set task_id for grounding component
+        if hasattr(self, 'grounding') and hasattr(self.grounding, 'set_task_id'):
+            self.grounding.set_task_id(self.task_id)
 
     def set_task_id(self, task_id: str) -> None:
         """
         Store the task identifier on the agent and propagate it to subcomponents that use it.
-        
+
         Parameters:
             task_id (str): Identifier for the active task; assigned to this agent and, if present, to
                 `fast_action_generator` and `reflection_agent`.
@@ -839,6 +847,9 @@ class AgentSFast(UIAgent):
             self.fast_action_generator.task_id = task_id
         if hasattr(self, 'reflection_agent') and self.reflection_agent:
             self.reflection_agent.task_id = task_id
+        # Set task_id for grounding component
+        if hasattr(self, 'grounding') and hasattr(self.grounding, 'set_task_id'):
+            self.grounding.set_task_id(task_id)
 
     def predict(self, instruction: str, observation: Dict) -> Tuple[Dict, List[str]]:
         """

gui_agents/agents/grounding.py

@@ -92,8 +92,13 @@ class Grounding(ACI):
         self.text_span_agent = Tools()
         _register(self.text_span_agent, "text_span")
 
-        self.global_state: GlobalState = Registry.get(
-            "GlobalStateStore")  # type: ignore
+        # GlobalState will be initialized when task_id is set
+        self.global_state: GlobalState = None  # type: ignore
+
+    def set_task_id(self, task_id: str) -> None:
+        """Set the task identifier and update global state reference"""
+        # Update global state reference with task-specific registry
+        self.global_state = Registry.get_from_context("GlobalStateStore", task_id)  # type: ignore
 
     def generate_coords(self, ref_expr: str, obs: Dict) -> List[int]:
         grounding_start_time = time.time()
@@ -453,8 +458,13 @@ class FastGrounding(ACI):
         self.height = height
         self.grounding_width = grounding_width
         self.grounding_height = grounding_height
-        self.global_state: GlobalState = Registry.get(
-            "GlobalStateStore")  # type: ignore
+        # GlobalState will be initialized when task_id is set
+        self.global_state: GlobalState = None  # type: ignore
+
+    def set_task_id(self, task_id: str) -> None:
+        """Set the task identifier and update global state reference"""
+        # Update global state reference with task-specific registry
+        self.global_state = Registry.get_from_context("GlobalStateStore", task_id)  # type: ignore
 
     def reset_screen_size(self, width: int, height: int):
         self.width = width

gui_agents/agents/manager.py

@@ -102,7 +102,8 @@ class Manager:
             Tools_dict=KB_Tools_dict,
         )
 
-        self.global_state: GlobalState = Registry.get("GlobalStateStore") # type: ignore
+        # GlobalState will be initialized in reset() method when task_id is available
+        self.global_state: GlobalState = None  # type: ignore
 
         self.planner_history = []
 
@@ -118,6 +119,12 @@ class Manager:
 
         self.multi_round = multi_round
 
+    def set_task_id(self, task_id: str) -> None:
+        """Set the task identifier and update global state reference"""
+        self.task_id = task_id
+        # Update global state reference with task-specific registry
+        self.global_state = Registry.get_from_context("GlobalStateStore", task_id)  # type: ignore
+
     def _send_stream_message(self, task_id: str, stage: str, message: str) -> None:
         """
         Enqueue a stream message for the given task if a task ID is provided.

gui_agents/agents/worker.py

@@ -67,8 +67,8 @@ class Worker:
 
         self.enable_reflection = enable_reflection
         self.use_subtask_experience = use_subtask_experience
-        self.global_state: GlobalState = Registry.get(
-            "GlobalStateStore")  # type: ignore
+        # GlobalState will be initialized in reset() method when task_id is available
+        self.global_state: GlobalState = None  # type: ignore
         self.reset()
 
     def reset(self):
@@ -159,6 +159,12 @@ class Worker:
         self.max_trajector_length = 8
         self.task_id = None  # Will be set by agent
 
+    def set_task_id(self, task_id: str) -> None:
+        """Set the task identifier and update global state reference"""
+        self.task_id = task_id
+        # Update global state reference with task-specific registry
+        self.global_state = Registry.get_from_context("GlobalStateStore", task_id)  # type: ignore
+
     def generate_next_action(
         self,
         Tu: str,