41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
from __future__ import annotations
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class AgentRegistry:
|
|
def __init__(self):
|
|
self._agents: dict = {} # agent_key -> BaseAgent instance
|
|
self._active: set = set()
|
|
self._capabilities: dict = {}
|
|
|
|
async def load_from_odoo(self, odoo_client):
|
|
try:
|
|
rows = await odoo_client.search_read(
|
|
'ab.ai.agent.registry',
|
|
[['is_active', '=', True]],
|
|
['agent_key', 'name', 'capabilities_summary', 'sweep_enabled', 'sweep_interval_hours'])
|
|
self._active = {r['agent_key'] for r in rows}
|
|
self._capabilities = {r['agent_key']: r.get('capabilities_summary', '') for r in rows}
|
|
logger.info('AgentRegistry loaded: active=%s', list(self._active))
|
|
except Exception as exc:
|
|
logger.error('AgentRegistry.load_from_odoo failed: %s', exc)
|
|
|
|
async def get_active_agents(self):
|
|
return [{'agent_key': k, 'capabilities_summary': self._capabilities.get(k, '')}
|
|
for k in self._active]
|
|
|
|
async def is_active(self, agent_key):
|
|
return agent_key in self._active
|
|
|
|
async def sync(self, active_keys):
|
|
self._active = set(active_keys)
|
|
logger.info('AgentRegistry synced: active=%s', active_keys)
|
|
|
|
def register(self, agent_key, agent_instance):
|
|
self._agents[agent_key] = agent_instance
|
|
|
|
def get_agent_instance(self, agent_key):
|
|
return self._agents.get(agent_key)
|