@team-agent/installer 0.1.11 → 0.2.1

package/src/team_agent/messaging/delivery.py

@@ -0,0 +1,138 @@
+from __future__ import annotations
+
+from team_agent.messaging.deps import (
+    EventLog,
+    MessageStore,
+    _message_by_id,
+    _message_payload,
+    _tmux_inject_text,
+    _tmux_window_exists,
+    core_render_message,
+)
+
+from pathlib import Path
+from typing import Any
+
+def _deliver_pending_message(
+    workspace: Path,
+    state: dict[str, Any],
+    message_id: str,
+    wait_visible: bool = True,
+    timeout: float = 30.0,
+) -> dict[str, Any]:
+    store = MessageStore(workspace)
+    row = next((m for m in store.messages() if m["message_id"] == message_id), None)
+    if not row:
+        return {"ok": False, "status": "failed", "reason": "message_missing"}
+    agent_state = state.get("agents", {}).get(row["recipient"])
+    if not agent_state:
+        store.mark(message_id, "failed", "unknown recipient")
+        return {"ok": False, "status": "failed", "reason": "unknown_recipient"}
+    session_name = state.get("session_name")
+    window = agent_state.get("window", row["recipient"])
+    payload = _message_payload(row)
+    rendered = core_render_message(payload)
+    text = rendered["text"]
+    if not session_name or not _tmux_window_exists(session_name, window):
+        store.mark(message_id, "failed", "tmux target missing")
+        EventLog(workspace).write("send.failed", message_id=message_id, reason="tmux target missing", target=f"{session_name}:{window}")
+        return {"ok": False, "status": "failed", "reason": "tmux_target_missing"}
+    target = f"{session_name}:{window}"
+    if not store.claim_for_delivery(message_id):
+        current = _message_by_id(store, message_id)
+        status = current["status"] if current else "missing"
+        EventLog(workspace).write("send.delivery_claim_skipped", message_id=message_id, target=target, status=status)
+        return {
+            "ok": status in {"injected", "visible", "submitted", "submitted_unverified", "delivered", "acknowledged"},
+            "status": status,
+            "reason": "message_already_claimed",
+        }
+    recipient_status = agent_state.get("status")
+    EventLog(workspace).write(
+        "send.deliver_attempt",
+        message_id=message_id,
+        target=target,
+        payload=payload,
+        recipient_status=recipient_status,
+        recipient_busy=recipient_status == "busy",
+        visible_token=rendered.get("token"),
+    )
+    injection = _tmux_inject_text(
+        target,
+        text,
+        "Enter",
+        f"team-agent-send-{message_id}",
+        attempts=3 if wait_visible else 1,
+        provider=agent_state.get("provider", "fake"),
+    )
+    if injection["ok"]:
+        store.mark(message_id, "submitted")
+        EventLog(workspace).write(
+            "send.submitted",
+            message_id=message_id,
+            target=target,
+            status="submitted",
+            verification=injection.get("verification"),
+            submit_verification=injection.get("submit_verification"),
+            turn_verification=injection.get("turn_verification"),
+            paste_attempts=injection.get("attempts"),
+            submit_attempts=injection.get("submit_attempts"),
+        )
+        return {
+            "ok": True,
+            "status": "delivered",
+            "message_status": "submitted",
+            "verification": injection.get("verification"),
+            "submit_verification": injection.get("submit_verification"),
+            "turn_verification": injection.get("turn_verification"),
+            "paste_attempts": injection.get("attempts"),
+            "submit_attempts": injection.get("submit_attempts"),
+        }
+    reason = injection.get("error") or injection.get("verification") or "tmux injection failed"
+    store.mark(message_id, "failed", reason)
+    EventLog(workspace).write(
+        "send.failed",
+        message_id=message_id,
+        reason=reason,
+        target=target,
+        stage=injection.get("stage"),
+        verification=injection.get("verification"),
+        submit_verification=injection.get("submit_verification"),
+        turn_verification=injection.get("turn_verification"),
+        paste_attempts=injection.get("attempts"),
+        submit_attempts=injection.get("submit_attempts"),
+    )
+    return {
+        "ok": False,
+        "status": "failed",
+        "reason": reason,
+        "stage": injection.get("stage"),
+        "verification": injection.get("verification"),
+        "submit_verification": injection.get("submit_verification"),
+        "turn_verification": injection.get("turn_verification"),
+        "paste_attempts": injection.get("attempts"),
+        "submit_attempts": injection.get("submit_attempts"),
+    }
+
+
+def _deliver_pending_messages(workspace: Path, state: dict[str, Any], event_log: EventLog) -> list[str]:
+    store = MessageStore(workspace)
+    delivered: list[str] = []
+    for row in store.messages():
+        if row["status"] not in {"pending", "accepted"}:
+            continue
+        agent_state = state.get("agents", {}).get(row["recipient"]) or {}
+        if str(agent_state.get("status") or "").lower() == "busy":
+            event_log.write(
+                "send.deferred_busy",
+                message_id=row["message_id"],
+                sender=row.get("sender"),
+                recipient=row["recipient"],
+                reason="recipient_busy",
+            )
+            continue
+        result = _deliver_pending_message(workspace, state, row["message_id"], wait_visible=True, timeout=30.0)
+        if result.get("ok"):
+            delivered.append(row["message_id"])
+            event_log.write("send.pending_delivered", message_id=row["message_id"], agent_id=row["recipient"])
+    return delivered

package/src/team_agent/messaging/deps.py

@@ -0,0 +1,263 @@
+from __future__ import annotations
+
+import copy
+import json
+import os
+import re
+import subprocess
+import time
+from datetime import datetime, timedelta, timezone
+from typing import Any
+
+from team_agent import runtime as _runtime
+from team_agent.errors import RuntimeError, ValidationError
+from team_agent.events import EventLog
+from team_agent.message_store import MessageStore
+from team_agent.paths import runtime_dir
+from team_agent.permissions import missing_tools
+from team_agent.routing import route_task
+from team_agent.spec import load_spec, validate_result_envelope
+from team_agent.state import (
+    ambiguous_team_target_result,
+    check_team_owner,
+    load_runtime_state,
+    save_runtime_state,
+    save_team_scoped_state,
+    select_runtime_state,
+    team_state_key,
+    write_team_state,
+)
+from team_agent.task_graph import update_task_status
+
+# Explicit runtime dependency surface for messaging extraction. Wrappers keep
+# runtime monkeypatch points stable while avoiding module-wide globals sync.
+_RUNTIME_CONSTANTS = (
+    "DELIVERY_CAPTURE_LINES",
+    "PASTED_CONTENT_PROMPT_RE",
+    "TMUX_PANE_FORMAT",
+    "TMUX_PASTE_BYTES_PER_SECOND",
+    "TMUX_PASTE_MAX_READY_TIMEOUT",
+    "TMUX_PASTE_MIN_READY_TIMEOUT",
+    "TMUX_STDIN_BUFFER_THRESHOLD",
+    "TMUX_SUBMIT_BYTES_PER_SECOND",
+    "TMUX_SUBMIT_MAX_SETTLE_TIMEOUT",
+    "TMUX_SUBMIT_MIN_SETTLE_TIMEOUT",
+)
+_RUNTIME_PATCH_POINTS = (
+    "_capture_has_pasted_content_prompt",
+    "_capture_missing_sessions",
+    "_capture_tmux_pane_text",
+    "_choose_leader_submit_key",
+    "_current_task_for_agent",
+    "_deliver_pending_message",
+    "_deliver_pending_messages",
+    "_find_agent",
+    "_find_task",
+    "_find_task_or_none",
+    "_format_team_agent_message",
+    "_handle_provider_runtime_prompts",
+    "_handle_provider_startup_prompts",
+    "_infer_active_tmux_pane",
+    "_infer_workspace_tmux_pane",
+    "_is_leader_sender",
+    "_is_leader_target",
+    "_is_message_scoped_result",
+    "_is_runtime_team_agent",
+    "_leader_id",
+    "_leader_receiver_is_direct",
+    "_message_by_id",
+    "_message_payload",
+    "_mirror_peer_message_to_leader",
+    "_notify_leader_of_report_result",
+    "_rediscover_leader_receiver",
+    "_refresh_agent_runtime_statuses",
+    "_result_status_to_task_status",
+    "_runtime_lock",
+    "_runtime_team_agent_ids",
+    "_send_to_leader_receiver",
+    "_submit_worker_prompt",
+    "_tmux_current_client_pane_info",
+    "_tmux_inject_text",
+    "_tmux_list_panes",
+    "_tmux_load_buffer_stdin",
+    "_tmux_pane_info",
+    "_tmux_paste_ready_timeout",
+    "_tmux_set_buffer_text",
+    "_tmux_submit_settle_timeout",
+    "_tmux_window_exists",
+    "_validate_leader_receiver",
+    "_wait_for_message_ready",
+    "_wait_for_worker_message_ready",
+    "core_list_targets",
+    "core_render_message",
+    "run_cmd",
+    "send_message",
+    "start_coordinator",
+)
+for _name in (*_RUNTIME_CONSTANTS, *_RUNTIME_PATCH_POINTS):
+    if not hasattr(_runtime, _name):
+        raise ImportError(f"team_agent.runtime missing messaging dependency: {_name}")
+
+DELIVERY_CAPTURE_LINES = _runtime.DELIVERY_CAPTURE_LINES
+PASTED_CONTENT_PROMPT_RE = _runtime.PASTED_CONTENT_PROMPT_RE
+TMUX_PANE_FORMAT = _runtime.TMUX_PANE_FORMAT
+TMUX_PASTE_BYTES_PER_SECOND = _runtime.TMUX_PASTE_BYTES_PER_SECOND
+TMUX_PASTE_MAX_READY_TIMEOUT = _runtime.TMUX_PASTE_MAX_READY_TIMEOUT
+TMUX_PASTE_MIN_READY_TIMEOUT = _runtime.TMUX_PASTE_MIN_READY_TIMEOUT
+TMUX_STDIN_BUFFER_THRESHOLD = _runtime.TMUX_STDIN_BUFFER_THRESHOLD
+TMUX_SUBMIT_BYTES_PER_SECOND = _runtime.TMUX_SUBMIT_BYTES_PER_SECOND
+TMUX_SUBMIT_MAX_SETTLE_TIMEOUT = _runtime.TMUX_SUBMIT_MAX_SETTLE_TIMEOUT
+TMUX_SUBMIT_MIN_SETTLE_TIMEOUT = _runtime.TMUX_SUBMIT_MIN_SETTLE_TIMEOUT
+
+
+def _runtime_symbol(name: str) -> Any:
+    return getattr(_runtime, name)
+
+def _capture_has_pasted_content_prompt(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_capture_has_pasted_content_prompt")(*args, **kwargs)
+
+def _capture_missing_sessions(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_capture_missing_sessions")(*args, **kwargs)
+
+def _capture_tmux_pane_text(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_capture_tmux_pane_text")(*args, **kwargs)
+
+def _choose_leader_submit_key(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_choose_leader_submit_key")(*args, **kwargs)
+
+def _current_task_for_agent(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_current_task_for_agent")(*args, **kwargs)
+
+def _deliver_pending_message(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_deliver_pending_message")(*args, **kwargs)
+
+def _deliver_pending_messages(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_deliver_pending_messages")(*args, **kwargs)
+
+def _find_agent(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_find_agent")(*args, **kwargs)
+
+def _find_task(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_find_task")(*args, **kwargs)
+
+def _find_task_or_none(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_find_task_or_none")(*args, **kwargs)
+
+def _format_team_agent_message(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_format_team_agent_message")(*args, **kwargs)
+
+def _handle_provider_runtime_prompts(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_handle_provider_runtime_prompts")(*args, **kwargs)
+
+def _handle_provider_startup_prompts(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_handle_provider_startup_prompts")(*args, **kwargs)
+
+def _is_leader_sender(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_is_leader_sender")(*args, **kwargs)
+
+def _is_leader_target(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_is_leader_target")(*args, **kwargs)
+
+def _is_message_scoped_result(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_is_message_scoped_result")(*args, **kwargs)
+
+def _is_runtime_team_agent(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_is_runtime_team_agent")(*args, **kwargs)
+
+def _leader_id(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_leader_id")(*args, **kwargs)
+
+def _leader_receiver_is_direct(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_leader_receiver_is_direct")(*args, **kwargs)
+
+def _message_by_id(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_message_by_id")(*args, **kwargs)
+
+def _message_payload(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_message_payload")(*args, **kwargs)
+
+def _mirror_peer_message_to_leader(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_mirror_peer_message_to_leader")(*args, **kwargs)
+
+def _notify_leader_of_report_result(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_notify_leader_of_report_result")(*args, **kwargs)
+
+def _rediscover_leader_receiver(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_rediscover_leader_receiver")(*args, **kwargs)
+
+def _refresh_agent_runtime_statuses(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_refresh_agent_runtime_statuses")(*args, **kwargs)
+
+def _result_status_to_task_status(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_result_status_to_task_status")(*args, **kwargs)
+
+def _runtime_lock(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_runtime_lock")(*args, **kwargs)
+
+def _runtime_team_agent_ids(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_runtime_team_agent_ids")(*args, **kwargs)
+
+def _send_to_leader_receiver(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_send_to_leader_receiver")(*args, **kwargs)
+
+def _submit_worker_prompt(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_submit_worker_prompt")(*args, **kwargs)
+
+def _tmux_inject_text(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_tmux_inject_text")(*args, **kwargs)
+
+def _tmux_current_client_pane_info(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_tmux_current_client_pane_info")(*args, **kwargs)
+
+def _tmux_list_panes(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_tmux_list_panes")(*args, **kwargs)
+
+def _infer_active_tmux_pane(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_infer_active_tmux_pane")(*args, **kwargs)
+
+def _tmux_pane_info(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_tmux_pane_info")(*args, **kwargs)
+
+def _infer_workspace_tmux_pane(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_infer_workspace_tmux_pane")(*args, **kwargs)
+
+def _tmux_load_buffer_stdin(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_tmux_load_buffer_stdin")(*args, **kwargs)
+
+def _tmux_paste_ready_timeout(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_tmux_paste_ready_timeout")(*args, **kwargs)
+
+def _tmux_set_buffer_text(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_tmux_set_buffer_text")(*args, **kwargs)
+
+def _tmux_submit_settle_timeout(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_tmux_submit_settle_timeout")(*args, **kwargs)
+
+def _tmux_window_exists(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_tmux_window_exists")(*args, **kwargs)
+
+def _validate_leader_receiver(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_validate_leader_receiver")(*args, **kwargs)
+
+def _wait_for_message_ready(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_wait_for_message_ready")(*args, **kwargs)
+
+def _wait_for_worker_message_ready(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("_wait_for_worker_message_ready")(*args, **kwargs)
+
+def run_cmd(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("run_cmd")(*args, **kwargs)
+
+def core_list_targets(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("core_list_targets")(*args, **kwargs)
+
+def core_render_message(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("core_render_message")(*args, **kwargs)
+
+def send_message(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("send_message")(*args, **kwargs)
+
+def start_coordinator(*args: Any, **kwargs: Any) -> Any:
+    return _runtime_symbol("start_coordinator")(*args, **kwargs)
+
+__all__ = ['DELIVERY_CAPTURE_LINES', 'EventLog', 'MessageStore', 'PASTED_CONTENT_PROMPT_RE', 'RuntimeError', 'TMUX_PANE_FORMAT', 'TMUX_PASTE_BYTES_PER_SECOND', 'TMUX_PASTE_MAX_READY_TIMEOUT', 'TMUX_PASTE_MIN_READY_TIMEOUT', 'TMUX_STDIN_BUFFER_THRESHOLD', 'TMUX_SUBMIT_BYTES_PER_SECOND', 'TMUX_SUBMIT_MAX_SETTLE_TIMEOUT', 'TMUX_SUBMIT_MIN_SETTLE_TIMEOUT', 'ValidationError', '_capture_has_pasted_content_prompt', '_capture_missing_sessions', '_capture_tmux_pane_text', '_choose_leader_submit_key', '_current_task_for_agent', '_deliver_pending_message', '_deliver_pending_messages', '_find_agent', '_find_task', '_find_task_or_none', '_format_team_agent_message', '_handle_provider_runtime_prompts', '_handle_provider_startup_prompts', '_infer_active_tmux_pane', '_infer_workspace_tmux_pane', '_is_leader_sender', '_is_leader_target', '_is_message_scoped_result', '_is_runtime_team_agent', '_leader_id', '_leader_receiver_is_direct', '_message_by_id', '_message_payload', '_mirror_peer_message_to_leader', '_notify_leader_of_report_result', '_rediscover_leader_receiver', '_refresh_agent_runtime_statuses', '_result_status_to_task_status', '_runtime_lock', '_runtime_team_agent_ids', '_send_to_leader_receiver', '_submit_worker_prompt', '_tmux_current_client_pane_info', '_tmux_inject_text', '_tmux_list_panes', '_tmux_load_buffer_stdin', '_tmux_pane_info', '_tmux_paste_ready_timeout', '_tmux_set_buffer_text', '_tmux_submit_settle_timeout', '_tmux_window_exists', '_validate_leader_receiver', '_wait_for_message_ready', '_wait_for_worker_message_ready', 'ambiguous_team_target_result', 'check_team_owner', 'copy', 'core_list_targets', 'core_render_message', 'datetime', 'json', 'load_runtime_state', 'load_spec', 'missing_tools', 'os', 're', 'route_task', 'run_cmd', 'runtime_dir', 'save_runtime_state', 'save_team_scoped_state', 'select_runtime_state', 'send_message', 'start_coordinator', 'subprocess', 'team_state_key', 'time', 'timedelta', 'timezone', 'update_task_status', 'validate_result_envelope', 'write_team_state']