i 6dZddlZddlZddlZddlmZmZddlmZddlZejjddddl m Z dZ dZehd Zehd ZGd d eZGd dZy)u QueenRegistry — AIVA's identity store. Reads Queen identity from Postgres with Redis cache. Logs capability gains and retrieves capability history. AIVA RLM Nexus — Stories 2.01 + 2.02, Track A File: core/registry/queen_registry.py RULE COMPLIANCE: - Rule 7: No SQLite — all storage via Elestio Postgres/Redis - Rule 6: E: drive only — module lives at /mnt/e/genesis-system/core/registry/ - Rule 3: Verification stamp present at bottom of file N)datetimetimezone)Optionalz/mnt/e/genesis-system) connectionsz aiva:identityi,> bug_fixed new_skillimproved_recallpattern_learned>namerole voice_model last_evolvedactive_capabilitiesrecent_improvementstotal_conversationsceZdZdZy) RegistryErrorzHRaised when registry operations fail unrecoverably (e.g. Postgres down).N)__name__ __module__ __qualname____doc__5/mnt/e/genesis-system/core/registry/queen_registry.pyrr8sRrrc eZdZdZdddZdefdZddZ ddeded ed edef d Z dd e de fd Z de efdZdeddfdZdefdZy) QueenRegistryu AIVA Queen identity registry. Provides a single, authoritative view of AIVA's identity. Read path: 1. Try Redis cache (CACHE_KEY). Hit → decode JSON and return. 2. Miss → query Postgres (royal_conversations + aiva_capability_log). 3. Write result back to Redis (CACHE_TTL seconds). Failure semantics: - Redis failure at any point: non-fatal, silently skipped. - Postgres failure: raises RegistryError. Nreturnc"|xst|_y)z Args: connection_factory: A ConnectionFactory instance. Defaults to the module-level singleton ``connections``. Inject a mock for unit tests. N)r_cf)selfconnection_factorys r__init__zQueenRegistry.__init__Qs&4rcp|j}||S|j}|j||S)u Return AIVA's current identity object. Tries Redis cache first; falls back to Postgres on miss or Redis error. Returns: dict with exactly these keys: name (str) — always "AIVA" role (str) — human-readable role label voice_model (str) — env-overrideable voice model ID total_conversations (int) — COUNT(*) from royal_conversations last_evolved (str) — ISO-8601 timestamp of most recent capability log active_capabilities (list) — list[str] of currently active capability names recent_improvements (list) — list[str] of the 5 most recent capability descriptions Raises: RegistryError: when Postgres is unavailable or returns an unexpected error. )_try_cache_read_read_from_postgres_try_cache_write)r cachedidentitys r get_identityzQueenRegistry.get_identity^sB,%%'  M ++- h'rc |jj}|jty#t$rYywxYw)u Delete the Redis identity cache entry. Redis failure is silently swallowed — invalidation is best-effort. N)r get_redisdelete CACHE_KEY Exception)r rediss rinvalidate_cachezQueenRegistry.invalidate_caches7  HH&&(E LL #   s /2 >> descriptioncapability_typemetricsepoch_idc *|tvrtd|dtttt j }t jtj}||ni} |jj}|j} | jd|||||tj|f|j!| j#|j)|S#t$$r} t'd| | d} ~ wwxYw)a Log a new capability gain to ``aiva_capability_log``. Validates ``capability_type`` before touching the database. After a successful write, invalidates the Redis identity cache so that the next ``get_identity()`` call reflects the new log entry. Args: description: Human-readable description of the capability gained. capability_type: One of VALID_CAPABILITY_TYPES. metrics: Optional dict of supporting metrics. Stored as JSONB. Defaults to ``{}`` (never NULL in Postgres). epoch_id: Optional epoch identifier string. Returns: log_id: UUID string for the newly inserted row. Raises: ValueError: If ``capability_type`` is not in VALID_CAPABILITY_TYPES. RegistryError: If the Postgres write fails. zInvalid capability_type 'z'. Must be one of: Nz INSERT INTO aiva_capability_log (log_id, logged_at, capability_type, description, epoch_id, metrics) VALUES (%s, %s, %s, %s, %s, %s) zFailed to log capability gain: )VALID_CAPABILITY_TYPES ValueErrorsortedstruuiduuid4rnowrutcr get_postgrescursorexecutejsondumpscommitcloser.rr0) r r1r2r3r4log_id logged_at safe_metricsconncurexcs rlog_capability_gainz!QueenRegistry.log_capability_gains: "8 8+O+<=##)*@#A"BD  TZZ\"LL. ")"5w2  R88((*D++-C KK#JJ|,   KKM IIK   R"A# GHc Q Rs.A5C55 D>D  Dlast_nc  |jj}|j}|jd|f|j }|j g}|D]v}|\}} } } } t| dr| j} n t| } t| tr| }n| rtj| }ni}|j|| | | |dx|S#t $r}td||d}~wwxYw)u Return the last N capability gains, ordered newest first. Args: last_n: Maximum number of rows to return. Defaults to 10. Returns: List of dicts, each containing: log_id (str) — UUID of the log entry description (str) — human-readable capability description capability_type (str) — one of VALID_CAPABILITY_TYPES logged_at (str) — ISO-8601 timestamp metrics (dict) — supporting metrics (empty dict if none) Raises: RegistryError: If the Postgres query fails. z SELECT log_id, description, capability_type, logged_at, metrics FROM aiva_capability_log ORDER BY logged_at DESC LIMIT %s z$Failed to fetch capability history: N isoformat)rEr1r2rFr3)rr>r?r@fetchallrDr.rhasattrrNr9 isinstancedictrAloadsappend)r rLrHrIrowsrJresultrowrEr1r2rFr3 logged_at_str metrics_dicts rget_capability_historyz$QueenRegistry.get_capability_historys$ W88((*D++-C KK   <<>D IIK CGJ DFK)Wy+. ) 3 3 5 #I '4(& #zz'2 ! MM *#2*'  # 2 ; W"Fse LMSV V WsAC C;'C66C;c |jj}|jt}|rt j |S y#t $rYywxYw)z Attempt to read identity from Redis. Returns: Parsed dict if a valid cache entry exists, otherwise None. N)rr+getr-rArSr.)r r/raws rr$zQueenRegistry._try_cache_readsY HH&&(E))I&Czz#&   sAA AAr(c |jj}|jttt j |ty#t$rYywxYw)zk Attempt to write identity dict to Redis with TTL. Failure is silently swallowed. )defaultN) rr+setexr- CACHE_TTLrArBr9r.)r r(r/s rr&zQueenRegistry._try_cache_write'sF  HH&&(E KK 9djj3.O P   sAA AAcx |jj}|j}|jd|j d}|jd|j }|Dcgc]}|d }}|r1|dd}t |dr|j}n=t|}n1tjtjj}|jdd t!j"d d ||gd |d Scc}w#t$r} td| | d} ~ wwxYw)u Build the identity dict from Postgres tables. Queries: royal_conversations → COUNT(*) for total_conversations aiva_capability_log → 5 most-recent rows for recent_improvements + last_evolved timestamp Returns: Fully-populated identity dict. Raises: RegistryError: wraps any Postgres exception. z(SELECT COUNT(*) FROM royal_conversationsrz SELECT description, logged_at FROM aiva_capability_log ORDER BY logged_at DESC LIMIT 5 rNzFailed to read Queen identity: NAIVAuQueen AI — Lead OrchestratorAIVA_VOICE_MODELztelnyx-eucalyptus)voice_telephonymemory_persistenceknowledge_graph web_researchtelegram_interfacecron_scheduling)r r r rrrr)rr>r?r@fetchonerOrPrNr9rr<rr=rDr.rosgetenv) r rHrIrrUrWrraw_tsrrJs rr%z!QueenRegistry._read_from_postgres3s5# R88((*D++-C KKB C'*||~a'8  KK <<>D;?(@CQ(@ (@a6;/(.(8(8(:L#&v;L'||HLL9CCE IIK 499%79LM#6($$7  %)A R"A# GHc Q Rs+A3D5 DA6DD D9%D44D9)N)rN)NN) )rrrrr"rRr)r0r9rKintlistrZrr$r&r%rrrrrAs 5$d$L   BBB B  B BH=S=$=F$  $ C TC rr)rrArmr:rrtypingrsyspathinsertcore.db.connectionsrr-ra frozensetr6REQUIRED_FIELDSr.rrrrrrzs{  ' *++   #$ I u u r