@juspay/neurolink 9.65.0 → 9.65.2

package/dist/client/reactHooks.js

@@ -254,8 +254,9 @@ export function useChat(options = {}) {
      */
     const handleSubmit = useCallback((e, submitOptions) => {
         e?.preventDefault?.();
-        if (!input.trim())
+        if (!input.trim()) {
             return;
+        }
         const message = {
             role: "user",
             content: input,
@@ -271,8 +272,9 @@ export function useChat(options = {}) {
         // Read from the ref so we always have the latest messages
         const currentMessages = messagesRef.current;
         const lastUserMessageIndex = currentMessages.findLastIndex((m) => m.role === "user");
-        if (lastUserMessageIndex === -1)
+        if (lastUserMessageIndex === -1) {
             return null;
+        }
         const lastUserMessage = currentMessages[lastUserMessageIndex];
         // Remove messages after the last user message
         setMessages((prev) => prev.slice(0, lastUserMessageIndex));
@@ -742,8 +744,9 @@ export function useVoice(options = {}) {
     const synthesisRef = useRef(null);
     // Check browser support
     const isSupported = useMemo(() => {
-        if (typeof window === "undefined")
+        if (typeof window === "undefined") {
             return false;
+        }
         const hasSpeechRecognition = "SpeechRecognition" in window || "webkitSpeechRecognition" in window;
         const hasSpeechSynthesis = "speechSynthesis" in window;
         return hasSpeechRecognition || hasSpeechSynthesis;
@@ -752,12 +755,14 @@ export function useVoice(options = {}) {
      * Initialize speech recognition
      */
     const initRecognition = useCallback(() => {
-        if (typeof window === "undefined")
+        if (typeof window === "undefined") {
             return null;
+        }
         const SpeechRecognitionCtor = window.SpeechRecognition ||
             window.webkitSpeechRecognition;
-        if (!SpeechRecognitionCtor)
+        if (!SpeechRecognitionCtor) {
             return null;
+        }
         const recognition = new SpeechRecognitionCtor();
         recognition.continuous = true;
         recognition.interimResults = true;
@@ -795,8 +800,9 @@ export function useVoice(options = {}) {
      * Start listening for voice input
      */
     const startListening = useCallback(() => {
-        if (!enableSpeechRecognition)
+        if (!enableSpeechRecognition) {
             return;
+        }
         if (!recognitionRef.current) {
             recognitionRef.current = initRecognition();
         }
@@ -820,8 +826,9 @@ export function useVoice(options = {}) {
      * Speak text using TTS
      */
     const speak = useCallback(async (text) => {
-        if (typeof window === "undefined")
+        if (typeof window === "undefined") {
             return;
+        }
         setIsSpeaking(true);
         onSpeechStart?.();
         try {
@@ -1002,8 +1009,9 @@ export function useStream(options = {}) {
             let fullText = "";
             while (true) {
                 const { done, value } = await reader.read();
-                if (done)
+                if (done) {
                     break;
+                }
                 buffer += decoder.decode(value, { stream: true });
                 const lines = buffer.split("\n");
                 buffer = lines.pop() ?? "";

package/dist/client/reactHooks.tsx

@@ -27,7 +27,6 @@ import {
 import type { ReactNode } from "react";
 
 import type {
-  ClientConfig,
   ClientApiError,
   NeuroLinkProviderProps,
   UseChatOptions,
@@ -361,7 +360,9 @@ export function useChat(options: UseChatOptions = {}): UseChatReturn {
     ) => {
       e?.preventDefault?.();
 
-      if (!input.trim()) return;
+      if (!input.trim()) {
+        return;
+      }
 
       const message: Omit<ChatMessage, "id" | "createdAt"> = {
         role: "user",
@@ -384,7 +385,9 @@ export function useChat(options: UseChatOptions = {}): UseChatReturn {
     const lastUserMessageIndex = currentMessages.findLastIndex(
       (m: ChatMessage) => m.role === "user",
     );
-    if (lastUserMessageIndex === -1) return null;
+    if (lastUserMessageIndex === -1) {
+      return null;
+    }
 
     const lastUserMessage = currentMessages[lastUserMessageIndex];
 
@@ -977,7 +980,9 @@ export function useVoice(options: UseVoiceOptions = {}): UseVoiceReturn {
 
   // Check browser support
   const isSupported = useMemo(() => {
-    if (typeof window === "undefined") return false;
+    if (typeof window === "undefined") {
+      return false;
+    }
     const hasSpeechRecognition =
       "SpeechRecognition" in window || "webkitSpeechRecognition" in window;
     const hasSpeechSynthesis = "speechSynthesis" in window;
@@ -988,7 +993,9 @@ export function useVoice(options: UseVoiceOptions = {}): UseVoiceReturn {
    * Initialize speech recognition
    */
   const initRecognition = useCallback(() => {
-    if (typeof window === "undefined") return null;
+    if (typeof window === "undefined") {
+      return null;
+    }
 
     const SpeechRecognitionCtor =
       (
@@ -1002,7 +1009,9 @@ export function useVoice(options: UseVoiceOptions = {}): UseVoiceReturn {
         }
       ).webkitSpeechRecognition;
 
-    if (!SpeechRecognitionCtor) return null;
+    if (!SpeechRecognitionCtor) {
+      return null;
+    }
 
     const recognition = new SpeechRecognitionCtor();
     recognition.continuous = true;
@@ -1047,7 +1056,9 @@ export function useVoice(options: UseVoiceOptions = {}): UseVoiceReturn {
    * Start listening for voice input
    */
   const startListening = useCallback(() => {
-    if (!enableSpeechRecognition) return;
+    if (!enableSpeechRecognition) {
+      return;
+    }
 
     if (!recognitionRef.current) {
       recognitionRef.current = initRecognition();
@@ -1076,7 +1087,9 @@ export function useVoice(options: UseVoiceOptions = {}): UseVoiceReturn {
    */
   const speak = useCallback(
     async (text: string): Promise<void> => {
-      if (typeof window === "undefined") return;
+      if (typeof window === "undefined") {
+        return;
+      }
 
       setIsSpeaking(true);
       onSpeechStart?.();
@@ -1286,7 +1299,9 @@ export function useStream(options: UseStreamOptions = {}): UseStreamReturn {
 
         while (true) {
           const { done, value } = await reader.read();
-          if (done) break;
+          if (done) {
+            break;
+          }
 
           buffer += decoder.decode(value, { stream: true });
           const lines = buffer.split("\n");

package/dist/core/baseProvider.d.ts

@@ -1,9 +1,9 @@
-import type { LanguageModel, ModelMessage, Tool, ToolCallRepairFunction, ToolSet } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import type { EvaluationData } from "../index.js";
 import type { NeuroLink } from "../neurolink.js";
 import type { UnknownRecord, MiddlewareFactoryOptions, StreamOptions, StreamResult, AIProvider, AnalyticsData, EnhancedGenerateResult, TextGenerationOptions, TextGenerationResult, ValidationSchema } from "../types/index.js";
 import { TelemetryHandler } from "./modules/TelemetryHandler.js";
+import type { LanguageModel, ModelMessage, Tool, ToolCallRepairFunction, ToolSet } from "../types/index.js";
 /**
  * Abstract base class for all AI providers
  * Tools are integrated as first-class citizens - always available by default
@@ -20,11 +20,6 @@ export declare abstract class BaseProvider implements AIProvider {
     protected sessionId?: string;
     protected userId?: string;
     protected neurolink?: NeuroLink;
-    /** @internal Trace context propagated from NeuroLink SDK for span hierarchy */
-    protected _traceContext: {
-        traceId: string;
-        parentSpanId: string;
-    } | null;
     setTraceContext(ctx: {
         traceId: string;
         parentSpanId: string;

package/dist/core/baseProvider.js

@@ -1,5 +1,4 @@
 import { context, SpanKind, SpanStatusCode, trace } from "@opentelemetry/api";
-import { generateText } from "ai";
 import { directAgentTools } from "../agent/directTools.js";
 import { IMAGE_GENERATION_MODELS } from "../core/constants.js";
 import { MiddlewareFactory } from "../middleware/factory.js";
@@ -21,6 +20,7 @@ import { StreamHandler } from "./modules/StreamHandler.js";
 import { TelemetryHandler } from "./modules/TelemetryHandler.js";
 import { ToolsManager } from "./modules/ToolsManager.js";
 import { Utilities } from "./modules/Utilities.js";
+import { generateText } from "../utils/generation.js";
 /**
  * Abstract base class for all AI providers
  * Tools are integrated as first-class citizens - always available by default

package/dist/core/constants.d.ts

@@ -21,6 +21,7 @@ export declare const DEFAULT_TEMPERATURE = 0.7;
 export declare const DEFAULT_TIMEOUT = 60000;
 export declare const DEFAULT_MAX_STEPS = 200;
 export declare const DEFAULT_TOOL_MAX_RETRIES = 2;
+export declare const TOOL_STORAGE_TIMEOUT_MS = 5000;
 export declare const STEP_LIMITS: {
     min: number;
     max: number;

package/dist/core/constants.js

@@ -107,6 +107,9 @@ export const DEFAULT_TEMPERATURE = 0.7;
 export const DEFAULT_TIMEOUT = 60000;
 export const DEFAULT_MAX_STEPS = 200;
 export const DEFAULT_TOOL_MAX_RETRIES = 2; // Maximum retries per tool before permanently failing
+// Fire-and-forget tool storage writes (Redis). 5s is generous for a single
+// Redis write; if breached, the .catch logs a warning.
+export const TOOL_STORAGE_TIMEOUT_MS = 5000;
 // Step execution limits
 export const STEP_LIMITS = {
     min: 1,

package/dist/core/modules/GenerationHandler.d.ts

@@ -12,9 +12,9 @@
  *
  * @module core/modules/GenerationHandler
  */
-import type { LanguageModel, ModelMessage, Tool } from "ai";
-import { generateText } from "ai";
 import type { AIProviderName, EnhancedGenerateResult, NeuroLinkEvents, StandardRecord, TextGenerationOptions, TypedEventEmitter } from "../../types/index.js";
+import type { LanguageModel, ModelMessage, Tool } from "../../types/index.js";
+import { generateText } from "../../utils/generation.js";
 /**
  * GenerationHandler class - Handles text generation operations for AI providers
  */

package/dist/core/modules/GenerationHandler.js

@@ -13,7 +13,6 @@
  * @module core/modules/GenerationHandler
  */
 import { SpanKind, SpanStatusCode } from "@opentelemetry/api";
-import { generateText, NoObjectGeneratedError, Output, stepCountIs } from "ai";
 import { getModelId } from "../../providers/providerTypeUtils.js";
 import { tracers } from "../../telemetry/tracers.js";
 import { logger } from "../../utils/logger.js";
@@ -22,6 +21,9 @@ import { calculateCost } from "../../utils/pricing.js";
 import { withProviderRetry } from "../../utils/providerRetry.js";
 import { calculateCacheSavingsPercent, extractCacheCreationTokens, extractCacheReadTokens, extractTokenUsage, } from "../../utils/tokenUtils.js";
 import { DEFAULT_MAX_STEPS } from "../constants.js";
+import { NoObjectGeneratedError } from "../../utils/generationErrors.js";
+import { Output, stepCountIs } from "../../utils/tool.js";
+import { generateText } from "../../utils/generation.js";
 const genTracer = tracers.generation;
 /**
  * Safely preview-serialize a value for debug logging.

package/dist/core/modules/MessageBuilder.d.ts

@@ -1,19 +1,5 @@
-/**
- * Message Builder Module
- *
- * Handles all message construction logic for AI providers.
- * Extracted from BaseProvider to follow Single Responsibility Principle.
- *
- * Responsibilities:
- * - Building messages from text generation options
- * - Building messages from stream options
- * - Multimodal input detection
- * - Message format conversion (to ModelMessage[])
- *
- * @module core/modules/MessageBuilder
- */
-import type { ModelMessage } from "ai";
 import type { AIProviderName, TextGenerationOptions, StreamOptions } from "../../types/index.js";
+import type { ModelMessage } from "../../types/index.js";
 /**
  * MessageBuilder class - Handles message construction for AI providers
  */

package/dist/core/modules/MessageBuilder.js

@@ -1,17 +1,3 @@
-/**
- * Message Builder Module
- *
- * Handles all message construction logic for AI providers.
- * Extracted from BaseProvider to follow Single Responsibility Principle.
- *
- * Responsibilities:
- * - Building messages from text generation options
- * - Building messages from stream options
- * - Multimodal input detection
- * - Message format conversion (to ModelMessage[])
- *
- * @module core/modules/MessageBuilder
- */
 import { tracers, ATTR, withSpan } from "../../telemetry/index.js";
 import { logger } from "../../utils/logger.js";
 import { buildMessagesArray, buildMultimodalMessagesArray, } from "../../utils/messageBuilder.js";

package/dist/core/modules/StreamHandler.js

@@ -20,7 +20,7 @@ import { buildNoOutputSentinel, detectPostStreamNoOutput, stampNoOutputSpan, } f
 import { STEP_LIMITS } from "../constants.js";
 import { createAnalytics } from "../analytics.js";
 import { nanoid } from "nanoid";
-import { NoOutputGeneratedError } from "ai";
+import { NoOutputGeneratedError } from "../../utils/generationErrors.js";
 /**
  * StreamHandler class - Handles streaming operations for AI providers
  */

package/dist/core/modules/ToolsManager.d.ts

@@ -1,22 +1,6 @@
-/**
- * Tools Manager Module
- *
- * Handles all tool registration, discovery, and execution for AI providers.
- * Extracted from BaseProvider to follow Single Responsibility Principle.
- *
- * Responsibilities:
- * - Tool registration (direct, custom, MCP, external MCP)
- * - Tool discovery and aggregation
- * - Tool creation from definitions and schemas
- * - Tool executor setup
- * - Session context management for MCP tools
- * - Event emission wrapping for tool execution
- *
- * @module core/modules/ToolsManager
- */
-import type { Tool } from "ai";
 import type { AIProviderName, ToolUtilities } from "../../types/index.js";
 import type { NeuroLink } from "../../neurolink.js";
+import type { Tool } from "../../types/index.js";
 /**
  * ToolsManager class - Handles all tool management operations
  */

package/dist/core/modules/ToolsManager.js

@@ -1,20 +1,3 @@
-/**
- * Tools Manager Module
- *
- * Handles all tool registration, discovery, and execution for AI providers.
- * Extracted from BaseProvider to follow Single Responsibility Principle.
- *
- * Responsibilities:
- * - Tool registration (direct, custom, MCP, external MCP)
- * - Tool discovery and aggregation
- * - Tool creation from definitions and schemas
- * - Tool executor setup
- * - Session context management for MCP tools
- * - Event emission wrapping for tool execution
- *
- * @module core/modules/ToolsManager
- */
-import { tool as createAISDKTool, jsonSchema } from "ai";
 import { z } from "zod";
 import { createToolEventPayload } from "../toolEvents.js";
 import { tracers, ATTR, withSpan } from "../../telemetry/index.js";
@@ -23,6 +6,7 @@ import { logger } from "../../utils/logger.js";
 import { getKeyCount } from "../../utils/transformationUtils.js";
 import { convertJsonSchemaToZod } from "../../utils/schemaConversion.js";
 import { generateToolOutputPreview } from "../../context/toolOutputLimits.js";
+import { tool as createAISDKTool, jsonSchema } from "../../utils/tool.js";
 /**
  * ToolsManager class - Handles all tool management operations
  */

package/dist/core/redisConversationMemoryManager.js

@@ -951,12 +951,6 @@ User message: "${userMessage}"`;
                 provider: this.config.summarizationProvider || "vertex",
                 model: this.config.summarizationModel || "gemini-2.5-flash",
                 disableTools: true, // Title generation doesn't need tools — saves ~600 tokens of tool descriptions
-                // A 20-25 letter title is well under 64 tokens; capping prevents
-                // Vertex 400 INVALID_ARGUMENT — Gemini 2.5 Flash on the native
-                // @google/genai SDK rejects requests with no maxOutputTokens cap
-                // when the prompt is short, which silently broke every
-                // `conversationMemory.enabled + context` test path.
-                maxTokens: 64,
             });
             // Clean up the generated title
             let title = result.content?.trim() || "New Conversation";

package/dist/core/streamAnalytics.js

@@ -1,7 +1,7 @@
 import { createAnalytics } from "./analytics.js";
 import { logger } from "../utils/logger.js";
 import { extractTokenUsage, createEmptyTokenUsage, } from "../utils/tokenUtils.js";
-import { NoOutputGeneratedError } from "ai";
+import { NoOutputGeneratedError } from "../utils/generationErrors.js";
 /**
  * Base implementation for collecting analytics from Vercel AI SDK stream results
  */

package/dist/evaluation/contextBuilder.d.ts

@@ -1,8 +1,5 @@
-/**
- * @file Implements the ContextBuilder class for creating rich evaluation context.
- */
-import type { LanguageModelV3CallOptions } from "@ai-sdk/provider";
 import type { EnhancedEvaluationContext, EvaluationResult, GenerateResult } from "../types/index.js";
+import type { LanguageModelV3CallOptions } from "../types/index.js";
 /**
  * Builds the enhanced context required for a RAGAS-style evaluation.
  * This class gathers data from the generation options and results to create a

package/dist/evaluation/contextBuilder.js

@@ -1,6 +1,3 @@
-/**
- * @file Implements the ContextBuilder class for creating rich evaluation context.
- */
 import { logger } from "../utils/logger.js";
 /**
  * Builds the enhanced context required for a RAGAS-style evaluation.

package/dist/evaluation/index.d.ts

@@ -1,8 +1,5 @@
-/**
- * @file This file exports the main Evaluator class, which serves as the central entry point for the evaluation system.
- */
-import type { LanguageModelV3CallOptions } from "@ai-sdk/provider";
 import type { EvaluationData, GenerateResult, AutoEvaluationConfig, EvaluationConfig } from "../types/index.js";
+import type { LanguageModelV3CallOptions } from "../types/index.js";
 export * from "./errors/index.js";
 export * from "./hooks/index.js";
 export * from "./pipeline/index.js";

package/dist/evaluation/index.js

@@ -1,6 +1,3 @@
-/**
- * @file This file exports the main Evaluator class, which serves as the central entry point for the evaluation system.
- */
 import { ContextBuilder } from "./contextBuilder.js";
 import { RAGASEvaluator } from "./ragasEvaluator.js";
 import { mapToEvaluationData } from "./scoring.js";

package/dist/files/fileTools.d.ts

@@ -1,27 +1,11 @@
-/**
- * On-Demand File Access Tools
- *
- * Exposes file operations as LLM tools via the Vercel AI SDK `tool()` pattern.
- * These tools allow the LLM to access file content lazily — only reading
- * the specific sections, lines, or search results it needs.
- *
- * Tools:
- * - list_attached_files: List all registered files with metadata
- * - read_file_section: Read specific line range with token budget
- * - search_in_file: Search for patterns within a file
- * - get_file_preview: Get the preview/summary of a file
- * - extract_file_content: Targeted extraction for any file type (video frames,
- *     PDF pages, spreadsheet ranges, archive entries, PPTX slides)
- *
- * @module files/fileTools
- */
 import type { FileReferenceRegistry } from "./fileReferenceRegistry.js";
 /**
  * Create file access tools bound to a FileReferenceRegistry instance.
  *
  * These tools follow the same pattern as the existing directAgentTools
  * (getCurrentTime, readFile, etc.) in src/lib/agent/directTools.ts.
- * They use `tool()` from the Vercel AI SDK with zod parameter schemas.
+ * They use NeuroLink's `tool()` helper from `../utils/tool.js` with Zod
+ * parameter schemas.
  *
  * @param registry - The FileReferenceRegistry instance to bind to
  * @returns Record of tool name to tool definition

package/dist/files/fileTools.js

@@ -1,28 +1,12 @@
-/**
- * On-Demand File Access Tools
- *
- * Exposes file operations as LLM tools via the Vercel AI SDK `tool()` pattern.
- * These tools allow the LLM to access file content lazily — only reading
- * the specific sections, lines, or search results it needs.
- *
- * Tools:
- * - list_attached_files: List all registered files with metadata
- * - read_file_section: Read specific line range with token budget
- * - search_in_file: Search for patterns within a file
- * - get_file_preview: Get the preview/summary of a file
- * - extract_file_content: Targeted extraction for any file type (video frames,
- *     PDF pages, spreadsheet ranges, archive entries, PPTX slides)
- *
- * @module files/fileTools
- */
-import { tool } from "ai";
 import { z } from "zod";
+import { tool } from "../utils/tool.js";
 /**
  * Create file access tools bound to a FileReferenceRegistry instance.
  *
  * These tools follow the same pattern as the existing directAgentTools
  * (getCurrentTime, readFile, etc.) in src/lib/agent/directTools.ts.
- * They use `tool()` from the Vercel AI SDK with zod parameter schemas.
+ * They use NeuroLink's `tool()` helper from `../utils/tool.js` with Zod
+ * parameter schemas.
  *
  * @param registry - The FileReferenceRegistry instance to bind to
  * @returns Record of tool name to tool definition

package/dist/lib/adapters/video/videoAnalyzer.d.ts

@@ -7,7 +7,7 @@
  * @module adapters/video/geminiVideoAnalyzer
  */
 import { AIProviderName } from "../../constants/enums.js";
-import type { ModelMessage } from "ai";
+import type { ModelMessage } from "../../types/index.js";
 export declare function analyzeVideoWithVertexAI(messages: ModelMessage[], options?: {
     project?: string;
     location?: string;

package/dist/lib/agent/directTools.d.ts

@@ -1,14 +1,10 @@
-/**
- * Direct Tool Definitions for NeuroLink CLI Agent
- * Simple, reliable tools that work immediately with Vercel AI SDK
- */
-import type { AllToolsMap, BasicToolsMap, FilesystemToolsMap, UtilityToolsMap } from "../types/index.js";
+import type { AllToolsMap, BasicToolsMap, FilesystemToolsMap, Tool, UtilityToolsMap } from "../types/index.js";
 /**
  * Direct tool definitions that work immediately with Gemini/AI SDK
  * These bypass MCP complexity and provide reliable agent functionality
  */
 export declare const directAgentTools: {
-    getCurrentTime: import("ai").Tool<{
+    getCurrentTime: Tool<{
         timezone?: string | undefined;
     }, {
         success: boolean;
@@ -32,7 +28,7 @@ export declare const directAgentTools: {
         iso?: undefined;
         timestamp?: undefined;
     }>;
-    readFile: import("ai").Tool<{
+    readFile: Tool<{
         path: string;
     }, {
         success: boolean;
@@ -56,7 +52,7 @@ export declare const directAgentTools: {
         size?: undefined;
         lastModified?: undefined;
     }>;
-    listDirectory: import("ai").Tool<{
+    listDirectory: Tool<{
         path: string;
         includeHidden: boolean;
     }, {
@@ -77,7 +73,7 @@ export declare const directAgentTools: {
         items?: undefined;
         count?: undefined;
     }>;
-    calculateMath: import("ai").Tool<{
+    calculateMath: Tool<{
         expression: string;
         precision: number;
     }, {
@@ -99,7 +95,7 @@ export declare const directAgentTools: {
         result?: undefined;
         type?: undefined;
     }>;
-    writeFile: import("ai").Tool<{
+    writeFile: Tool<{
         path: string;
         content: string;
         mode: "append" | "create" | "overwrite";
@@ -125,7 +121,7 @@ export declare const directAgentTools: {
         size?: undefined;
         written?: undefined;
     }>;
-    analyzeCSV: import("ai").Tool<{
+    analyzeCSV: Tool<{
         filePath: string;
         operation: "count_by_column" | "sum_by_column" | "average_by_column" | "min_max_by_column" | "describe";
         column: string;
@@ -147,7 +143,7 @@ export declare const directAgentTools: {
         operation: "count_by_column" | "sum_by_column" | "average_by_column" | "min_max_by_column" | "describe";
         column: string;
     }>;
-    websearchGrounding: import("ai").Tool<{
+    websearchGrounding: Tool<{
         query: string;
         maxResults: number;
         maxWords: number;
@@ -221,11 +217,7 @@ export declare const directAgentTools: {
  * Import this directly when you need bash execution capabilities:
  *   import { bashTool } from '../agent/directTools.js';
  */
-export declare const bashTool: import("ai").Tool<{
-    command: string;
-    timeout: number;
-    cwd?: string | undefined;
-}, unknown>;
+export declare const bashTool: Tool;
 /**
  * Get a subset of tools for specific use cases with improved type safety
  */

package/dist/lib/agent/directTools.js

@@ -1,8 +1,3 @@
-/**
- * Direct Tool Definitions for NeuroLink CLI Agent
- * Simple, reliable tools that work immediately with Vercel AI SDK
- */
-import { tool } from "ai";
 import { z } from "zod";
 import * as fs from "fs";
 import * as path from "path";
@@ -11,6 +6,7 @@ import { logger } from "../utils/logger.js";
 import { VertexAI } from "@google-cloud/vertexai";
 import { CSVProcessor } from "../utils/csvProcessor.js";
 import { shouldEnableBashTool } from "../utils/toolUtils.js";
+import { tool } from "../utils/tool.js";
 const MAX_OUTPUT_BYTES = 102400; // 100KB
 function truncateOutput(output) {
     if (output.length > MAX_OUTPUT_BYTES) {
@@ -547,9 +543,16 @@ export const directAgentTools = {
     // It is only included in directAgentTools when NEUROLINK_ENABLE_BASH_TOOL=true or
     // toolConfig.enableBashTool is explicitly set to true. See shouldEnableBashTool() in toolUtils.ts.
     websearchGrounding: tool({
-        description: "Search the web for current information using Google Search grounding. Returns raw search data for AI processing.",
+        description: "Performs a Google Search and returns a summarized answer with source citations. Always check the current date before constructing the query. Use whenever the answer depends on time-sensitive facts or requires verification against real-world sources.",
         inputSchema: z.object({
-            query: z.string().describe("Search query to find information about"),
+            query: z
+                .string()
+                .trim()
+                .min(1, { message: "must be a non-empty search string" })
+                .refine((v) => v.toLowerCase() !== "undefined", {
+                message: 'must not be the literal string "undefined" — pass a real search query',
+            })
+                .describe("The search query string to look up on the web."),
             maxResults: z
                 .number()
                 .optional()
@@ -581,7 +584,8 @@ export const directAgentTools = {
                     project: hasProjectId,
                     location: projectLocation,
                 });
-                const websearchModel = "gemini-2.5-flash-lite";
+                const websearchModel = process.env.NEUROLINK_WEBSEARCH_MODEL?.trim() ||
+                    "gemini-2.5-flash-lite";
                 const model = vertex_ai.getGenerativeModel({
                     model: websearchModel,
                     tools: createGoogleSearchTools(),