@juspay/neurolink 2.1.0 → 3.0.1

package/dist/lib/providers/agent-enhanced-provider.js

@@ -66,14 +66,15 @@ export class AgentEnhancedProvider {
         try {
             mcpLogger.info("[AgentEnhancedProvider] Initializing MCP integration...");
             this.mcpSystem = new UnifiedMCPSystem({
-                configPath: this.config.mcpDiscoveryOptions?.configFiles?.[0] || ".mcp-config.json",
+                configPath: this.config.mcpDiscoveryOptions?.configFiles?.[0] ||
+                    ".mcp-config.json",
                 enableExternalServers: true,
                 enableInternalServers: true,
-                autoInitialize: false
+                autoInitialize: false,
             });
             // ADD TIMEOUT to prevent hanging forever
             const initPromise = this.mcpSystem.initialize();
-            const timeoutPromise = new Promise((_, reject) => setTimeout(() => reject(new Error('MCP initialization timeout after 15 seconds')), this.config.mcpInitTimeoutMs || 15000));
+            const timeoutPromise = new Promise((_, reject) => setTimeout(() => reject(new Error("MCP initialization timeout after 15 seconds")), this.config.mcpInitTimeoutMs || 15000));
             await Promise.race([initPromise, timeoutPromise]);
             this.mcpInitialized = true;
             mcpLogger.info("[AgentEnhancedProvider] MCP integration initialized successfully");
@@ -101,10 +102,13 @@ export class AgentEnhancedProvider {
             return directTools;
         }
         // Get MCP tools if available
-        let mcpTools = {};
+        const mcpTools = {};
         try {
             // Skip if MCP failed to initialize or is still initializing
-            if (this.mcpInitFailed || this.mcpInitializing || !this.mcpInitialized || !this.mcpSystem) {
+            if (this.mcpInitFailed ||
+                this.mcpInitializing ||
+                !this.mcpInitialized ||
+                !this.mcpSystem) {
                 return directTools;
             }
             const mcpToolInfos = await this.mcpSystem.listTools();
@@ -123,7 +127,7 @@ export class AgentEnhancedProvider {
                                 ? new AbortController()
                                 : undefined;
                             if (toolAbortController && toolTimeout) {
-                                const timeoutMs = typeof toolTimeout === 'string'
+                                const timeoutMs = typeof toolTimeout === "string"
                                     ? parseTimeout(toolTimeout)
                                     : toolTimeout;
                                 timeoutId = setTimeout(() => {
@@ -131,31 +135,35 @@ export class AgentEnhancedProvider {
                                 }, timeoutMs);
                             }
                             const context = {
-                                sessionId: 'cli-session',
-                                userId: 'cli-user',
+                                sessionId: "cli-session",
+                                userId: "cli-user",
                                 secureFS: {
                                     readFile: async (path, encoding) => {
-                                        const fs = await import('fs/promises');
-                                        return encoding ? fs.readFile(path, { encoding: encoding }) : fs.readFile(path);
+                                        const fs = await import("fs/promises");
+                                        return encoding
+                                            ? fs.readFile(path, {
+                                                encoding: encoding,
+                                            })
+                                            : fs.readFile(path);
                                     },
                                     writeFile: async (path, content) => {
-                                        const fs = await import('fs/promises');
+                                        const fs = await import("fs/promises");
                                         await fs.writeFile(path, content);
                                     },
                                     readdir: async (path) => {
-                                        const fs = await import('fs/promises');
+                                        const fs = await import("fs/promises");
                                         return fs.readdir(path);
                                     },
                                     stat: async (path) => {
-                                        const fs = await import('fs/promises');
+                                        const fs = await import("fs/promises");
                                         return fs.stat(path);
                                     },
                                     mkdir: async (path, options) => {
-                                        const fs = await import('fs/promises');
+                                        const fs = await import("fs/promises");
                                         await fs.mkdir(path, options);
                                     },
                                     exists: async (path) => {
-                                        const fs = await import('fs/promises');
+                                        const fs = await import("fs/promises");
                                         try {
                                             await fs.access(path);
                                             return true;
@@ -163,34 +171,34 @@ export class AgentEnhancedProvider {
                                         catch {
                                             return false;
                                         }
-                                    }
+                                    },
                                 },
                                 path: {
                                     join: (...paths) => {
-                                        const path = require('path');
+                                        const path = require("path");
                                         return path.join(...paths);
                                     },
                                     resolve: (...paths) => {
-                                        const path = require('path');
+                                        const path = require("path");
                                         return path.resolve(...paths);
                                     },
                                     relative: (from, to) => {
-                                        const path = require('path');
+                                        const path = require("path");
                                         return path.relative(from, to);
                                     },
                                     dirname: (path) => {
-                                        const pathLib = require('path');
+                                        const pathLib = require("path");
                                         return pathLib.dirname(path);
                                     },
                                     basename: (path, ext) => {
-                                        const pathLib = require('path');
+                                        const pathLib = require("path");
                                         return pathLib.basename(path, ext);
-                                    }
+                                    },
                                 },
-                                grantedPermissions: ['read', 'write', 'execute'],
+                                grantedPermissions: ["read", "write", "execute"],
                                 log: (level, message, data) => {
                                     const logFn = mcpLogger[level];
-                                    if (typeof logFn === 'function') {
+                                    if (typeof logFn === "function") {
                                         if (data) {
                                             logFn(`${message} ${JSON.stringify(data)}`);
                                         }
@@ -198,7 +206,7 @@ export class AgentEnhancedProvider {
                                             logFn(message);
                                         }
                                     }
-                                }
+                                },
                             };
                             const toolPromise = this.mcpSystem.executeTool(toolInfo.name, args, context);
                             let result;
@@ -207,10 +215,10 @@ export class AgentEnhancedProvider {
                                 result = await Promise.race([
                                     toolPromise,
                                     new Promise((_, reject) => {
-                                        toolAbortController.signal.addEventListener('abort', () => {
+                                        toolAbortController.signal.addEventListener("abort", () => {
                                             reject(new Error(`Tool ${toolInfo.name} timed out after ${this.config.toolExecutionTimeout}`));
                                         });
-                                    })
+                                    }),
                                 ]);
                             }
                             else {
@@ -230,7 +238,7 @@ export class AgentEnhancedProvider {
                             mcpLogger.error(`MCP tool ${toolInfo.name} execution failed:`, error);
                             throw error;
                         }
-                    }
+                    },
                 };
             }
             mcpLogger.info(`[AgentEnhancedProvider] Loaded ${Object.keys(mcpTools).length} MCP tools`);
@@ -246,22 +254,20 @@ export class AgentEnhancedProvider {
             : optionsOrPrompt;
         const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt, schema, timeout, } = options;
         // Get combined tools (direct + MCP) if enabled
-        const tools = this.config.enableTools
-            ? await this.getCombinedTools()
-            : {};
+        const tools = this.config.enableTools ? await this.getCombinedTools() : {};
         const log = (msg, data) => {
-            mcpLogger.info(`[AgentEnhancedProvider] ${msg}`, data ? JSON.stringify(data, null, 2) : '');
+            mcpLogger.info(`[AgentEnhancedProvider] ${msg}`, data ? JSON.stringify(data, null, 2) : "");
         };
-        log('Starting text generation', {
+        log("Starting text generation", {
             prompt: prompt.substring(0, 100),
             toolsCount: Object.keys(tools).length,
-            maxSteps: this.config.maxSteps
+            maxSteps: this.config.maxSteps,
         });
         try {
             // Parse timeout if provided
             let abortSignal;
             if (timeout) {
-                const timeoutMs = typeof timeout === 'string' ? parseTimeout(timeout) : timeout;
+                const timeoutMs = typeof timeout === "string" ? parseTimeout(timeout) : timeout;
                 if (timeoutMs !== undefined) {
                     abortSignal = AbortSignal.timeout(timeoutMs);
                 }
@@ -279,24 +285,26 @@ export class AgentEnhancedProvider {
                 toolChoice: this.shouldForceToolUsage(prompt) ? "required" : "auto",
                 abortSignal, // Pass abort signal for timeout support
             });
-            log('Generation completed', {
+            log("Generation completed", {
                 text: result.text?.substring(0, 200),
                 finishReason: result.finishReason,
                 toolCallsCount: result.toolCalls?.length || 0,
                 toolResultsCount: result.toolResults?.length || 0,
-                stepsCount: result.steps?.length || 0
+                stepsCount: result.steps?.length || 0,
             });
             // Check if tools were called but no final text was generated
-            if (result.finishReason === 'tool-calls' && !result.text && result.toolResults?.length > 0) {
-                log('Tools called but no final text generated, creating summary response');
+            if (result.finishReason === "tool-calls" &&
+                !result.text &&
+                result.toolResults?.length > 0) {
+                log("Tools called but no final text generated, creating summary response");
                 try {
                     // Extract tool results and create a summary prompt
-                    let toolResultsSummary = '';
+                    let toolResultsSummary = "";
                     if (result.toolResults) {
                         for (const toolResult of result.toolResults) {
                             const resultData = toolResult.result || toolResult;
                             // Try to extract meaningful data from the result
-                            if (typeof resultData === 'object' && resultData !== null) {
+                            if (typeof resultData === "object" && resultData !== null) {
                                 if (resultData.success && resultData.items) {
                                     // This looks like a filesystem listing
                                     toolResultsSummary += `Directory listing for ${resultData.path}:\n`;
@@ -311,34 +319,36 @@ export class AgentEnhancedProvider {
                             else {
                                 toolResultsSummary += String(resultData);
                             }
-                            toolResultsSummary += '\n\n';
+                            toolResultsSummary += "\n\n";
                         }
                     }
-                    log('Tool results extracted', {
+                    log("Tool results extracted", {
                         summaryLength: toolResultsSummary.length,
-                        preview: toolResultsSummary.substring(0, 200)
+                        preview: toolResultsSummary.substring(0, 200),
                     });
                     // Create a simple, direct summary
                     const finalText = `Based on the user request "${prompt}", here's what I found:\n\n${toolResultsSummary}`;
-                    log('Final text created', {
+                    log("Final text created", {
                         textLength: finalText.length,
-                        preview: finalText.substring(0, 200)
+                        preview: finalText.substring(0, 200),
                     });
                     // Return result with the formatted text
                     return {
                         ...result,
                         text: finalText,
-                        finishReason: 'stop'
+                        finishReason: "stop",
                     };
                 }
                 catch (error) {
-                    log('Error in summary generation', { error: error instanceof Error ? error.message : String(error) });
+                    log("Error in summary generation", {
+                        error: error instanceof Error ? error.message : String(error),
+                    });
                     // Fallback: return raw tool results
                     const fallbackText = `Tool execution completed. Raw results: ${JSON.stringify(result.toolResults, null, 2)}`;
                     return {
                         ...result,
                         text: fallbackText,
-                        finishReason: 'stop'
+                        finishReason: "stop",
                     };
                 }
             }
@@ -356,14 +366,12 @@ export class AgentEnhancedProvider {
             : optionsOrPrompt;
         const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt, timeout, } = options;
         // Get combined tools (direct + MCP) if enabled
-        const tools = this.config.enableTools
-            ? await this.getCombinedTools()
-            : {};
+        const tools = this.config.enableTools ? await this.getCombinedTools() : {};
         try {
             // Parse timeout if provided
             let abortSignal;
             if (timeout) {
-                const timeoutMs = typeof timeout === 'string' ? parseTimeout(timeout) : timeout;
+                const timeoutMs = typeof timeout === "string" ? parseTimeout(timeout) : timeout;
                 if (timeoutMs !== undefined) {
                     abortSignal = AbortSignal.timeout(timeoutMs);
                 }

package/dist/lib/providers/amazonBedrock.js

@@ -1,7 +1,7 @@
 import { createAmazonBedrock } from "@ai-sdk/amazon-bedrock";
 import { streamText, generateText, Output, } from "ai";
 import { logger } from "../utils/logger.js";
-import { createTimeoutController, TimeoutError, getDefaultTimeout } from "../utils/timeout.js";
+import { createTimeoutController, TimeoutError, getDefaultTimeout, } from "../utils/timeout.js";
 // Default system context
 const DEFAULT_SYSTEM_CONTEXT = {
     systemPrompt: "You are a helpful AI assistant.",
@@ -129,7 +129,7 @@ export class AmazonBedrock {
             const options = typeof optionsOrPrompt === "string"
                 ? { prompt: optionsOrPrompt }
                 : optionsOrPrompt;
-            const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = DEFAULT_SYSTEM_CONTEXT.systemPrompt, schema, timeout = getDefaultTimeout(provider, 'stream'), } = options;
+            const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = DEFAULT_SYSTEM_CONTEXT.systemPrompt, schema, timeout = getDefaultTimeout(provider, "stream"), } = options;
             // Use schema from options or fallback parameter
             const finalSchema = schema || analysisSchema;
             logger.debug(`[${functionTag}] Stream request started`, {
@@ -141,7 +141,7 @@ export class AmazonBedrock {
                 timeout,
             });
             // Create timeout controller if timeout is specified
-            const timeoutController = createTimeoutController(timeout, provider, 'stream');
+            const timeoutController = createTimeoutController(timeout, provider, "stream");
             const streamOptions = {
                 model: this.model,
                 prompt: prompt,
@@ -149,7 +149,9 @@ export class AmazonBedrock {
                 temperature,
                 maxTokens,
                 // Add abort signal if available
-                ...(timeoutController && { abortSignal: timeoutController.controller.signal }),
+                ...(timeoutController && {
+                    abortSignal: timeoutController.controller.signal,
+                }),
                 onError: (event) => {
                     const error = event.error;
                     const errorMessage = error instanceof Error ? error.message : String(error);
@@ -234,7 +236,7 @@ export class AmazonBedrock {
             const options = typeof optionsOrPrompt === "string"
                 ? { prompt: optionsOrPrompt }
                 : optionsOrPrompt;
-            const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = DEFAULT_SYSTEM_CONTEXT.systemPrompt, schema, timeout = getDefaultTimeout(provider, 'generate'), } = options;
+            const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = DEFAULT_SYSTEM_CONTEXT.systemPrompt, schema, timeout = getDefaultTimeout(provider, "generate"), } = options;
             // Use schema from options or fallback parameter
             const finalSchema = schema || analysisSchema;
             logger.debug(`[${functionTag}] Generate text started`, {
@@ -247,7 +249,7 @@ export class AmazonBedrock {
                 timeout,
             });
             // Create timeout controller if timeout is specified
-            const timeoutController = createTimeoutController(timeout, provider, 'generate');
+            const timeoutController = createTimeoutController(timeout, provider, "generate");
             const generateOptions = {
                 model: this.model,
                 prompt: prompt,
@@ -255,7 +257,9 @@ export class AmazonBedrock {
                 temperature,
                 maxTokens,
                 // Add abort signal if available
-                ...(timeoutController && { abortSignal: timeoutController.controller.signal }),
+                ...(timeoutController && {
+                    abortSignal: timeoutController.controller.signal,
+                }),
             };
             if (finalSchema) {
                 generateOptions.experimental_output = Output.object({

package/dist/lib/providers/anthropic.js

@@ -6,7 +6,8 @@
  */
 import { AIProviderName } from "../core/types.js";
 import { logger } from "../utils/logger.js";
-import { createTimeoutController, TimeoutError, getDefaultTimeout } from "../utils/timeout.js";
+import { createTimeoutController, TimeoutError, getDefaultTimeout, } from "../utils/timeout.js";
+import { createProxyFetch } from "../proxy/proxy-fetch.js";
 export class AnthropicProvider {
     name = AIProviderName.ANTHROPIC;
     apiKey;
@@ -40,7 +41,8 @@ export class AnthropicProvider {
         };
         logger.debug(`[AnthropicProvider.makeRequest] ${stream ? "Streaming" : "Non-streaming"} request to ${url}`);
         logger.debug(`[AnthropicProvider.makeRequest] Model: ${body.model}, Max tokens: ${body.max_tokens}`);
-        const response = await fetch(url, {
+        const proxyFetch = createProxyFetch();
+        const response = await proxyFetch(url, {
             method: "POST",
             headers,
             body: JSON.stringify(body),
@@ -61,7 +63,7 @@ export class AnthropicProvider {
         const options = typeof optionsOrPrompt === "string"
             ? { prompt: optionsOrPrompt }
             : optionsOrPrompt;
-        const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = "You are Claude, an AI assistant created by Anthropic. You are helpful, harmless, and honest.", timeout = getDefaultTimeout(provider, 'generate'), } = options;
+        const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = "You are Claude, an AI assistant created by Anthropic. You are helpful, harmless, and honest.", timeout = getDefaultTimeout(provider, "generate"), } = options;
         logger.debug(`[${functionTag}] Prompt: "${prompt.substring(0, 100)}...", Temperature: ${temperature}, Max tokens: ${maxTokens}, Timeout: ${timeout}`);
         const requestBody = {
             model: this.getModel(),
@@ -76,7 +78,7 @@ export class AnthropicProvider {
             system: systemPrompt,
         };
         // Create timeout controller if timeout is specified
-        const timeoutController = createTimeoutController(timeout, provider, 'generate');
+        const timeoutController = createTimeoutController(timeout, provider, "generate");
         try {
             const response = await this.makeRequest("messages", requestBody, false, timeoutController?.controller.signal);
             const data = await response.json();
@@ -107,9 +109,9 @@ export class AnthropicProvider {
                     message: error.message,
                 });
             }
-            else if (error?.name === 'AbortError') {
+            else if (error?.name === "AbortError") {
                 // Convert AbortError to TimeoutError
-                const timeoutError = new TimeoutError(`${provider} generate operation timed out after ${timeout}`, timeoutController?.timeoutMs || 0, provider, 'generate');
+                const timeoutError = new TimeoutError(`${provider} generate operation timed out after ${timeout}`, timeoutController?.timeoutMs || 0, provider, "generate");
                 logger.error(`[${functionTag}] Timeout error`, {
                     provider,
                     timeout: timeoutController?.timeoutMs,
@@ -131,7 +133,7 @@ export class AnthropicProvider {
         const options = typeof optionsOrPrompt === "string"
             ? { prompt: optionsOrPrompt }
             : optionsOrPrompt;
-        const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = "You are Claude, an AI assistant created by Anthropic. You are helpful, harmless, and honest.", timeout = getDefaultTimeout(provider, 'stream'), } = options;
+        const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = "You are Claude, an AI assistant created by Anthropic. You are helpful, harmless, and honest.", timeout = getDefaultTimeout(provider, "stream"), } = options;
         logger.debug(`[${functionTag}] Streaming prompt: "${prompt.substring(0, 100)}...", Timeout: ${timeout}`);
         const requestBody = {
             model: this.getModel(),
@@ -147,7 +149,7 @@ export class AnthropicProvider {
             stream: true,
         };
         // Create timeout controller if timeout is specified
-        const timeoutController = createTimeoutController(timeout, provider, 'stream');
+        const timeoutController = createTimeoutController(timeout, provider, "stream");
         try {
             const response = await this.makeRequest("messages", requestBody, true, timeoutController?.controller.signal);
             if (!response.body) {
@@ -174,9 +176,9 @@ export class AnthropicProvider {
                     message: error.message,
                 });
             }
-            else if (error?.name === 'AbortError') {
+            else if (error?.name === "AbortError") {
                 // Convert AbortError to TimeoutError
-                const timeoutError = new TimeoutError(`${provider} stream operation timed out after ${timeout}`, timeoutController?.timeoutMs || 0, provider, 'stream');
+                const timeoutError = new TimeoutError(`${provider} stream operation timed out after ${timeout}`, timeoutController?.timeoutMs || 0, provider, "stream");
                 logger.error(`[${functionTag}] Timeout error`, {
                     provider,
                     timeout: timeoutController?.timeoutMs,
@@ -198,7 +200,7 @@ export class AnthropicProvider {
             while (true) {
                 // Check if aborted
                 if (signal?.aborted) {
-                    throw new Error('AbortError');
+                    throw new Error("AbortError");
                 }
                 const { done, value } = await reader.read();
                 if (done) {

package/dist/lib/providers/azureOpenAI.js

@@ -6,7 +6,7 @@
  */
 import { AIProviderName } from "../core/types.js";
 import { logger } from "../utils/logger.js";
-import { createTimeoutController, TimeoutError, getDefaultTimeout } from "../utils/timeout.js";
+import { createTimeoutController, TimeoutError, getDefaultTimeout, } from "../utils/timeout.js";
 export class AzureOpenAIProvider {
     name = AIProviderName.AZURE;
     apiKey;
@@ -74,7 +74,7 @@ export class AzureOpenAIProvider {
         const options = typeof optionsOrPrompt === "string"
             ? { prompt: optionsOrPrompt }
             : optionsOrPrompt;
-        const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = "You are a helpful AI assistant.", timeout = getDefaultTimeout(provider, 'generate'), } = options;
+        const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = "You are a helpful AI assistant.", timeout = getDefaultTimeout(provider, "generate"), } = options;
         logger.debug(`[${functionTag}] Prompt: "${prompt.substring(0, 100)}...", Temperature: ${temperature}, Max tokens: ${maxTokens}, Timeout: ${timeout}`);
         const messages = [];
         if (systemPrompt) {
@@ -93,7 +93,7 @@ export class AzureOpenAIProvider {
             max_tokens: maxTokens,
         };
         // Create timeout controller if timeout is specified
-        const timeoutController = createTimeoutController(timeout, provider, 'generate');
+        const timeoutController = createTimeoutController(timeout, provider, "generate");
         try {
             const response = await this.makeRequest(requestBody, false, timeoutController?.controller.signal);
             const data = await response.json();
@@ -124,9 +124,9 @@ export class AzureOpenAIProvider {
                     message: error.message,
                 });
             }
-            else if (error?.name === 'AbortError') {
+            else if (error?.name === "AbortError") {
                 // Convert AbortError to TimeoutError
-                const timeoutError = new TimeoutError(`${provider} generate operation timed out after ${timeout}`, timeoutController?.timeoutMs || 0, provider, 'generate');
+                const timeoutError = new TimeoutError(`${provider} generate operation timed out after ${timeout}`, timeoutController?.timeoutMs || 0, provider, "generate");
                 logger.error(`[${functionTag}] Timeout error`, {
                     provider,
                     timeout: timeoutController?.timeoutMs,
@@ -148,7 +148,7 @@ export class AzureOpenAIProvider {
         const options = typeof optionsOrPrompt === "string"
             ? { prompt: optionsOrPrompt }
             : optionsOrPrompt;
-        const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = "You are a helpful AI assistant.", timeout = getDefaultTimeout(provider, 'stream'), } = options;
+        const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = "You are a helpful AI assistant.", timeout = getDefaultTimeout(provider, "stream"), } = options;
         logger.debug(`[${functionTag}] Streaming prompt: "${prompt.substring(0, 100)}...", Timeout: ${timeout}`);
         const messages = [];
         if (systemPrompt) {
@@ -168,7 +168,7 @@ export class AzureOpenAIProvider {
             stream: true,
         };
         // Create timeout controller if timeout is specified
-        const timeoutController = createTimeoutController(timeout, provider, 'stream');
+        const timeoutController = createTimeoutController(timeout, provider, "stream");
         try {
             const response = await this.makeRequest(requestBody, true, timeoutController?.controller.signal);
             if (!response.body) {
@@ -195,9 +195,9 @@ export class AzureOpenAIProvider {
                     message: error.message,
                 });
             }
-            else if (error?.name === 'AbortError') {
+            else if (error?.name === "AbortError") {
                 // Convert AbortError to TimeoutError
-                const timeoutError = new TimeoutError(`${provider} stream operation timed out after ${timeout}`, timeoutController?.timeoutMs || 0, provider, 'stream');
+                const timeoutError = new TimeoutError(`${provider} stream operation timed out after ${timeout}`, timeoutController?.timeoutMs || 0, provider, "stream");
                 logger.error(`[${functionTag}] Timeout error`, {
                     provider,
                     timeout: timeoutController?.timeoutMs,
@@ -219,7 +219,7 @@ export class AzureOpenAIProvider {
             while (true) {
                 // Check if aborted
                 if (signal?.aborted) {
-                    throw new Error('AbortError');
+                    throw new Error("AbortError");
                 }
                 const { done, value } = await reader.read();
                 if (done) {

package/dist/lib/providers/googleAIStudio.js

@@ -1,7 +1,8 @@
 import { createGoogleGenerativeAI } from "@ai-sdk/google";
 import { streamText, generateText, Output, } from "ai";
 import { logger } from "../utils/logger.js";
-import { createTimeoutController, TimeoutError, getDefaultTimeout } from "../utils/timeout.js";
+import { createTimeoutController, TimeoutError, getDefaultTimeout, } from "../utils/timeout.js";
+import { createProxyFetch } from "../proxy/proxy-fetch.js";
 // CRITICAL: Setup environment variables early for AI SDK compatibility
 // The AI SDK specifically looks for GOOGLE_GENERATIVE_AI_API_KEY
 // We need to ensure this is set before any AI SDK operations
@@ -39,8 +40,10 @@ let _google = null;
 function getGoogleInstance() {
     if (!_google) {
         const apiKey = getGoogleAIApiKey();
+        const proxyFetch = createProxyFetch();
         _google = createGoogleGenerativeAI({
             apiKey: apiKey,
+            fetch: proxyFetch,
             headers: {
                 "X-Powered-By": "NeuroLink",
             },
@@ -110,7 +113,7 @@ export class GoogleAIStudio {
             const options = typeof optionsOrPrompt === "string"
                 ? { prompt: optionsOrPrompt }
                 : optionsOrPrompt;
-            const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = DEFAULT_SYSTEM_CONTEXT.systemPrompt, schema, tools, timeout = getDefaultTimeout(provider, 'stream'), } = options;
+            const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = DEFAULT_SYSTEM_CONTEXT.systemPrompt, schema, tools, timeout = getDefaultTimeout(provider, "stream"), } = options;
             // Use schema from options or fallback parameter
             const finalSchema = schema || analysisSchema;
             logger.debug(`[${functionTag}] Stream request started`, {
@@ -126,7 +129,7 @@ export class GoogleAIStudio {
             });
             const model = this.getModel();
             // Create timeout controller if timeout is specified
-            const timeoutController = createTimeoutController(timeout, provider, 'stream');
+            const timeoutController = createTimeoutController(timeout, provider, "stream");
             const streamOptions = {
                 model: model,
                 prompt: prompt,
@@ -135,7 +138,9 @@ export class GoogleAIStudio {
                 maxTokens,
                 ...(tools && { tools }), // Add tools if provided
                 // Add abort signal if available
-                ...(timeoutController && { abortSignal: timeoutController.controller.signal }),
+                ...(timeoutController && {
+                    abortSignal: timeoutController.controller.signal,
+                }),
                 onError: (event) => {
                     const error = event.error;
                     const errorMessage = error instanceof Error ? error.message : String(error);
@@ -219,7 +224,7 @@ export class GoogleAIStudio {
             const options = typeof optionsOrPrompt === "string"
                 ? { prompt: optionsOrPrompt }
                 : optionsOrPrompt;
-            const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = DEFAULT_SYSTEM_CONTEXT.systemPrompt, schema, tools, timeout = getDefaultTimeout(provider, 'generate'), } = options;
+            const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt = DEFAULT_SYSTEM_CONTEXT.systemPrompt, schema, tools, timeout = getDefaultTimeout(provider, "generate"), } = options;
             // Use schema from options or fallback parameter
             const finalSchema = schema || analysisSchema;
             logger.debug(`[${functionTag}] Generate request started`, {
@@ -234,7 +239,7 @@ export class GoogleAIStudio {
             });
             const model = this.getModel();
             // Create timeout controller if timeout is specified
-            const timeoutController = createTimeoutController(timeout, provider, 'generate');
+            const timeoutController = createTimeoutController(timeout, provider, "generate");
             const generateOptions = {
                 model: model,
                 prompt: prompt,
@@ -246,7 +251,9 @@ export class GoogleAIStudio {
                     maxSteps: 5, // Allow multiple steps for tool execution and response generation
                 }), // Add tools if provided
                 // Add abort signal if available
-                ...(timeoutController && { abortSignal: timeoutController.controller.signal }),
+                ...(timeoutController && {
+                    abortSignal: timeoutController.controller.signal,
+                }),
             };
             if (finalSchema) {
                 generateOptions.experimental_output = Output.object({