hypermind 0.11.0__py3-none-any.whl

hypermind/simlab/_persona_registry.py

@@ -0,0 +1,93 @@
+"""Tiny in-process registry: (sim_id, agent_name) → persona metadata.
+
+Used by the SSE event stream to enrich live records with persona +
+emoji so the frontend Activity canvas can color-cluster bubbles by
+persona without a second round-trip. Populated by ``_scenario_impl``
+when AgentProfiles are constructed; read by ``app._handle_sim_events_stream``.
+
+Bounded: cleared on sim completion via ``forget_sim`` to avoid leaking
+across long-running servers. If the registry grows past the soft cap
+(default 4096 entries) the oldest sims are evicted.
+"""
+
+from __future__ import annotations
+
+from collections import OrderedDict
+from threading import RLock
+
+_LOCK = RLock()
+_REGISTRY: OrderedDict[str, dict[str, dict[str, str]]] = OrderedDict()
+_MAX_SIMS = 64  # keep last 64 sims around for late event tails
+
+
+def remember(
+    sim_id: str,
+    agent_name: str,
+    *,
+    persona: str,
+    persona_emoji: str = "",
+    display_name: str = "",
+    role: str = "",
+) -> None:
+    """Record a (sim_id, agent_name) → persona/role mapping.
+
+    ``agent_name`` may be either the agent's friendly name (e.g.
+    ``"analyst-01"``) or its kid hex; both are typically registered
+    pointing at the same metadata so the SSE handler can look up by
+    whichever identifier the underlying event carries.
+
+    ``role`` (S5): one of scout/sceptic/synthesist/mediator/oracle/
+    generalist (or "" for unknown). Drives the canvas role-ring colour.
+    """
+    if not sim_id or not agent_name:
+        return
+    with _LOCK:
+        bucket = _REGISTRY.get(sim_id)
+        if bucket is None:
+            bucket = {}
+            _REGISTRY[sim_id] = bucket
+            while len(_REGISTRY) > _MAX_SIMS:
+                _REGISTRY.popitem(last=False)
+        _REGISTRY.move_to_end(sim_id)
+        existing = bucket.get(agent_name, {})
+        bucket[agent_name] = {
+            "persona": persona,
+            "persona_emoji": persona_emoji or "",
+            "display_name": display_name or agent_name,
+            # Preserve existing role if not overridden — _scenario_impl
+            # registers personas first, then sets oracle role later.
+            "role": role or existing.get("role", ""),
+        }
+
+
+def set_role(sim_id: str, agent_name: str, role: str) -> None:
+    """Late-binding role update. Used after _scenario_impl assigns oracle
+    and any other post-construction roles."""
+    if not sim_id or not agent_name:
+        return
+    with _LOCK:
+        bucket = _REGISTRY.get(sim_id)
+        if not bucket or agent_name not in bucket:
+            return
+        bucket[agent_name]["role"] = role
+
+
+def lookup(sim_id: str, agent_name: str) -> dict[str, str] | None:
+    if not sim_id or not agent_name:
+        return None
+    with _LOCK:
+        bucket = _REGISTRY.get(sim_id)
+        if not bucket:
+            return None
+        return bucket.get(agent_name)
+
+
+def forget_sim(sim_id: str) -> None:
+    with _LOCK:
+        _REGISTRY.pop(sim_id, None)
+
+
+def reset() -> None:
+    """Test-only: clear all entries."""
+    with _LOCK:
+        _REGISTRY.clear()