Merge pull request #55 from runbasehq/dev

Release/0.4.1

Author: PineAppleGrits

README.md

@@ -2,7 +2,7 @@
 ⚠️ It's not usable yet, but docs and a stable release are coming in the next few weeks. Stay tuned.
 
 <p align="center">
-  <a href="https://x.com/fveiras_">
+  <a href="https://www.npmjs.com/package/mcp-check">
   <img width="1033" height="204" alt="ascii-art-image (1)" src="https://github.com/user-attachments/assets/e77a1383-c0af-4c04-92f6-ed4b941043a3" />
   </a>
 </p>

packages/mcp-check/CHANGELOG.md

@@ -1,5 +1,11 @@
 # mcp-testing-library
 
+## 0.4.1
+
+### Patch Changes
+
+- fix chunk duplication
+
 ## 0.4.0
 
 ### Minor Changes

packages/mcp-check/package.json

@@ -1,7 +1,7 @@
 {
   "name": "mcp-check",
   "module": "dist/src/index.js",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "type": "module",
   "main": "dist/src/index.js",
   "types": "dist/src/index.d.ts",

packages/mcp-check/src/agents/client.ts

@@ -31,7 +31,7 @@ export class AgentsClient<T extends ModelName = ModelName> {
   constructor(mcpServer: McpServer, models: T[], config: ProviderConfig = {}) {
     this.models = models;
     this.mcpServer = mcpServer;
-    this.config = config;
+    this.config = { silent: true, ...config };
   }
 
   /**

packages/mcp-check/src/index.ts

@@ -295,7 +295,7 @@ export class AgentsClient<T extends ModelName = ModelName> {
     this.models = models;
     this.mcpServer = mcpServer;
     this._scorers = scorers || [];
-    this.config = config;
+    this.config = { silent: true, ...config };
   }
 
   prompt(text: string): Promise<AgentsResult<T>> {

packages/mcp-check/src/providers/anthropic.ts

@@ -368,33 +368,7 @@ export class AnthropicProvider extends Provider {
   protected async processNormalizedChunk(
     normalizedChunk: NormalizedChunk,
   ): Promise<void> {
-    // Handle Anthropic-specific chunk types
-    if (normalizedChunk.originalChunk?.type === "content_block_delta") {
-      const handlers = this.getChunkHandlers();
-      if (handlers.anthropic?.onContentBlockDelta) {
-        await handlers.anthropic.onContentBlockDelta(
-          normalizedChunk.originalChunk,
-        );
-      }
-    }
-
-    if (normalizedChunk.originalChunk?.type === "content_block_start") {
-      const handlers = this.getChunkHandlers();
-      if (handlers.anthropic?.onContentBlockStart) {
-        await handlers.anthropic.onContentBlockStart(
-          normalizedChunk.originalChunk,
-        );
-      }
-    }
-
-    if (normalizedChunk.originalChunk?.type === "content_block_stop") {
-      const handlers = this.getChunkHandlers();
-      if (handlers.anthropic?.onContentBlockStop) {
-        await handlers.anthropic.onContentBlockStop(
-          normalizedChunk.originalChunk,
-        );
-      }
-    }
+    // Provider-specific handlers are invoked centrally by the normalizer
 
     // Handle text delta accumulation
     if (normalizedChunk.type === "text_delta") {

packages/mcp-check/src/providers/openai.ts

@@ -277,14 +277,10 @@ export class OpenAIProvider extends Provider {
   protected async processNormalizedChunk(
     normalizedChunk: NormalizedChunk,
   ): Promise<void> {
-    await super.processNormalizedChunk(normalizedChunk);
-
-    // Backward compatibility
+    // Backward compatibility: keep logs but avoid duplicating state updates
     if (normalizedChunk.type === "tool_call_start") {
       const toolName = normalizedChunk.data.toolName;
       if (toolName) {
-        this.usedTools.push(toolName);
-
         if (!this.config.silent) {
           process.stdout.write(
             JSON.stringify({
@@ -294,55 +290,21 @@ export class OpenAIProvider extends Provider {
             }) + "\n"
           );
         }
-
-        if (!this.toolCalls[toolName]) {
-          this.toolCalls[toolName] = [];
-        }
-
-        this.toolCalls[toolName].push({
-          args: normalizedChunk.data.toolArgs || {},
-          result: null,
-        });
       }
     } else if (normalizedChunk.type === "tool_call_done") {
       const toolName = normalizedChunk.data.toolName;
-      if (
-        toolName &&
-        this.toolCalls[toolName] &&
-        this.toolCalls[toolName].length > 0
-      ) {
-        const lastCall =
-          this.toolCalls[toolName][this.toolCalls[toolName].length - 1];
-        if (lastCall) {
-          lastCall.result = { id: `result_${Date.now()}`, status: "completed" };
-
-          if (!this.config.silent) {
-            process.stdout.write(
-              JSON.stringify({
-                type: "model_stream",
-                model: this.currentModel,
-                text: `Tool ${toolName} completed\n`,
-              }) + "\n"
-            );
-          }
-        }
+      if (!this.config.silent && toolName) {
+        process.stdout.write(
+          JSON.stringify({
+            type: "model_stream",
+            model: this.currentModel,
+            text: `Tool ${toolName} completed\n`,
+          }) + "\n"
+        );
       }
     }
 
-    // Handle OpenAI-specific chunk types
-    if (normalizedChunk.originalChunk?.type === "response.output_item.added") {
-      const handlers = this.getChunkHandlers();
-      if (handlers.openai?.onResponseOutputItemAdded) {
-        await handlers.openai.onResponseOutputItemAdded(normalizedChunk.originalChunk);
-      }
-    }
-
-    if (normalizedChunk.originalChunk?.type === "response.output_item.done") {
-      const handlers = this.getChunkHandlers();
-      if (handlers.openai?.onResponseOutputItemDone) {
-        await handlers.openai.onResponseOutputItemDone(normalizedChunk.originalChunk);
-      }
-    }
+    // Provider-specific handlers are invoked centrally by the normalizer
 
     // Handle text delta accumulation
     if (normalizedChunk.type === "text_delta") {

packages/mcp-check/src/providers/provider.ts

@@ -48,7 +48,7 @@ export abstract class Provider {
   constructor(mcpServer: McpServer, promptText: string, config: ProviderConfig = {}) {
     this.mcpServer = mcpServer;
     this.promptText = promptText;
-    this.config = config;
+    this.config = { silent: true, ...config };
   }
 
   /**