213 lines
No EOL
8 KiB
Python
213 lines
No EOL
8 KiB
Python
"""
|
|
Chat Agent Registry Modul.
|
|
Stellt ein zentrales Registry-System für alle verfügbaren Agenten bereit.
|
|
"""
|
|
|
|
import os
|
|
import logging
|
|
import importlib
|
|
from typing import Dict, Any, List, Optional
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class AgentRegistry:
|
|
"""Zentrale Registry für alle verfügbaren Agenten im System."""
|
|
|
|
_instance = None
|
|
|
|
@classmethod
|
|
def get_instance(cls):
|
|
"""Gibt eine Singleton-Instanz der Agent-Registry zurück."""
|
|
if cls._instance is None:
|
|
cls._instance = cls()
|
|
return cls._instance
|
|
|
|
def __init__(self):
|
|
"""Initialisiert die Agent-Registry."""
|
|
if AgentRegistry._instance is not None:
|
|
raise RuntimeError("Singleton-Instanz existiert bereits - verwende get_instance()")
|
|
|
|
self.agents = {}
|
|
self.ai_service = None
|
|
self._load_agents()
|
|
|
|
def _load_agents(self):
|
|
"""Lädt alle verfügbaren Agenten aus den Modulen."""
|
|
logger.info("Lade Agent-Module...")
|
|
|
|
# Liste der zu ladenden Agent-Module
|
|
agent_modules = []
|
|
agent_dir = os.path.dirname(__file__)
|
|
|
|
# Durchsuche das Verzeichnis nach Agent-Modulen
|
|
for filename in os.listdir(agent_dir):
|
|
if filename.startswith("chat_agent_") and filename.endswith(".py"):
|
|
agent_modules.append(filename[:-3]) # Entferne .py-Endung
|
|
|
|
if not agent_modules:
|
|
logger.warning("Keine Agent-Module gefunden")
|
|
return
|
|
|
|
logger.info(f"{len(agent_modules)} Agent-Module gefunden")
|
|
|
|
# Lade jedes Agent-Modul
|
|
for module_name in agent_modules:
|
|
try:
|
|
# Importiere das Modul
|
|
module = importlib.import_module(f"modules.{module_name}")
|
|
|
|
# Suche nach der Agent-Klasse oder einer get_*_agent-Funktion
|
|
agent_name= module_name.split('_')[-1]
|
|
class_name = f"Agent{agent_name.capitalize()}"
|
|
getter_name = f"get_{agent_name}_agent"
|
|
|
|
agent = None
|
|
|
|
# Versuche, den Agenten über die get_*_agent-Funktion zu erhalten
|
|
if hasattr(module, getter_name):
|
|
getter_func = getattr(module, getter_name)
|
|
agent = getter_func()
|
|
logger.info(f"Agent '{agent.name}' über {getter_name}() geladen")
|
|
|
|
# Alternativ versuche, den Agenten direkt zu instanziieren
|
|
elif hasattr(module, class_name):
|
|
agent_class = getattr(module, class_name)
|
|
agent = agent_class()
|
|
logger.info(f"Agent '{agent.name}' (Typ: {agent.name}) direkt instanziert")
|
|
|
|
if agent:
|
|
# Registriere den Agenten
|
|
self.register_agent(agent)
|
|
else:
|
|
logger.warning(f"Keine Agent-Klasse oder Getter-Funktion in Modul {module_name} gefunden")
|
|
|
|
except ImportError as e:
|
|
logger.error(f"Modul {module_name} konnte nicht importiert werden: {e}")
|
|
except Exception as e:
|
|
logger.error(f"Fehler beim Laden des Agenten aus Modul {module_name}: {e}")
|
|
|
|
def set_ai_service(self, ai_service):
|
|
self.ai_service = ai_service
|
|
self.update_agent_dependencies()
|
|
|
|
def update_agent_dependencies(self):
|
|
"""Aktualisiert die Abhängigkeiten für alle registrierten Agenten."""
|
|
for agent_id, agent in self.agents.items():
|
|
if hasattr(agent, 'set_dependencies'):
|
|
agent.set_dependencies(ai_service=self.ai_service)
|
|
|
|
def register_agent(self, agent):
|
|
"""
|
|
Registriert einen Agenten in der Registry.
|
|
|
|
Args:
|
|
agent: Der zu registrierende Agent
|
|
"""
|
|
agent_id = getattr(agent, 'name', "unknown_agent")
|
|
# Initialisiere Agenten mit Abhängigkeiten
|
|
if hasattr(agent, 'set_dependencies'):
|
|
agent.set_dependencies(ai_service=self.ai_service)
|
|
self.agents[agent_id] = agent
|
|
logger.debug(f"Agent '{agent.name}' (Typ: {agent_id}, Name: {agent_id}) registriert")
|
|
|
|
def get_agent(self, agent_identifier: str):
|
|
"""
|
|
Gibt eine Agenten-Instanz zurück
|
|
Args:
|
|
agent_identifier: ID oder Typ des gewünschten Agenten
|
|
Returns:
|
|
Agenten-Instanz oder None, falls nicht gefunden
|
|
"""
|
|
if agent_identifier in self.agents:
|
|
return self.agents[agent_identifier]
|
|
logger.error(f"Agent mit Kennung '{agent_identifier}' nicht gefunden")
|
|
return None
|
|
|
|
def get_all_agents(self) -> Dict[str, Any]:
|
|
"""Gibt alle registrierten Agenten zurück."""
|
|
return self.agents
|
|
|
|
def get_agent_infos(self) -> List[Dict[str, Any]]:
|
|
"""Gibt Informationen über alle registrierten Agenten zurück."""
|
|
agent_infos = []
|
|
seen_agents = set()
|
|
for agent in self.agents.values():
|
|
if agent not in seen_agents:
|
|
# Verwende get_agent_info oder erstelle manuell die Info
|
|
if hasattr(agent, 'get_agent_info'):
|
|
agent_infos.append(agent.get_agent_info())
|
|
else:
|
|
agent_infos.append({
|
|
"name": agent.name,
|
|
"capabilities": getattr(agent, 'capabilities', ""),
|
|
"result_format": getattr(agent, 'result_format', "Text")
|
|
})
|
|
logger.error(f"Agent mit Kennung '{agent.name}' hat keine vollständigen Daten")
|
|
seen_agents.add(agent)
|
|
return agent_infos
|
|
|
|
|
|
# Basis-Agent-Klasse
|
|
class AgentBase:
|
|
"""
|
|
Basis-Klasse für alle Chat-Agenten.
|
|
Definiert die grundlegende Schnittstelle und Funktionalität.
|
|
"""
|
|
|
|
def __init__(self):
|
|
"""Initialisiere den Basis-Agenten."""
|
|
self.name = "Basis-Agent"
|
|
self.capabilities = "Grundlegende Agentenfunktionen"
|
|
self.result_format = "Text"
|
|
self.ai_service = None
|
|
|
|
def set_dependencies(self, ai_service=None):
|
|
self.ai_service = ai_service
|
|
|
|
def get_config(self) -> Dict[str, Any]:
|
|
return {
|
|
"name": self.name,
|
|
"capabilities": self.capabilities,
|
|
"result_format": self.result_format
|
|
}
|
|
|
|
async def process_message(self, message: Dict[str, Any], context: Dict[str, Any] = None) -> Dict[str, Any]:
|
|
# Basisimplementierung - sollte von spezialisierten Agenten überschrieben werden
|
|
if not self.ai_service:
|
|
logger.warning(f"Agent {self.id} hat keinen konfigurierten AI-Service")
|
|
return {
|
|
"role": "assistant",
|
|
"content": f"Ich bin {self.name}, aber ich bin nicht richtig konfiguriert. Bitte den AI-Service einrichten.",
|
|
"agent_name": self.name,
|
|
"result_format": "Text"
|
|
}
|
|
|
|
# Einfachen Prompt erstellen
|
|
prompt = message.get("content", "")
|
|
|
|
# Antwort generieren
|
|
try:
|
|
response_content = self.ai_service.call_api([
|
|
{"role": "system", "content": f"Du bist {self.name}, ein spezialisierter {self.name}-Agent mit Fähigkeiten in: {self.capabilities}"},
|
|
{"role": "user", "content": prompt}
|
|
])
|
|
|
|
return {
|
|
"role": "assistant",
|
|
"content": response_content,
|
|
"agent_name": self.name,
|
|
"result_format": self.result_format
|
|
}
|
|
except Exception as e:
|
|
logger.error(f"Fehler in Agent {self.id}: {str(e)}")
|
|
return {
|
|
"role": "assistant",
|
|
"content": f"Ich habe einen Fehler festgestellt: {str(e)}",
|
|
"agent_name": self.name,
|
|
"result_format": "Text"
|
|
}
|
|
|
|
|
|
# Singleton-Factory für die Agent-Registry
|
|
def get_agent_registry():
|
|
return AgentRegistry.get_instance() |