@probelabs/probe 0.6.0-rc233 → 0.6.0-rc235

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc233",
+  "version": "0.6.0-rc235",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",
@@ -79,7 +79,7 @@
     "@ai-sdk/openai": "^2.0.10",
     "@anthropic-ai/claude-agent-sdk": "^0.1.46",
     "@modelcontextprotocol/sdk": "^1.0.0",
-    "@nyariv/sandboxjs": "^0.8.32",
+    "@nyariv/sandboxjs": "github:probelabs/SandboxJS",
     "@probelabs/maid": "^0.0.24",
     "acorn": "^8.15.0",
     "acorn-walk": "^8.3.4",

package/src/agent/ProbeAgent.js

@@ -81,7 +81,8 @@ import {
   generateSchemaInstructions,
   isJsonSchemaDefinition,
   createSchemaDefinitionCorrectionPrompt,
-  validateAndFixMermaidResponse
+  validateAndFixMermaidResponse,
+  tryAutoWrapForSimpleSchema
 } from './schemaUtils.js';
 import { removeThinkingTags } from './xmlParsingUtils.js';
 import { predefinedPrompts } from './shared/prompts.js';
@@ -3679,6 +3680,17 @@ Follow these instructions carefully:
                 const executeToolCall = async () => {
                   // For delegate tool, pass current iteration, max iterations, session ID, and config
                   if (toolName === 'delegate') {
+                    // Reconstruct allowedTools array preserving all modes (same logic as clone())
+                    let allowedToolsForDelegate = null;
+                    if (this.allowedTools.mode === 'whitelist') {
+                      allowedToolsForDelegate = [...this.allowedTools.allowed];
+                    } else if (this.allowedTools.mode === 'none') {
+                      allowedToolsForDelegate = [];
+                    } else if (this.allowedTools.mode === 'all' && this.allowedTools.exclusions?.length > 0) {
+                      allowedToolsForDelegate = ['*', ...this.allowedTools.exclusions.map(t => '!' + t)];
+                    }
+                    // If mode is 'all' with no exclusions, leave as null (default)
+
                     const enhancedParams = {
                       ...toolParams,
                       currentIteration,
@@ -3689,6 +3701,12 @@ Follow these instructions carefully:
                       model: this.model,                // Inherit model
                       searchDelegate: this.searchDelegate,
                       enableTasks: this.enableTasks,    // Inherit task management (subagent gets isolated TaskManager)
+                      enableMcp: !!this.mcpBridge,      // Inherit MCP enablement
+                      mcpConfig: this.mcpConfig,        // Inherit MCP configuration
+                      mcpConfigPath: this.mcpConfigPath, // Inherit MCP config path
+                      enableBash: this.enableBash,      // Inherit bash enablement
+                      bashConfig: this.bashConfig,      // Inherit bash configuration
+                      allowedTools: allowedToolsForDelegate,  // Inherit allowed tools from parent
                       debug: this.debug,
                       tracer: this.tracer
                     };
@@ -4501,6 +4519,19 @@ Convert your previous response content into actual JSON data that follows this s
               retryCount = 1; // Start at 1 since we already did one correction
             }
             
+            // Before entering correction loop, try auto-wrapping for simple schemas
+            // This avoids re-invoking AI for schemas like {text: string} where we can just wrap programmatically
+            if (!validation.isValid) {
+              const autoWrapped = tryAutoWrapForSimpleSchema(finalResult, options.schema, { debug: this.debug });
+              if (autoWrapped) {
+                if (this.debug) {
+                  console.log(`[DEBUG] JSON validation: Auto-wrapped plain text for simple schema`);
+                }
+                finalResult = autoWrapped;
+                validation = validateJsonResponse(finalResult, { debug: this.debug });
+              }
+            }
+
             while (!validation.isValid && retryCount < maxRetries) {
               if (this.debug) {
                 console.log(`[DEBUG] JSON validation: attempt_completion validation failed (attempt ${retryCount + 1}/${maxRetries}):`, validation.error);

package/src/agent/schemaUtils.js

@@ -771,6 +771,78 @@ export function isJsonSchemaDefinition(jsonString, options = {}) {
   }
 }
 
+/**
+ * Check if schema is a simple text wrapper (e.g., {text: string} or {response: string})
+ * These schemas can be auto-wrapped without re-invoking the AI
+ * @param {string} schema - The schema string
+ * @returns {Object|null} - Object with field name if simple wrapper, null otherwise
+ */
+export function isSimpleTextWrapperSchema(schema) {
+  if (!schema || typeof schema !== 'string') {
+    return null;
+  }
+
+  const trimmed = schema.trim();
+
+  // Match patterns like: {text: string}, {"text": "string"}, {response: string}, etc.
+  // These are simple wrappers that just need a single text field
+  const simplePatterns = [
+    /^\{\s*["']?(\w+)["']?\s*:\s*["']?string["']?\s*\}$/i,
+    /^\{\s*["']?type["']?\s*:\s*["']?object["']?\s*,\s*["']?properties["']?\s*:\s*\{\s*["']?(\w+)["']?\s*:\s*\{\s*["']?type["']?\s*:\s*["']?string["']?\s*\}\s*\}\s*\}$/i
+  ];
+
+  for (const pattern of simplePatterns) {
+    const match = trimmed.match(pattern);
+    if (match) {
+      return { fieldName: match[1] };
+    }
+  }
+
+  return null;
+}
+
+/**
+ * Try to auto-wrap plain text for simple wrapper schemas
+ * Returns wrapped JSON if successful, null if not applicable
+ * @param {string} response - The response to wrap
+ * @param {string} schema - The schema string
+ * @param {Object} options - Options
+ * @param {boolean} [options.debug=false] - Enable debug logging
+ * @returns {string|null} - Wrapped JSON string or null
+ */
+export function tryAutoWrapForSimpleSchema(response, schema, options = {}) {
+  const { debug = false } = options;
+
+  const wrapperInfo = isSimpleTextWrapperSchema(schema);
+  if (!wrapperInfo) {
+    if (debug) {
+      console.log(`[DEBUG] Auto-wrap: Schema is not a simple text wrapper`);
+    }
+    return null;
+  }
+
+  // Check if response is already valid JSON
+  try {
+    JSON.parse(response);
+    // Already valid JSON, don't wrap
+    if (debug) {
+      console.log(`[DEBUG] Auto-wrap: Response is already valid JSON, skipping`);
+    }
+    return null;
+  } catch {
+    // Not valid JSON, proceed with wrapping
+  }
+
+  // Wrap the plain text in the schema structure
+  const wrapped = JSON.stringify({ [wrapperInfo.fieldName]: response });
+
+  if (debug) {
+    console.log(`[DEBUG] Auto-wrap: Wrapped plain text in {"${wrapperInfo.fieldName}": ...} (${response.length} chars)`);
+  }
+
+  return wrapped;
+}
+
 /**
  * Create a correction prompt for invalid JSON
  * @param {string} invalidResponse - The invalid JSON response
@@ -817,9 +889,10 @@ export function createJsonCorrectionPrompt(invalidResponse, schema, errorOrValid
   const level = Math.min(retryCount, strengthLevels.length - 1);
   const currentLevel = strengthLevels[level];
 
+  // Include full response - truncating loses context the AI needs to fix the issue
   let prompt = `${currentLevel.prefix} Your previous response is not valid JSON and cannot be parsed. Here's what you returned:
 
-${invalidResponse.substring(0, 500)}${invalidResponse.length > 500 ? '...' : ''}
+${invalidResponse}
 
 Error: ${enhancedError}
 

package/src/agent/tasks/taskTool.js

@@ -42,7 +42,7 @@ Manage tasks for tracking progress during code exploration and problem-solving.
 
 Parameters:
 - action: (required) The action to perform: create, update, complete, delete, list
-- tasks: (optional) JSON array for batch operations - alternative to single-task params
+- tasks: (optional) Array of task objects for batch operations. Place raw JSON array directly between tags.
 - id: (optional) Task ID for single operations (e.g., "task-1")
 - title: (optional) Task title for create/update
 - description: (optional) Task description for create/update
@@ -51,6 +51,11 @@ Parameters:
 - dependencies: (optional) JSON array of task IDs that must be completed first
 - after: (optional) Task ID to insert the new task after (for ordering). By default, new tasks are appended to the end
 
+IMPORTANT - JSON Format:
+Place raw JSON arrays directly between tags without quotes or escaping:
+  CORRECT:   <tasks>[{"title": "Do X"}]</tasks>
+  INCORRECT: <tasks>"[{\"title\": \"Do X\"}]"</tasks>
+
 Usage Examples:
 
 Creating a single task: