@corbat-tech/coco 2.27.5 → 2.28.0

package/dist/index.js

@@ -16483,6 +16483,17 @@ function extractSseEventData(rawEvent) {
   }
   return dataLines.join("\n");
 }
+function getToolCallFingerprint(part) {
+  const name = part.functionCall?.name ?? "unknown_function";
+  let argsSerialized = "{}";
+  try {
+    argsSerialized = JSON.stringify(part.functionCall?.args ?? {});
+  } catch {
+    argsSerialized = "{}";
+  }
+  const thoughtSignature = part.thoughtSignature ?? part.thought_signature ?? part.functionCall?.thoughtSignature ?? part.functionCall?.thought_signature ?? "";
+  return `${name}:${argsSerialized}:${thoughtSignature}`;
+}
 var VertexProvider = class {
   id = "vertex";
   name = "Google Vertex AI Gemini";
@@ -16558,6 +16569,7 @@ var VertexProvider = class {
     );
     let stopReason = "end_turn";
     let streamToolCallCounter = 0;
+    const emittedToolFingerprints = /* @__PURE__ */ new Set();
     for await (const chunk of stream) {
       const candidate = chunk.candidates?.[0];
       const parts = candidate?.content?.parts ?? [];
@@ -16566,6 +16578,11 @@ var VertexProvider = class {
           yield { type: "text", text: part.text };
         }
         if (part.functionCall) {
+          const fingerprint = getToolCallFingerprint(part);
+          if (emittedToolFingerprints.has(fingerprint)) {
+            continue;
+          }
+          emittedToolFingerprints.add(fingerprint);
           streamToolCallCounter++;
           const geminiThoughtSignature = part.thoughtSignature ?? part.thought_signature ?? part.functionCall.thoughtSignature ?? part.functionCall.thought_signature;
           yield {
@@ -28526,6 +28543,25 @@ var MCPServerManager = class _MCPServerManager {
   connections = /* @__PURE__ */ new Map();
   logger = getLogger();
   static STOP_TIMEOUT_MS = 5e3;
+  /**
+   * Run an async operation with a timeout, always clearing timer resources.
+   */
+  async runWithTimeout(operation, timeoutMs, timeoutMessage) {
+    let timeoutId;
+    const timeoutPromise = new Promise((_, reject) => {
+      timeoutId = setTimeout(() => reject(new Error(timeoutMessage)), timeoutMs);
+      if (typeof timeoutId.unref === "function") {
+        timeoutId.unref();
+      }
+    });
+    try {
+      return await Promise.race([operation, timeoutPromise]);
+    } finally {
+      if (timeoutId) {
+        clearTimeout(timeoutId);
+      }
+    }
+  }
   /**
    * Create transport for a server config
    */
@@ -28612,15 +28648,11 @@ var MCPServerManager = class _MCPServerManager {
     }
     this.logger.info(`Stopping MCP server: ${name}`);
     try {
-      await Promise.race([
+      await this.runWithTimeout(
         connection.transport.disconnect(),
-        new Promise(
-          (_, reject) => setTimeout(
-            () => reject(new Error("MCP disconnect timeout")),
-            _MCPServerManager.STOP_TIMEOUT_MS
-          )
-        )
-      ]);
+        _MCPServerManager.STOP_TIMEOUT_MS,
+        "MCP disconnect timeout"
+      );
     } catch (error) {
       this.logger.error(
         `Error disconnecting server '${name}': ${error instanceof Error ? error.message : String(error)}`
@@ -28657,12 +28689,11 @@ var MCPServerManager = class _MCPServerManager {
     }
     const startTime = performance.now();
     try {
-      const { tools } = await Promise.race([
+      const { tools } = await this.runWithTimeout(
         connection.client.listTools(),
-        new Promise(
-          (_, reject) => setTimeout(() => reject(new Error("Health check timeout")), 5e3)
-        )
-      ]);
+        5e3,
+        "Health check timeout"
+      );
       const latencyMs = performance.now() - startTime;
       connection.healthy = true;
       connection.toolCount = tools.length;