@juspay/neurolink 9.68.1 → 9.68.3

package/dist/lib/providers/fireworks.d.ts

@@ -1,33 +1,22 @@
 import type { AIProviderName } from "../constants/enums.js";
-import { BaseProvider } from "../core/baseProvider.js";
-import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
-import type { LanguageModel } from "../types/index.js";
+import type { NeurolinkCredentials } from "../types/index.js";
+import { OpenAIChatCompletionsProvider } from "./openaiChatCompletionsBase.js";
 /**
- * Fireworks AI Provider
+ * Fireworks AI Provider — direct HTTP, no AI SDK.
  *
  * Hosted open-model serving at api.fireworks.ai/inference/v1
  * (OpenAI-compatible). Best for low-latency at scale on Llama / Mixtral /
- * Qwen / DeepSeek.
+ * Qwen / DeepSeek. All request/stream/tool-loop orchestration lives in
+ * `OpenAIChatCompletionsProvider`; this class only declares configuration
+ * and provider-specific error mapping.
  *
  * @see https://docs.fireworks.ai/api-reference/introduction
  */
-export declare class FireworksProvider extends BaseProvider {
-    private model;
-    private apiKey;
-    private baseURL;
+export declare class FireworksProvider extends OpenAIChatCompletionsProvider {
     constructor(modelName?: string, sdk?: unknown, _region?: string, credentials?: NeurolinkCredentials["fireworks"]);
-    protected executeStream(options: StreamOptions, _analysisSchema?: ValidationSchema): Promise<StreamResult>;
-    private executeStreamInner;
     protected getProviderName(): AIProviderName;
     protected getDefaultModel(): string;
-    protected getAISDKModel(): LanguageModel;
+    protected getFallbackModelName(): string;
+    protected getFallbackModels(): string[];
     protected formatProviderError(error: unknown): Error;
-    validateConfiguration(): Promise<boolean>;
-    getConfiguration(): {
-        provider: AIProviderName;
-        model: string;
-        defaultModel: string;
-        baseURL: string;
-    };
 }
-export default FireworksProvider;

package/dist/lib/providers/fireworks.js

@@ -1,132 +1,60 @@
-import { createOpenAI } from "@ai-sdk/openai";
 import { FireworksModels } from "../constants/enums.js";
-import { BaseProvider } from "../core/baseProvider.js";
-import { DEFAULT_MAX_STEPS } from "../core/constants.js";
-import { streamAnalyticsCollector } from "../core/streamAnalytics.js";
-import { isNeuroLink } from "../neurolink.js";
-import { createLoggingFetch } from "../utils/loggingFetch.js";
-import { tracers, ATTR, withClientStreamSpan } from "../telemetry/index.js";
 import { AuthenticationError, InvalidModelError, NetworkError, ProviderError, RateLimitError, } from "../types/index.js";
 import { logger } from "../utils/logger.js";
 import { createFireworksConfig, getProviderModel, validateApiKey, } from "../utils/providerConfig.js";
-import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../utils/timeout.js";
-import { emitToolEndFromStepFinish } from "../utils/toolEndEmitter.js";
-import { resolveToolChoice } from "../utils/toolChoice.js";
-import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
-import { stepCountIs } from "../utils/tool.js";
-import { streamText } from "../utils/generation.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());
 const getDefaultFireworksModel = () => getProviderModel("FIREWORKS_MODEL", FireworksModels.DEEPSEEK_V4_PRO);
 /**
- * Fireworks AI Provider
+ * Fireworks AI Provider — direct HTTP, no AI SDK.
  *
  * Hosted open-model serving at api.fireworks.ai/inference/v1
  * (OpenAI-compatible). Best for low-latency at scale on Llama / Mixtral /
- * Qwen / DeepSeek.
+ * Qwen / DeepSeek. All request/stream/tool-loop orchestration lives in
+ * `OpenAIChatCompletionsProvider`; this class only declares configuration
+ * and provider-specific error mapping.
  *
  * @see https://docs.fireworks.ai/api-reference/introduction
  */
-export class FireworksProvider extends BaseProvider {
-    model;
-    apiKey;
-    baseURL;
+export class FireworksProvider extends OpenAIChatCompletionsProvider {
     constructor(modelName, sdk, _region, credentials) {
-        const validatedNeurolink = isNeuroLink(sdk) ? sdk : undefined;
-        super(modelName, "fireworks", validatedNeurolink);
+        // Trim the override before applying precedence. A blank/whitespace
+        // `credentials.apiKey` must NOT bypass the env key — that would build a
+        // client with an unusable bearer token and fail at request time.
         const overrideApiKey = credentials?.apiKey?.trim();
-        this.apiKey =
-            overrideApiKey && overrideApiKey.length > 0
-                ? overrideApiKey
-                : getFireworksApiKey();
-        this.baseURL =
-            credentials?.baseURL ??
-                process.env.FIREWORKS_BASE_URL ??
-                FIREWORKS_DEFAULT_BASE_URL;
-        const fireworks = createOpenAI({
-            apiKey: this.apiKey,
-            baseURL: this.baseURL,
-            fetch: createLoggingFetch("fireworks"),
-        });
-        this.model = fireworks.chat(this.modelName);
+        const apiKey = overrideApiKey && overrideApiKey.length > 0
+            ? overrideApiKey
+            : getFireworksApiKey();
+        const baseURL = credentials?.baseURL?.trim() ||
+            process.env.FIREWORKS_BASE_URL?.trim() ||
+            FIREWORKS_DEFAULT_BASE_URL;
+        super("fireworks", modelName, sdk, { baseURL, apiKey });
         logger.debug("Fireworks Provider initialized", {
             modelName: this.modelName,
             providerName: this.providerName,
-            baseURL: this.baseURL,
+            baseURL: this.config.baseURL,
         });
     }
-    async executeStream(options, _analysisSchema) {
-        return withClientStreamSpan({
-            name: "neurolink.provider.stream",
-            tracer: tracers.provider,
-            attributes: {
-                [ATTR.GEN_AI_SYSTEM]: "fireworks",
-                [ATTR.GEN_AI_MODEL]: this.modelName,
-                [ATTR.GEN_AI_OPERATION]: "stream",
-                [ATTR.NL_STREAM_MODE]: true,
-            },
-        }, async () => this.executeStreamInner(options), (r) => r.stream, (r, wrapped) => ({ ...r, stream: wrapped }));
-    }
-    async executeStreamInner(options) {
-        this.validateStreamOptions(options);
-        const startTime = Date.now();
-        const timeout = this.getTimeout(options);
-        const timeoutController = createTimeoutController(timeout, this.providerName, "stream");
-        try {
-            const shouldUseTools = !options.disableTools && this.supportsTools();
-            const tools = shouldUseTools
-                ? options.tools || (await this.getAllTools())
-                : {};
-            const messages = await this.buildMessagesForStream(options);
-            const model = await this.getAISDKModelWithMiddleware(options);
-            const result = await streamText({
-                model,
-                messages,
-                temperature: options.temperature,
-                maxOutputTokens: options.maxTokens,
-                tools,
-                stopWhen: stepCountIs(options.maxSteps || DEFAULT_MAX_STEPS),
-                toolChoice: resolveToolChoice(options, tools, shouldUseTools),
-                abortSignal: composeAbortSignals(options.abortSignal, timeoutController?.controller.signal),
-                experimental_telemetry: this.telemetryHandler.getTelemetryConfig(options),
-                experimental_repairToolCall: this.getToolCallRepairFn(options),
-                onStepFinish: ({ toolCalls, toolResults }) => {
-                    emitToolEndFromStepFinish(this.neurolink?.getEventEmitter(), toolResults);
-                    this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
-                        logger.warn("[FireworksProvider] Failed to store tool executions", {
-                            provider: this.providerName,
-                            error: error instanceof Error ? error.message : String(error),
-                        });
-                    });
-                },
-            });
-            timeoutController?.cleanup();
-            const transformedStream = this.createTextStream(result);
-            const analyticsPromise = streamAnalyticsCollector.createAnalytics(this.providerName, this.modelName, toAnalyticsStreamResult(result), Date.now() - startTime, {
-                requestId: `fireworks-stream-${Date.now()}`,
-                streamingMode: true,
-            });
-            return {
-                stream: transformedStream,
-                provider: this.providerName,
-                model: this.modelName,
-                analytics: analyticsPromise,
-                metadata: { startTime, streamId: `fireworks-${Date.now()}` },
-            };
-        }
-        catch (error) {
-            timeoutController?.cleanup();
-            throw this.handleProviderError(error);
-        }
-    }
     getProviderName() {
-        return this.providerName;
+        return "fireworks";
     }
     getDefaultModel() {
         return getDefaultFireworksModel();
     }
-    getAISDKModel() {
-        return this.model;
+    getFallbackModelName() {
+        return FireworksModels.DEEPSEEK_V4_PRO;
+    }
+    getFallbackModels() {
+        return [
+            FireworksModels.DEEPSEEK_V4_PRO,
+            FireworksModels.GLM_5P1,
+            FireworksModels.GLM_5,
+            FireworksModels.KIMI_K2P6,
+            FireworksModels.KIMI_K2P5,
+            FireworksModels.GPT_OSS_120B,
+        ];
     }
     formatProviderError(error) {
         if (error instanceof TimeoutError) {
@@ -149,17 +77,5 @@ export class FireworksProvider extends BaseProvider {
         }
         return new ProviderError(`Fireworks error: ${message}`, "fireworks");
     }
-    async validateConfiguration() {
-        return typeof this.apiKey === "string" && this.apiKey.trim().length > 0;
-    }
-    getConfiguration() {
-        return {
-            provider: this.providerName,
-            model: this.modelName,
-            defaultModel: getDefaultFireworksModel(),
-            baseURL: this.baseURL,
-        };
-    }
 }
-export default FireworksProvider;
 //# sourceMappingURL=fireworks.js.map

package/dist/lib/providers/xai.d.ts

@@ -1,33 +1,22 @@
 import type { AIProviderName } from "../constants/enums.js";
-import { BaseProvider } from "../core/baseProvider.js";
-import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
-import type { LanguageModel } from "../types/index.js";
+import type { NeurolinkCredentials } from "../types/index.js";
+import { OpenAIChatCompletionsProvider } from "./openaiChatCompletionsBase.js";
 /**
- * xAI Grok Provider
+ * xAI Grok Provider — direct HTTP, no AI SDK.
  *
- * OpenAI-compatible chat completions at api.x.ai/v1. Supports the Grok
- * family: grok-2, grok-3, grok-3-mini, grok-2-vision-latest (multimodal),
- * and grok-beta. Streaming and tool calling supported.
+ * OpenAI-compatible chat completions at api.x.ai/v1 (Grok family:
+ * grok-3, grok-3-mini, grok-2-latest, grok-2-vision-latest, grok-beta).
+ * All request/stream/tool-loop orchestration lives in
+ * `OpenAIChatCompletionsProvider`; this class only declares configuration
+ * and provider-specific error mapping.
  *
  * @see https://docs.x.ai/api
  */
-export declare class XaiProvider extends BaseProvider {
-    private model;
-    private apiKey;
-    private baseURL;
+export declare class XaiProvider extends OpenAIChatCompletionsProvider {
     constructor(modelName?: string, sdk?: unknown, _region?: string, credentials?: NeurolinkCredentials["xai"]);
-    protected executeStream(options: StreamOptions, _analysisSchema?: ValidationSchema): Promise<StreamResult>;
-    private executeStreamInner;
     protected getProviderName(): AIProviderName;
     protected getDefaultModel(): string;
-    protected getAISDKModel(): LanguageModel;
+    protected getFallbackModelName(): string;
+    protected getFallbackModels(): string[];
     protected formatProviderError(error: unknown): Error;
-    validateConfiguration(): Promise<boolean>;
-    getConfiguration(): {
-        provider: AIProviderName;
-        model: string;
-        defaultModel: string;
-        baseURL: string;
-    };
 }
-export default XaiProvider;

package/dist/lib/providers/xai.js

@@ -1,135 +1,59 @@
-import { createOpenAI } from "@ai-sdk/openai";
 import { XaiModels } from "../constants/enums.js";
-import { BaseProvider } from "../core/baseProvider.js";
-import { DEFAULT_MAX_STEPS } from "../core/constants.js";
-import { streamAnalyticsCollector } from "../core/streamAnalytics.js";
-import { isNeuroLink } from "../neurolink.js";
-import { createLoggingFetch } from "../utils/loggingFetch.js";
-import { tracers, ATTR, withClientStreamSpan } from "../telemetry/index.js";
 import { AuthenticationError, InvalidModelError, NetworkError, ProviderError, RateLimitError, } from "../types/index.js";
 import { logger } from "../utils/logger.js";
 import { createXaiConfig, getProviderModel, validateApiKey, } from "../utils/providerConfig.js";
-import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../utils/timeout.js";
-import { emitToolEndFromStepFinish } from "../utils/toolEndEmitter.js";
-import { resolveToolChoice } from "../utils/toolChoice.js";
-import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
-import { stepCountIs } from "../utils/tool.js";
-import { streamText } from "../utils/generation.js";
-/**
- * Logging fetch wrapper — masks the proxy URL on non-2xx responses so
- * stack traces and log lines don't leak the upstream's internal token /
- * tenant id (mirrors the deepseek/groq/etc. providers).
- */
+import { TimeoutError } from "../utils/timeout.js";
+import { OpenAIChatCompletionsProvider } from "./openaiChatCompletionsBase.js";
 const XAI_DEFAULT_BASE_URL = "https://api.x.ai/v1";
 const getXaiApiKey = () => validateApiKey(createXaiConfig());
 const getDefaultXaiModel = () => getProviderModel("XAI_MODEL", XaiModels.GROK_3);
 /**
- * xAI Grok Provider
+ * xAI Grok Provider — direct HTTP, no AI SDK.
  *
- * OpenAI-compatible chat completions at api.x.ai/v1. Supports the Grok
- * family: grok-2, grok-3, grok-3-mini, grok-2-vision-latest (multimodal),
- * and grok-beta. Streaming and tool calling supported.
+ * OpenAI-compatible chat completions at api.x.ai/v1 (Grok family:
+ * grok-3, grok-3-mini, grok-2-latest, grok-2-vision-latest, grok-beta).
+ * All request/stream/tool-loop orchestration lives in
+ * `OpenAIChatCompletionsProvider`; this class only declares configuration
+ * and provider-specific error mapping.
  *
  * @see https://docs.x.ai/api
  */
-export class XaiProvider extends BaseProvider {
-    model;
-    apiKey;
-    baseURL;
+export class XaiProvider extends OpenAIChatCompletionsProvider {
     constructor(modelName, sdk, _region, credentials) {
-        const validatedNeurolink = isNeuroLink(sdk) ? sdk : undefined;
-        super(modelName, "xai", validatedNeurolink);
+        // Trim the override before applying precedence. A blank/whitespace
+        // `credentials.apiKey` must NOT bypass the env key — that would build a
+        // client with an unusable bearer token and fail at request time.
         const overrideApiKey = credentials?.apiKey?.trim();
-        this.apiKey =
-            overrideApiKey && overrideApiKey.length > 0
-                ? overrideApiKey
-                : getXaiApiKey();
-        this.baseURL =
-            credentials?.baseURL ?? process.env.XAI_BASE_URL ?? XAI_DEFAULT_BASE_URL;
-        const xai = createOpenAI({
-            apiKey: this.apiKey,
-            baseURL: this.baseURL,
-            fetch: createLoggingFetch("xai"),
-        });
-        this.model = xai.chat(this.modelName);
+        const apiKey = overrideApiKey && overrideApiKey.length > 0
+            ? overrideApiKey
+            : getXaiApiKey();
+        const baseURL = credentials?.baseURL?.trim() ||
+            process.env.XAI_BASE_URL?.trim() ||
+            XAI_DEFAULT_BASE_URL;
+        super("xai", modelName, sdk, { baseURL, apiKey });
         logger.debug("xAI Provider initialized", {
             modelName: this.modelName,
             providerName: this.providerName,
-            baseURL: this.baseURL,
+            baseURL: this.config.baseURL,
         });
     }
-    async executeStream(options, _analysisSchema) {
-        return withClientStreamSpan({
-            name: "neurolink.provider.stream",
-            tracer: tracers.provider,
-            attributes: {
-                [ATTR.GEN_AI_SYSTEM]: "xai",
-                [ATTR.GEN_AI_MODEL]: this.modelName,
-                [ATTR.GEN_AI_OPERATION]: "stream",
-                [ATTR.NL_STREAM_MODE]: true,
-            },
-        }, async () => this.executeStreamInner(options), (r) => r.stream, (r, wrapped) => ({ ...r, stream: wrapped }));
-    }
-    async executeStreamInner(options) {
-        this.validateStreamOptions(options);
-        const startTime = Date.now();
-        const timeout = this.getTimeout(options);
-        const timeoutController = createTimeoutController(timeout, this.providerName, "stream");
-        try {
-            const shouldUseTools = !options.disableTools && this.supportsTools();
-            const tools = shouldUseTools
-                ? options.tools || (await this.getAllTools())
-                : {};
-            const messages = await this.buildMessagesForStream(options);
-            const model = await this.getAISDKModelWithMiddleware(options);
-            const result = await streamText({
-                model,
-                messages,
-                temperature: options.temperature,
-                maxOutputTokens: options.maxTokens,
-                tools,
-                stopWhen: stepCountIs(options.maxSteps || DEFAULT_MAX_STEPS),
-                toolChoice: resolveToolChoice(options, tools, shouldUseTools),
-                abortSignal: composeAbortSignals(options.abortSignal, timeoutController?.controller.signal),
-                experimental_telemetry: this.telemetryHandler.getTelemetryConfig(options),
-                experimental_repairToolCall: this.getToolCallRepairFn(options),
-                onStepFinish: ({ toolCalls, toolResults }) => {
-                    emitToolEndFromStepFinish(this.neurolink?.getEventEmitter(), toolResults);
-                    this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
-                        logger.warn("[XaiProvider] Failed to store tool executions", {
-                            provider: this.providerName,
-                            error: error instanceof Error ? error.message : String(error),
-                        });
-                    });
-                },
-            });
-            timeoutController?.cleanup();
-            const transformedStream = this.createTextStream(result);
-            const analyticsPromise = streamAnalyticsCollector.createAnalytics(this.providerName, this.modelName, toAnalyticsStreamResult(result), Date.now() - startTime, {
-                requestId: `xai-stream-${Date.now()}`,
-                streamingMode: true,
-            });
-            return {
-                stream: transformedStream,
-                provider: this.providerName,
-                model: this.modelName,
-                analytics: analyticsPromise,
-                metadata: { startTime, streamId: `xai-${Date.now()}` },
-            };
-        }
-        catch (error) {
-            timeoutController?.cleanup();
-            throw this.handleProviderError(error);
-        }
-    }
     getProviderName() {
-        return this.providerName;
+        return "xai";
     }
     getDefaultModel() {
         return getDefaultXaiModel();
     }
-    getAISDKModel() {
-        return this.model;
+    getFallbackModelName() {
+        return XaiModels.GROK_3_MINI;
+    }
+    getFallbackModels() {
+        return [
+            XaiModels.GROK_3,
+            XaiModels.GROK_3_MINI,
+            XaiModels.GROK_2_LATEST,
+            XaiModels.GROK_2_VISION_LATEST,
+            XaiModels.GROK_BETA,
+        ];
     }
     formatProviderError(error) {
         if (error instanceof TimeoutError) {
@@ -157,17 +81,5 @@ export class XaiProvider extends BaseProvider {
         }
         return new ProviderError(`xAI error: ${message}`, "xai");
     }
-    async validateConfiguration() {
-        return typeof this.apiKey === "string" && this.apiKey.trim().length > 0;
-    }
-    getConfiguration() {
-        return {
-            provider: this.providerName,
-            model: this.modelName,
-            defaultModel: getDefaultXaiModel(),
-            baseURL: this.baseURL,
-        };
-    }
 }
-export default XaiProvider;
 //# sourceMappingURL=xai.js.map

package/dist/providers/fireworks.d.ts

@@ -1,33 +1,22 @@
 import type { AIProviderName } from "../constants/enums.js";
-import { BaseProvider } from "../core/baseProvider.js";
-import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
-import type { LanguageModel } from "../types/index.js";
+import type { NeurolinkCredentials } from "../types/index.js";
+import { OpenAIChatCompletionsProvider } from "./openaiChatCompletionsBase.js";
 /**
- * Fireworks AI Provider
+ * Fireworks AI Provider — direct HTTP, no AI SDK.
  *
  * Hosted open-model serving at api.fireworks.ai/inference/v1
  * (OpenAI-compatible). Best for low-latency at scale on Llama / Mixtral /
- * Qwen / DeepSeek.
+ * Qwen / DeepSeek. All request/stream/tool-loop orchestration lives in
+ * `OpenAIChatCompletionsProvider`; this class only declares configuration
+ * and provider-specific error mapping.
  *
  * @see https://docs.fireworks.ai/api-reference/introduction
  */
-export declare class FireworksProvider extends BaseProvider {
-    private model;
-    private apiKey;
-    private baseURL;
+export declare class FireworksProvider extends OpenAIChatCompletionsProvider {
     constructor(modelName?: string, sdk?: unknown, _region?: string, credentials?: NeurolinkCredentials["fireworks"]);
-    protected executeStream(options: StreamOptions, _analysisSchema?: ValidationSchema): Promise<StreamResult>;
-    private executeStreamInner;
     protected getProviderName(): AIProviderName;
     protected getDefaultModel(): string;
-    protected getAISDKModel(): LanguageModel;
+    protected getFallbackModelName(): string;
+    protected getFallbackModels(): string[];
     protected formatProviderError(error: unknown): Error;
-    validateConfiguration(): Promise<boolean>;
-    getConfiguration(): {
-        provider: AIProviderName;
-        model: string;
-        defaultModel: string;
-        baseURL: string;
-    };
 }
-export default FireworksProvider;