autobyteus 1.2.0__py3-none-any.whl → 1.2.1__py3-none-any.whl

autobyteus/task_management/tools/{update_task_status.py → task_tools/update_task_status.py}

@@ -7,7 +7,7 @@ from autobyteus.tools.base_tool import BaseTool
 from autobyteus.tools.tool_category import ToolCategory
 from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
 from autobyteus.tools.pydantic_schema_converter import pydantic_to_parameter_schema
-from autobyteus.task_management.base_task_board import TaskStatus
+from autobyteus.task_management.base_task_plan import TaskStatus
 from autobyteus.task_management.deliverable import FileDeliverable
 from autobyteus.task_management.schemas import FileDeliverableSchema
 
@@ -18,18 +18,18 @@ if TYPE_CHECKING:
 logger = logging.getLogger(__name__)
 
 class UpdateTaskStatus(BaseTool):
-    """A tool for member agents to update their progress and submit file deliverables on the TaskBoard."""
+    """A tool for member agents to update their progress and submit file deliverables on the TaskPlan."""
 
     CATEGORY = ToolCategory.TASK_MANAGEMENT
 
     @classmethod
     def get_name(cls) -> str:
-        return "UpdateTaskStatus"
+        return "update_task_status"
 
     @classmethod
     def get_description(cls) -> str:
         return (
-            "Updates the status of a specific task on the team's shared task board. "
+            "Updates the status of a specific task on the team's shared task plan. "
             "When completing a task, this tool can also be used to formally submit a list of file deliverables."
         )
 
@@ -60,29 +60,29 @@ class UpdateTaskStatus(BaseTool):
 
     async def _execute(self, context: 'AgentContext', task_name: str, status: str, deliverables: Optional[List[Dict[str, Any]]] = None) -> str:
         agent_name = context.config.name
-        log_msg = f"Agent '{agent_name}' is executing UpdateTaskStatus for task '{task_name}' to status '{status}'"
+        log_msg = f"Agent '{agent_name}' is executing update_task_status for task '{task_name}' to status '{status}'"
         if deliverables:
             log_msg += f" with {len(deliverables)} deliverable(s)."
         logger.info(log_msg)
         
         team_context: Optional['AgentTeamContext'] = context.custom_data.get("team_context")
         if not team_context:
-            error_msg = "Error: Team context is not available. Cannot access the task board."
+            error_msg = "Error: Team context is not available. Cannot access the task plan."
             logger.error(f"Agent '{agent_name}': {error_msg}")
             return error_msg
             
-        task_board = getattr(team_context.state, 'task_board', None)
-        if not task_board:
-            error_msg = "Error: Task board has not been initialized for this team."
+        task_plan = getattr(team_context.state, 'task_plan', None)
+        if not task_plan:
+            error_msg = "Error: Task plan has not been initialized for this team."
             logger.error(f"Agent '{agent_name}': {error_msg}")
             return error_msg
         
-        if not task_board.tasks:
-            error_msg = "Error: No tasks are currently loaded on the task board."
-            logger.warning(f"Agent '{agent_name}' tried to update task status, but the board is empty.")
+        if not task_plan.tasks:
+            error_msg = "Error: No tasks are currently loaded on the task plan."
+            logger.warning(f"Agent '{agent_name}' tried to update task status, but the plan is empty.")
             return error_msg
 
-        target_task = next((t for t in task_board.tasks if t.task_name == task_name), None)
+        target_task = next((t for t in task_plan.tasks if t.task_name == task_name), None)
 
         if not target_task:
             error_msg = f"Failed to update status for task '{task_name}'. The task name does not exist on the current plan."
@@ -93,7 +93,7 @@ class UpdateTaskStatus(BaseTool):
             status_enum = TaskStatus(status)
         except ValueError:
             error_msg = f"Invalid status '{status}'. Must be one of: {', '.join([s.value for s in TaskStatus])}."
-            logger.warning(f"Agent '{agent_name}' provided invalid status for UpdateTaskStatus: {status}")
+            logger.warning(f"Agent '{agent_name}' provided invalid status for update_task_status: {status}")
             return f"Error: {error_msg}"
         
         if deliverables:
@@ -108,8 +108,8 @@ class UpdateTaskStatus(BaseTool):
                 logger.warning(f"Agent '{agent_name}': {error_msg}")
                 return f"Error: {error_msg}"
 
-        if not task_board.update_task_status(target_task.task_id, status_enum, agent_name):
-            error_msg = f"Failed to update status for task '{task_name}'. An unexpected error occurred on the task board."
+        if not task_plan.update_task_status(target_task.task_id, status_enum, agent_name):
+            error_msg = f"Failed to update status for task '{task_name}'. An unexpected error occurred on the task plan."
             logger.error(f"Agent '{agent_name}': {error_msg}")
             return f"Error: {error_msg}"
 

autobyteus/task_management/tools/todo_tools/__init__.py

@@ -0,0 +1,18 @@
+# file: autobyteus/task_management/tools/todo_tools/__init__.py
+"""
+To-do management tool package containing list creation and update utilities.
+"""
+from .create_todo_list import CreateToDoList
+from .add_todo import AddToDo
+from .get_todo_list import GetToDoList
+from .update_todo_status import UpdateToDoStatus
+
+UpdateToDoStatusTool = UpdateToDoStatus
+
+__all__ = [
+    "CreateToDoList",
+    "AddToDo",
+    "GetToDoList",
+    "UpdateToDoStatus",
+    "UpdateToDoStatusTool",
+]

autobyteus/task_management/tools/todo_tools/add_todo.py

@@ -0,0 +1,78 @@
+# file: autobyteus/autobyteus/task_management/tools/todo_tools/add_todo.py
+import logging
+from typing import TYPE_CHECKING, Any
+
+from pydantic import ValidationError
+
+from autobyteus.tools.base_tool import BaseTool
+from autobyteus.tools.tool_category import ToolCategory
+from autobyteus.tools.pydantic_schema_converter import pydantic_to_parameter_schema
+from autobyteus.task_management.schemas.todo_definition import ToDoDefinitionSchema
+from autobyteus.task_management.todo_list import ToDoList
+
+if TYPE_CHECKING:
+    from autobyteus.agent.context import AgentContext
+
+logger = logging.getLogger(__name__)
+
+def _notify_todo_update(context: 'AgentContext'):
+    if context.phase_manager and context.phase_manager.notifier:
+        todo_list = context.state.todo_list
+        if todo_list:
+            todos_for_llm = [todo.model_dump(mode='json') for todo in todo_list.get_all_todos()]
+            context.phase_manager.notifier.notify_agent_data_todo_list_updated(todos_for_llm)
+            logger.debug(f"Agent '{context.agent_id}': Notified ToDo list update with {len(todos_for_llm)} items.")
+
+class AddToDo(BaseTool):
+    """A tool for an agent to add a new item to its personal to-do list."""
+
+    CATEGORY = ToolCategory.TASK_MANAGEMENT
+
+    @classmethod
+    def get_name(cls) -> str:
+        return "add_todo"
+
+    @classmethod
+    def get_description(cls) -> str:
+        return (
+            "Adds a single new item to your personal to-do list. "
+            "Use this if you discover a new step is needed to complete your task."
+        )
+
+    @classmethod
+    def get_argument_schema(cls) -> Any:
+        return pydantic_to_parameter_schema(ToDoDefinitionSchema)
+
+    async def _execute(self, context: 'AgentContext', **kwargs: Any) -> str:
+        agent_id = context.agent_id
+        logger.info(f"Agent '{agent_id}' is executing add_todo.")
+
+        if context.state.todo_list is None:
+            # If no list exists, create one.
+            context.state.todo_list = ToDoList(agent_id=agent_id)
+            logger.info(f"Agent '{agent_id}': No existing to-do list found, created a new one.")
+        
+        todo_list = context.state.todo_list
+
+        try:
+            todo_def_schema = ToDoDefinitionSchema(**kwargs)
+        except ValidationError as e:
+            error_msg = f"Invalid to-do item definition provided: {e}"
+            logger.warning(f"Agent '{agent_id}' provided an invalid definition for add_todo: {error_msg}")
+            return f"Error: {error_msg}"
+
+        # The add_todo method now takes the definition and returns the created ToDo object
+        new_todo = todo_list.add_todo(todo_def_schema)
+
+        if new_todo:
+            # Notify about the update
+            _notify_todo_update(context)
+
+            success_msg = f"Successfully added new item to your to-do list: '{new_todo.description}' (ID: {new_todo.todo_id})."
+            logger.info(f"Agent '{agent_id}': {success_msg}")
+            return success_msg
+        else:
+            # This path is unlikely given the refactoring but kept for robustness.
+            error_msg = "Failed to add item to the to-do list for an unknown reason."
+            logger.error(f"Agent '{agent_id}': {error_msg}")
+            return f"Error: {error_msg}"

autobyteus/task_management/tools/todo_tools/create_todo_list.py

@@ -0,0 +1,79 @@
+# file: autobyteus/autobyteus/task_management/tools/todo_tools/create_todo_list.py
+import json
+import logging
+from typing import TYPE_CHECKING, Any
+
+from pydantic import ValidationError
+
+from autobyteus.tools.base_tool import BaseTool
+from autobyteus.tools.tool_category import ToolCategory
+from autobyteus.tools.pydantic_schema_converter import pydantic_to_parameter_schema
+from autobyteus.task_management.schemas.todo_definition import ToDosDefinitionSchema
+from autobyteus.task_management.todo_list import ToDoList
+
+if TYPE_CHECKING:
+    from autobyteus.agent.context import AgentContext
+
+logger = logging.getLogger(__name__)
+
+def _notify_todo_update(context: 'AgentContext'):
+    if context.phase_manager and context.phase_manager.notifier:
+        todo_list = context.state.todo_list
+        if todo_list:
+            todos_for_llm = [todo.model_dump(mode='json') for todo in todo_list.get_all_todos()]
+            context.phase_manager.notifier.notify_agent_data_todo_list_updated(todos_for_llm)
+            logger.debug(f"Agent '{context.agent_id}': Notified ToDo list update with {len(todos_for_llm)} items.")
+
+class CreateToDoList(BaseTool):
+    """A tool for an agent to create or overwrite its own personal to-do list."""
+
+    CATEGORY = ToolCategory.TASK_MANAGEMENT
+
+    @classmethod
+    def get_name(cls) -> str:
+        return "create_todo_list"
+
+    @classmethod
+    def get_description(cls) -> str:
+        return (
+            "Creates a new personal to-do list for you to manage your own sub-tasks. "
+            "This will overwrite any existing to-do list you have. Use this to break down a larger task into smaller steps. "
+            "Returns the full list of created to-do items with their new IDs."
+        )
+
+    @classmethod
+    def get_argument_schema(cls) -> Any:
+        return pydantic_to_parameter_schema(ToDosDefinitionSchema)
+
+    async def _execute(self, context: 'AgentContext', **kwargs: Any) -> str:
+        agent_id = context.agent_id
+        logger.info(f"Agent '{agent_id}' is executing create_todo_list.")
+
+        try:
+            todos_def_schema = ToDosDefinitionSchema(**kwargs)
+        except ValidationError as e:
+            error_msg = f"Invalid to-do list definition provided: {e}"
+            logger.warning(f"Agent '{agent_id}' provided an invalid definition for create_todo_list: {error_msg}")
+            return f"Error: {error_msg}"
+
+        # Create a new ToDoList, which overwrites any existing one.
+        todo_list = ToDoList(agent_id=agent_id)
+        # Add items from definitions; this now returns the created ToDo objects.
+        new_todos = todo_list.add_todos(todos_def_schema.todos)
+
+        # Set the new list on the agent's state.
+        context.state.todo_list = todo_list
+
+        # Notify any UI components about the update.
+        _notify_todo_update(context)
+
+        # Return the created list to the LLM so it knows the new IDs.
+        try:
+            todos_for_llm = [todo.model_dump(mode='json') for todo in new_todos]
+            logger.info(f"Agent '{agent_id}' successfully created a new to-do list with {len(new_todos)} items.")
+            return json.dumps(todos_for_llm, indent=2)
+        except Exception as e:
+            error_msg = f"An unexpected error occurred while formatting the new to-do list: {e}"
+            logger.error(f"Agent '{agent_id}': {error_msg}", exc_info=True)
+            # Fallback message
+            return f"Successfully created {len(new_todos)} to-do items, but failed to return them in the response."

autobyteus/task_management/tools/todo_tools/get_todo_list.py

@@ -0,0 +1,55 @@
+# file: autobyteus/autobyteus/task_management/tools/todo_tools/get_todo_list.py
+import logging
+import json
+from typing import TYPE_CHECKING, Any
+
+from autobyteus.tools.base_tool import BaseTool
+from autobyteus.tools.tool_category import ToolCategory
+
+if TYPE_CHECKING:
+    from autobyteus.agent.context import AgentContext
+
+logger = logging.getLogger(__name__)
+
+class GetToDoList(BaseTool):
+    """A tool for an agent to retrieve its own personal to-do list."""
+
+    CATEGORY = ToolCategory.TASK_MANAGEMENT
+
+    @classmethod
+    def get_name(cls) -> str:
+        return "get_todo_list"
+
+    @classmethod
+    def get_description(cls) -> str:
+        return (
+            "Retrieves your current personal to-do list. "
+            "Use this to see your plan, check the status of your steps, and decide what to do next."
+        )
+
+    @classmethod
+    def get_argument_schema(cls) -> Any:
+        return None
+
+    async def _execute(self, context: 'AgentContext') -> str:
+        agent_id = context.agent_id
+        logger.info(f"Agent '{agent_id}' is executing get_todo_list.")
+
+        if context.state.todo_list is None or not context.state.todo_list.get_all_todos():
+            return "Your to-do list is empty."
+        
+        todo_list = context.state.todo_list
+        
+        try:
+            # Convert the internal ToDo objects to a JSON-friendly list of dicts
+            todos_for_llm = [
+                todo.model_dump(mode='json') for todo in todo_list.get_all_todos()
+            ]
+            
+            logger.info(f"Agent '{agent_id}' retrieved {len(todos_for_llm)} items from their to-do list.")
+            return json.dumps(todos_for_llm, indent=2)
+            
+        except Exception as e:
+            error_msg = f"An unexpected error occurred while formatting your to-do list: {e}"
+            logger.error(f"Agent '{agent_id}': {error_msg}", exc_info=True)
+            return f"Error: {error_msg}"

autobyteus/task_management/tools/todo_tools/update_todo_status.py

@@ -0,0 +1,85 @@
+# file: autobyteus/autobyteus/task_management/tools/todo_tools/update_todo_status.py
+import logging
+from typing import TYPE_CHECKING, Any
+
+from autobyteus.tools.base_tool import BaseTool
+from autobyteus.tools.tool_category import ToolCategory
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.task_management.todo import ToDoStatus
+
+if TYPE_CHECKING:
+    from autobyteus.agent.context import AgentContext
+
+logger = logging.getLogger(__name__)
+
+def _notify_todo_update(context: 'AgentContext'):
+    if context.phase_manager and context.phase_manager.notifier:
+        todo_list = context.state.todo_list
+        if todo_list:
+            todos_for_llm = [todo.model_dump(mode='json') for todo in todo_list.get_all_todos()]
+            context.phase_manager.notifier.notify_agent_data_todo_list_updated(todos_for_llm)
+            logger.debug(f"Agent '{context.agent_id}': Notified ToDo list update with {len(todos_for_llm)} items.")
+
+class UpdateToDoStatus(BaseTool):
+    """A tool for an agent to update the status of an item on its personal to-do list."""
+
+    CATEGORY = ToolCategory.TASK_MANAGEMENT
+
+    @classmethod
+    def get_name(cls) -> str:
+        return "update_todo_status"
+
+    @classmethod
+    def get_description(cls) -> str:
+        return "Updates the status of a specific item on your personal to-do list."
+
+    @classmethod
+    def get_argument_schema(cls) -> Any:
+        schema = ParameterSchema()
+        schema.add_parameter(ParameterDefinition(
+            name="todo_id",
+            param_type=ParameterType.STRING,
+            description="The unique ID of the to-do item to update (e.g., 'todo_...').",
+            required=True
+        ))
+        schema.add_parameter(ParameterDefinition(
+            name="status",
+            param_type=ParameterType.ENUM,
+            description=f"The new status. Must be one of: {', '.join([s.value for s in ToDoStatus])}.",
+            required=True,
+            enum_values=[s.value for s in ToDoStatus]
+        ))
+        return schema
+
+    async def _execute(self, context: 'AgentContext', todo_id: str, status: str) -> str:
+        agent_id = context.agent_id
+        logger.info(f"Agent '{agent_id}' is executing update_todo_status for item '{todo_id}' to status '{status}'.")
+
+        if context.state.todo_list is None:
+            return "Error: You do not have a to-do list to update."
+            
+        todo_list = context.state.todo_list
+        
+        try:
+            status_enum = ToDoStatus(status)
+        except ValueError:
+            error_msg = f"Invalid status '{status}'. Must be one of: {', '.join([s.value for s in ToDoStatus])}."
+            logger.warning(f"Agent '{agent_id}' provided invalid status for update_todo_status: {status}")
+            return f"Error: {error_msg}"
+
+        if not todo_list.get_todo_by_id(todo_id):
+            error_msg = f"Failed to update status. A to-do item with ID '{todo_id}' does not exist on your list."
+            logger.warning(f"Agent '{agent_id}' failed to update status for non-existent to-do item '{todo_id}'.")
+            return f"Error: {error_msg}"
+
+        if todo_list.update_todo_status(todo_id, status_enum):
+            # Notify about the update
+            _notify_todo_update(context)
+            
+            success_msg = f"Successfully updated status of to-do item '{todo_id}' to '{status}'."
+            logger.info(f"Agent '{agent_id}': {success_msg}")
+            return success_msg
+        else:
+            error_msg = f"Failed to update status for item '{todo_id}'. An unexpected error occurred."
+            logger.error(f"Agent '{agent_id}': {error_msg}")
+            return f"Error: {error_msg}"

autobyteus/tools/__init__.py

@@ -21,16 +21,18 @@ logger = logging.getLogger(__name__)
 
 # Functional tools (decorated functions are now instances)
 from .bash.bash_executor import bash_executor
-from .file.file_reader import file_reader
-from .file.file_writer import file_writer
-from .file.file_editor import file_edit
+from .file.read_file import read_file
+from .file.write_file import write_file
+from .file.edit_file import edit_file
+from .file.search_files import search_files
+from .file.list_directory import list_directory
 
 # General Class-based tools
 try:
-    from .google_search import GoogleSearch
+    from .search_tool import Search
 except ModuleNotFoundError as import_err:
-    logger.warning("GoogleSearch tool not available: %s", import_err)
-    GoogleSearch = None
+    logger.warning("Search tool not available: %s", import_err)
+    Search = None
 from .timer import Timer
 try:
     from .multimedia.image_tools import GenerateImageTool, EditImageTool
@@ -44,7 +46,7 @@ except ModuleNotFoundError as import_err:
     logger.warning("Media reader tool not available: %s", import_err)
     ReadMediaFile = None
 try:
-    from .download_media_tool import DownloadMediaTool
+    from autobyteus.multimedia.download_media_tool import DownloadMediaTool
 except ModuleNotFoundError as import_err:
     logger.warning("Download media tool not available: %s", import_err)
     DownloadMediaTool = None
@@ -91,12 +93,14 @@ __all__ = [
 
     # Re-exported functional tool instances
     "bash_executor",
-    "file_reader",
-    "file_writer",
-    "file_edit",
+    "read_file",
+    "write_file",
+    "edit_file",
+    "search_files",
+    "list_directory",
 
     # Re-exported general class-based tools
-    "GoogleSearch",
+    "Search",
     "Timer",
     "GenerateImageTool",
     "EditImageTool",

autobyteus/tools/bash/bash_executor.py

@@ -13,7 +13,7 @@ if TYPE_CHECKING:
 
 logger = logging.getLogger(__name__)
 
-@tool(name="BashExecutor", category=ToolCategory.SYSTEM)
+@tool(name="execute_bash", category=ToolCategory.SYSTEM)
 async def bash_executor(context: Optional['AgentContext'], command: str) -> str:
     """
     Executes bash commands using the '/bin/bash' interpreter.
@@ -25,7 +25,7 @@ async def bash_executor(context: Optional['AgentContext'], command: str) -> str:
     The command is executed in the agent's workspace directory if available.
     """
     if not shutil.which("bash"):
-        error_msg = "'bash' executable not found in system PATH. The BashExecutor tool cannot be used."
+        error_msg = "'bash' executable not found in system PATH. The execute_bash tool cannot be used."
         logger.error(error_msg)
         raise FileNotFoundError(error_msg)
         
@@ -51,7 +51,7 @@ async def bash_executor(context: Optional['AgentContext'], command: str) -> str:
         effective_cwd = tempfile.gettempdir()
         log_cwd_source = f"system temporary directory: {effective_cwd}"
 
-    logger.debug(f"Functional BashExecutor tool executing for '{agent_id_str}': {command} in cwd from {log_cwd_source}")
+    logger.debug(f"Functional execute_bash tool executing for '{agent_id_str}': {command} in cwd from {log_cwd_source}")
 
     try:
         # Explicitly use 'bash -c' for reliable execution

autobyteus/tools/browser/session_aware/browser_session_aware_navigate_to.py

@@ -21,11 +21,11 @@ class BrowserSessionAwareNavigateTo(BrowserSessionAwareTool):
 
     def __init__(self, config: Optional[ToolConfig] = None):
         super().__init__(config=config)
-        logger.debug("BrowserSessionAwareNavigateTo tool initialized.")
+        logger.debug("navigate_to tool (session-aware) initialized.")
 
     @classmethod
     def get_name(cls) -> str:
-        return "NavigateTo"
+        return "navigate_to"
 
     @classmethod
     def get_description(cls) -> str:
@@ -44,7 +44,7 @@ class BrowserSessionAwareNavigateTo(BrowserSessionAwareTool):
         return schema
     
     async def perform_action(self, shared_session: SharedBrowserSession, webpage_url: str) -> str:
-        logger.info(f"BrowserSessionAwareNavigateTo performing action for URL: {webpage_url}")
+        logger.info(f"navigate_to (session-aware) performing action for URL: {webpage_url}")
 
         if not self._is_valid_url(webpage_url):
             raise ValueError(f"Invalid URL format: {webpage_url}. Must include scheme and netloc.")
@@ -53,12 +53,12 @@ class BrowserSessionAwareNavigateTo(BrowserSessionAwareTool):
             response = await shared_session.page.goto(webpage_url, wait_until="networkidle", timeout=60000)
             
             if response and response.ok:
-                success_msg = f"The NavigateTo command to {webpage_url} is executed successfully."
+                success_msg = f"The navigate_to command to {webpage_url} is executed successfully."
                 logger.info(success_msg)
                 return success_msg
             else:
                 status = response.status if response else "Unknown"
-                failure_msg = f"The NavigationTo command to {webpage_url} failed with status {status}."
+                failure_msg = f"The navigate_to command to {webpage_url} failed with status {status}."
                 logger.warning(failure_msg)
                 return failure_msg
         except Exception as e:

autobyteus/tools/browser/session_aware/browser_session_aware_web_element_trigger.py

@@ -24,11 +24,11 @@ class BrowserSessionAwareWebElementTrigger(BrowserSessionAwareTool):
     
     def __init__(self, config: Optional[ToolConfig] = None):
         super().__init__(config=config)
-        logger.debug("BrowserSessionAwareWebElementTrigger tool initialized.")
+        logger.debug("trigger_web_element (session-aware) tool initialized.")
 
     @classmethod
     def get_name(cls) -> str:
-        return "WebElementTrigger"
+        return "trigger_web_element"
 
     @classmethod
     def get_description(cls) -> str:
@@ -75,7 +75,7 @@ class BrowserSessionAwareWebElementTrigger(BrowserSessionAwareTool):
         webpage_url: str,
         params: Optional[str] = ""
     ) -> str:
-        logger.info(f"WebElementTrigger performing action '{action}' on selector '{css_selector}' for page related to URL '{webpage_url}'. Params: '{params[:50]}...'")
+        logger.info(f"trigger_web_element performing action '{action}' on selector '{css_selector}' for page related to URL '{webpage_url}'. Params: '{params[:50]}...'")
 
         try:
             action_enum = WebElementAction.from_string(action)
@@ -124,7 +124,7 @@ class BrowserSessionAwareWebElementTrigger(BrowserSessionAwareTool):
         else:
             raise ValueError(f"Unsupported action: {action_enum}")
 
-        success_msg = f"The WebElementTrigger action '{action_enum}' on selector '{css_selector}' was executed."
+        success_msg = f"The trigger_web_element action '{action_enum}' on selector '{css_selector}' was executed."
         logger.info(success_msg)
         return success_msg
 

autobyteus/tools/browser/session_aware/browser_session_aware_webpage_reader.py

@@ -37,11 +37,11 @@ class BrowserSessionAwareWebPageReader(BrowserSessionAwareTool):
                     cleaning_mode_to_use = cleaning_mode_value
         
         self.cleaning_mode = cleaning_mode_to_use
-        logger.debug(f"BrowserSessionAwareWebPageReader initialized with cleaning_mode: {self.cleaning_mode}")
+        logger.debug(f"read_webpage (session-aware) tool initialized with cleaning_mode: {self.cleaning_mode}")
 
     @classmethod
     def get_name(cls) -> str: 
-        return "WebPageReader"
+        return "read_webpage"
 
     @classmethod
     def get_description(cls) -> str:
@@ -77,7 +77,7 @@ class BrowserSessionAwareWebPageReader(BrowserSessionAwareTool):
         shared_session: SharedBrowserSession,
         webpage_url: str 
     ) -> str: 
-        logger.info(f"BrowserSessionAwareWebPageReader performing action. Current page URL: {shared_session.page.url}, cleaning_mode: {self.cleaning_mode}")
+        logger.info(f"read_webpage (session-aware) performing action. Current page URL: {shared_session.page.url}, cleaning_mode: {self.cleaning_mode}")
         
         try:
             page_content = await shared_session.page.content()

autobyteus/tools/browser/session_aware/browser_session_aware_webpage_screenshot_taker.py

@@ -32,11 +32,11 @@ class BrowserSessionAwareWebPageScreenshotTaker(BrowserSessionAwareTool):
             if self.image_format not in ["png", "jpeg"]:
                 logger.warning(f"Invalid image_format '{self.image_format}' in config. Defaulting to 'png'.")
                 self.image_format = "png"
-        logger.debug(f"BrowserSessionAwareWebPageScreenshotTaker initialized. Full page: {self.full_page}, Format: {self.image_format}")
+        logger.debug(f"take_webpage_screenshot (session-aware) initialized. Full page: {self.full_page}, Format: {self.image_format}")
 
     @classmethod
     def get_name(cls) -> str: 
-        return "WebPageScreenshotTaker" 
+        return "take_webpage_screenshot" 
 
     @classmethod
     def get_description(cls) -> str:
@@ -87,7 +87,7 @@ class BrowserSessionAwareWebPageScreenshotTaker(BrowserSessionAwareTool):
         file_name: str, 
         webpage_url: str 
     ) -> str: 
-        logger.info(f"BrowserSessionAwareWebPageScreenshotTaker performing action. Saving to '{file_name}'. Current page: {shared_session.page.url}")
+        logger.info(f"take_webpage_screenshot (session-aware) performing action. Saving to '{file_name}'. Current page: {shared_session.page.url}")
 
         output_dir = os.path.dirname(file_name)
         if output_dir: