@probelabs/probe 0.6.0-rc264 → 0.6.0-rc266

package/build/tools/edit.js

@@ -448,6 +448,13 @@ Parameters:
           return `Error editing file: Multiple occurrences found - the old_string appears ${occurrences} times in ${file_path}. To fix: (1) Set replace_all=true to replace all occurrences, or (2) Include more surrounding lines in old_string to make the match unique (add the full line or adjacent lines for context).`;
         }
 
+        // Guard against over-scoped text edits (replacing large blocks with tiny replacements)
+        const oldLines = matchTarget.split('\n').length;
+        const newLines = new_string.split('\n').length;
+        if (oldLines >= 20 && newLines < oldLines * 0.5) {
+          return `Error editing file: Edit scope too large — replacing ${oldLines} lines with ${newLines} lines risks accidental content deletion. To fix: (1) Use line-targeted editing (start_line/end_line) instead to constrain scope, or (2) Split into smaller, focused edits that each target only the lines you intend to change.`;
+        }
+
         // Perform the replacement
         let newContent;
         if (replace_all) {

package/build/tools/executePlan.js

@@ -393,7 +393,7 @@ export function createExecutePlanTool(options) {
     description: 'Execute a JavaScript DSL program to orchestrate tool calls. ' +
       'Use for batch processing, paginated APIs, multi-step workflows where intermediate data is large. ' +
       'Write simple synchronous-looking code — do NOT use async/await.',
-    parameters: executePlanSchema,
+    inputSchema: executePlanSchema,
     execute: async ({ code, description }) => {
       // Generate a unique session ID for this execute_plan invocation
       // This ensures search pagination is isolated per execute_plan call
@@ -1051,7 +1051,7 @@ export function createCleanupExecutePlanTool(options) {
   return tool({
     description: 'Clean up output buffer and session store from previous execute_plan calls. ' +
       'Use this when a previous execute_plan failed and left stale data, or before starting a fresh analysis.',
-    parameters: cleanupExecutePlanSchema,
+    inputSchema: cleanupExecutePlanSchema,
     execute: async ({ clearOutputBuffer = true, clearSessionStore = false }) => {
       const span = tracer?.createToolSpan?.('cleanup_execute_plan', {
         'cleanup.clear_output_buffer': clearOutputBuffer,

package/build/tools/index.js

@@ -12,7 +12,7 @@ export { editTool, createTool, multiEditTool } from './edit.js';
 export { createSearchTool, createQueryTool, createExtractTool } from './langchain.js';
 
 // Export execute_plan and cleanup_execute_plan tools
-export { createExecutePlanTool, getExecutePlanToolDefinition, createCleanupExecutePlanTool, getCleanupExecutePlanToolDefinition } from './executePlan.js';
+export { createExecutePlanTool, createCleanupExecutePlanTool } from './executePlan.js';
 
 // Export common schemas and utilities
 export {
@@ -24,15 +24,15 @@ export {
 	executePlanSchema,
 	cleanupExecutePlanSchema,
 	delegateDescription,
-	delegateToolDefinition,
 	bashDescription,
-	bashToolDefinition,
 	attemptCompletionSchema,
-	attemptCompletionToolDefinition,
 	parseAndResolvePaths,
 	resolveTargetPath,
-	DEFAULT_VALID_TOOLS,
-	buildToolTagPattern
+	listFilesSchema,
+	searchFilesSchema,
+	readImageSchema,
+	listSkillsSchema,
+	useSkillSchema
 } from './common.js';
 
 // Export edit and create schemas
@@ -42,10 +42,7 @@ export {
 	multiEditSchema,
 	editDescription,
 	createDescription,
-	multiEditDescription,
-	editToolDefinition,
-	createToolDefinition,
-	multiEditToolDefinition
+	multiEditDescription
 } from './edit.js';
 
 // Export system message
@@ -66,4 +63,4 @@ const tools = {
 	DEFAULT_SYSTEM_MESSAGE
 };
 
-export { tools };
+export { tools };