b60439152b
- harness/es.py: append-only event store + проекции (образец — кооп-портал) - harness/levels.py: уровни 1с/1м/1ч/1д, контекст ∝ масштабу - harness/scheduler.py: уровневые потоки, идемпотентные тики, absorbed-стабы - harness/isolated_runner.py + _llm_worker.py: LLM-тик в сабпроцессе (глобальный asyncio-рантайм Strands не переживает 2+ create_harness) - cli.py: init/task/tasks/status/events + requeue (написан самим минутным агентом) - 9router: отдельный ключ sk-simbiot, модель auto (glm-5.3 первым)
119 lines
4.3 KiB
Python
119 lines
4.3 KiB
Python
"""MultiAgentPlugin base class for extending multi-agent orchestrator functionality.
|
|
|
|
This module defines the MultiAgentPlugin base class, which provides a composable way to
|
|
add behavior changes to multi-agent orchestrators (Swarm, Graph) through automatic hook
|
|
registration and custom initialization.
|
|
|
|
MultiAgentPlugin is the orchestrator-level counterpart to Plugin (which targets individual agents).
|
|
A class can implement both Plugin and MultiAgentPlugin to provide functionality at both levels.
|
|
"""
|
|
|
|
from abc import ABC, abstractmethod
|
|
from collections.abc import Awaitable
|
|
from typing import TYPE_CHECKING
|
|
|
|
from ..hooks.registry import HookCallback
|
|
from ._discovery import discover_hooks
|
|
|
|
if TYPE_CHECKING:
|
|
from ..multiagent.base import MultiAgentBase
|
|
|
|
|
|
class MultiAgentPlugin(ABC):
|
|
"""Base class for objects that extend multi-agent orchestrator functionality.
|
|
|
|
MultiAgentPlugins provide a composable way to add behavior changes to orchestrators
|
|
(Swarm, Graph). They support automatic discovery and registration of methods decorated
|
|
with @hook.
|
|
|
|
Unlike agent-level Plugin, MultiAgentPlugin does not support @tool decorated methods
|
|
since orchestrators do not have tool registries.
|
|
|
|
Attributes:
|
|
name: A stable string identifier for the plugin (must be provided by subclass)
|
|
hooks: Hooks attached to the orchestrator, auto-discovered from @hook decorated methods
|
|
|
|
Example using decorators (recommended):
|
|
```python
|
|
from strands.plugins import MultiAgentPlugin, hook
|
|
from strands.hooks import BeforeNodeCallEvent, AfterNodeCallEvent
|
|
|
|
class MonitoringPlugin(MultiAgentPlugin):
|
|
name = "monitoring"
|
|
|
|
@hook
|
|
def on_before_node(self, event: BeforeNodeCallEvent):
|
|
print(f"Node {event.node_id} starting")
|
|
|
|
@hook
|
|
def on_after_node(self, event: AfterNodeCallEvent):
|
|
print(f"Node {event.node_id} completed")
|
|
```
|
|
|
|
Example with custom initialization:
|
|
```python
|
|
class MyPlugin(MultiAgentPlugin):
|
|
name = "my-plugin"
|
|
|
|
def init_multi_agent(self, orchestrator: MultiAgentBase) -> None:
|
|
# Custom initialization logic
|
|
pass
|
|
```
|
|
|
|
Dual-use example (both agent and orchestrator):
|
|
```python
|
|
from strands.plugins import Plugin, MultiAgentPlugin, hook
|
|
from strands.hooks import BeforeInvocationEvent, BeforeNodeCallEvent
|
|
|
|
class ObservabilityPlugin(Plugin, MultiAgentPlugin):
|
|
name = "observability"
|
|
|
|
@hook
|
|
def on_agent_invocation(self, event: BeforeInvocationEvent):
|
|
print("Agent invocation started")
|
|
|
|
@hook
|
|
def on_node_call(self, event: BeforeNodeCallEvent):
|
|
print(f"Node {event.node_id} starting")
|
|
|
|
def init_agent(self, agent):
|
|
pass # Agent-level setup
|
|
|
|
def init_multi_agent(self, orchestrator):
|
|
pass # Orchestrator-level setup
|
|
```
|
|
"""
|
|
|
|
@property
|
|
@abstractmethod
|
|
def name(self) -> str:
|
|
"""A stable string identifier for the plugin."""
|
|
...
|
|
|
|
def __init__(self) -> None:
|
|
"""Initialize the plugin and discover decorated hook methods.
|
|
|
|
Scans the class for methods decorated with @hook and stores references
|
|
for later registration when the plugin is attached to an orchestrator.
|
|
|
|
Uses a guard to prevent double-discovery when used with multiple inheritance
|
|
(e.g., a class that inherits from both Plugin and MultiAgentPlugin).
|
|
"""
|
|
if not hasattr(self, "_hooks"):
|
|
self._hooks: list[HookCallback] = discover_hooks(self, self.name)
|
|
|
|
@property
|
|
def hooks(self) -> list[HookCallback]:
|
|
"""List of hooks the plugin provides, auto-discovered from @hook decorated methods."""
|
|
return self._hooks
|
|
|
|
def init_multi_agent(self, orchestrator: "MultiAgentBase") -> None | Awaitable[None]:
|
|
"""Initialize the plugin with the orchestrator instance.
|
|
|
|
Override this method to add custom initialization logic. Decorated
|
|
hooks are automatically registered by the plugin registry.
|
|
|
|
Args:
|
|
orchestrator: The multi-agent orchestrator instance to initialize with.
|
|
"""
|
|
return None
|