sylvain
67256f83ae
Previously delegate returned immediately after sending the task, causing the agent's result to be lost. Now subscribes to a results topic and waits up to 120s for the actual response before returning it to Nexus LLM. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
"""
|
|
Skill DELEGATE — déléguer une tâche à un agent spécialisé via MQTT.
|
|
Attend la réponse de l'agent (synchrone) avant de retourner.
|
|
|
|
Usage LLM : SKILL:delegate ARGS:<agent_id> | <tâche>
|
|
"""
|
|
import threading
|
|
import uuid
|
|
|
|
DESCRIPTION = "Déléguer une tâche à un agent spécialisé et retourner son résultat"
|
|
USAGE = "SKILL:delegate ARGS:<agent_id> | <tâche>"
|
|
|
|
TIMEOUT = 120 # secondes max d'attente de la réponse
|
|
|
|
|
|
def run(args: str, context) -> str:
|
|
if "|" not in args:
|
|
return "Format invalide. Usage : SKILL:delegate ARGS:<agent_id> | <tâche>"
|
|
|
|
agent_id, task = args.split("|", 1)
|
|
agent_id = agent_id.strip()
|
|
task = task.strip()
|
|
|
|
# Vérifier que l'agent est connu
|
|
caps = context.registry.get(agent_id)
|
|
if caps is None:
|
|
known = [a.agent_id for a in context.registry.all_agents()]
|
|
return f"Agent '{agent_id}' inconnu. Agents connus : {', '.join(known)}"
|
|
|
|
# Préparer la réception de la réponse
|
|
corr_id = str(uuid.uuid4())
|
|
reply_topic = context.mqtt.topic_results(corr_id)
|
|
|
|
result_event = threading.Event()
|
|
result_holder = []
|
|
|
|
def on_result(msg, topic):
|
|
payload = msg.payload if hasattr(msg, "payload") else str(msg)
|
|
result_holder.append(payload)
|
|
result_event.set()
|
|
|
|
# S'abonner avant d'envoyer pour ne pas manquer la réponse
|
|
context.mqtt.subscribe(reply_topic, on_result)
|
|
|
|
# Envoyer la tâche
|
|
context.mqtt.send_to(
|
|
recipient_id=agent_id,
|
|
payload=task,
|
|
correlation_id=corr_id,
|
|
reply_to=reply_topic,
|
|
)
|
|
|
|
# Attendre la réponse
|
|
got_reply = result_event.wait(timeout=TIMEOUT)
|
|
context.mqtt.unsubscribe(reply_topic)
|
|
|
|
if not got_reply:
|
|
return f"⏱ Timeout : {agent_id} n'a pas répondu en {TIMEOUT}s."
|
|
|
|
return result_holder[0] if result_holder else "Réponse vide de l'agent."
|