autobyteus 1.0.6__py3-none-any.whl → 1.1.1__py3-none-any.whl

autobyteus/agent/handlers/tool_invocation_request_event_handler.py

@@ -0,0 +1,211 @@
+# file: autobyteus/autobyteus/agent/handlers/tool_invocation_request_event_handler.py
+import logging
+import json 
+import traceback 
+from typing import TYPE_CHECKING, Optional 
+
+from autobyteus.agent.handlers.base_event_handler import AgentEventHandler
+from autobyteus.agent.events import PendingToolInvocationEvent, ToolResultEvent 
+from autobyteus.agent.tool_invocation import ToolInvocation
+
+if TYPE_CHECKING:
+    from autobyteus.agent.context import AgentContext 
+    from autobyteus.agent.events.notifiers import AgentExternalEventNotifier 
+
+logger = logging.getLogger(__name__)
+
+class ToolInvocationRequestEventHandler(AgentEventHandler):
+    """
+    Handles PendingToolInvocationEvents.
+    If 'auto_execute_tools' (from AgentConfig) is False, it stores the invocation,
+    updates history, and emits an AGENT_REQUEST_TOOL_INVOCATION_APPROVAL event via the notifier.
+    If 'auto_execute_tools' is True, it executes the tool directly, emits
+    AGENT_DATA_TOOL_LOG events for call and result/error, 
+    and queues a ToolResultEvent.
+    """
+    def __init__(self): # pragma: no cover
+        logger.info("ToolInvocationRequestEventHandler initialized.")
+
+    async def _execute_tool_directly(self, 
+                                     tool_invocation: ToolInvocation, 
+                                     context: 'AgentContext',
+                                     notifier: Optional['AgentExternalEventNotifier']) -> None: # pragma: no cover
+        agent_id = context.agent_id 
+        tool_name = tool_invocation.name
+        arguments = tool_invocation.arguments
+        invocation_id = tool_invocation.id
+
+        if notifier:
+            try:
+                auto_exec_data = {
+                    "invocation_id": invocation_id,
+                    "tool_name": tool_name,
+                    "arguments": arguments,
+                }
+                notifier.notify_agent_tool_invocation_auto_executing(auto_exec_data)
+            except Exception as e_notify:
+                logger.error(f"Agent '{agent_id}': Error notifying tool auto-execution: {e_notify}", exc_info=True)
+        
+        logger.info(f"Agent '{agent_id}' executing tool directly: '{tool_name}' (ID: {invocation_id}) with args: {arguments}")
+        
+        try:
+            args_str = json.dumps(arguments)
+        except TypeError:
+            args_str = str(arguments) 
+
+        log_msg_call = f"[TOOL_CALL_DIRECT] Agent_ID: {agent_id}, Tool: {tool_name}, Invocation_ID: {invocation_id}, Arguments: {args_str}"
+        if notifier:
+            try:
+                log_data = {
+                    "log_entry": log_msg_call,
+                    "tool_invocation_id": invocation_id,
+                    "tool_name": tool_name,
+                }
+                notifier.notify_agent_data_tool_log(log_data)
+            except Exception as e_notify: 
+                 logger.error(f"Agent '{agent_id}': Error notifying tool call log: {e_notify}", exc_info=True)
+
+        tool_instance = context.get_tool(tool_name) 
+        result_event: ToolResultEvent
+
+        if not tool_instance:
+            error_message = f"Tool '{tool_name}' not found or configured for agent '{agent_id}'."
+            logger.error(error_message)
+            result_event = ToolResultEvent(tool_name=tool_name, result=None, error=error_message, tool_invocation_id=invocation_id)
+            context.add_message_to_history({ 
+                "role": "tool",
+                "tool_call_id": invocation_id,
+                "name": tool_name,
+                "content": f"Error: Tool '{tool_name}' execution failed. Reason: {error_message}",
+            })
+            log_msg_error = f"[TOOL_ERROR_DIRECT] {error_message}"
+            if notifier:
+                try:
+                    # Log entry
+                    log_data = { "log_entry": log_msg_error, "tool_invocation_id": invocation_id, "tool_name": tool_name, }
+                    notifier.notify_agent_data_tool_log(log_data)
+                    # Generic output error
+                    notifier.notify_agent_error_output_generation(
+                        error_source=f"ToolExecutionDirect.ToolNotFound.{tool_name}",
+                        error_message=error_message
+                    )
+                except Exception as e_notify: 
+                    logger.error(f"Agent '{agent_id}': Error notifying tool error log/output error: {e_notify}", exc_info=True)
+        else:
+            try:
+                logger.debug(f"Executing tool '{tool_name}' for agent '{agent_id}'. Invocation ID: {invocation_id}")
+                execution_result = await tool_instance.execute(context=context, **arguments) 
+                
+                try:
+                    result_json_for_log = json.dumps(execution_result)
+                except (TypeError, ValueError): 
+                    result_json_for_log = json.dumps(str(execution_result))
+
+                logger.info(f"Tool '{tool_name}' (ID: {invocation_id}) executed by agent '{agent_id}'.")
+                result_event = ToolResultEvent(tool_name=tool_name, result=execution_result, error=None, tool_invocation_id=invocation_id)
+                
+                history_content = str(execution_result) 
+                context.add_message_to_history({ 
+                    "role": "tool",
+                    "tool_call_id": invocation_id,
+                    "name": tool_name,
+                    "content": history_content,
+                })
+                log_msg_result = f"[TOOL_RESULT_DIRECT] {result_json_for_log}"
+                if notifier:
+                    try:
+                        # Log entry with embedded JSON result
+                        log_data = { "log_entry": log_msg_result, "tool_invocation_id": invocation_id, "tool_name": tool_name }
+                        notifier.notify_agent_data_tool_log(log_data)
+                    except Exception as e_notify: 
+                        logger.error(f"Agent '{agent_id}': Error notifying tool result log: {e_notify}", exc_info=True)
+
+            except Exception as e: 
+                error_message = f"Error executing tool '{tool_name}' (ID: {invocation_id}): {str(e)}"
+                error_details = traceback.format_exc()
+                logger.error(f"Agent '{agent_id}' {error_message}", exc_info=True)
+                result_event = ToolResultEvent(tool_name=tool_name, result=None, error=error_message, tool_invocation_id=invocation_id)
+                context.add_message_to_history({ 
+                    "role": "tool",
+                    "tool_call_id": invocation_id,
+                    "name": tool_name,
+                    "content": f"Error: Tool '{tool_name}' execution failed. Reason: {error_message}",
+                })
+                log_msg_exception = f"[TOOL_EXCEPTION_DIRECT] {error_message}\nDetails:\n{error_details}"
+                if notifier:
+                    try:
+                        # Log entry
+                        log_data = { "log_entry": log_msg_exception, "tool_invocation_id": invocation_id, "tool_name": tool_name }
+                        notifier.notify_agent_data_tool_log(log_data)
+                        # Generic output error
+                        notifier.notify_agent_error_output_generation(
+                            error_source=f"ToolExecutionDirect.Exception.{tool_name}",
+                            error_message=error_message,
+                            error_details=error_details
+                        )
+                    except Exception as e_notify: 
+                        logger.error(f"Agent '{agent_id}': Error notifying tool exception log/output error: {e_notify}", exc_info=True)
+        
+        await context.input_event_queues.enqueue_tool_result(result_event) 
+        logger.debug(f"Agent '{agent_id}' enqueued ToolResultEvent (direct exec) for '{tool_name}' (ID: {invocation_id}).")
+
+
+    async def handle(self,
+                     event: PendingToolInvocationEvent, 
+                     context: 'AgentContext') -> None: # pragma: no cover
+        if not isinstance(event, PendingToolInvocationEvent): 
+            logger.warning(f"ToolInvocationRequestEventHandler received non-PendingToolInvocationEvent: {type(event)}. Skipping.")
+            return
+
+        tool_invocation: ToolInvocation = event.tool_invocation
+        agent_id = context.agent_id 
+        
+        notifier: Optional['AgentExternalEventNotifier'] = None
+        if context.phase_manager:
+            notifier = context.phase_manager.notifier
+        
+        if not notifier:
+            logger.error(f"Agent '{agent_id}': Notifier not available in ToolInvocationRequestEventHandler. Output events for tool approval/logging will be lost.")
+            if not context.auto_execute_tools:
+                logger.critical(f"Agent '{agent_id}': Notifier is REQUIRED for manual tool approval flow but is unavailable. Tool '{tool_invocation.name}' cannot be processed for approval.")
+                return
+
+        if not context.auto_execute_tools:
+            logger.info(f"Agent '{agent_id}': Tool '{tool_invocation.name}' (ID: {tool_invocation.id}) requires approval. Storing pending invocation and emitting request.")
+            
+            context.store_pending_tool_invocation(tool_invocation) 
+
+            try:
+                arguments_json_str = json.dumps(tool_invocation.arguments or {})
+            except TypeError: 
+                logger.warning(f"Could not serialize args for history tool_call for '{tool_invocation.name}'. Using empty dict string.")
+                arguments_json_str = "{}"
+
+            context.add_message_to_history({ 
+                "role": "assistant",
+                "content": None, 
+                "tool_calls": [{
+                    "id": tool_invocation.id,
+                    "type": "function", 
+                    "function": {
+                        "name": tool_invocation.name,
+                        "arguments": arguments_json_str 
+                    }
+                }]
+            })
+            
+            approval_data = {
+                "invocation_id": tool_invocation.id,
+                "tool_name": tool_invocation.name,
+                "arguments": tool_invocation.arguments,
+            }
+            if notifier:
+                try:
+                    notifier.notify_agent_request_tool_invocation_approval(approval_data)
+                    logger.debug(f"Agent '{agent_id}': Emitted AGENT_REQUEST_TOOL_INVOCATION_APPROVAL for '{tool_invocation.name}' (ID: {tool_invocation.id}).")
+                except Exception as e_notify: 
+                    logger.error(f"Agent '{agent_id}': Error emitting AGENT_REQUEST_TOOL_INVOCATION_APPROVAL: {e_notify}", exc_info=True)
+            
+        else: 
+            logger.info(f"Agent '{agent_id}': Tool '{tool_invocation.name}' (ID: {tool_invocation.id}) executing automatically (auto_execute_tools=True).")
+            await self._execute_tool_directly(tool_invocation, context, notifier)

autobyteus/agent/handlers/tool_result_event_handler.py

@@ -0,0 +1,101 @@
+# file: autobyteus/autobyteus/agent/handlers/tool_result_event_handler.py
+import logging
+import json 
+from typing import TYPE_CHECKING, Optional 
+
+from autobyteus.agent.handlers.base_event_handler import AgentEventHandler 
+from autobyteus.agent.events import ToolResultEvent, LLMUserMessageReadyEvent 
+from autobyteus.llm.user_message import LLMUserMessage 
+
+if TYPE_CHECKING:
+    from autobyteus.agent.context import AgentContext 
+    from autobyteus.agent.events.notifiers import AgentExternalEventNotifier 
+
+logger = logging.getLogger(__name__)
+
+class ToolResultEventHandler(AgentEventHandler):
+    """
+    Handles ToolResultEvents by formatting the tool's output (or error)
+    as a new LLMUserMessage, emitting AGENT_DATA_TOOL_LOG event for this outcome,
+    and enqueuing an LLMUserMessageReadyEvent for further LLM processing.
+    """
+    def __init__(self):
+        logger.info("ToolResultEventHandler initialized.")
+
+    async def handle(self,
+                     event: ToolResultEvent,
+                     context: 'AgentContext') -> None:
+        if not isinstance(event, ToolResultEvent): 
+            logger.warning(f"ToolResultEventHandler received non-ToolResultEvent: {type(event)}. Skipping.")
+            return
+
+        agent_id = context.agent_id 
+        tool_invocation_id = event.tool_invocation_id if event.tool_invocation_id else 'N/A'
+
+        logger.info(f"Agent '{agent_id}' handling ToolResultEvent from tool: '{event.tool_name}' (Invocation ID: {tool_invocation_id}). Error: {event.error is not None}")
+        
+        notifier: Optional['AgentExternalEventNotifier'] = None
+        if context.phase_manager:
+            notifier = context.phase_manager.notifier
+        
+        if not notifier: # pragma: no cover
+            logger.error(f"Agent '{agent_id}': Notifier not available in ToolResultEventHandler. Tool result processing logs will not be emitted.")
+
+        if event.error:
+            logger.debug(f"Agent '{agent_id}' tool '{event.tool_name}' (ID: {tool_invocation_id}) raw error details: {event.error}")
+        else:
+            try:
+                raw_result_str_for_debug_log = json.dumps(event.result, indent=2)
+            except TypeError: # pragma: no cover
+                raw_result_str_for_debug_log = str(event.result)
+            logger.debug(f"Agent '{agent_id}' tool '{event.tool_name}' (ID: {tool_invocation_id}) raw result:\n---\n{raw_result_str_for_debug_log}\n---")
+
+
+        content_for_llm: str
+        if event.error:
+            content_for_llm = (
+                f"The tool '{event.tool_name}' (invocation ID: {tool_invocation_id}) encountered an error.\n"
+                f"Error details: {event.error}\n"
+                f"Please analyze this error and decide the next course of action."
+            )
+            log_msg_error_processed = f"[TOOL_RESULT_ERROR_PROCESSED] Agent_ID: {agent_id}, Tool: {event.tool_name}, Invocation_ID: {tool_invocation_id}, Error: {event.error}"
+            if notifier:
+                try:
+                    log_data = {
+                        "log_entry": log_msg_error_processed,
+                        "tool_invocation_id": tool_invocation_id,
+                        "tool_name": event.tool_name,
+                    }
+                    notifier.notify_agent_data_tool_log(log_data)
+                except Exception as e_notify: 
+                    logger.error(f"Agent '{agent_id}': Error notifying tool result error log: {e_notify}", exc_info=True)
+        else:
+            try:
+                result_str_for_llm = json.dumps(event.result, indent=2) if not isinstance(event.result, str) else event.result
+            except TypeError: # pragma: no cover
+                result_str_for_llm = str(event.result)
+
+            content_for_llm = (
+                f"The tool '{event.tool_name}' (invocation ID: {tool_invocation_id}) has executed.\n"
+                f"Result:\n{result_str_for_llm}\n" 
+                f"Based on this result, what is the next step or final answer?"
+            )
+            log_msg_success_processed = f"[TOOL_RESULT_SUCCESS_PROCESSED] Agent_ID: {agent_id}, Tool: {event.tool_name}, Invocation_ID: {tool_invocation_id}, Result (first 200 chars of stringified): {str(event.result)[:200]}"
+            if notifier:
+                try:
+                    log_data = {
+                        "log_entry": log_msg_success_processed,
+                        "tool_invocation_id": tool_invocation_id,
+                        "tool_name": event.tool_name,
+                    }
+                    notifier.notify_agent_data_tool_log(log_data)
+                except Exception as e_notify: 
+                    logger.error(f"Agent '{agent_id}': Error notifying tool result success log: {e_notify}", exc_info=True)
+        
+        logger.debug(f"Agent '{agent_id}' preparing message for LLM based on tool '{event.tool_name}' (ID: {tool_invocation_id}) result:\n---\n{content_for_llm}\n---")
+        llm_user_message = LLMUserMessage(content=content_for_llm)
+        
+        next_event = LLMUserMessageReadyEvent(llm_user_message=llm_user_message) 
+        await context.input_event_queues.enqueue_internal_system_event(next_event)
+        
+        logger.info(f"Agent '{agent_id}' enqueued LLMUserMessageReadyEvent for LLM based on tool '{event.tool_name}' (ID: {tool_invocation_id}) result summary.") 

autobyteus/agent/handlers/user_input_message_event_handler.py

@@ -0,0 +1,77 @@
+# file: autobyteus/autobyteus/agent/handlers/user_input_message_event_handler.py
+import logging
+from typing import TYPE_CHECKING
+
+from autobyteus.agent.handlers.base_event_handler import AgentEventHandler
+from autobyteus.agent.events import UserMessageReceivedEvent, LLMUserMessageReadyEvent 
+from autobyteus.agent.message.agent_input_user_message import AgentInputUserMessage 
+from autobyteus.agent.input_processor import BaseAgentUserInputMessageProcessor
+from autobyteus.llm.user_message import LLMUserMessage
+
+
+if TYPE_CHECKING:
+    from autobyteus.agent.context import AgentContext 
+
+logger = logging.getLogger(__name__)
+
+class UserInputMessageEventHandler(AgentEventHandler):
+    """
+    Handles UserMessageReceivedEvents by first applying any configured
+    AgentUserInputMessageProcessors (provided as instances) to the AgentInputUserMessage,
+    then converting the processed message into an LLMUserMessage, and finally
+    enqueuing an LLMUserMessageReadyEvent for further processing by the LLM.
+    """
+
+    def __init__(self):
+        logger.info("UserInputMessageEventHandler initialized.")
+
+    async def handle(self,
+                     event: UserMessageReceivedEvent, 
+                     context: 'AgentContext') -> None:
+        if not isinstance(event, UserMessageReceivedEvent): 
+            logger.warning(f"UserInputMessageEventHandler received non-UserMessageReceivedEvent: {type(event)}. Skipping.")
+            return
+
+        original_agent_input_user_msg: AgentInputUserMessage = event.agent_input_user_message 
+        processed_agent_input_user_msg: AgentInputUserMessage = original_agent_input_user_msg 
+        
+        logger.info(f"Agent '{context.agent_id}' handling UserMessageReceivedEvent: '{original_agent_input_user_msg.content[:100]}...'") 
+        
+        processor_instances = context.config.input_processors
+        if processor_instances:
+            processor_names = [p.get_name() for p in processor_instances]
+            logger.debug(f"Agent '{context.agent_id}': Applying input processors: {processor_names}")
+            for processor_instance in processor_instances:
+                processor_name_for_log = "unknown"
+                try:
+                    if not isinstance(processor_instance, BaseAgentUserInputMessageProcessor):
+                        logger.error(f"Agent '{context.agent_id}': Invalid input processor type in config: {type(processor_instance)}. Skipping.")
+                        continue
+                    
+                    processor_name_for_log = processor_instance.get_name()
+                    logger.debug(f"Agent '{context.agent_id}': Applying input processor '{processor_name_for_log}'.")
+                    msg_before_this_processor = processed_agent_input_user_msg
+                    # Pass the original event to the processor
+                    processed_agent_input_user_msg = await processor_instance.process(
+                        message=msg_before_this_processor, 
+                        context=context, 
+                        triggering_event=event
+                    )
+                    logger.info(f"Agent '{context.agent_id}': Input processor '{processor_name_for_log}' applied successfully.")
+
+                except Exception as e:
+                    logger.error(f"Agent '{context.agent_id}': Error applying input processor '{processor_name_for_log}': {e}. "
+                                 f"Skipping this processor and continuing with message from before this processor.", exc_info=True)
+                    processed_agent_input_user_msg = msg_before_this_processor
+        else:
+            logger.debug(f"Agent '{context.agent_id}': No input processors configured in agent config.")
+
+        llm_user_message = LLMUserMessage( 
+            content=processed_agent_input_user_msg.content,
+            image_urls=processed_agent_input_user_msg.image_urls 
+        )
+        
+        llm_user_message_ready_event = LLMUserMessageReadyEvent(llm_user_message=llm_user_message) 
+        await context.input_event_queues.enqueue_internal_system_event(llm_user_message_ready_event)
+        
+        logger.info(f"Agent '{context.agent_id}' processed AgentInputUserMessage and enqueued LLMUserMessageReadyEvent.")

autobyteus/agent/hooks/__init__.py

@@ -0,0 +1,16 @@
+# file: autobyteus/autobyteus/agent/hooks/__init__.py
+"""
+Components for defining and running lifecycle hooks based on agent phase transitions.
+"""
+from .base_phase_hook import BasePhaseHook
+from .hook_definition import PhaseHookDefinition
+from .hook_meta import PhaseHookMeta
+from .hook_registry import PhaseHookRegistry, default_phase_hook_registry
+
+__all__ = [
+    "BasePhaseHook",
+    "PhaseHookDefinition",
+    "PhaseHookMeta",
+    "PhaseHookRegistry",
+    "default_phase_hook_registry",
+]

autobyteus/agent/hooks/base_phase_hook.py

@@ -0,0 +1,61 @@
+# file: autobyteus/autobyteus/agent/hooks/base_phase_hook.py
+import logging
+from abc import ABC, abstractmethod
+from typing import TYPE_CHECKING
+
+from autobyteus.agent.phases import AgentOperationalPhase
+from .hook_meta import PhaseHookMeta
+
+if TYPE_CHECKING:
+    from autobyteus.agent.context import AgentContext
+
+logger = logging.getLogger(__name__)
+
+class BasePhaseHook(ABC, metaclass=PhaseHookMeta):
+    """
+    Abstract base class for creating hooks that execute on specific agent
+    phase transitions.
+
+    Subclasses must define the `source_phase` and `target_phase` to specify
+    the exact transition they are interested in, and implement the `execute`
+    method for their custom logic.
+    """
+
+    @classmethod
+    def get_name(cls) -> str:
+        """
+        Returns the unique registration name for this hook.
+        Defaults to the class name. Can be overridden by subclasses.
+        """
+        return cls.__name__
+
+    @property
+    @abstractmethod
+    def source_phase(self) -> AgentOperationalPhase:
+        """The source phase for the transition this hook targets."""
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def target_phase(self) -> AgentOperationalPhase:
+        """The target phase for the transition this hook targets."""
+        raise NotImplementedError
+
+    @abstractmethod
+    async def execute(self, context: 'AgentContext') -> None:
+        """
+        The method executed when the specified phase transition occurs.
+
+        Args:
+            context: The agent's context at the time of the transition.
+        """
+        raise NotImplementedError
+
+    def __repr__(self) -> str:
+        # Use try-except in case properties are not yet implemented during introspection
+        try:
+            return (f"<{self.__class__.__name__} "
+                    f"source='{self.source_phase.value}' "
+                    f"target='{self.target_phase.value}'>")
+        except (NotImplementedError, AttributeError):
+            return f"<{self.__class__.__name__} (unconfigured)>"

autobyteus/agent/hooks/hook_definition.py

@@ -0,0 +1,36 @@
+# file: autobyteus/autobyteus/agent/hooks/hook_definition.py
+import logging
+from typing import Type, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .base_phase_hook import BasePhaseHook
+
+logger = logging.getLogger(__name__)
+
+class PhaseHookDefinition:
+    """
+    Represents the definition of a phase hook.
+    Contains its registered name and the class itself.
+    """
+    def __init__(self, name: str, hook_class: Type['BasePhaseHook']):
+        """
+        Initializes the PhaseHookDefinition.
+
+        Args:
+            name: The unique registered name of the hook.
+            hook_class: The class of the phase hook.
+
+        Raises:
+            ValueError: If name is empty or hook_class is not a type.
+        """
+        if not name or not isinstance(name, str):
+            raise ValueError("Hook name must be a non-empty string.")
+        if not isinstance(hook_class, type):
+            raise ValueError("hook_class must be a class type.")
+        
+        self.name: str = name
+        self.hook_class: Type['BasePhaseHook'] = hook_class
+        logger.debug(f"PhaseHookDefinition created: name='{name}', class='{hook_class.__name__}'.")
+
+    def __repr__(self) -> str:
+        return f"<PhaseHookDefinition name='{self.name}', class='{self.hook_class.__name__}'>"

autobyteus/agent/hooks/hook_meta.py

@@ -0,0 +1,37 @@
+# file: autobyteus/autobyteus/agent/hooks/hook_meta.py
+import logging
+from abc import ABCMeta
+
+from .hook_registry import default_phase_hook_registry
+from .hook_definition import PhaseHookDefinition
+
+logger = logging.getLogger(__name__)
+
+class PhaseHookMeta(ABCMeta):
+    """
+    Metaclass for BasePhaseHook that automatically registers concrete
+    hook subclasses with the default_phase_hook_registry.
+    Registration uses the name obtained from the class method `get_name()`.
+    """
+    def __init__(cls, name, bases, dct):
+        super().__init__(name, bases, dct)
+
+        if name == 'BasePhaseHook' or getattr(cls, "__abstractmethods__", None):
+            logger.debug(f"Skipping registration for abstract phase hook class: {name}")
+            return
+
+        try:
+            hook_name = cls.get_name()
+
+            if not hook_name or not isinstance(hook_name, str):
+                logger.error(f"Phase hook class {name} must return a valid string from static get_name(). Skipping registration.")
+                return
+            
+            definition = PhaseHookDefinition(name=hook_name, hook_class=cls)
+            default_phase_hook_registry.register_hook(definition)
+            logger.info(f"Auto-registered phase hook: '{hook_name}' from class {name} (no schema).")
+
+        except AttributeError as e:
+            logger.error(f"Phase hook class {name} is missing required static/class method 'get_name' ({e}). Skipping registration.")
+        except Exception as e:
+            logger.error(f"Failed to auto-register phase hook class {name}: {e}", exc_info=True)

autobyteus/agent/hooks/hook_registry.py

@@ -0,0 +1,118 @@
+# file: autobyteus/autobyteus/agent/hooks/hook_registry.py
+import logging
+from typing import TYPE_CHECKING, Dict, List, Optional
+
+from autobyteus.utils.singleton import SingletonMeta
+from .hook_definition import PhaseHookDefinition
+
+if TYPE_CHECKING:
+    from .base_phase_hook import BasePhaseHook
+
+logger = logging.getLogger(__name__)
+
+class PhaseHookRegistry(metaclass=SingletonMeta):
+    """
+    A singleton registry for PhaseHookDefinition objects.
+    Hooks are typically auto-registered via PhaseHookMeta.
+    """
+
+    def __init__(self):
+        """Initializes the PhaseHookRegistry with an empty store."""
+        self._definitions: Dict[str, PhaseHookDefinition] = {}
+        logger.info("PhaseHookRegistry initialized.")
+
+    def register_hook(self, definition: PhaseHookDefinition) -> None:
+        """
+        Registers a phase hook definition.
+        If a definition with the same name already exists, it will be overwritten,
+        and a warning will be logged.
+
+        Args:
+            definition: The PhaseHookDefinition object to register.
+
+        Raises:
+            TypeError: If the definition is not an instance of PhaseHookDefinition.
+        """
+        if not isinstance(definition, PhaseHookDefinition):
+            raise TypeError(f"Expected PhaseHookDefinition instance, got {type(definition).__name__}.")
+
+        hook_name = definition.name
+        if hook_name in self._definitions:
+            logger.warning(f"Overwriting existing phase hook definition for name: '{hook_name}'.")
+        
+        self._definitions[hook_name] = definition
+        logger.info(f"Phase hook definition '{hook_name}' (class: '{definition.hook_class.__name__}') registered successfully.")
+
+    def get_hook_definition(self, name: str) -> Optional[PhaseHookDefinition]:
+        """
+        Retrieves a phase hook definition by its name.
+
+        Args:
+            name: The name of the phase hook definition to retrieve.
+
+        Returns:
+            The PhaseHookDefinition object if found, otherwise None.
+        """
+        if not isinstance(name, str):
+            logger.warning(f"Attempted to retrieve hook definition with non-string name: {type(name).__name__}.")
+            return None
+        definition = self._definitions.get(name)
+        if not definition:
+            logger.debug(f"Phase hook definition with name '{name}' not found in registry.")
+        return definition
+
+    def get_hook(self, name: str) -> Optional['BasePhaseHook']:
+        """
+        Retrieves an instance of a phase hook by its name.
+
+        Args:
+            name: The name of the phase hook to retrieve.
+
+        Returns:
+            An instance of the BasePhaseHook if found and instantiable, otherwise None.
+        """
+        definition = self.get_hook_definition(name)
+        if definition:
+            try:
+                return definition.hook_class()
+            except Exception as e:
+                logger.error(f"Failed to instantiate phase hook '{name}' from class '{definition.hook_class.__name__}': {e}", exc_info=True)
+                return None
+        return None
+
+    def list_hook_names(self) -> List[str]:
+        """
+        Returns a list of names of all registered phase hook definitions.
+
+        Returns:
+            A list of strings, where each string is a registered hook name.
+        """
+        return list(self._definitions.keys())
+
+    def get_all_definitions(self) -> Dict[str, PhaseHookDefinition]:
+        """
+        Returns a shallow copy of the dictionary containing all registered phase hook definitions.
+
+        Returns:
+            A dictionary where keys are hook names and values are PhaseHookDefinition objects.
+        """
+        return dict(self._definitions)
+
+    def clear(self) -> None:
+        """Removes all definitions from the registry."""
+        count = len(self._definitions)
+        self._definitions.clear()
+        logger.info(f"Cleared {count} definitions from the PhaseHookRegistry.")
+        
+    def __len__(self) -> int:
+        """Returns the number of registered hook definitions."""
+        return len(self._definitions)
+
+    def __contains__(self, name: str) -> bool:
+        """Checks if a hook definition is in the registry by name."""
+        if isinstance(name, str):
+            return name in self._definitions
+        return False
+
+# Default instance of the registry
+default_phase_hook_registry = PhaseHookRegistry()