@cfio/cohort-sync 0.2.2 → 0.2.3

package/dist/index.js

@@ -12766,6 +12766,18 @@ function registerHooks(api, cfg) {
     if (m && typeof m === "object" && "primary" in m) return String(m.primary);
     return "unknown";
   }
+  function getModelContextLimit(model) {
+    const m = model.toLowerCase();
+    if (m.includes("opus") || m.includes("sonnet") || m.includes("haiku")) return 2e5;
+    if (m.includes("gpt-4o")) return 128e3;
+    if (m.includes("gpt-4-turbo") || m.includes("gpt-4-1")) return 128e3;
+    if (m.includes("gpt-4")) return 8192;
+    if (m.includes("o3") || m.includes("o4-mini")) return 2e5;
+    if (m.includes("gemini-2")) return 1e6;
+    if (m.includes("gemini")) return 128e3;
+    if (m.includes("deepseek")) return 128e3;
+    return 2e5;
+  }
   async function pushHeartbeat() {
     const allAgentIds = ["main", ...(config?.agents?.list ?? []).map((a) => a.id)];
     for (const agentId of allAgentIds) {
@@ -12891,19 +12903,30 @@ function registerHooks(api, cfg) {
     }
   });
   api.on("llm_output", async (event, ctx) => {
-    diag("HOOK_llm_output", { ctx: dumpCtx(ctx), model: event.model, tokensIn: event.usage?.input, tokensOut: event.usage?.output });
+    const usage = event.usage ?? {};
+    const contextTokens = (usage.input ?? 0) + (usage.cacheRead ?? 0) + (usage.cacheWrite ?? 0);
+    const model = event.model ?? resolveModel(ctx.agentId ?? "main");
+    const contextLimit = getModelContextLimit(model);
+    diag("HOOK_llm_output", {
+      ctx: dumpCtx(ctx),
+      model,
+      tokensIn: usage.input,
+      tokensOut: usage.output,
+      cacheRead: usage.cacheRead,
+      cacheWrite: usage.cacheWrite,
+      contextTokens,
+      contextLimit
+    });
     const agentId = ctx.agentId ?? "main";
     const agentName = resolveAgentName(agentId);
     try {
-      const usage = event.usage ?? {};
-      const ext = event;
       const sessionKey = ctx.sessionKey;
       tracker.updateFromLlmOutput(agentName, sessionKey, {
-        model: event.model ?? resolveModel(agentId),
+        model,
         tokensIn: usage.input ?? 0,
         tokensOut: usage.output ?? 0,
-        contextTokens: ext.contextTokens ?? usage.total ?? 0,
-        contextLimit: ext.contextLimit ?? 0
+        contextTokens,
+        contextLimit
       });
       if (sessionKey && !tracker.hasSession(agentName, sessionKey)) {
         tracker.addSession(agentName, sessionKey);
@@ -12933,10 +12956,9 @@ function registerHooks(api, cfg) {
     const agentId = ctx.agentId ?? "main";
     const agentName = resolveAgentName(agentId);
     try {
-      const ext = event;
       tracker.updateFromCompaction(agentName, {
-        contextTokens: ext.contextTokens ?? event.tokenCount ?? 0,
-        contextLimit: ext.contextLimit ?? 0
+        contextTokens: event.tokenCount ?? 0,
+        contextLimit: getModelContextLimit(resolveModel(agentId))
       });
       if (tracker.shouldPushTelemetry(agentName)) {
         const snapshot = tracker.getTelemetrySnapshot(agentName);
@@ -12947,7 +12969,7 @@ function registerHooks(api, cfg) {
       }
       const entry = buildActivityEntry(agentName, "after_compaction", {
         messageCount: event.messageCount,
-        compactedCount: ext.compactedCount ?? event.compactedCount,
+        compactedCount: event.compactedCount,
         sessionKey: ctx.sessionKey
       });
       if (entry) addActivityToHot(entry);

package/dist/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cfio/cohort-sync",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "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.2",
+  "version": "0.2.3",
   "description": "Syncs agent status and skills to Cohort dashboard",
   "license": "MIT",
   "homepage": "https://docs.cohort.bot/gateway",