@probelabs/probe 0.6.0-rc250 → 0.6.0-rc251

package/build/agent/ProbeAgent.js

@@ -4383,7 +4383,8 @@ Convert your previous response content into actual JSON data that follows this s
           // Call answer recursively with _schemaFormatted flag to prevent infinite loop
           finalResult = await this.answer(schemaPrompt, [], {
             ...options,
-            _schemaFormatted: true
+            _schemaFormatted: true,
+            _completionPromptProcessed: true  // Prevent cascading completion prompts in retry calls
           });
 
           // Step 2: Validate and fix Mermaid diagrams if present (BEFORE cleaning schema)
@@ -4642,7 +4643,8 @@ Convert your previous response content into actual JSON data that follows this s
               finalResult = await this.answer(schemaDefinitionPrompt, [], {
                 ...options,
                 _schemaFormatted: true,
-                _skipValidation: true  // Skip validation in recursive correction calls to prevent loops
+                _skipValidation: true,  // Skip validation in recursive correction calls to prevent loops
+                _completionPromptProcessed: true  // Prevent cascading completion prompts in retry calls
               });
               finalResult = cleanSchemaResponse(finalResult);
               validation = validateJsonResponse(finalResult);
@@ -4702,7 +4704,8 @@ Convert your previous response content into actual JSON data that follows this s
                 ...options,
                 _schemaFormatted: true,
                 _skipValidation: true,  // Skip validation in recursive correction calls to prevent loops
-                _disableTools: true     // Only allow attempt_completion - prevent AI from using search/query tools
+                _disableTools: true,    // Only allow attempt_completion - prevent AI from using search/query tools
+                _completionPromptProcessed: true  // Prevent cascading completion prompts in retry calls
               });
               finalResult = cleanSchemaResponse(finalResult);
               
@@ -4787,7 +4790,8 @@ Convert your previous response content into actual JSON data that follows this s
       }
 
       // Append DSL output buffer directly to response (bypasses LLM rewriting)
-      if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted) {
+      // Skip during _completionPromptProcessed — only the parent answer() should append the buffer.
+      if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted && !options._completionPromptProcessed) {
         const outputContent = this._outputBuffer.items.join('\n\n');
         if (options.schema) {
           // Schema response — the finalResult is JSON. Wrap output in RAW_OUTPUT

package/build/agent/index.js

@@ -84647,7 +84647,9 @@ NOT: {"type": "object", "properties": {"name": {"type": "string"}}}
 Convert your previous response content into actual JSON data that follows this schema structure.`;
               finalResult = await this.answer(schemaPrompt, [], {
                 ...options,
-                _schemaFormatted: true
+                _schemaFormatted: true,
+                _completionPromptProcessed: true
+                // Prevent cascading completion prompts in retry calls
               });
               if (!this.disableMermaidValidation) {
                 try {
@@ -84851,8 +84853,10 @@ Convert your previous response content into actual JSON data that follows this s
                   finalResult = await this.answer(schemaDefinitionPrompt, [], {
                     ...options,
                     _schemaFormatted: true,
-                    _skipValidation: true
+                    _skipValidation: true,
                     // Skip validation in recursive correction calls to prevent loops
+                    _completionPromptProcessed: true
+                    // Prevent cascading completion prompts in retry calls
                   });
                   finalResult = cleanSchemaResponse(finalResult);
                   validation = validateJsonResponse(finalResult);
@@ -84905,8 +84909,10 @@ Convert your previous response content into actual JSON data that follows this s
                     _schemaFormatted: true,
                     _skipValidation: true,
                     // Skip validation in recursive correction calls to prevent loops
-                    _disableTools: true
+                    _disableTools: true,
                     // Only allow attempt_completion - prevent AI from using search/query tools
+                    _completionPromptProcessed: true
+                    // Prevent cascading completion prompts in retry calls
                   });
                   finalResult = cleanSchemaResponse(finalResult);
                   validation = validateJsonResponse(finalResult, { debug: this.debug });
@@ -84976,7 +84982,7 @@ Convert your previous response content into actual JSON data that follows this s
               console.log(`[DEBUG] Removed thinking tags from final result`);
             }
           }
-          if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted) {
+          if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted && !options._completionPromptProcessed) {
             const outputContent = this._outputBuffer.items.join("\n\n");
             if (options.schema) {
               finalResult = (finalResult || "") + "\n<<<RAW_OUTPUT>>>\n" + outputContent + "\n<<<END_RAW_OUTPUT>>>";

package/cjs/agent/ProbeAgent.cjs

@@ -111346,7 +111346,9 @@ NOT: {"type": "object", "properties": {"name": {"type": "string"}}}
 Convert your previous response content into actual JSON data that follows this schema structure.`;
               finalResult = await this.answer(schemaPrompt, [], {
                 ...options,
-                _schemaFormatted: true
+                _schemaFormatted: true,
+                _completionPromptProcessed: true
+                // Prevent cascading completion prompts in retry calls
               });
               if (!this.disableMermaidValidation) {
                 try {
@@ -111550,8 +111552,10 @@ Convert your previous response content into actual JSON data that follows this s
                   finalResult = await this.answer(schemaDefinitionPrompt, [], {
                     ...options,
                     _schemaFormatted: true,
-                    _skipValidation: true
+                    _skipValidation: true,
                     // Skip validation in recursive correction calls to prevent loops
+                    _completionPromptProcessed: true
+                    // Prevent cascading completion prompts in retry calls
                   });
                   finalResult = cleanSchemaResponse(finalResult);
                   validation = validateJsonResponse(finalResult);
@@ -111604,8 +111608,10 @@ Convert your previous response content into actual JSON data that follows this s
                     _schemaFormatted: true,
                     _skipValidation: true,
                     // Skip validation in recursive correction calls to prevent loops
-                    _disableTools: true
+                    _disableTools: true,
                     // Only allow attempt_completion - prevent AI from using search/query tools
+                    _completionPromptProcessed: true
+                    // Prevent cascading completion prompts in retry calls
                   });
                   finalResult = cleanSchemaResponse(finalResult);
                   validation = validateJsonResponse(finalResult, { debug: this.debug });
@@ -111675,7 +111681,7 @@ Convert your previous response content into actual JSON data that follows this s
               console.log(`[DEBUG] Removed thinking tags from final result`);
             }
           }
-          if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted) {
+          if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted && !options._completionPromptProcessed) {
             const outputContent = this._outputBuffer.items.join("\n\n");
             if (options.schema) {
               finalResult = (finalResult || "") + "\n<<<RAW_OUTPUT>>>\n" + outputContent + "\n<<<END_RAW_OUTPUT>>>";

package/cjs/index.cjs

@@ -109699,7 +109699,9 @@ NOT: {"type": "object", "properties": {"name": {"type": "string"}}}
 Convert your previous response content into actual JSON data that follows this schema structure.`;
               finalResult = await this.answer(schemaPrompt, [], {
                 ...options,
-                _schemaFormatted: true
+                _schemaFormatted: true,
+                _completionPromptProcessed: true
+                // Prevent cascading completion prompts in retry calls
               });
               if (!this.disableMermaidValidation) {
                 try {
@@ -109903,8 +109905,10 @@ Convert your previous response content into actual JSON data that follows this s
                   finalResult = await this.answer(schemaDefinitionPrompt, [], {
                     ...options,
                     _schemaFormatted: true,
-                    _skipValidation: true
+                    _skipValidation: true,
                     // Skip validation in recursive correction calls to prevent loops
+                    _completionPromptProcessed: true
+                    // Prevent cascading completion prompts in retry calls
                   });
                   finalResult = cleanSchemaResponse(finalResult);
                   validation = validateJsonResponse(finalResult);
@@ -109957,8 +109961,10 @@ Convert your previous response content into actual JSON data that follows this s
                     _schemaFormatted: true,
                     _skipValidation: true,
                     // Skip validation in recursive correction calls to prevent loops
-                    _disableTools: true
+                    _disableTools: true,
                     // Only allow attempt_completion - prevent AI from using search/query tools
+                    _completionPromptProcessed: true
+                    // Prevent cascading completion prompts in retry calls
                   });
                   finalResult = cleanSchemaResponse(finalResult);
                   validation = validateJsonResponse(finalResult, { debug: this.debug });
@@ -110028,7 +110034,7 @@ Convert your previous response content into actual JSON data that follows this s
               console.log(`[DEBUG] Removed thinking tags from final result`);
             }
           }
-          if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted) {
+          if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted && !options._completionPromptProcessed) {
             const outputContent = this._outputBuffer.items.join("\n\n");
             if (options.schema) {
               finalResult = (finalResult || "") + "\n<<<RAW_OUTPUT>>>\n" + outputContent + "\n<<<END_RAW_OUTPUT>>>";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc250",
+  "version": "0.6.0-rc251",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",

package/src/agent/ProbeAgent.js

@@ -4383,7 +4383,8 @@ Convert your previous response content into actual JSON data that follows this s
           // Call answer recursively with _schemaFormatted flag to prevent infinite loop
           finalResult = await this.answer(schemaPrompt, [], {
             ...options,
-            _schemaFormatted: true
+            _schemaFormatted: true,
+            _completionPromptProcessed: true  // Prevent cascading completion prompts in retry calls
           });
 
           // Step 2: Validate and fix Mermaid diagrams if present (BEFORE cleaning schema)
@@ -4642,7 +4643,8 @@ Convert your previous response content into actual JSON data that follows this s
               finalResult = await this.answer(schemaDefinitionPrompt, [], {
                 ...options,
                 _schemaFormatted: true,
-                _skipValidation: true  // Skip validation in recursive correction calls to prevent loops
+                _skipValidation: true,  // Skip validation in recursive correction calls to prevent loops
+                _completionPromptProcessed: true  // Prevent cascading completion prompts in retry calls
               });
               finalResult = cleanSchemaResponse(finalResult);
               validation = validateJsonResponse(finalResult);
@@ -4702,7 +4704,8 @@ Convert your previous response content into actual JSON data that follows this s
                 ...options,
                 _schemaFormatted: true,
                 _skipValidation: true,  // Skip validation in recursive correction calls to prevent loops
-                _disableTools: true     // Only allow attempt_completion - prevent AI from using search/query tools
+                _disableTools: true,    // Only allow attempt_completion - prevent AI from using search/query tools
+                _completionPromptProcessed: true  // Prevent cascading completion prompts in retry calls
               });
               finalResult = cleanSchemaResponse(finalResult);
               
@@ -4787,7 +4790,8 @@ Convert your previous response content into actual JSON data that follows this s
       }
 
       // Append DSL output buffer directly to response (bypasses LLM rewriting)
-      if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted) {
+      // Skip during _completionPromptProcessed — only the parent answer() should append the buffer.
+      if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted && !options._completionPromptProcessed) {
         const outputContent = this._outputBuffer.items.join('\n\n');
         if (options.schema) {
           // Schema response — the finalResult is JSON. Wrap output in RAW_OUTPUT