@cfio/cohort-sync 0.2.2 → 0.2.4

package/dist/index.js

@@ -11506,41 +11506,59 @@ var _systemSchema = defineSchema({
 function deriveConvexUrl(apiUrl) {
   return apiUrl.replace(/\.convex\.site\/?$/, ".convex.cloud");
 }
+var DEFAULT_BEHAVIORAL_PROMPT = `BEFORE RESPONDING:
+- Does your planned response address the task's stated scope? If not, do not comment.
+- Do not post acknowledgment-only responses ("got it", "sounds good", "confirmed"). If you have no new information to add, do not comment.
+- If the work is complete, transition the task to "waiting" and set noReply=true on your final comment, then stop working on this task.`;
 function buildNotificationMessage(n) {
   let header;
   let cta;
   switch (n.type) {
     case "comment":
       if (n.isMentioned) {
-        header = `\u{1F4AC} You were @mentioned on task #${n.taskNumber} "${n.taskTitle}"
+        header = `You were @mentioned on task #${n.taskNumber} "${n.taskTitle}"
 By: ${n.actorName}`;
         cta = "You were directly mentioned. Read the comment and respond using the cohort_comment tool.";
       } else {
-        header = `\u{1F4AC} New comment on task #${n.taskNumber} "${n.taskTitle}"
+        header = `New comment on task #${n.taskNumber} "${n.taskTitle}"
 From: ${n.actorName}`;
         cta = "Read the comment thread and respond using the cohort_comment tool if appropriate.";
       }
       break;
     case "assignment":
-      header = `\u{1F4CC} You were assigned to task #${n.taskNumber} "${n.taskTitle}"
+      header = `You were assigned to task #${n.taskNumber} "${n.taskTitle}"
 By: ${n.actorName}`;
       cta = "Review the task description and begin working on it.";
       break;
     case "status_change":
-      header = `\u{1F504} Status changed on task #${n.taskNumber} "${n.taskTitle}"
+      header = `Status changed on task #${n.taskNumber} "${n.taskTitle}"
 By: ${n.actorName}`;
       cta = "Review the status change and take any follow-up action needed.";
       break;
     default:
-      header = `\u{1F4CB} Notification on task #${n.taskNumber} "${n.taskTitle}"
+      header = `Notification on task #${n.taskNumber} "${n.taskTitle}"
 From: ${n.actorName}`;
       cta = "Check the task and respond if needed.";
   }
   const body = n.preview ? `
 Comment: "${n.preview}"` : "";
-  return `${header}${body}
+  let scope = "";
+  if (n.taskDescription && n.type === "comment") {
+    const truncated = n.taskDescription.length > 500 ? n.taskDescription.slice(0, 500) + "..." : n.taskDescription;
+    scope = `
 
-${cta}`;
+Scope: ${truncated}`;
+  }
+  const prompt = n.behavioralPrompt ? `${n.behavioralPrompt}
+
+${DEFAULT_BEHAVIORAL_PROMPT}` : DEFAULT_BEHAVIORAL_PROMPT;
+  const promptBlock = n.type === "comment" ? `
+
+---
+${prompt}` : "";
+  return `${header}${scope}${body}
+
+${cta}${promptBlock}`;
 }
 async function injectNotification(port, hooksToken, n, agentId = "main") {
   const response = await fetch(`http://localhost:${port}/hooks/agent`, {
@@ -11566,6 +11584,9 @@ var markDeliveredByPlugin = makeFunctionReference("notifications:markDeliveredBy
 var upsertTelemetryFromPlugin = makeFunctionReference("telemetryPlugin:upsertTelemetryFromPlugin");
 var upsertSessionsFromPlugin = makeFunctionReference("telemetryPlugin:upsertSessionsFromPlugin");
 var pushActivityFromPluginRef = makeFunctionReference("activityFeed:pushActivityFromPlugin");
+var getPendingCommandsForPlugin = makeFunctionReference("gatewayCommands:getPendingForPlugin");
+var acknowledgeCommandRef = makeFunctionReference("gatewayCommands:acknowledgeCommand");
+var addCommentFromPluginRef = makeFunctionReference("comments:addCommentFromPlugin");
 var HOT_KEY = "__cohort_sync__";
 function getHotState() {
   let state = globalThis[HOT_KEY];
@@ -11705,6 +11726,61 @@ async function initSubscription(port, cfg, hooksToken, logger) {
   }
   state.unsubscribers = [...unsubscribers];
 }
+function initCommandSubscription(cfg, logger) {
+  const c = getOrCreateClient();
+  if (!c) {
+    logger.warn("cohort-sync: no ConvexClient \u2014 command subscription skipped");
+    return;
+  }
+  let processing = false;
+  const unsubscribe = c.onUpdate(
+    getPendingCommandsForPlugin,
+    { apiKey: cfg.apiKey },
+    async (commands) => {
+      if (processing) return;
+      if (commands.length === 0) return;
+      processing = true;
+      const cmd = commands[commands.length - 1];
+      logger.info(`cohort-sync: received gateway command: ${cmd.type} (${cmd._id})`);
+      if (cmd.type === "restart") {
+        try {
+          await c.mutation(acknowledgeCommandRef, {
+            commandId: cmd._id,
+            apiKey: cfg.apiKey
+          });
+          logger.info("cohort-sync: restart command acknowledged, terminating in 500ms");
+          await new Promise((r) => setTimeout(r, 500));
+          process.kill(process.pid, "SIGTERM");
+        } catch (err) {
+          logger.error(`cohort-sync: failed to process restart command: ${String(err)}`);
+          processing = false;
+        }
+      } else {
+        logger.warn(`cohort-sync: unknown command type: ${cmd.type}`);
+        processing = false;
+      }
+    },
+    (err) => {
+      logger.error(`cohort-sync: command subscription error: ${String(err)}`);
+    }
+  );
+  unsubscribers.push(unsubscribe);
+  getHotState().unsubscribers = [...unsubscribers];
+  logger.info("cohort-sync: command subscription active");
+}
+async function callAddCommentFromPlugin(apiKey, args) {
+  const c = getOrCreateClient();
+  if (!c) {
+    throw new Error("Convex client not initialized \u2014 subscription may not be active");
+  }
+  return await c.mutation(addCommentFromPluginRef, {
+    apiKey,
+    taskNumber: args.taskNumber,
+    agentName: args.agentName,
+    content: args.content,
+    noReply: args.noReply
+  });
+}
 function closeSubscription() {
   for (const unsub of unsubscribers) {
     try {
@@ -12731,31 +12807,75 @@ function registerHooks(api, cfg) {
       description: "Post a comment on a Cohort task. Use this to respond to @mentions or collaborate on tasks.",
       parameters: Type.Object({
         task_number: Type.Number({ description: "Task number (e.g. 312)" }),
-        comment: Type.String({ description: "Comment text to post" })
+        comment: Type.String({ description: "Comment text to post" }),
+        no_reply: Type.Optional(Type.Boolean({
+          description: "If true, no notifications will be sent for this comment. Use for final/closing comments."
+        }))
       }),
       async execute(_toolCallId, params) {
-        const apiUrl = cfg.apiUrl.replace(/\/+$/, "");
-        const res = await fetch(
-          `${apiUrl}/api/v1/tasks/${params.task_number}/comments`,
-          {
-            method: "POST",
-            headers: {
-              Authorization: `Bearer ${cfg.apiKey}`,
-              "Content-Type": "application/json"
-            },
-            body: JSON.stringify({ body: params.comment }),
-            signal: AbortSignal.timeout(1e4)
+        try {
+          const result = await callAddCommentFromPlugin(cfg.apiKey, {
+            taskNumber: params.task_number,
+            agentName,
+            content: params.comment,
+            noReply: params.no_reply ?? false
+          });
+          const lines = [`Comment posted on task #${params.task_number}.`];
+          if (result.stats) {
+            lines.push("");
+            lines.push(`This task has ${result.stats.totalComments} comments. ${result.stats.myRecentCount}/${result.stats.threshold} hourly limit used on this task.`);
           }
-        );
-        if (!res.ok) {
-          const text = await res.text().catch(() => "");
-          throw new Error(`POST /api/v1/tasks/${params.task_number}/comments \u2192 ${res.status}: ${text}`);
+          if (result.budget) {
+            lines.push(`Daily budget: ${result.budget.used}/${result.budget.limit}`);
+          }
+          return {
+            content: [{ type: "text", text: lines.join("\n") }],
+            details: result
+          };
+        } catch (err) {
+          const msg = err instanceof Error ? err.message : String(err);
+          if (msg.includes("AGENT_COMMENTS_LOCKED")) {
+            return {
+              content: [{
+                type: "text",
+                text: `Cannot comment on task #${params.task_number}.
+Reason: Agent comments are locked on this task.
+Do not re-attempt to comment on this task.`
+              }],
+              details: { error: "AGENT_COMMENTS_LOCKED", taskNumber: params.task_number }
+            };
+          }
+          if (msg.includes("TASK_HOUR_LIMIT_REACHED")) {
+            const parts = msg.split("|");
+            const count = parts[1] ?? "?";
+            const limit = parts[2] ?? "?";
+            return {
+              content: [{
+                type: "text",
+                text: `Cannot comment on task #${params.task_number}.
+Reason: You have posted ${count} comments on this task in the last hour (limit: ${limit}).
+Step back from this task. Do not comment again until the next hour.`
+              }],
+              details: { error: "TASK_HOUR_LIMIT_REACHED", count, limit, taskNumber: params.task_number }
+            };
+          }
+          if (msg.includes("DAILY_LIMIT_REACHED")) {
+            const parts = msg.split("|");
+            const used = parts[1] ?? "?";
+            const max = parts[2] ?? "?";
+            const resetAt = parts[3] ?? "tomorrow";
+            return {
+              content: [{
+                type: "text",
+                text: `Cannot comment on task #${params.task_number}.
+Reason: Daily comment limit reached (${used}/${max}).
+Do not attempt to make more comments until ${resetAt}.`
+              }],
+              details: { error: "DAILY_LIMIT_REACHED", used, max, resetAt, taskNumber: params.task_number }
+            };
+          }
+          throw err;
         }
-        const result = await res.json();
-        return {
-          content: [{ type: "text", text: `Comment posted on task #${params.task_number}` }],
-          details: result
-        };
       }
     };
   });
@@ -12766,6 +12886,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) {
@@ -12846,6 +12978,7 @@ function registerHooks(api, cfg) {
     ).catch((err) => {
       logger.error(`cohort-sync: subscription init failed: ${String(err)}`);
     });
+    initCommandSubscription(cfg, logger);
     const allAgentIds = ["main", ...(config?.agents?.list ?? []).map((a) => a.id)];
     for (const agentId of allAgentIds) {
       const agentName = resolveAgentName(agentId);
@@ -12891,19 +13024,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 +13077,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 +13090,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.4",
   "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.4",
   "description": "Syncs agent status and skills to Cohort dashboard",
   "license": "MIT",
   "homepage": "https://docs.cohort.bot/gateway",