@juspay/neurolink 11.0.0 → 11.1.0

package/dist/providers/amazonBedrock/client.js

@@ -4,6 +4,7 @@ import { createAnalytics } from "../../core/analytics.js";
 import { BaseProvider } from "../../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../../core/constants.js";
 import { AuthenticationError, ProviderError, RateLimitError, } from "../../types/index.js";
+import { classifyProviderError } from "../../utils/errorClassifier.js";
 import { isAbortError, withTimeout } from "../../utils/errorHandling.js";
 import { emitToolEndFromStepFinish } from "../../utils/toolEndEmitter.js";
 import { logger } from "../../utils/logger.js";
@@ -1724,22 +1725,39 @@ export class AmazonBedrockProvider extends BaseProvider {
         }
     }
     formatProviderError(error) {
-        // Handle AWS SDK specific errors
-        const message = error instanceof Error ? error.message : String(error);
-        if (message.includes("AccessDeniedException")) {
-            return new AuthenticationError("AWS Bedrock access denied. Check your credentials and permissions.", this.providerName);
-        }
-        if (message.includes("ValidationException")) {
-            return new ProviderError(`Validation error: ${message}`, this.providerName);
-        }
-        // Check for AWS-specific throttling BEFORE generic mapping
-        const errName = error?.name ?? "";
-        const errCode = error?.code ?? "";
-        if (errName === "ThrottlingException" ||
-            errCode === "ThrottlingException") {
-            return new RateLimitError(`Bedrock rate limit (throttled): ${error instanceof Error ? error.message : String(error)}`, "bedrock");
-        }
-        return new ProviderError(`AWS Bedrock error: ${message}`, this.providerName);
+        const rules = [
+            {
+                match: (ctx) => /AccessDeniedException/i.test(ctx.message),
+                errorClass: AuthenticationError,
+                message: "AWS Bedrock access denied. Check your credentials and permissions.",
+            },
+            {
+                // Checked before the generic ValidationException fallback — a
+                // throttling error can be shaped like a validation error in some
+                // AWS SDK error message templates, but .name/.code are authoritative.
+                match: (ctx) => ctx.errorName === "ThrottlingException" ||
+                    ctx.errorCode === "ThrottlingException",
+                errorClass: RateLimitError,
+                message: (ctx) => `Bedrock rate limit (throttled): ${ctx.message}`,
+            },
+            {
+                match: (ctx) => /ValidationException/i.test(ctx.message),
+                errorClass: ProviderError,
+                message: (ctx) => `Validation error: ${ctx.message}`,
+            },
+            {
+                match: () => true,
+                errorClass: ProviderError,
+                message: (ctx) => `AWS Bedrock error: ${ctx.message}`,
+            },
+        ];
+        // this.providerName resolves to the literal "bedrock" (getProviderName()
+        // returns it directly, and the constructor only overrides providerName
+        // when an explicit providerName is passed in) — verified matches the
+        // pre-migration hardcoded "bedrock" literal in the throttling branch, so
+        // a single classifyProviderError call using this.providerName for every
+        // rule (rather than a per-rule literal) is not a behavior change.
+        return classifyProviderError(error, rules, this.providerName, this.modelName);
     }
     /**
      * Generate embeddings for text using Amazon Bedrock embedding models

package/dist/providers/amazonSagemaker.d.ts

@@ -69,7 +69,7 @@ export declare class AmazonSageMakerProvider extends BaseProvider {
         provider: string;
         specificationVersion: "v2";
         endpointName: string;
-        modelType: "huggingface" | "mistral" | "custom" | "llama" | "claude" | "jumpstart" | undefined;
+        modelType: "huggingface" | "mistral" | "custom" | "claude" | "llama" | "jumpstart" | undefined;
         region: string;
     } | {
         capabilities: {

package/dist/providers/anthropic/client.js

@@ -13,6 +13,7 @@ import { createOAuthFetch } from "../../proxy/oauthFetch.js";
 import { createProxyFetch } from "../../proxy/proxyFetch.js";
 import { getCapturedLimitSnapshot, getCapturedResponseHeaders, logClaudeLimitSnapshot, runInLimitCaptureScope, setLimitSpanAttributes, withLimitCapture, wrapFetchWithLimitCapture, } from "./rateLimitCapture.js";
 import { AuthenticationError, NetworkError, ProviderError, RateLimitError, } from "../../types/index.js";
+import { classifyProviderError } from "../../utils/errorClassifier.js";
 import { logger } from "../../utils/logger.js";
 import { ANTHROPIC_ELISION_NOTE, planAnthropicLoopReclaim, previewAnthropicToolResultText, } from "../../context/anthropicLoopGuard.js";
 import { getAvailableInputTokens } from "../../constants/contextWindows.js";
@@ -22,13 +23,14 @@ import { ANTHROPIC_MAX_CACHE_BREAKPOINTS, applyAnthropicHistoryCacheBreakpoints,
 import { calculateCost } from "../../utils/pricing.js";
 import { resolveDeferredTool } from "../../tools/toolDiscovery.js";
 import { createAnthropicConfig, getProviderModel, validateApiKey, } from "../../utils/providerConfig.js";
-import { composeAbortSignals, createTimeoutController, mergeAbortSignals, TimeoutError, } from "../../utils/timeout.js";
+import { composeAbortSignals, createTimeoutController, mergeAbortSignals, } from "../../utils/timeout.js";
 import { resolveToolChoice } from "../../utils/toolChoice.js";
 import { emitToolEndFromStepFinish } from "../../utils/toolEndEmitter.js";
 import { NoOutputGeneratedError } from "../../utils/generationErrors.js";
 import { buildNoOutputSentinel, stampNoOutputSpan, } from "../../utils/noOutputSentinel.js";
 import { convertZodToJsonSchema } from "../../utils/schemaConversion.js";
 import { resolveClaudeMaxTokens } from "../../utils/tokenLimits.js";
+import { withProviderRetry } from "../../utils/providerRetry.js";
 import { toAnthropicImageBlock, fileToAnthropicBlock, } from "../anthropicImageBlocks.js";
 import { resolveSamplingParams } from "../../models/modelRegistry.js";
 import { createChunkQueue, createDeferredAnalytics, stringifyToolInput, } from "../openaiChatCompletionsClient.js";
@@ -1315,37 +1317,40 @@ export class AnthropicProvider extends BaseProvider {
         return delegatingModel;
     }
     formatProviderError(error) {
-        if (error instanceof TimeoutError) {
-            return new NetworkError(`Request timed out after ${error.timeout}ms`, this.providerName);
-        }
-        const errorRecord = error;
-        const message = typeof errorRecord?.message === "string"
-            ? errorRecord.message
-            : "Unknown error";
-        if (message.includes("API_KEY_INVALID") ||
-            message.includes("Invalid API key")) {
-            return new AuthenticationError("Invalid Anthropic API key. Please check your ANTHROPIC_API_KEY environment variable.", this.providerName);
-        }
-        if (message.includes("rate limit") ||
-            message.includes("too_many_requests") ||
-            message.includes("429")) {
-            return new RateLimitError("Anthropic rate limit exceeded. Please try again later.", this.providerName);
-        }
-        if (message.includes("ECONNRESET") ||
-            message.includes("ENOTFOUND") ||
-            message.includes("ECONNREFUSED") ||
-            message.includes("network") ||
-            message.includes("connection")) {
-            return new NetworkError(`Connection error: ${message}`, this.providerName);
-        }
-        if (message.includes("500") ||
-            message.includes("502") ||
-            message.includes("503") ||
-            message.includes("504") ||
-            message.includes("server error")) {
-            return new ProviderError(`Server error: ${message}`, this.providerName);
-        }
-        return new ProviderError(`Anthropic error: ${message}`, this.providerName);
+        const rules = [
+            {
+                // Plan 02's mocked-contract work documented a pre-existing gap: the
+                // Anthropic SDK formats auth failures as a bare "401 <msg>" string,
+                // which the message-text regex alone does not catch — mirrors
+                // Vertex's statusCode === 401 fallback in this same task.
+                match: (ctx) => ctx.statusCode === 401 ||
+                    /API_KEY_INVALID|Invalid API key/i.test(ctx.message),
+                errorClass: AuthenticationError,
+                message: "Invalid Anthropic API key. Please check your ANTHROPIC_API_KEY environment variable.",
+            },
+            {
+                match: (ctx) => ctx.statusCode === 429 ||
+                    /rate limit|too_many_requests|429/i.test(ctx.message),
+                errorClass: RateLimitError,
+                message: "Anthropic rate limit exceeded. Please try again later.",
+            },
+            {
+                match: (ctx) => /ECONNRESET|ENOTFOUND|ECONNREFUSED|network|connection/i.test(ctx.message),
+                errorClass: NetworkError,
+                message: (ctx) => `Connection error: ${ctx.message}`,
+            },
+            {
+                match: (ctx) => /500|502|503|504|server error/i.test(ctx.message),
+                errorClass: ProviderError,
+                message: (ctx) => `Server error: ${ctx.message}`,
+            },
+            {
+                match: () => true,
+                errorClass: ProviderError,
+                message: (ctx) => `Anthropic error: ${ctx.message}`,
+            },
+        ];
+        return classifyProviderError(error, rules, this.providerName, this.modelName);
     }
     // executeGenerate removed - BaseProvider handles all generation with tools
     /**
@@ -1662,9 +1667,9 @@ export class AnthropicProvider extends BaseProvider {
                     ...(anthropicToolChoice ? { tool_choice: anthropicToolChoice } : {}),
                     ...(thinking ? { thinking } : {}),
                 };
-                const events = await client.messages.create(params, {
+                const events = await withProviderRetry(() => client.messages.create(params, {
                     signal: abortSignal ?? undefined,
-                });
+                }), trace.getActiveSpan() ?? undefined, `${this.providerName} stream`);
                 // Per-step accumulators, keyed by content-block index so blocks are
                 // replayed to the conversation in order (thinking blocks must be
                 // passed back with their signatures when tool use continues a turn).

package/dist/providers/azureOpenai.js

@@ -1,8 +1,8 @@
 import { APIVersions } from "../constants/enums.js";
-import { AuthenticationError, NetworkError, ProviderError, } from "../types/index.js";
+import { AuthenticationError } from "../types/index.js";
 import { logger } from "../utils/logger.js";
+import { classifyProviderError, DEFAULT_ERROR_RULES, } from "../utils/errorClassifier.js";
 import { createAzureAPIKeyConfig, createAzureEndpointConfig, validateApiKey, } from "../utils/providerConfig.js";
-import { TimeoutError } from "../utils/timeout.js";
 import { transformParamsForLogging } from "../utils/transformationUtils.js";
 import { OpenAIChatCompletionsProvider } from "./openaiChatCompletionsBase.js";
 import { requiresMaxCompletionTokens } from "./openaiChatCompletionsClient.js";
@@ -158,19 +158,21 @@ export class AzureOpenAIProvider extends OpenAIChatCompletionsProvider {
         return this.azureDeployment;
     }
     formatProviderError(error) {
-        if (error instanceof TimeoutError) {
-            return new NetworkError(`Request timed out: ${error.message}`, "azure");
-        }
-        const errorObj = error;
-        if (errorObj?.message &&
-            typeof errorObj.message === "string" &&
-            errorObj.message.includes("401")) {
-            return new AuthenticationError("Invalid Azure OpenAI API key or endpoint.", "azure");
-        }
-        const message = errorObj?.message && typeof errorObj.message === "string"
-            ? errorObj.message
-            : "Unknown error";
-        return new ProviderError(`Azure OpenAI error: ${message}`, "azure");
+        // RULING 1 fix: previously only a raw "401" substring was special-cased —
+        // everything else (429, 404, network errors, 5xx) fell through to one
+        // generic ProviderError. The 401 branch below is preserved byte-for-byte
+        // (message text + the raw-substring match style); DEFAULT_ERROR_RULES now
+        // gives 429/404/network/5xx their correct classifications instead of the
+        // old undifferentiated catch-all.
+        const rules = [
+            {
+                match: (ctx) => /401/.test(ctx.message),
+                errorClass: AuthenticationError,
+                message: "Invalid Azure OpenAI API key or endpoint.",
+            },
+            ...DEFAULT_ERROR_RULES,
+        ];
+        return classifyProviderError(error, rules, "azure", this.modelName);
     }
     // ===========================================================================
     // New overridable hooks (provided by the base-enhancement branch)

package/dist/providers/cloudflare.js

@@ -1,9 +1,9 @@
 import { CloudflareModels } from "../constants/enums.js";
-import { AuthenticationError, InvalidModelError, NetworkError, ProviderError, RateLimitError, } from "../types/index.js";
+import { AuthenticationError } from "../types/index.js";
 import { logger } from "../utils/logger.js";
 import { redactUrlCredentials } from "../utils/logSanitize.js";
+import { classifyProviderError, DEFAULT_ERROR_RULES, } from "../utils/errorClassifier.js";
 import { createCloudflareConfig, getProviderModel, validateApiKey, } from "../utils/providerConfig.js";
-import { TimeoutError } from "../utils/timeout.js";
 import { OpenAIChatCompletionsProvider } from "./openaiChatCompletionsBase.js";
 /**
  * Cloudflare Workers AI exposes an OpenAI-compatible endpoint scoped per
@@ -68,24 +68,15 @@ export class CloudflareProvider extends OpenAIChatCompletionsProvider {
         ];
     }
     formatProviderError(error) {
-        if (error instanceof TimeoutError) {
-            return new NetworkError(`Request timed out: ${error.message}`, "cloudflare");
-        }
-        const errorRecord = error;
-        const message = typeof errorRecord?.message === "string"
-            ? errorRecord.message
-            : "Unknown error";
-        if (message.includes("Invalid API key") ||
-            message.includes("Authentication") ||
-            message.includes("401")) {
-            return new AuthenticationError("Invalid Cloudflare API key. Use a token with Workers AI Read+Write scope. Get one at https://dash.cloudflare.com/profile/api-tokens", "cloudflare");
-        }
-        if (message.includes("rate limit") || message.includes("429")) {
-            return new RateLimitError("Cloudflare Workers AI rate limit exceeded. Free-tier neurons reset daily.", "cloudflare");
-        }
-        if (message.includes("model_not_found") || message.includes("404")) {
-            return new InvalidModelError(`Cloudflare model '${this.modelName}' not found. Browse https://developers.cloudflare.com/workers-ai/models/`, "cloudflare");
-        }
-        return new ProviderError(`Cloudflare Workers AI error: ${message}`, "cloudflare");
+        const rules = [
+            {
+                match: (ctx) => ctx.statusCode === 401 ||
+                    /Invalid API key|Authentication/i.test(ctx.message),
+                errorClass: AuthenticationError,
+                message: "Invalid Cloudflare API key. Use a token with Workers AI Read+Write scope. Get one at https://dash.cloudflare.com/profile/api-tokens",
+            },
+            ...DEFAULT_ERROR_RULES,
+        ];
+        return classifyProviderError(error, rules, "cloudflare", this.modelName);
     }
 }

package/dist/providers/cohere.js

@@ -1,10 +1,11 @@
 import { CohereModels } from "../constants/enums.js";
-import { AuthenticationError, InvalidModelError, NetworkError, ProviderError, RateLimitError, } from "../types/index.js";
+import { AuthenticationError, InvalidModelError, ProviderError, RateLimitError, } from "../types/index.js";
+import { classifyProviderError, DEFAULT_ERROR_RULES, } from "../utils/errorClassifier.js";
 import { logger } from "../utils/logger.js";
 import { redactUrlCredentials } from "../utils/logSanitize.js";
 import { createProxyFetch } from "../proxy/proxyFetch.js";
 import { createCohereConfig, getProviderModel, validateApiKey, } from "../utils/providerConfig.js";
-import { createTimeoutController, TimeoutError } from "../utils/timeout.js";
+import { createTimeoutController } from "../utils/timeout.js";
 import { stripTrailingSlash } from "./openaiChatCompletionsClient.js";
 import { OpenAIChatCompletionsProvider } from "./openaiChatCompletionsBase.js";
 const COHERE_DEFAULT_BASE_URL = "https://api.cohere.com/compatibility/v1";
@@ -59,29 +60,34 @@ export class CohereProvider extends OpenAIChatCompletionsProvider {
         ];
     }
     formatProviderError(error) {
-        if (error instanceof TimeoutError) {
-            return new NetworkError(`Request timed out: ${error.message}`, "cohere");
-        }
-        const errorRecord = error;
-        const message = typeof errorRecord?.message === "string"
-            ? errorRecord.message
-            : "Unknown error";
-        if (message.includes("invalid api token") ||
-            message.includes("Authentication") ||
-            message.includes("401") ||
-            message.includes("invalid_api_token")) {
-            return new AuthenticationError("Invalid Cohere API key. Check COHERE_API_KEY. Get one at https://dashboard.cohere.com/api-keys", "cohere");
-        }
-        if (message.includes("rate limit") || message.includes("429")) {
-            return new RateLimitError("Cohere rate limit exceeded. Back off and retry.", "cohere");
-        }
-        if (message.includes("model_not_found") || message.includes("404")) {
-            return new InvalidModelError(`Cohere model '${this.modelName}' not found. Use command-r, command-r-plus, or command-r7b-12-2024.`, "cohere");
-        }
-        if (message.includes("trial limit") || message.includes("trial_limit")) {
-            return new ProviderError("Cohere trial usage limit exceeded. Upgrade at https://dashboard.cohere.com/billing.", "cohere");
-        }
-        return new ProviderError(`Cohere error: ${message}`, "cohere");
+        const rules = [
+            // "invalid api token" is checked lowercase (case-sensitive substring, as
+            // in the original) — deliberately NOT normalized to match the family's
+            // more common "Invalid API key" capitalization, since Cohere's actual
+            // wire message differs.
+            {
+                match: (ctx) => /invalid api token|Authentication|401|invalid_api_token/.test(ctx.message),
+                errorClass: AuthenticationError,
+                message: "Invalid Cohere API key. Check COHERE_API_KEY. Get one at https://dashboard.cohere.com/api-keys",
+            },
+            {
+                match: (ctx) => /rate limit|429/.test(ctx.message),
+                errorClass: RateLimitError,
+                message: "Cohere rate limit exceeded. Back off and retry.",
+            },
+            {
+                match: (ctx) => /model_not_found|404/.test(ctx.message),
+                errorClass: InvalidModelError,
+                message: () => `Cohere model '${this.modelName}' not found. Use command-r, command-r-plus, or command-r7b-12-2024.`,
+            },
+            {
+                match: (ctx) => /trial limit|trial_limit/.test(ctx.message),
+                errorClass: ProviderError,
+                message: "Cohere trial usage limit exceeded. Upgrade at https://dashboard.cohere.com/billing.",
+            },
+            ...DEFAULT_ERROR_RULES,
+        ];
+        return classifyProviderError(error, rules, "cohere", this.modelName);
     }
     async validateConfiguration() {
         return (typeof this.config.apiKey === "string" &&

package/dist/providers/deepseek.js

@@ -1,9 +1,9 @@
 import { DeepSeekModels } from "../constants/enums.js";
-import { AuthenticationError, InvalidModelError, NetworkError, ProviderError, RateLimitError, } from "../types/index.js";
+import { AuthenticationError, InvalidModelError, ProviderError, } from "../types/index.js";
 import { logger } from "../utils/logger.js";
 import { redactUrlCredentials } from "../utils/logSanitize.js";
+import { classifyProviderError, DEFAULT_ERROR_RULES, } from "../utils/errorClassifier.js";
 import { createDeepSeekConfig, getProviderModel, validateApiKey, } from "../utils/providerConfig.js";
-import { TimeoutError } from "../utils/timeout.js";
 import { OpenAIChatCompletionsProvider } from "./openaiChatCompletionsBase.js";
 const DEEPSEEK_BASE_URL = "https://api.deepseek.com";
 const getDeepSeekApiKey = () => {
@@ -69,30 +69,27 @@ export class DeepSeekProvider extends OpenAIChatCompletionsProvider {
         return getDefaultDeepSeekModel();
     }
     formatProviderError(error) {
-        if (error instanceof TimeoutError) {
-            return new NetworkError(`Request timed out: ${error.message}`, "deepseek");
-        }
-        const errorRecord = error;
-        const message = typeof errorRecord?.message === "string"
-            ? errorRecord.message
-            : "Unknown error";
-        if (message.includes("Invalid API key") ||
-            message.includes("Authentication") ||
-            message.includes("401")) {
-            return new AuthenticationError("Invalid DeepSeek API key. Please check your DEEPSEEK_API_KEY environment variable.", "deepseek");
-        }
-        if (message.includes("rate limit") || message.includes("429")) {
-            return new RateLimitError("DeepSeek rate limit exceeded", "deepseek");
-        }
-        if (message.includes("Insufficient Balance") ||
-            message.includes("insufficient_balance") ||
-            message.includes("402")) {
-            return new ProviderError("DeepSeek account has insufficient balance. Top up at https://platform.deepseek.com/usage", "deepseek");
-        }
-        if (message.includes("model_not_found") || message.includes("404")) {
-            return new InvalidModelError(`DeepSeek model '${this.modelName}' not found. Use 'deepseek-chat' or 'deepseek-reasoner'.`, "deepseek");
-        }
-        return new ProviderError(`DeepSeek error: ${message}`, "deepseek");
+        const rules = [
+            {
+                match: (ctx) => ctx.statusCode === 401 ||
+                    /Invalid API key|Authentication/i.test(ctx.message),
+                errorClass: AuthenticationError,
+                message: "Invalid DeepSeek API key. Please check your DEEPSEEK_API_KEY environment variable.",
+            },
+            {
+                match: (ctx) => ctx.statusCode === 402 ||
+                    /Insufficient Balance|insufficient_balance/i.test(ctx.message),
+                errorClass: ProviderError,
+                message: "DeepSeek account has insufficient balance. Top up at https://platform.deepseek.com/usage",
+            },
+            {
+                match: (ctx) => /model_not_found/i.test(ctx.message),
+                errorClass: InvalidModelError,
+                message: (ctx) => `DeepSeek model '${ctx.modelName}' not found. Use 'deepseek-chat' or 'deepseek-reasoner'.`,
+            },
+            ...DEFAULT_ERROR_RULES,
+        ];
+        return classifyProviderError(error, rules, "deepseek", this.modelName);
     }
     // ===========================================================================
     // Optional hooks — provider-specific quirks

package/dist/providers/fireworks.js

@@ -1,9 +1,9 @@
 import { FireworksModels } from "../constants/enums.js";
-import { AuthenticationError, InvalidModelError, NetworkError, ProviderError, RateLimitError, } from "../types/index.js";
+import { AuthenticationError } from "../types/index.js";
 import { logger } from "../utils/logger.js";
 import { redactUrlCredentials } from "../utils/logSanitize.js";
+import { classifyProviderError, DEFAULT_ERROR_RULES, } from "../utils/errorClassifier.js";
 import { createFireworksConfig, getProviderModel, validateApiKey, } from "../utils/providerConfig.js";
-import { TimeoutError } from "../utils/timeout.js";
 import { OpenAIChatCompletionsProvider } from "./openaiChatCompletionsBase.js";
 const FIREWORKS_DEFAULT_BASE_URL = "https://api.fireworks.ai/inference/v1";
 const getFireworksApiKey = () => validateApiKey(createFireworksConfig());
@@ -58,24 +58,15 @@ export class FireworksProvider extends OpenAIChatCompletionsProvider {
         ];
     }
     formatProviderError(error) {
-        if (error instanceof TimeoutError) {
-            return new NetworkError(`Request timed out: ${error.message}`, "fireworks");
-        }
-        const errorRecord = error;
-        const message = typeof errorRecord?.message === "string"
-            ? errorRecord.message
-            : "Unknown error";
-        if (message.includes("Invalid API key") ||
-            message.includes("Authentication") ||
-            message.includes("401")) {
-            return new AuthenticationError("Invalid Fireworks API key. Get one at https://fireworks.ai/account/api-keys", "fireworks");
-        }
-        if (message.includes("rate limit") || message.includes("429")) {
-            return new RateLimitError("Fireworks rate limit exceeded. Back off and retry.", "fireworks");
-        }
-        if (message.includes("model_not_found") || message.includes("404")) {
-            return new InvalidModelError(`Fireworks model '${this.modelName}' not found. Browse https://fireworks.ai/models`, "fireworks");
-        }
-        return new ProviderError(`Fireworks error: ${message}`, "fireworks");
+        const rules = [
+            {
+                match: (ctx) => ctx.statusCode === 401 ||
+                    /Invalid API key|Authentication/i.test(ctx.message),
+                errorClass: AuthenticationError,
+                message: "Invalid Fireworks API key. Get one at https://fireworks.ai/account/api-keys",
+            },
+            ...DEFAULT_ERROR_RULES,
+        ];
+        return classifyProviderError(error, rules, "fireworks", this.modelName);
     }
 }

package/dist/providers/googleAiStudio/client.d.ts

@@ -6,6 +6,7 @@ export declare class GoogleAIStudioProvider extends BaseProvider {
     private credentials?;
     constructor(modelName?: string, sdk?: unknown, credentials?: {
         apiKey?: string;
+        baseURL?: string;
     });
     getProviderName(): AIProviderName;
     getDefaultModel(): string;
@@ -86,5 +87,6 @@ export declare class GoogleAIStudioProvider extends BaseProvider {
      */
     embedMany(texts: string[], modelName?: string): Promise<number[][]>;
     private getApiKey;
+    private getBaseURL;
 }
 export default GoogleAIStudioProvider;