@probelabs/probe 0.6.0-rc299 → 0.6.0-rc301

package/build/agent/ProbeAgent.js

@@ -38,6 +38,7 @@ import { existsSync } from 'fs';
 import { readFile, stat, readdir } from 'fs/promises';
 import { resolve, isAbsolute, dirname, basename, normalize, sep } from 'path';
 import { TokenCounter } from './tokenCounter.js';
+import { truncateForSpan } from './simpleTelemetry.js';
 import { InMemoryStorageAdapter } from './storage/InMemoryStorageAdapter.js';
 import { HookManager, HOOK_TYPES } from './hooks/HookManager.js';
 import { SUPPORTED_IMAGE_EXTENSIONS, IMAGE_MIME_TYPES, isFormatSupportedByProvider } from './imageConfig.js';
@@ -4327,9 +4328,7 @@ Double-check your response based on the criteria above. If everything looks good
 
           let aiResult;
           if (this.tracer) {
-            const inputPreview = message.length > 1000
-              ? message.substring(0, 1000) + '... [truncated]'
-              : message;
+            const inputPreview = truncateForSpan(message, 4096);
 
             aiResult = await this.tracer.withSpan('ai.request', executeAIRequest, {
               'ai.model': this.model,
@@ -4340,6 +4339,12 @@ Double-check your response based on the criteria above. If everything looks good
               'max_tokens': maxResponseTokens,
               'temperature': 0.3,
               'message_count': currentMessages.length
+            }, (span, result) => {
+              const text = result?.finalText || '';
+              span.setAttributes({
+                'ai.output': truncateForSpan(text),
+                'ai.output_length': text.length
+              });
             });
           } else {
             aiResult = await executeAIRequest();
@@ -4367,6 +4372,14 @@ Double-check your response based on the criteria above. If everything looks good
             finalResult = aiResult.finalText;
           }
 
+          // Ensure finalResult is always a string (#533).
+          // When both structured output and finalText are empty (e.g., timeout,
+          // model returns nothing), downstream callers like delegate() expect a
+          // string and would otherwise throw "not a string".
+          if (finalResult === null || finalResult === undefined) {
+            finalResult = '';
+          }
+
           // Graceful timeout handling: ensure the response clearly indicates
           // the research was interrupted and may be incomplete.
           if (gracefulTimeoutState.triggered) {

package/build/agent/simpleTelemetry.js

@@ -2,6 +2,21 @@ import { existsSync, mkdirSync, createWriteStream } from 'fs';
 import { dirname } from 'path';
 import { patchConsole } from './otelLogBridge.js';
 
+/**
+ * Truncate text for span attributes, preserving head and tail for context.
+ * For text <= maxLen, returns as-is. For longer text, shows first half and
+ * last half of the budget with a separator indicating omitted chars.
+ * @param {string} text - The text to truncate
+ * @param {number} [maxLen=4096] - Maximum output length
+ * @returns {string} The truncated text
+ */
+export function truncateForSpan(text, maxLen = 4096) {
+  if (!text || text.length <= maxLen) return text || '';
+  const half = Math.floor((maxLen - 40) / 2); // 40 chars reserved for separator
+  const omitted = text.length - half * 2;
+  return text.substring(0, half) + `\n... [${omitted} chars omitted] ...\n` + text.substring(text.length - half);
+}
+
 /**
  * Simple telemetry implementation for probe-agent
  * This provides basic tracing functionality without complex OpenTelemetry dependencies
@@ -463,7 +478,7 @@ export class SimpleAppTracer {
     });
   }
 
-  async withSpan(spanName, fn, attributes = {}) {
+  async withSpan(spanName, fn, attributes = {}, onResult = null) {
     if (!this.isEnabled()) {
       return fn();
     }
@@ -476,12 +491,19 @@ export class SimpleAppTracer {
     try {
       const result = await fn();
       span.setStatus('OK');
+      if (onResult) {
+        try {
+          onResult(span, result);
+        } catch (_) {
+          // Don't let span enrichment errors break the flow
+        }
+      }
       return result;
     } catch (error) {
       span.setStatus('ERROR');
-      span.addEvent('exception', { 
+      span.addEvent('exception', {
         'exception.message': error.message,
-        'exception.stack': error.stack 
+        'exception.stack': error.stack
       });
       throw error;
     } finally {

package/build/tools/vercel.js

@@ -13,6 +13,7 @@ import { searchSchema, querySchema, extractSchema, delegateSchema, analyzeAllSch
 import { existsSync } from 'fs';
 import { formatErrorForAI } from '../utils/error-types.js';
 import { annotateOutputWithHashes } from './hashline.js';
+import { truncateForSpan } from '../agent/simpleTelemetry.js';
 
 /**
  * Auto-quote search query terms that contain mixed case or underscores.
@@ -539,6 +540,7 @@ export const searchTool = (options = {}) => {
 					tracer: options.tracer || null,
 					enableBash: false,
 					bashConfig: null,
+					allowEdit: options.allowEdit || false,
 					architectureFileName: options.architectureFileName || null,
 					promptType: 'code-searcher',
 					allowedTools: ['search', 'extract', 'listFiles'],
@@ -551,6 +553,12 @@ export const searchTool = (options = {}) => {
 					? await options.tracer.withSpan('search.delegate', runDelegation, {
 						'search.query': searchQuery,
 						'search.path': searchPath
+					}, (span, result) => {
+						const text = typeof result === 'string' ? result : '';
+						span.setAttributes({
+							'search.delegate.output': truncateForSpan(text),
+							'search.delegate.output_length': text.length
+						});
 					})
 					: await runDelegation();
 
@@ -888,7 +896,7 @@ export const extractTool = (options = {}) => {
  * @returns {Object} Configured delegate tool
  */
 export const delegateTool = (options = {}) => {
-	const { debug = false, timeout = 300, cwd, allowedFolders, workspaceRoot, enableBash = false, bashConfig, architectureFileName, enableMcp = false, mcpConfig = null, mcpConfigPath = null, delegationManager = null,
+	const { debug = false, timeout = 300, cwd, allowedFolders, workspaceRoot, enableBash = false, bashConfig, allowEdit = false, architectureFileName, enableMcp = false, mcpConfig = null, mcpConfigPath = null, delegationManager = null,
 		// Timeout settings inherited from parent agent
 		timeoutBehavior, maxOperationTimeout, requestTimeout, gracefulTimeoutBonusSteps,
 		negotiatedTimeoutBudget, negotiatedTimeoutMaxRequests, negotiatedTimeoutMaxPerRequest,
@@ -1007,6 +1015,7 @@ export const delegateTool = (options = {}) => {
 				model,
 				tracer,
 				enableBash,
+				allowEdit,
 				bashConfig,
 				architectureFileName,
 				searchDelegate,