mirror of
https://github.com/andrewyng/openworker.git
synced 2026-09-03 23:03:22 +00:00
Rohit C PrasadandDevika
2b45018ffa
Imported from andrewyng/aisuite@1b4bbf303e (contents of its platform/ directory, hoisted to the repo root). Development history prior to this commit lives in that repository. Co-authored-by: Devika <devikaverma11@gmail.com>
182 lines
8.0 KiB
Python
182 lines
8.0 KiB
Python
"""Connection hierarchy (UI-REFRESH §4) — the per-persona + per-session connector layers.
|
|
|
|
Three layers gate whether a connector is *effective* for a session:
|
|
|
|
1. **account-connected** — a connector profile with valid creds exists (``connector_list[].connected``).
|
|
Owned by the SecretStore; not stored here.
|
|
2. **persona-default-enabled** — per persona, which connected connectors are on by default for its
|
|
sessions (``PersonaConnectionStore``). Seeded from the persona manifest's ``recommends`` and then
|
|
user-editable.
|
|
3. **session-override** — per session, an explicit on/off that overrides the persona default
|
|
(``SessionConnectionStore``). Absence of an override means *inherit the persona default*.
|
|
|
|
``effective(connector)`` = **connected** AND (``session_override`` if present, else the persona
|
|
default if present, else inherit-on). A connector that is not connected is never effective. A
|
|
connector with no persona opinion and no session override inherits *on* — the persona's
|
|
``recommends`` curates what to *suggest*/seed-on, it is not an exhaustive allow-list, so a connected
|
|
connector the persona never mentions stays available unless something explicitly turns it off.
|
|
|
|
Both stores are tiny JSON files mirroring ``SubscriptionStore`` (optional path, ``_load``/``_save``,
|
|
``indent=2``); the manager owns one of each and resolves via :func:`effective`.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import threading
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
|
|
class PersonaConnectionStore:
|
|
"""``{persona_id: {connector: bool}}`` — the per-persona default on/off for each connector."""
|
|
|
|
def __init__(self, path: Optional[str | Path] = None) -> None:
|
|
self.path = Path(path) if path else None
|
|
self._lock = threading.Lock()
|
|
self._rows: dict[str, dict[str, bool]] = {}
|
|
self._load()
|
|
|
|
def _load(self) -> None:
|
|
if self.path and self.path.is_file():
|
|
data = json.loads(self.path.read_text(encoding="utf-8"))
|
|
self._rows = {
|
|
pid: {str(c): bool(v) for c, v in (row or {}).items()}
|
|
for pid, row in data.get("personas", {}).items()
|
|
}
|
|
|
|
def _save(self) -> None:
|
|
if not self.path:
|
|
return
|
|
self.path.parent.mkdir(parents=True, exist_ok=True)
|
|
self.path.write_text(
|
|
json.dumps({"personas": self._rows}, indent=2),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
# -- queries ----------------------------------------------------------------
|
|
def get(self, persona_id: str) -> dict[str, bool]:
|
|
"""The persona's stored row (a copy). Empty dict if it was never seeded/edited — this does
|
|
NOT seed; use :meth:`defaults_for` to seed from a manifest."""
|
|
return dict(self._rows.get(persona_id, {}))
|
|
|
|
def defaults_for(
|
|
self, persona_id: str, manifest, *, connected: set[str]
|
|
) -> dict[str, bool]:
|
|
"""The persona's default connector map, seeding it from the manifest on first read.
|
|
|
|
Seeding rule: a ``recommends`` item of kind ``connector`` with ``tier == "core"`` defaults
|
|
**True**; every other recommended connector (optional) defaults **False**. (mcp recommends
|
|
and non-connector kinds are ignored.) The seeded row is persisted on first read so the seed
|
|
is stable thereafter — a later edit/toggle persists over it. A persona with no manifest
|
|
(e.g. a builtin) seeds an empty row.
|
|
|
|
NOTE: this intentionally deviates from §4.2's literal "whose connector is connected" wording
|
|
to honor its intent. A core connector seeds True even when not connected yet:
|
|
:func:`effective` already gates on ``connected``, so it stays filtered out while
|
|
disconnected and **self-lights when it later connects** — rather than being frozen False
|
|
forever (a stale seed that would break the "connect a core connector → on by default"
|
|
flow). ``connected`` is kept in the signature for back-compat but is no longer read here,
|
|
leaving :func:`effective`'s connected-gate the single source of truth for connectedness.
|
|
"""
|
|
with self._lock:
|
|
if persona_id in self._rows:
|
|
return dict(self._rows[persona_id])
|
|
seeded: dict[str, bool] = {}
|
|
recommends = list(getattr(manifest, "recommends", None) or [])
|
|
for rec in recommends:
|
|
if getattr(rec, "kind", None) != "connector":
|
|
continue
|
|
# core → on by default (connectedness is enforced later by effective()).
|
|
seeded[rec.ref] = getattr(rec, "tier", "") == "core"
|
|
self._rows[persona_id] = seeded
|
|
self._save()
|
|
return dict(seeded)
|
|
|
|
# -- mutations --------------------------------------------------------------
|
|
def set(self, persona_id: str, connector: str, enabled: bool) -> None:
|
|
with self._lock:
|
|
self._rows.setdefault(persona_id, {})[connector] = bool(enabled)
|
|
self._save()
|
|
|
|
|
|
class SessionConnectionStore:
|
|
"""``{session_id: {connector: bool}}`` — per-session overrides only; an absent entry means the
|
|
session inherits the persona default."""
|
|
|
|
def __init__(self, path: Optional[str | Path] = None) -> None:
|
|
self.path = Path(path) if path else None
|
|
self._lock = threading.Lock()
|
|
self._rows: dict[str, dict[str, bool]] = {}
|
|
self._load()
|
|
|
|
def _load(self) -> None:
|
|
if self.path and self.path.is_file():
|
|
data = json.loads(self.path.read_text(encoding="utf-8"))
|
|
self._rows = {
|
|
sid: {str(c): bool(v) for c, v in (row or {}).items()}
|
|
for sid, row in data.get("sessions", {}).items()
|
|
}
|
|
|
|
def _save(self) -> None:
|
|
if not self.path:
|
|
return
|
|
self.path.parent.mkdir(parents=True, exist_ok=True)
|
|
self.path.write_text(
|
|
json.dumps({"sessions": self._rows}, indent=2),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
# -- queries ----------------------------------------------------------------
|
|
def get(self, session_id: str) -> dict[str, bool]:
|
|
return dict(self._rows.get(session_id, {}))
|
|
|
|
# -- mutations --------------------------------------------------------------
|
|
def set(self, session_id: str, connector: str, enabled: bool) -> None:
|
|
with self._lock:
|
|
self._rows.setdefault(session_id, {})[connector] = bool(enabled)
|
|
self._save()
|
|
|
|
def clear(self, session_id: str, connector: str) -> None:
|
|
"""Drop a single override so the session inherits the persona default again."""
|
|
with self._lock:
|
|
row = self._rows.get(session_id)
|
|
if row and connector in row:
|
|
del row[connector]
|
|
if not row:
|
|
del self._rows[session_id]
|
|
self._save()
|
|
|
|
def remove_session(self, session_id: str) -> None:
|
|
"""Drop all of a session's overrides (called when the session is deleted)."""
|
|
with self._lock:
|
|
if session_id in self._rows:
|
|
del self._rows[session_id]
|
|
self._save()
|
|
|
|
|
|
def effective(
|
|
*,
|
|
connected: set[str],
|
|
persona_defaults: dict[str, bool],
|
|
session_overrides: dict[str, bool],
|
|
) -> dict[str, bool]:
|
|
"""Resolve the effective-enabled connectors for a session — the §4 invariant.
|
|
|
|
For each **connected** connector: a session override (if present) wins; otherwise the persona
|
|
default (if present) applies; otherwise it inherits *on*. Not-connected connectors are never
|
|
effective. Returns only the effective-**enabled** connectors, each mapped to ``True`` (muted /
|
|
off connectors are omitted), so the result reads as the session's live connector set.
|
|
"""
|
|
out: dict[str, bool] = {}
|
|
for connector in connected:
|
|
if connector in session_overrides:
|
|
enabled = session_overrides[connector]
|
|
elif connector in persona_defaults:
|
|
enabled = persona_defaults[connector]
|
|
else:
|
|
enabled = True # connected, no opinion → inherit on
|
|
if enabled:
|
|
out[connector] = True
|
|
return out
|