titan-agent 5.5.26 → 5.5.27

package/dist/agent/agent.js

@@ -1075,7 +1075,27 @@ CRITICAL: Widgets are NOT files. Do NOT use write_file, edit_file, or any file t
   } catch {
   }
   const { resolveActivePersona: __resolvePersona } = await import("./personaProfiles.js");
-  const earlyPersona = __resolvePersona({ channel });
+  let earlyPersona = __resolvePersona({ channel });
+  let __cohortAssignment = null;
+  if (earlyPersona) {
+    try {
+      const { pickCohortAssignment } = await import("./personaRollout.js");
+      const assignment = pickCohortAssignment(earlyPersona.id, { sessionId: session.id, channel });
+      if (assignment) {
+        __cohortAssignment = { cohortId: assignment.cohortId, role: assignment.role };
+        if (assignment.role === "candidate" && assignment.personaId !== earlyPersona.id) {
+          const { resolveActivePersona } = await import("./personaProfiles.js");
+          const candidate = resolveActivePersona({ forceId: assignment.personaId });
+          if (candidate) {
+            logger.info(COMPONENT, `[Cohort:${assignment.cohortId}] swap ${earlyPersona.id} \u2192 ${candidate.id}`);
+            earlyPersona = candidate;
+          }
+        }
+      }
+    } catch (err) {
+      logger.warn(COMPONENT, `cohort assignment skipped: ${err.message}`);
+    }
+  }
   if (earlyPersona?.systemPromptAppendix) {
     enrichedSystemPrompt += earlyPersona.systemPromptAppendix;
     logger.info(COMPONENT, `[Persona:${earlyPersona.id}] system prompt appendix injected`);
@@ -1332,6 +1352,30 @@ Synthesize these results into a coherent response for the user.`
   recordUsage(session.id, providerName, modelUsed, totalPromptTokens, totalCompletionTokens);
   const durationMs = Date.now() - startTime;
   logger.info(COMPONENT, `Response generated in ${durationMs}ms (${totalPromptTokens + totalCompletionTokens} tokens)`);
+  if (__cohortAssignment) {
+    void (async () => {
+      try {
+        const { recordOutcome } = await import("./personaRollout.js");
+        let safetySat = null;
+        try {
+          const { loadDriveHistory } = await import("../organism/drives.js");
+          const persisted = loadDriveHistory();
+          const safety = persisted?.latest?.drives?.find((d) => d.id === "safety");
+          safetySat = safety ? safety.satisfaction : null;
+        } catch {
+        }
+        recordOutcome({
+          cohortId: __cohortAssignment.cohortId,
+          role: __cohortAssignment.role,
+          sessionId: session.id,
+          success: !budgetExhausted && finalContent.length > 0,
+          latencyMs: durationMs,
+          safetySat
+        });
+      } catch {
+      }
+    })();
+  }
   if (toolsUsed.length > 0) {
     const uniqueTools = [...new Set(toolsUsed)];
     learnFact(