@juspay/neurolink 9.67.0 → 9.67.2

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

@@ -0,0 +1,67 @@
+/**
+ * Shared OpenAI chat-completions wire-format helpers used by providers that
+ * talk to an OpenAI-shaped /chat/completions endpoint (openai-compatible,
+ * litellm, groq, perplexity, xai, fireworks, togetherAi, cohere, cloudflare,
+ * huggingFace, llamaCpp, lmStudio, deepseek, nvidiaNim, and openAI itself).
+ *
+ * Everything in this module is provider-agnostic: pure functions that convert
+ * between NeuroLink-shaped values and the OpenAI wire format, plus the SSE
+ * parser + queue primitives a streaming provider needs. Provider classes own
+ * their own orchestration (executeStream + runStreamLoop) for now — that
+ * extraction is a follow-up PR.
+ *
+ * Nothing here imports from "ai" or "@ai-sdk/*". The whole point of this
+ * module is to be the native replacement for the AI SDK's OpenAI wrapper.
+ */
+import type { OpenAICompatBuildBodyArgs, OpenAICompatChatMessage, OpenAICompatChatRequest, OpenAICompatChatTool, OpenAICompatMessage, OpenAICompatMessageContent, OpenAICompatResponseFormat, OpenAICompatSSEResult, OpenAICompatStreamChunk, OpenAICompatToolChoiceWire, OpenAICompatV3CallToolChoice, OpenAICompatV3CallTools, Tool } from "../types/index.js";
+export declare const stripTrailingSlash: (s: string) => string;
+export declare const safeStringify: (value: unknown) => string;
+export declare const stringifyToolInput: (input: unknown) => string;
+export declare const stringifyToolOutput: (output: unknown) => string;
+export declare const imageDataToURL: (data: unknown) => string | undefined;
+export declare const convertContentForOpenAI: (content: unknown) => string | OpenAICompatMessageContent[];
+export declare const messageBuilderToOpenAI: (messages: ReadonlyArray<OpenAICompatMessage>) => OpenAICompatChatMessage[];
+export declare const buildToolsForOpenAI: (tools: Record<string, Tool>) => OpenAICompatChatTool[] | undefined;
+export declare const v3ToolsToOpenAI: (tools: OpenAICompatV3CallTools | undefined) => OpenAICompatChatTool[] | undefined;
+export declare const v3ToolChoiceToOpenAI: (choice: OpenAICompatV3CallToolChoice) => OpenAICompatToolChoiceWire | undefined;
+export declare const v3ResponseFormatToOpenAI: (rf: {
+    type: "text" | "json";
+    schema?: Record<string, unknown>;
+    name?: string;
+    description?: string;
+}) => OpenAICompatResponseFormat | undefined;
+export declare const mapNeuroLinkToolChoice: (choice: unknown) => OpenAICompatToolChoiceWire | undefined;
+export declare const buildBody: (args: OpenAICompatBuildBodyArgs) => OpenAICompatChatRequest;
+export declare const parseSSEStream: (body: ReadableStream<Uint8Array>, onTextDelta: (delta: string) => void) => Promise<OpenAICompatSSEResult>;
+export declare const buildAPIError: (url: string, body: OpenAICompatChatRequest, res: Response) => Promise<Error>;
+export declare const createDeferredAnalytics: () => {
+    usagePromise: Promise<{
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    }>;
+    finishPromise: Promise<string>;
+    resolveUsage: (u: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    }) => void;
+    resolveFinish: (reason: string) => void;
+};
+export declare const createChunkQueue: () => {
+    pushChunk: (c: OpenAICompatStreamChunk) => void;
+    nextChunk: () => Promise<OpenAICompatStreamChunk>;
+};
+export declare const mergeUsage: (a: {
+    prompt_tokens?: number;
+    completion_tokens?: number;
+    total_tokens?: number;
+} | undefined, b: {
+    prompt_tokens?: number;
+    completion_tokens?: number;
+    total_tokens?: number;
+} | undefined) => {
+    prompt_tokens?: number;
+    completion_tokens?: number;
+    total_tokens?: number;
+} | undefined;

package/dist/lib/providers/openaiChatCompletionsClient.js

@@ -0,0 +1,526 @@
+/**
+ * Shared OpenAI chat-completions wire-format helpers used by providers that
+ * talk to an OpenAI-shaped /chat/completions endpoint (openai-compatible,
+ * litellm, groq, perplexity, xai, fireworks, togetherAi, cohere, cloudflare,
+ * huggingFace, llamaCpp, lmStudio, deepseek, nvidiaNim, and openAI itself).
+ *
+ * Everything in this module is provider-agnostic: pure functions that convert
+ * between NeuroLink-shaped values and the OpenAI wire format, plus the SSE
+ * parser + queue primitives a streaming provider needs. Provider classes own
+ * their own orchestration (executeStream + runStreamLoop) for now — that
+ * extraction is a follow-up PR.
+ *
+ * Nothing here imports from "ai" or "@ai-sdk/*". The whole point of this
+ * module is to be the native replacement for the AI SDK's OpenAI wrapper.
+ */
+import { createParser } from "eventsource-parser";
+import { convertZodToJsonSchema } from "../utils/schemaConversion.js";
+export const stripTrailingSlash = (s) => s.replace(/\/+$/, "");
+export const safeStringify = (value) => {
+    try {
+        return JSON.stringify(value ?? "");
+    }
+    catch {
+        return String(value ?? "");
+    }
+};
+export const stringifyToolInput = (input) => {
+    if (typeof input === "string") {
+        return input;
+    }
+    try {
+        return JSON.stringify(input ?? {});
+    }
+    catch {
+        return "{}";
+    }
+};
+// V3 tool-result `output` is a tagged union ({type:"text"|"json"|...}).
+// Serialize each variant the way an OpenAI-compatible endpoint expects
+// to read it as the `content` of a `role: "tool"` message.
+export const stringifyToolOutput = (output) => {
+    if (output === null || output === undefined) {
+        return "";
+    }
+    if (typeof output === "string") {
+        return output;
+    }
+    if (typeof output !== "object") {
+        return String(output);
+    }
+    const o = output;
+    switch (o.type) {
+        case "text":
+            return typeof o.value === "string" ? o.value : safeStringify(o.value);
+        case "json":
+            return safeStringify(o.value);
+        case "execution-denied":
+            return `Tool execution denied${o.reason ? `: ${o.reason}` : ""}`;
+        case "error-text":
+            return typeof o.value === "string" ? o.value : safeStringify(o.value);
+        case "error-json":
+            return safeStringify(o.value);
+        case "content":
+            if (Array.isArray(o.value)) {
+                return o.value
+                    .map((p) => {
+                    if (p &&
+                        typeof p === "object" &&
+                        p.type === "text") {
+                        return String(p.text ?? "");
+                    }
+                    return "";
+                })
+                    .filter((s) => s.length > 0)
+                    .join("\n");
+            }
+            return "";
+        default:
+            return safeStringify(output);
+    }
+};
+export const imageDataToURL = (data) => {
+    if (typeof data === "string") {
+        if (data.startsWith("data:") || /^https?:\/\//i.test(data)) {
+            return data;
+        }
+        return `data:image/png;base64,${data}`;
+    }
+    if (data instanceof URL) {
+        return data.toString();
+    }
+    if (data instanceof Uint8Array) {
+        return `data:image/png;base64,${Buffer.from(data).toString("base64")}`;
+    }
+    return undefined;
+};
+export const convertContentForOpenAI = (content) => {
+    if (typeof content === "string") {
+        return content;
+    }
+    if (!Array.isArray(content)) {
+        return safeStringify(content);
+    }
+    const out = [];
+    for (const part of content) {
+        if (typeof part === "string") {
+            out.push({ type: "text", text: part });
+            continue;
+        }
+        if (!part || typeof part !== "object") {
+            continue;
+        }
+        const p = part;
+        if (p.type === "text") {
+            out.push({
+                type: "text",
+                text: part.text ?? "",
+            });
+        }
+        else if (p.type === "image" || p.type === "image_url") {
+            const data = part.image ??
+                part.data ??
+                part.url;
+            const url = imageDataToURL(data);
+            if (url) {
+                out.push({ type: "image_url", image_url: { url } });
+            }
+        }
+    }
+    if (out.length === 1 && out[0].type === "text") {
+        return out[0].text;
+    }
+    return out;
+};
+export const messageBuilderToOpenAI = (messages) => {
+    const out = [];
+    for (const msg of messages) {
+        switch (msg.role) {
+            case "system":
+                out.push({
+                    role: "system",
+                    content: typeof msg.content === "string"
+                        ? msg.content
+                        : safeStringify(msg.content),
+                });
+                break;
+            case "user":
+                out.push({
+                    role: "user",
+                    content: convertContentForOpenAI(msg.content),
+                });
+                break;
+            case "assistant": {
+                const parts = Array.isArray(msg.content) ? msg.content : [msg.content];
+                const text = [];
+                const toolCalls = [];
+                for (const part of parts) {
+                    if (part && typeof part === "object") {
+                        const p = part;
+                        if (p.type === "text") {
+                            text.push({
+                                type: "text",
+                                text: part.text ?? "",
+                            });
+                        }
+                        else if (p.type === "tool-call") {
+                            const tc = part;
+                            toolCalls.push({
+                                id: tc.toolCallId ?? "",
+                                type: "function",
+                                function: {
+                                    name: tc.toolName ?? "",
+                                    arguments: stringifyToolInput(tc.input),
+                                },
+                            });
+                        }
+                    }
+                    else if (typeof part === "string") {
+                        text.push({ type: "text", text: part });
+                    }
+                }
+                const flat = text.length === 0
+                    ? null
+                    : text.length === 1 && text[0].type === "text"
+                        ? text[0].text
+                        : text;
+                out.push({
+                    role: "assistant",
+                    content: flat,
+                    ...(toolCalls.length > 0 ? { tool_calls: toolCalls } : {}),
+                });
+                break;
+            }
+            case "tool": {
+                // V3 tool messages carry `{ toolCallId, output }` per content[] entry,
+                // not at the top-level. Emit one OpenAI `role: "tool"` message per
+                // tool-result part so the model can correlate by tool_call_id.
+                if (Array.isArray(msg.content)) {
+                    for (const part of msg.content) {
+                        if (!part || typeof part !== "object") {
+                            continue;
+                        }
+                        const p = part;
+                        if (p.type === "tool-result") {
+                            out.push({
+                                role: "tool",
+                                tool_call_id: p.toolCallId ?? "",
+                                content: stringifyToolOutput(p.output),
+                            });
+                        }
+                    }
+                }
+                else if (typeof msg.content === "string") {
+                    out.push({
+                        role: "tool",
+                        tool_call_id: msg.toolCallId ?? "",
+                        content: msg.content,
+                    });
+                }
+                break;
+            }
+        }
+    }
+    return out;
+};
+export const buildToolsForOpenAI = (tools) => {
+    const entries = Object.entries(tools);
+    if (entries.length === 0) {
+        return undefined;
+    }
+    const out = [];
+    for (const [name, tool] of entries) {
+        const t = tool;
+        const rawSchema = t.inputSchema ?? t.parameters;
+        // tool.inputSchema may be a Zod schema, an AI SDK jsonSchema() wrapper,
+        // or plain JSON Schema — convertZodToJsonSchema normalizes all three.
+        // Sending raw Zod internals (with `_def`) gets rejected by most
+        // OpenAI-compatible endpoints.
+        const parameters = rawSchema
+            ? convertZodToJsonSchema(rawSchema)
+            : { type: "object", properties: {} };
+        out.push({
+            type: "function",
+            function: {
+                name,
+                ...(t.description ? { description: t.description } : {}),
+                parameters,
+            },
+        });
+    }
+    return out;
+};
+// V3 → OpenAI conversion helpers used by the non-streaming `doGenerate`
+// path that BaseProvider's `generate()` still drives via the AI SDK's
+// `generateText`. The streaming path doesn't need these — it consumes
+// NeuroLink-shaped options directly.
+export const v3ToolsToOpenAI = (tools) => {
+    if (!tools || tools.length === 0) {
+        return undefined;
+    }
+    const out = [];
+    for (const t of tools) {
+        if (t.type === "function") {
+            out.push({
+                type: "function",
+                function: {
+                    name: t.name,
+                    ...(t.description ? { description: t.description } : {}),
+                    parameters: t.inputSchema,
+                    ...(t.strict !== undefined ? { strict: t.strict } : {}),
+                },
+            });
+        }
+        // provider-defined V3 tools are silently dropped here — they have no
+        // OpenAI chat-completions equivalent.
+    }
+    return out.length > 0 ? out : undefined;
+};
+export const v3ToolChoiceToOpenAI = (choice) => {
+    switch (choice.type) {
+        case "auto":
+        case "none":
+        case "required":
+            return choice.type;
+        case "tool":
+            return { type: "function", function: { name: choice.toolName } };
+    }
+};
+export const v3ResponseFormatToOpenAI = (rf) => {
+    if (rf.type === "text") {
+        return { type: "text" };
+    }
+    if (!rf.schema) {
+        return { type: "json_object" };
+    }
+    return {
+        type: "json_schema",
+        json_schema: {
+            name: rf.name ?? "response",
+            schema: rf.schema,
+            ...(rf.description ? { description: rf.description } : {}),
+            strict: true,
+        },
+    };
+};
+export const mapNeuroLinkToolChoice = (choice) => {
+    if (!choice) {
+        return undefined;
+    }
+    if (choice === "auto" || choice === "none" || choice === "required") {
+        return choice;
+    }
+    if (typeof choice === "object" && choice !== null) {
+        const c = choice;
+        if (c.type === "tool" && c.toolName) {
+            return { type: "function", function: { name: c.toolName } };
+        }
+    }
+    return undefined;
+};
+export const buildBody = (args) => {
+    const { modelId, messages, options, tools, toolChoice, streaming, responseFormat, } = args;
+    const body = {
+        model: modelId,
+        messages,
+        ...(streaming ? { stream: true } : {}),
+        ...(streaming ? { stream_options: { include_usage: true } } : {}),
+    };
+    if (options.maxTokens !== undefined && options.maxTokens !== null) {
+        body.max_tokens = options.maxTokens;
+    }
+    if (options.temperature !== undefined && options.temperature !== null) {
+        body.temperature = options.temperature;
+    }
+    if (options.topP !== undefined && options.topP !== null) {
+        body.top_p = options.topP;
+    }
+    if (options.presencePenalty !== undefined &&
+        options.presencePenalty !== null) {
+        body.presence_penalty = options.presencePenalty;
+    }
+    if (options.frequencyPenalty !== undefined &&
+        options.frequencyPenalty !== null) {
+        body.frequency_penalty = options.frequencyPenalty;
+    }
+    if (options.seed !== undefined && options.seed !== null) {
+        body.seed = options.seed;
+    }
+    if (options.stopSequences && options.stopSequences.length > 0) {
+        body.stop = options.stopSequences;
+    }
+    if (tools) {
+        body.tools = tools;
+    }
+    if (toolChoice !== undefined) {
+        body.tool_choice = toolChoice;
+    }
+    if (responseFormat) {
+        body.response_format = responseFormat;
+    }
+    return body;
+};
+export const parseSSEStream = async (body, onTextDelta) => {
+    const result = {
+        text: "",
+        toolCalls: new Map(),
+        finishReason: null,
+        usage: undefined,
+    };
+    const decoder = new TextDecoder();
+    let parseErr;
+    const handleEvent = (msg) => {
+        const data = msg.data;
+        if (!data || data === "[DONE]") {
+            return;
+        }
+        let chunk;
+        try {
+            chunk = JSON.parse(data);
+        }
+        catch (err) {
+            parseErr = err instanceof Error ? err : new Error(String(err));
+            return;
+        }
+        if (chunk.usage) {
+            result.usage = chunk.usage;
+        }
+        const choice = chunk.choices?.[0];
+        if (!choice) {
+            return;
+        }
+        const delta = choice.delta;
+        if (delta?.content) {
+            result.text += delta.content;
+            onTextDelta(delta.content);
+        }
+        if (delta?.tool_calls) {
+            for (const tc of delta.tool_calls) {
+                let state = result.toolCalls.get(tc.index);
+                if (!state) {
+                    state = {
+                        id: tc.id ?? `call_${tc.index}_${Date.now()}`,
+                        name: tc.function?.name ?? "",
+                        argsBuffered: "",
+                    };
+                    result.toolCalls.set(tc.index, state);
+                }
+                else if (tc.id) {
+                    state.id = tc.id;
+                }
+                if (tc.function?.name) {
+                    state.name = tc.function.name;
+                }
+                if (tc.function?.arguments) {
+                    state.argsBuffered += tc.function.arguments;
+                }
+            }
+        }
+        if (choice.finish_reason) {
+            result.finishReason = choice.finish_reason;
+        }
+    };
+    const parser = createParser({ onEvent: handleEvent });
+    const reader = body.getReader();
+    try {
+        for (;;) {
+            const { done, value } = await reader.read();
+            if (done) {
+                break;
+            }
+            parser.feed(decoder.decode(value, { stream: true }));
+        }
+        parser.feed(decoder.decode());
+    }
+    finally {
+        reader.releaseLock();
+    }
+    if (parseErr) {
+        throw parseErr;
+    }
+    return result;
+};
+export const buildAPIError = async (url, body, res) => {
+    let bodyText;
+    let parsed;
+    try {
+        bodyText = await res.text();
+        parsed = bodyText
+            ? JSON.parse(bodyText)
+            : undefined;
+    }
+    catch {
+        parsed = undefined;
+    }
+    const msg = parsed?.error?.message ??
+        `OpenAI-compatible request failed with status ${res.status}`;
+    const err = new Error(msg);
+    err.statusCode = res.status;
+    err.url = url;
+    // Redacted summary only — never attach raw prompts, tool definitions, or
+    // tool arguments to the thrown error. Anything serialized by upstream
+    // logging would leak them otherwise.
+    err.requestBody = {
+        model: body.model,
+        stream: body.stream === true,
+        tool_count: body.tools?.length ?? 0,
+    };
+    if (bodyText !== undefined) {
+        err.responseBody = bodyText;
+    }
+    return err;
+};
+// Deferred-promise pair for `usage` and `finishReason` so the analytics
+// collector resolves with the actual aggregated values after the multi-step
+// loop ends, not the zeros they had at result-construction time.
+export const createDeferredAnalytics = () => {
+    let resolveUsage = () => { };
+    const usagePromise = new Promise((r) => {
+        resolveUsage = r;
+    });
+    let resolveFinish = () => { };
+    const finishPromise = new Promise((r) => {
+        resolveFinish = r;
+    });
+    return { usagePromise, finishPromise, resolveUsage, resolveFinish };
+};
+// Single-producer / single-consumer chunk queue. The streaming loop pushes
+// `{content}` deltas as they arrive from SSE and a final `{done:true}` when
+// it finishes; the consumer's AsyncIterable pulls from `nextChunk()`.
+export const createChunkQueue = () => {
+    const chunkQueue = [];
+    let pendingResolve;
+    const pushChunk = (c) => {
+        if (pendingResolve) {
+            const r = pendingResolve;
+            pendingResolve = undefined;
+            r(c);
+        }
+        else {
+            chunkQueue.push(c);
+        }
+    };
+    const nextChunk = () => new Promise((resolve) => {
+        if (chunkQueue.length > 0) {
+            resolve(chunkQueue.shift());
+        }
+        else {
+            pendingResolve = resolve;
+        }
+    });
+    return { pushChunk, nextChunk };
+};
+export const mergeUsage = (a, b) => {
+    if (!a) {
+        return b;
+    }
+    if (!b) {
+        return a;
+    }
+    return {
+        prompt_tokens: (a.prompt_tokens ?? 0) + (b.prompt_tokens ?? 0),
+        completion_tokens: (a.completion_tokens ?? 0) + (b.completion_tokens ?? 0),
+        total_tokens: (a.total_tokens ?? 0) + (b.total_tokens ?? 0),
+    };
+};
+//# sourceMappingURL=openaiChatCompletionsClient.js.map

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

@@ -1,16 +1,17 @@
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
-import type { StreamOptions, StreamResult, ZodUnknownSchema } from "../types/index.js";
-import type { LanguageModel, Schema } from "../types/index.js";
+import type { LanguageModel, Schema, StreamOptions, StreamResult, ZodUnknownSchema } from "../types/index.js";
 /**
- * OpenAI Compatible Provider - BaseProvider Implementation
- * Provides access to one of the OpenAI-compatible endpoint (OpenRouter, vLLM, LiteLLM, etc.)
+ * OpenAI Compatible Provider — direct HTTP, no AI SDK.
+ *
+ * Talks to any OpenAI chat-completions-shaped endpoint (LiteLLM, vLLM,
+ * OpenRouter, etc.). The entire request/stream/tool-loop is inline above;
+ * no `streamText`, no `LanguageModelV3`, no `@ai-sdk/openai`.
  */
 export declare class OpenAICompatibleProvider extends BaseProvider {
-    private model?;
     private config;
+    private resolvedModel?;
     private discoveredModel?;
-    private customOpenAI;
     constructor(modelName?: string, sdk?: unknown, _region?: string, credentials?: {
         apiKey?: string;
         baseURL?: string;
@@ -18,33 +19,59 @@ export declare class OpenAICompatibleProvider extends BaseProvider {
     protected getProviderName(): AIProviderName;
     protected getDefaultModel(): string;
     /**
-     * Returns the Vercel AI SDK model instance for OpenAI Compatible endpoints
-     * Handles auto-discovery if no model was specified
+     * Abstract from BaseProvider — used by the parent's generate() path which
+     * still goes through `generateText`. Returns a thin LanguageModelV3-shaped
+     * object that delegates to the same HTTP helpers used by executeStream.
+     * Stays inside this file so no AI-SDK-named import is needed here.
      */
     protected getAISDKModel(): Promise<LanguageModel>;
-    protected formatProviderError(error: unknown): Error;
+    private resolveModelName;
     /**
-     * OpenAI Compatible endpoints support tools for compatible models
+     * Returns a minimal V3-shaped model. Only used by BaseProvider's
+     * `generate()` non-streaming path which still relies on the parent's
+     * `generateText`. The streaming path bypasses this entirely.
      */
+    private buildDelegatingModel;
+    protected formatProviderError(error: unknown): Error;
     supportsTools(): boolean;
     /**
-     * Provider-specific streaming implementation
-     * Note: This is only used when tools are disabled
+     * Streaming path — drives the OpenAI endpoint directly. No streamText,
+     * no AI SDK orchestrator. Tool calls, multi-step loops, telemetry,
+     * abort handling all inline.
      */
     protected executeStream(options: StreamOptions, _analysisSchema?: ZodUnknownSchema | Schema<unknown>): Promise<StreamResult>;
     /**
-     * Get available models from OpenAI Compatible endpoint
+     * Multi-step streaming orchestrator. One iteration per model turn:
      *
-     * Fetches from the /v1/models endpoint to discover available models.
-     * This is useful for auto-discovery when no model is specified.
+     *   1. POST /chat/completions with stream:true
+     *   2. Parse SSE; push text deltas to the consumer queue
+     *   3. If the step emitted tool_calls → execute each, append to
+     *      conversation, loop again
+     *   4. Otherwise resolve the deferred analytics promises and exit
+     *
+     * Bounded by `args.maxSteps`. Any thrown error rejects loopPromise and
+     * is surfaced to the consumer via `await loopPromise` in the stream
+     * generator.
      */
-    getAvailableModels(): Promise<string[]>;
+    private runStreamLoop;
     /**
-     * Get the first available model for auto-selection
+     * One streaming round-trip: POST chat-completions, parse SSE, push text
+     * deltas to the consumer queue. Returns the accumulated SSE result so
+     * the caller can decide whether to run tools and re-stream.
      */
-    getFirstAvailableModel(): Promise<string>;
+    private streamOneStep;
     /**
-     * Fallback models when discovery fails
+     * Execute every tool_call collected from one streaming step:
+     *
+     *   - append an `assistant` turn carrying the tool_calls
+     *   - resolve each tool from the local registry and run it
+     *   - emit tool:start/tool:end events
+     *   - push per-execution summaries
+     *   - append a `tool` turn per result so the next step can see them
+     *   - mirror BaseProvider's tool-events + storage hooks
      */
+    private executeToolBatch;
+    getAvailableModels(): Promise<string[]>;
+    getFirstAvailableModel(): Promise<string>;
     private getFallbackModels;
 }