@cfio/cohort-sync 0.2.0 → 0.2.1

package/dist/index.js

@@ -11748,6 +11748,19 @@ function saveAgentStatesToHot(states) {
 function getAgentStatesFromHot() {
   return getHotState()?.agentStates ?? null;
 }
+function saveIntervalsToHot(heartbeat, activityFlush) {
+  const hot = getHotState();
+  if (hot) {
+    hot.intervals = { heartbeat, activityFlush };
+  }
+}
+function clearIntervalsFromHot() {
+  const hot = getHotState();
+  if (!hot?.intervals) return;
+  if (hot.intervals.heartbeat) clearInterval(hot.intervals.heartbeat);
+  if (hot.intervals.activityFlush) clearInterval(hot.intervals.activityFlush);
+  hot.intervals = { heartbeat: null, activityFlush: null };
+}
 
 // src/sync.ts
 function extractJson(raw) {
@@ -12014,6 +12027,7 @@ function shouldPushTelemetry(current, lastPushed) {
   if (!lastPushed) return true;
   if (current.status !== lastPushed.status) return true;
   if (current.compactions !== lastPushed.compactions) return true;
+  if (current.activeSessions !== lastPushed.activeSessions) return true;
   if (current.contextLimit > 0 && lastPushed.contextLimit > 0) {
     const currentPct = current.contextTokens / current.contextLimit;
     const lastPct = lastPushed.contextTokens / lastPushed.contextLimit;
@@ -12101,6 +12115,13 @@ function buildActivitySentence(agentName, event, context) {
 var AgentStateTracker = class {
   agents = /* @__PURE__ */ new Map();
   activityBuffer = [];
+  sessionKeyToAgent = /* @__PURE__ */ new Map();
+  setSessionAgent(sessionKey, agentName) {
+    this.sessionKeyToAgent.set(sessionKey, agentName);
+  }
+  getSessionAgent(sessionKey) {
+    return this.sessionKeyToAgent.get(sessionKey) ?? null;
+  }
   getOrCreate(agentName) {
     let state = this.agents.get(agentName);
     if (!state) {
@@ -12142,6 +12163,25 @@ var AgentStateTracker = class {
       state.sessions.delete(sessionKey);
     }
   }
+  hasSession(agentName, sessionKey) {
+    const state = this.agents.get(agentName);
+    if (!state) return false;
+    return state.sessions.has(sessionKey);
+  }
+  pruneStaleSessions(agentName, maxAgeMs) {
+    const state = this.agents.get(agentName);
+    if (!state) return [];
+    const now = Date.now();
+    const pruned = [];
+    for (const [key, session] of state.sessions) {
+      const lastMs = new Date(session.lastActivity).getTime();
+      if (now - lastMs > maxAgeMs) {
+        state.sessions.delete(key);
+        pruned.push(key);
+      }
+    }
+    return pruned;
+  }
   // --- Telemetry updates ---
   updateStatus(agentName, status) {
     this.getOrCreate(agentName).status = status;
@@ -12243,17 +12283,27 @@ var AgentStateTracker = class {
   clear() {
     this.agents.clear();
     this.activityBuffer = [];
+    this.sessionKeyToAgent.clear();
   }
   exportState() {
-    return new Map(this.agents);
+    return {
+      agents: new Map(this.agents),
+      sessionKeyToAgent: new Map(this.sessionKeyToAgent)
+    };
   }
   importState(states) {
-    for (const [name, state] of states) {
+    const agentMap = states instanceof Map ? states : states.agents;
+    for (const [name, state] of agentMap) {
       if (!(state.sessions instanceof Map)) {
         state.sessions = new Map(Object.entries(state.sessions));
       }
       this.agents.set(name, state);
     }
+    if (!(states instanceof Map) && states.sessionKeyToAgent) {
+      for (const [key, agent] of states.sessionKeyToAgent) {
+        this.sessionKeyToAgent.set(key, agent);
+      }
+    }
   }
 };
 
@@ -12315,6 +12365,30 @@ function registerHooks(api, cfg) {
   function resolveAgentName(agentId) {
     return (nameMap?.[agentId] ?? agentId).toLowerCase();
   }
+  function resolveAgentFromContext(ctx) {
+    if (ctx.agentId && typeof ctx.agentId === "string") {
+      return resolveAgentName(ctx.agentId);
+    }
+    const sessionKey = ctx.sessionKey ?? ctx.sessionId;
+    if (sessionKey && typeof sessionKey === "string") {
+      const mapped = tracker.getSessionAgent(sessionKey);
+      if (mapped) return mapped;
+    }
+    return resolveAgentName("main");
+  }
+  clearIntervalsFromHot();
+  heartbeatInterval = setInterval(() => {
+    pushHeartbeat().catch((err) => {
+      logger.warn(`cohort-sync: heartbeat tick failed: ${String(err)}`);
+    });
+  }, 12e4);
+  activityFlushInterval = setInterval(() => {
+    flushActivityBuffer().catch((err) => {
+      logger.warn(`cohort-sync: activity flush tick failed: ${String(err)}`);
+    });
+  }, 1500);
+  saveIntervalsToHot(heartbeatInterval, activityFlushInterval);
+  logger.info("cohort-sync: intervals created (heartbeat=2m, activityFlush=1.5s)");
   api.registerTool((toolCtx) => {
     const agentId = toolCtx.agentId ?? "main";
     const agentName = resolveAgentName(agentId);
@@ -12361,6 +12435,13 @@ function registerHooks(api, cfg) {
   }
   async function pushHeartbeat() {
     const allAgentIds = ["main", ...(config?.agents?.list ?? []).map((a) => a.id)];
+    for (const agentId of allAgentIds) {
+      const agentName = resolveAgentName(agentId);
+      const pruned = tracker.pruneStaleSessions(agentName, 6e5);
+      if (pruned.length > 0) {
+        logger.info(`cohort-sync: pruned ${pruned.length} stale sessions for ${agentName}`);
+      }
+    }
     for (const agentId of allAgentIds) {
       const agentName = resolveAgentName(agentId);
       try {
@@ -12374,6 +12455,18 @@ function registerHooks(api, cfg) {
         logger.warn(`cohort-sync: heartbeat push failed for ${agentName}: ${String(err)}`);
       }
     }
+    for (const agentId of allAgentIds) {
+      const agentName = resolveAgentName(agentId);
+      try {
+        if (tracker.shouldPushSessions(agentName)) {
+          const sessSnapshot = tracker.getSessionsSnapshot(agentName);
+          await pushSessions(cfg.apiKey, agentName, sessSnapshot);
+          tracker.markSessionsPushed(agentName);
+        }
+      } catch (err) {
+        logger.warn(`cohort-sync: heartbeat session push failed for ${agentName}: ${String(err)}`);
+      }
+    }
     logger.info(`cohort-sync: heartbeat pushed for ${allAgentIds.length} agents`);
   }
   async function flushActivityBuffer() {
@@ -12425,18 +12518,6 @@ function registerHooks(api, cfg) {
       }
     }
     logger.info(`cohort-sync: seeded telemetry for ${allAgentIds.length} agents`);
-    heartbeatInterval = setInterval(() => {
-      pushHeartbeat().catch((err) => {
-        logger.warn(`cohort-sync: heartbeat tick failed: ${String(err)}`);
-      });
-    }, 12e4);
-    logger.info("cohort-sync: heartbeat started (2m interval)");
-    activityFlushInterval = setInterval(() => {
-      flushActivityBuffer().catch((err) => {
-        logger.warn(`cohort-sync: activity flush tick failed: ${String(err)}`);
-      });
-    }, 1500);
-    logger.info("cohort-sync: activity flush started (1.5s interval)");
   });
   api.on("agent_end", async (_event, ctx) => {
     const agentId = ctx.agentId ?? "main";
@@ -12469,6 +12550,13 @@ function registerHooks(api, cfg) {
         contextTokens: ext.contextTokens ?? usage.total ?? 0,
         contextLimit: ext.contextLimit ?? 0
       });
+      if (sessionKey && !tracker.hasSession(agentName, sessionKey)) {
+        tracker.addSession(agentName, sessionKey);
+        logger.info(`cohort-sync: inferred session for ${agentName} from llm_output (${sessionKey})`);
+      }
+      if (sessionKey) {
+        tracker.setSessionAgent(sessionKey, agentName);
+      }
       if (tracker.shouldPushTelemetry(agentName)) {
         const snapshot = tracker.getTelemetrySnapshot(agentName);
         if (snapshot) {
@@ -12522,6 +12610,17 @@ function registerHooks(api, cfg) {
           tracker.markTelemetryPushed(agentName);
         }
       }
+      const sessionKey = ctx.sessionKey;
+      if (sessionKey && !tracker.hasSession(agentName, sessionKey)) {
+        tracker.addSession(agentName, sessionKey);
+        tracker.setSessionAgent(sessionKey, agentName);
+        logger.info(`cohort-sync: inferred session for ${agentName} (${sessionKey})`);
+        if (tracker.shouldPushSessions(agentName)) {
+          const sessSnapshot = tracker.getSessionsSnapshot(agentName);
+          await pushSessions(cfg.apiKey, agentName, sessSnapshot);
+          tracker.markSessionsPushed(agentName);
+        }
+      }
       const entry = buildActivitySentence(agentName, "before_agent_start", {
         channel: ctx.messageProvider ?? "unknown"
       });
@@ -12568,8 +12667,7 @@ function registerHooks(api, cfg) {
     }
   });
   api.on("before_tool_call", async (event, ctx) => {
-    const agentId = ctx.agentId ?? "main";
-    const agentName = resolveAgentName(agentId);
+    const agentName = resolveAgentFromContext(ctx);
     try {
       const entry = buildActivitySentence(agentName, "before_tool_call", {
         toolName: event.toolName ?? event.name,
@@ -12581,8 +12679,7 @@ function registerHooks(api, cfg) {
     }
   });
   api.on("message_received", async (_event, ctx) => {
-    const agentId = ctx.agentId ?? "main";
-    const agentName = resolveAgentName(agentId);
+    const agentName = resolveAgentFromContext(ctx);
     try {
       const entry = buildActivitySentence(agentName, "message_received", {
         channel: ctx.messageProvider ?? "unknown"
@@ -12593,8 +12690,7 @@ function registerHooks(api, cfg) {
     }
   });
   api.on("message_sent", async (_event, ctx) => {
-    const agentId = ctx.agentId ?? "main";
-    const agentName = resolveAgentName(agentId);
+    const agentName = resolveAgentFromContext(ctx);
     try {
       const entry = buildActivitySentence(agentName, "message_sent", {
         channel: ctx.messageProvider ?? "unknown"
@@ -12605,14 +12701,9 @@ function registerHooks(api, cfg) {
     }
   });
   api.on("gateway_stop", async () => {
-    if (heartbeatInterval) {
-      clearInterval(heartbeatInterval);
-      heartbeatInterval = null;
-    }
-    if (activityFlushInterval) {
-      clearInterval(activityFlushInterval);
-      activityFlushInterval = null;
-    }
+    clearIntervalsFromHot();
+    heartbeatInterval = null;
+    activityFlushInterval = null;
     await flushActivityBuffer().catch((err) => {
       logger.warn(`cohort-sync: final activity flush failed: ${String(err)}`);
     });

package/dist/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cfio/cohort-sync",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Syncs agent status and skills to Cohort dashboard",
   "type": "module",
   "main": "index.js",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cfio/cohort-sync",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Syncs agent status and skills to Cohort dashboard",
   "license": "MIT",
   "homepage": "https://docs.cohort.bot/gateway",