@node-llm/core 0.5.0 → 0.7.0

package/dist/providers/deepseek/Capabilities.js

@@ -0,0 +1,52 @@
+import { ModelRegistry } from "../../models/ModelRegistry.js";
+export class Capabilities {
+    static getCapabilities(modelId) {
+        const caps = ["streaming"];
+        if (/deepseek-chat/.test(modelId)) {
+            caps.push("function_calling");
+        }
+        if (/deepseek-reasoner/.test(modelId)) {
+            caps.push("reasoning");
+        }
+        return caps;
+    }
+    static getContextWindow(modelId) {
+        const val = ModelRegistry.getContextWindow(modelId, "deepseek");
+        if (val)
+            return val;
+        if (/deepseek-(?:chat|reasoner)/.test(modelId)) {
+            return 64_000;
+        }
+        return 32_768;
+    }
+    static getMaxOutputTokens(modelId) {
+        if (/deepseek-(?:chat|reasoner)/.test(modelId)) {
+            return 8_192;
+        }
+        return 4_096;
+    }
+    static supportsVision(modelId) {
+        return false;
+    }
+    static supportsTools(modelId) {
+        return /deepseek-chat/.test(modelId);
+    }
+    static supportsStructuredOutput(modelId) {
+        return /deepseek-(?:chat|reasoner)/.test(modelId);
+    }
+    static supportsEmbeddings(modelId) {
+        return false;
+    }
+    static supportsImageGeneration(modelId) {
+        return false;
+    }
+    static supportsTranscription(modelId) {
+        return false;
+    }
+    static supportsModeration(modelId) {
+        return false;
+    }
+    static supportsReasoning(modelId) {
+        return /deepseek-reasoner/.test(modelId);
+    }
+}

package/dist/providers/deepseek/Chat.d.ts

@@ -0,0 +1,8 @@
+import { ChatRequest, ChatResponse } from "../Provider.js";
+export declare class DeepSeekChat {
+    private readonly baseUrl;
+    private readonly apiKey;
+    constructor(baseUrl: string, apiKey: string);
+    execute(request: ChatRequest): Promise<ChatResponse>;
+}
+//# sourceMappingURL=Chat.d.ts.map

package/dist/providers/deepseek/Chat.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/deepseek/Chat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAS,MAAM,gBAAgB,CAAC;AAqBlE,qBAAa,YAAY;IACX,OAAO,CAAC,QAAQ,CAAC,OAAO;IAAU,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAAxC,OAAO,EAAE,MAAM,EAAmB,MAAM,EAAE,MAAM;IAEvE,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;CAwF3D"}

package/dist/providers/deepseek/Chat.js

@@ -0,0 +1,89 @@
+import { ModelRegistry } from "../../models/ModelRegistry.js";
+export class DeepSeekChat {
+    baseUrl;
+    apiKey;
+    constructor(baseUrl, apiKey) {
+        this.baseUrl = baseUrl;
+        this.apiKey = apiKey;
+    }
+    async execute(request) {
+        const { model, messages, tools, max_tokens, response_format, headers, ...rest } = request;
+        const body = {
+            model,
+            messages,
+            ...rest
+        };
+        if (max_tokens)
+            body.max_tokens = max_tokens;
+        if (tools && tools.length > 0)
+            body.tools = tools;
+        if (max_tokens)
+            body.max_tokens = max_tokens;
+        if (tools && tools.length > 0)
+            body.tools = tools;
+        // Handle structured output for DeepSeek
+        if (response_format) {
+            if (response_format.type === "json_schema") {
+                body.response_format = { type: "json_object" };
+                // Append schema instructions to the system prompt or convert to a new system message
+                const schema = response_format.json_schema;
+                const schemaString = JSON.stringify(schema.schema, null, 2);
+                const instruction = `\n\nIMPORTANT: You must output strictly valid JSON conforming to the following schema:\n${schemaString}\n\nOutput only the JSON object.`;
+                // Find system message or append to last user message
+                const systemMessage = body.messages.find((m) => m.role === "system");
+                if (systemMessage) {
+                    systemMessage.content += instruction;
+                }
+                else {
+                    // Insert system message at the beginning
+                    body.messages.unshift({
+                        role: "system",
+                        content: "You are a helpful assistant." + instruction
+                    });
+                }
+            }
+            else {
+                body.response_format = response_format;
+            }
+        }
+        if (process.env.NODELLM_DEBUG === "true") {
+            console.log(`[DeepSeek Request] ${JSON.stringify(body, null, 2)}`);
+        }
+        const response = await fetch(`${this.baseUrl}/chat/completions`, {
+            method: "POST",
+            headers: {
+                "Authorization": `Bearer ${this.apiKey}`,
+                "Content-Type": "application/json",
+                ...request.headers,
+            },
+            body: JSON.stringify(body),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`DeepSeek API error: ${response.status} - ${errorText}`);
+        }
+        const json = (await response.json());
+        const message = json.choices[0]?.message;
+        const content = message?.content ?? null;
+        const reasoning = message?.reasoning_content ?? null;
+        const usage = {
+            input_tokens: json.usage.prompt_tokens,
+            output_tokens: json.usage.completion_tokens,
+            total_tokens: json.usage.total_tokens,
+        };
+        const toolCalls = message?.tool_calls?.map((tc) => ({
+            id: tc.id,
+            type: tc.type,
+            function: {
+                name: tc.function.name,
+                arguments: tc.function.arguments
+            }
+        }));
+        if (!content && !reasoning && (!toolCalls || toolCalls.length === 0)) {
+            throw new Error("DeepSeek returned empty response");
+        }
+        // deepseek cost calculation if needed, otherwise just return usage
+        const calculatedUsage = ModelRegistry.calculateCost(usage, model, "deepseek");
+        return { content, reasoning, usage: calculatedUsage, tool_calls: toolCalls };
+    }
+}

package/dist/providers/deepseek/DeepSeekProvider.d.ts

@@ -0,0 +1,28 @@
+import { Provider, ChatRequest, ChatResponse, ModelInfo, ChatChunk } from "../Provider.js";
+export interface DeepSeekProviderOptions {
+    apiKey: string;
+    baseUrl?: string;
+}
+export declare class DeepSeekProvider implements Provider {
+    private readonly options;
+    private readonly baseUrl;
+    private readonly chatHandler;
+    private readonly streamingHandler;
+    private readonly modelsHandler;
+    capabilities: {
+        supportsVision: (model: string) => boolean;
+        supportsTools: (model: string) => boolean;
+        supportsStructuredOutput: (model: string) => boolean;
+        supportsEmbeddings: (model: string) => boolean;
+        supportsImageGeneration: (model: string) => boolean;
+        supportsTranscription: (model: string) => boolean;
+        supportsModeration: (model: string) => boolean;
+        supportsReasoning: (model: string) => boolean;
+        getContextWindow: (model: string) => number | null;
+    };
+    constructor(options: DeepSeekProviderOptions);
+    chat(request: ChatRequest): Promise<ChatResponse>;
+    stream(request: ChatRequest): AsyncGenerator<ChatChunk>;
+    listModels(): Promise<ModelInfo[]>;
+}
+//# sourceMappingURL=DeepSeekProvider.d.ts.map

package/dist/providers/deepseek/DeepSeekProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"DeepSeekProvider.d.ts","sourceRoot":"","sources":["../../../src/providers/deepseek/DeepSeekProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAM3F,MAAM,WAAW,uBAAuB;IACtC,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,gBAAiB,YAAW,QAAQ;IAkBnC,OAAO,CAAC,QAAQ,CAAC,OAAO;IAjBpC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAe;IAC3C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAoB;IACrD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAiB;IAExC,YAAY;gCACO,MAAM;+BACP,MAAM;0CACK,MAAM;oCACZ,MAAM;yCACD,MAAM;uCACR,MAAM;oCACT,MAAM;mCACP,MAAM;kCACP,MAAM;MAChC;gBAE2B,OAAO,EAAE,uBAAuB;IAOvD,IAAI,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAIhD,MAAM,CAAC,OAAO,EAAE,WAAW,GAAG,cAAc,CAAC,SAAS,CAAC;IAIxD,UAAU,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;CAGzC"}

package/dist/providers/deepseek/DeepSeekProvider.js

@@ -0,0 +1,38 @@
+import { DeepSeekChat } from "./Chat.js";
+import { DeepSeekModels } from "./Models.js";
+import { DeepSeekStreaming } from "./Streaming.js";
+import { Capabilities } from "./Capabilities.js";
+export class DeepSeekProvider {
+    options;
+    baseUrl;
+    chatHandler;
+    streamingHandler;
+    modelsHandler;
+    capabilities = {
+        supportsVision: (model) => Capabilities.supportsVision(model),
+        supportsTools: (model) => Capabilities.supportsTools(model),
+        supportsStructuredOutput: (model) => Capabilities.supportsStructuredOutput(model),
+        supportsEmbeddings: (model) => Capabilities.supportsEmbeddings(model),
+        supportsImageGeneration: (model) => Capabilities.supportsImageGeneration(model),
+        supportsTranscription: (model) => Capabilities.supportsTranscription(model),
+        supportsModeration: (model) => Capabilities.supportsModeration(model),
+        supportsReasoning: (model) => Capabilities.supportsReasoning(model),
+        getContextWindow: (model) => Capabilities.getContextWindow(model),
+    };
+    constructor(options) {
+        this.options = options;
+        this.baseUrl = options.baseUrl ?? "https://api.deepseek.com";
+        this.chatHandler = new DeepSeekChat(this.baseUrl, options.apiKey);
+        this.streamingHandler = new DeepSeekStreaming(this.baseUrl, options.apiKey);
+        this.modelsHandler = new DeepSeekModels(this.baseUrl, options.apiKey);
+    }
+    async chat(request) {
+        return this.chatHandler.execute(request);
+    }
+    async *stream(request) {
+        yield* this.streamingHandler.execute(request);
+    }
+    async listModels() {
+        return this.modelsHandler.execute();
+    }
+}

package/dist/providers/deepseek/Models.d.ts

@@ -0,0 +1,8 @@
+import { ModelInfo } from "../Provider.js";
+export declare class DeepSeekModels {
+    private readonly baseUrl;
+    private readonly apiKey;
+    constructor(baseUrl: string, apiKey: string);
+    execute(): Promise<ModelInfo[]>;
+}
+//# sourceMappingURL=Models.d.ts.map

package/dist/providers/deepseek/Models.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Models.d.ts","sourceRoot":"","sources":["../../../src/providers/deepseek/Models.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAe3C,qBAAa,cAAc;IACb,OAAO,CAAC,QAAQ,CAAC,OAAO;IAAU,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAAxC,OAAO,EAAE,MAAM,EAAmB,MAAM,EAAE,MAAM;IAEvE,OAAO,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;CA+DtC"}

package/dist/providers/deepseek/Models.js

@@ -0,0 +1,67 @@
+import { Capabilities } from "./Capabilities.js";
+import { ModelRegistry } from "../../models/ModelRegistry.js";
+export class DeepSeekModels {
+    baseUrl;
+    apiKey;
+    constructor(baseUrl, apiKey) {
+        this.baseUrl = baseUrl;
+        this.apiKey = apiKey;
+    }
+    async execute() {
+        const response = await fetch(`${this.baseUrl}/models`, {
+            method: "GET",
+            headers: {
+                "Authorization": `Bearer ${this.apiKey}`,
+            },
+        });
+        if (!response.ok) {
+            // Fallback to local registry
+            const localModels = ModelRegistry.all()
+                .filter(m => m.provider === "deepseek")
+                .map(m => ({
+                id: m.id,
+                name: m.name,
+                provider: "deepseek",
+                family: m.family ?? "deepseek",
+                context_window: m.context_window ?? null,
+                max_output_tokens: m.max_output_tokens ?? null,
+                modalities: m.modalities,
+                capabilities: m.capabilities,
+                pricing: m.pricing,
+                metadata: m.metadata
+            }));
+            return localModels;
+        }
+        const json = (await response.json());
+        const localRegistry = ModelRegistry.all().filter(m => m.provider === "deepseek");
+        return json.data.map(m => {
+            // Try to find in local registry for enriched data (pricing, limits)
+            const local = localRegistry.find(l => l.id === m.id);
+            if (local) {
+                return {
+                    id: local.id,
+                    name: local.name,
+                    provider: "deepseek", // Ensure literal type
+                    family: local.family ?? "deepseek", // Handle missing family
+                    context_window: local.context_window ?? null,
+                    max_output_tokens: local.max_output_tokens ?? null,
+                    modalities: local.modalities,
+                    capabilities: local.capabilities,
+                    pricing: local.pricing,
+                    metadata: local.metadata
+                };
+            }
+            return {
+                id: m.id,
+                name: m.id,
+                provider: "deepseek",
+                family: "deepseek",
+                context_window: Capabilities.getContextWindow(m.id),
+                max_output_tokens: Capabilities.getMaxOutputTokens(m.id),
+                modalities: { input: ["text"], output: ["text"] },
+                capabilities: Capabilities.getCapabilities(m.id),
+                pricing: {}
+            };
+        });
+    }
+}

package/dist/providers/deepseek/Streaming.d.ts

@@ -0,0 +1,8 @@
+import { ChatRequest, ChatChunk } from "../Provider.js";
+export declare class DeepSeekStreaming {
+    private readonly baseUrl;
+    private readonly apiKey;
+    constructor(baseUrl: string, apiKey: string);
+    execute(request: ChatRequest): AsyncGenerator<ChatChunk>;
+}
+//# sourceMappingURL=Streaming.d.ts.map

package/dist/providers/deepseek/Streaming.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Streaming.d.ts","sourceRoot":"","sources":["../../../src/providers/deepseek/Streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAExD,qBAAa,iBAAiB;IAChB,OAAO,CAAC,QAAQ,CAAC,OAAO;IAAU,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAAxC,OAAO,EAAE,MAAM,EAAmB,MAAM,EAAE,MAAM;IAEtE,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,cAAc,CAAC,SAAS,CAAC;CAwEhE"}

package/dist/providers/deepseek/Streaming.js

@@ -0,0 +1,74 @@
+export class DeepSeekStreaming {
+    baseUrl;
+    apiKey;
+    constructor(baseUrl, apiKey) {
+        this.baseUrl = baseUrl;
+        this.apiKey = apiKey;
+    }
+    async *execute(request) {
+        const { model, messages, tools, max_tokens, response_format, headers, ...rest } = request;
+        const body = {
+            model,
+            messages,
+            stream: true,
+            ...rest
+        };
+        if (max_tokens)
+            body.max_tokens = max_tokens;
+        if (tools && tools.length > 0)
+            body.tools = tools;
+        if (response_format)
+            body.response_format = response_format;
+        const response = await fetch(`${this.baseUrl}/chat/completions`, {
+            method: "POST",
+            headers: {
+                "Authorization": `Bearer ${this.apiKey}`,
+                "Content-Type": "application/json",
+                ...request.headers,
+            },
+            body: JSON.stringify(body),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`DeepSeek API error: ${response.status} - ${errorText}`);
+        }
+        if (!response.body) {
+            throw new Error("No response body for streaming");
+        }
+        // @ts-ignore
+        const reader = response.body.getReader();
+        const decoder = new TextDecoder();
+        let buffer = "";
+        while (true) {
+            const { value, done } = await reader.read();
+            if (done)
+                break;
+            const chunk = decoder.decode(value, { stream: true });
+            buffer += chunk;
+            const lines = buffer.split("\n\n");
+            buffer = lines.pop() || "";
+            for (const line of lines) {
+                const trimmed = line.trim();
+                if (!trimmed.startsWith("data: "))
+                    continue;
+                const data = trimmed.replace("data: ", "").trim();
+                if (data === "[DONE]")
+                    return;
+                try {
+                    const json = JSON.parse(data);
+                    const deltaContent = json.choices?.[0]?.delta?.content;
+                    const deltaReasoning = json.choices?.[0]?.delta?.reasoning_content;
+                    if (deltaContent || deltaReasoning) {
+                        yield {
+                            content: deltaContent || "",
+                            reasoning: deltaReasoning || ""
+                        };
+                    }
+                }
+                catch (e) {
+                    // ignore
+                }
+            }
+        }
+    }
+}

package/dist/providers/deepseek/index.d.ts

@@ -0,0 +1,7 @@
+export * from "./DeepSeekProvider.js";
+export * from "./Chat.js";
+export * from "./Models.js";
+export * from "./Capabilities.js";
+export declare function registerDeepSeekProvider(): void;
+export declare const ensureDeepSeekRegistered: typeof registerDeepSeekProvider;
+//# sourceMappingURL=index.d.ts.map

package/dist/providers/deepseek/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/providers/deepseek/index.ts"],"names":[],"mappings":"AAIA,cAAc,uBAAuB,CAAC;AACtC,cAAc,WAAW,CAAC;AAC1B,cAAc,aAAa,CAAC;AAC5B,cAAc,mBAAmB,CAAC;AAIlC,wBAAgB,wBAAwB,SAevC;AAED,eAAO,MAAM,wBAAwB,iCAA2B,CAAC"}

package/dist/providers/deepseek/index.js

@@ -0,0 +1,22 @@
+import { config } from "../../config.js";
+import { providerRegistry } from "../registry.js";
+import { DeepSeekProvider } from "./DeepSeekProvider.js";
+export * from "./DeepSeekProvider.js";
+export * from "./Chat.js";
+export * from "./Models.js";
+export * from "./Capabilities.js";
+let registered = false;
+export function registerDeepSeekProvider() {
+    if (registered)
+        return;
+    providerRegistry.register("deepseek", () => {
+        const apiKey = config.deepseekApiKey;
+        const baseUrl = config.deepseekApiBase; // Optional override
+        if (!apiKey) {
+            throw new Error("deepseek_api_key is not set in config or DEEPSEEK_API_KEY environment variable");
+        }
+        return new DeepSeekProvider({ apiKey, baseUrl });
+    });
+    registered = true;
+}
+export const ensureDeepSeekRegistered = registerDeepSeekProvider;

package/dist/providers/gemini/Capabilities.d.ts

@@ -4,27 +4,48 @@ export declare class Capabilities {
     static supportsVision(modelId: string): boolean;
     static supportsTools(modelId: string): boolean;
     static supportsStructuredOutput(modelId: string): boolean;
+    static supportsSystemInstructions(modelId: string): boolean;
     static supportsJsonMode(modelId: string): boolean;
     static supportsEmbeddings(modelId: string): boolean;
     static supportsImageGeneration(modelId: string): boolean;
     static supportsTranscription(modelId: string): boolean;
     static supportsModeration(modelId: string): boolean;
-    static normalizeTemperature(temperature: number | undefined, _modelId: string): number | undefined | null;
-    static getFamily(modelId: string): string;
+    static normalizeTemperature(temperature: number | undefined, model: string): number | undefined;
     static getModalities(modelId: string): {
         input: string[];
         output: string[];
     };
     static getCapabilities(modelId: string): string[];
     static getPricing(modelId: string): {
-        text_tokens: {
-            standard: {
-                input_per_million: number;
-                output_per_million: number;
+        text_tokens?: {
+            standard?: {
+                input_per_million?: number;
+                cached_input_per_million?: number;
+                output_per_million?: number;
+                reasoning_output_per_million?: number;
+            };
+            batch?: {
+                input_per_million?: number;
+                output_per_million?: number;
+            };
+        };
+        images?: {
+            standard?: {
+                input?: number;
+                output?: number;
+            };
+            batch?: {
+                input?: number;
+                output?: number;
+            };
+        };
+        audio_tokens?: {
+            standard?: {
+                input_per_million?: number;
+                output_per_million?: number;
             };
         };
     };
-    static formatDisplayName(modelId: string): string;
     private static normalizeModelId;
 }
 //# sourceMappingURL=Capabilities.d.ts.map

package/dist/providers/gemini/Capabilities.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/gemini/Capabilities.ts"],"names":[],"mappings":"AAAA,qBAAa,YAAY;IACvB,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAoBvD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAiBzD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQ/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQ9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQzD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIjD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKnD,MAAM,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKxD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKtD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAInD,MAAM,CAAC,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,SAAS,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,GAAG,IAAI;IAIzG,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAWzC,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE;IAY5E,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAUjD,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM;;;;;;;;IA0BjC,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAIjD,OAAO,CAAC,MAAM,CAAC,gBAAgB;CAGhC"}
+{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/gemini/Capabilities.ts"],"names":[],"mappings":"AAEA,qBAAa,YAAY;IACvB,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAuBvD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAoBzD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAW/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAW9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAWzD,MAAM,CAAC,0BAA0B,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAI3D,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIjD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQnD,MAAM,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQxD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQtD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAInD,MAAM,CAAC,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,SAAS,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAI/F,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE;IAY5E,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAUjD,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM;;oBArI2D,CAAC;iCAA6B,CAAC;wCAGrH,CAAC;kCACH,CAAC;4CAEgB,CAAC;;iBACd,CAAC;iCAEG,CAAC;kCAAoC,CAAC;;;;oBAG3B,CAAC;qBACV,CAAC;sBAEF,CAAC;;iBAGM,CAAC;qBAAiB,CAAC;sBACrC,CAAA;;;;oBAA6D,CAAC;iCACxC,CAAC;kCAEpB,CAAC;;;;IA4IN,OAAO,CAAC,MAAM,CAAC,gBAAgB;CAGhC"}

package/dist/providers/gemini/Capabilities.js

@@ -1,5 +1,9 @@
+import { ModelRegistry } from "../../models/ModelRegistry.js";
 export class Capabilities {
     static getContextWindow(modelId) {
+        const val = ModelRegistry.getContextWindow(modelId, "gemini");
+        if (val !== undefined && val !== null)
+            return val;
         const id = this.normalizeModelId(modelId);
         if (id.match(/gemini-2\.5-pro-exp-03-25|gemini-2\.0-flash|gemini-2\.0-flash-lite|gemini-1\.5-flash|gemini-1\.5-flash-8b/)) {
             return 1_048_576;
@@ -19,6 +23,9 @@ export class Capabilities {
         return 32_768;
     }
     static getMaxOutputTokens(modelId) {
+        const val = ModelRegistry.getMaxOutputTokens(modelId, "gemini");
+        if (val !== undefined && val !== null)
+            return val;
         const id = this.normalizeModelId(modelId);
         if (id.match(/gemini-2\.5-pro-exp-03-25/)) {
             return 64_000;
@@ -35,6 +42,9 @@ export class Capabilities {
         return 4_096;
     }
     static supportsVision(modelId) {
+        const model = ModelRegistry.find(modelId, "gemini");
+        if (model?.modalities?.input?.includes("image"))
+            return true;
         const id = this.normalizeModelId(modelId);
         if (id.match(/text-embedding|embedding-001|aqa/)) {
             return false;
@@ -42,6 +52,9 @@ export class Capabilities {
         return !!id.match(/gemini|flash|pro|imagen/);
     }
     static supportsTools(modelId) {
+        const model = ModelRegistry.find(modelId, "gemini");
+        if (model?.capabilities?.includes("function_calling"))
+            return true;
         const id = this.normalizeModelId(modelId);
         if (id.match(/text-embedding|embedding-001|aqa|flash-lite|imagen|gemini-2\.0-flash-lite/)) {
             return false;
@@ -49,49 +62,48 @@ export class Capabilities {
         return !!id.match(/gemini|pro|flash/);
     }
     static supportsStructuredOutput(modelId) {
+        const model = ModelRegistry.find(modelId, "gemini");
+        if (model?.capabilities?.includes("structured_output"))
+            return true;
         const id = this.normalizeModelId(modelId);
         if (id.match(/text-embedding|embedding-001|aqa|imagen/)) {
             return false;
         }
         return true;
     }
+    static supportsSystemInstructions(modelId) {
+        return true;
+    }
     static supportsJsonMode(modelId) {
         return this.supportsStructuredOutput(modelId);
     }
     static supportsEmbeddings(modelId) {
+        const model = ModelRegistry.find(modelId, "gemini");
+        if (model?.modalities?.output?.includes("embeddings"))
+            return true;
         const id = this.normalizeModelId(modelId);
         return !!id.match(/text-embedding|embedding|gemini-embedding/);
     }
     static supportsImageGeneration(modelId) {
+        const model = ModelRegistry.find(modelId, "gemini");
+        if (model?.capabilities?.includes("image_generation") || model?.modalities?.output?.includes("image"))
+            return true;
         const id = this.normalizeModelId(modelId);
         return !!id.match(/imagen/);
     }
     static supportsTranscription(modelId) {
+        const model = ModelRegistry.find(modelId, "gemini");
+        if (model?.modalities?.input?.includes("audio"))
+            return true;
         const id = this.normalizeModelId(modelId);
         return !!id.match(/gemini|flash|pro/);
     }
     static supportsModeration(modelId) {
         return false;
     }
-    static normalizeTemperature(temperature, _modelId) {
+    static normalizeTemperature(temperature, model) {
         return temperature;
     }
-    static getFamily(modelId) {
-        const id = this.normalizeModelId(modelId);
-        if (id.startsWith("gemini-1.5-pro"))
-            return "gemini-1.5-pro";
-        if (id.startsWith("gemini-1.5-flash"))
-            return "gemini-1.5-flash";
-        if (id.startsWith("gemini-2.0-flash"))
-            return "gemini-2.0-flash";
-        if (id.startsWith("gemini-2.0-flash-lite"))
-            return "gemini-2.0-flash-lite";
-        if (id.startsWith("text-embedding"))
-            return "text-embedding";
-        if (id.startsWith("imagen"))
-            return "imagen";
-        return "other";
-    }
     static getModalities(modelId) {
         const input = ["text"];
         const output = ["text"];
@@ -118,6 +130,9 @@ export class Capabilities {
         return caps;
     }
     static getPricing(modelId) {
+        const model = ModelRegistry.find(modelId, "gemini");
+        if (model?.pricing)
+            return model.pricing;
         const id = this.normalizeModelId(modelId);
         let input = 0;
         let output = 0;
@@ -142,9 +157,6 @@ export class Capabilities {
             }
         };
     }
-    static formatDisplayName(modelId) {
-        return modelId.replace("models/", "").replace(/-/g, " ").replace(/\b\w/g, l => l.toUpperCase());
-    }
     static normalizeModelId(modelId) {
         return modelId.replace("models/", "");
     }

package/dist/providers/gemini/Chat.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/gemini/Chat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAM3D,qBAAa,UAAU;IACT,OAAO,CAAC,QAAQ,CAAC,OAAO;IAAU,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAAxC,OAAO,EAAE,MAAM,EAAmB,MAAM,EAAE,MAAM;IAEvE,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IA0F1D,OAAO,CAAC,cAAc;CAwBvB"}
+{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/gemini/Chat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAS,MAAM,gBAAgB,CAAC;AAOlE,qBAAa,UAAU;IACT,OAAO,CAAC,QAAQ,CAAC,OAAO;IAAU,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAAxC,OAAO,EAAE,MAAM,EAAmB,MAAM,EAAE,MAAM;IAEvE,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAyF1D,OAAO,CAAC,cAAc;CAwBvB"}