@node-llm/core 0.5.0 → 0.6.0

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"}

package/dist/providers/gemini/Chat.js

@@ -1,6 +1,7 @@
 import { Capabilities } from "./Capabilities.js";
 import { handleGeminiError } from "./Errors.js";
 import { GeminiChatUtils } from "./ChatUtils.js";
+import { ModelRegistry } from "../../models/ModelRegistry.js";
 export class GeminiChat {
     baseUrl;
     apiKey;
@@ -19,24 +20,20 @@ export class GeminiChat {
         if (request.response_format?.type === "json_object") {
             generationConfig.responseMimeType = "application/json";
         }
-        else if (request.response_format?.type === "json_schema") {
-            generationConfig.responseMimeType = "application/json";
-            if (request.response_format.json_schema?.schema) {
-                generationConfig.responseSchema = request.response_format.json_schema.schema;
-            }
-        }
-        if (request.response_format?.type === "json_object") {
-            generationConfig.responseMimeType = "application/json";
-        }
         else if (request.response_format?.type === "json_schema") {
             generationConfig.responseMimeType = "application/json";
             if (request.response_format.json_schema?.schema) {
                 generationConfig.responseSchema = this.sanitizeSchema(request.response_format.json_schema.schema);
             }
         }
+        const { model: _model, messages: _messages, tools: _tools, temperature: _temp, max_tokens: _max, response_format: _format, headers: _headers, ...rest } = request;
         const payload = {
             contents,
-            generationConfig,
+            generationConfig: {
+                ...generationConfig,
+                ...(rest.generationConfig || {})
+            },
+            ...rest
         };
         if (systemInstructionParts.length > 0) {
             payload.systemInstruction = { parts: systemInstructionParts };
@@ -83,7 +80,8 @@ export class GeminiChat {
             output_tokens: json.usageMetadata.candidatesTokenCount,
             total_tokens: json.usageMetadata.totalTokenCount,
         } : undefined;
-        return { content, tool_calls, usage };
+        const calculatedUsage = usage ? ModelRegistry.calculateCost(usage, request.model, "gemini") : undefined;
+        return { content, tool_calls, usage: calculatedUsage };
     }
     sanitizeSchema(schema) {
         if (typeof schema !== "object" || schema === null)

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

@@ -4,5 +4,6 @@ export declare class GeminiModels {
     private readonly apiKey;
     constructor(baseUrl: string, apiKey: string);
     execute(): Promise<ModelInfo[]>;
+    find(modelId: string): import("../../models/types.js").Model | undefined;
 }
 //# sourceMappingURL=Models.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"Models.d.ts","sourceRoot":"","sources":["../../../src/providers/gemini/Models.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAI3C,qBAAa,YAAY;IACX,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;CAgCtC"}
+{"version":3,"file":"Models.d.ts","sourceRoot":"","sources":["../../../src/providers/gemini/Models.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAK3C,qBAAa,YAAY;IACX,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;IAkDrC,IAAI,CAAC,OAAO,EAAE,MAAM;CAGrB"}

package/dist/providers/gemini/Models.js

@@ -1,4 +1,5 @@
 import { Capabilities } from "./Capabilities.js";
+import { ModelRegistry } from "../../models/ModelRegistry.js";
 export class GeminiModels {
     baseUrl;
     apiKey;
@@ -7,32 +8,51 @@ export class GeminiModels {
         this.apiKey = apiKey;
     }
     async execute() {
-        const url = `${this.baseUrl}/models?key=${this.apiKey}`;
-        const response = await fetch(url);
-        if (!response.ok) {
-            const errorText = await response.text();
-            throw new Error(`Gemini error (${response.status}): ${errorText}`);
-        }
-        const json = (await response.json());
-        return json.models
-            .filter(m => m.supportedGenerationMethods.includes("generateContent"))
-            .map((model) => {
-            const id = model.name.replace("models/", "");
-            return {
-                id: id,
-                name: model.displayName || Capabilities.formatDisplayName(id),
-                provider: "gemini",
-                family: Capabilities.getFamily(id),
-                context_window: model.inputTokenLimit || Capabilities.getContextWindow(id),
-                max_output_tokens: model.outputTokenLimit || Capabilities.getMaxOutputTokens(id),
-                modalities: Capabilities.getModalities(id),
-                capabilities: Capabilities.getCapabilities(id),
-                pricing: Capabilities.getPricing(id),
-                metadata: {
-                    description: model.description,
-                    version: model.version,
+        try {
+            const response = await fetch(`${this.baseUrl}/models?key=${this.apiKey}`, {
+                method: "GET",
+                headers: {
+                    "Content-Type": "application/json",
                 },
-            };
-        });
+            });
+            if (response.ok) {
+                const { models } = await response.json();
+                return models.map(m => {
+                    const modelId = m.name.replace("models/", "");
+                    const registryModel = ModelRegistry.find(modelId, "gemini");
+                    const info = {
+                        id: modelId,
+                        name: registryModel?.name || m.displayName || modelId,
+                        provider: "gemini",
+                        family: registryModel?.family || modelId,
+                        context_window: registryModel?.context_window || Capabilities.getContextWindow(modelId),
+                        max_output_tokens: registryModel?.max_output_tokens || Capabilities.getMaxOutputTokens(modelId),
+                        modalities: registryModel?.modalities || Capabilities.getModalities(modelId),
+                        capabilities: Capabilities.getCapabilities(modelId),
+                        pricing: registryModel?.pricing || Capabilities.getPricing(modelId),
+                        metadata: {
+                            ...(registryModel?.metadata || {}),
+                            description: m.description,
+                            input_token_limit: m.inputTokenLimit,
+                            output_token_limit: m.outputTokenLimit,
+                            supported_generation_methods: m.supportedGenerationMethods
+                        }
+                    };
+                    return info;
+                });
+            }
+        }
+        catch (_error) {
+            // Fallback
+        }
+        return ModelRegistry.all()
+            .filter(m => m.provider === "gemini")
+            .map(m => ({
+            ...m,
+            capabilities: Capabilities.getCapabilities(m.id)
+        }));
+    }
+    find(modelId) {
+        return ModelRegistry.find(modelId, "gemini");
     }
 }

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

@@ -1,7 +1,4 @@
-import { ModelFamilyDefinition } from "./ModelDefinitions.js";
 export declare class Capabilities {
-    static getFamily(modelId: string): string;
-    static getDefinition(modelId: string): ModelFamilyDefinition;
     static getContextWindow(modelId: string): number | null;
     static getMaxOutputTokens(modelId: string): number | null;
     static supportsVision(modelId: string): boolean;
@@ -12,19 +9,14 @@ export declare class Capabilities {
     static supportsImageGeneration(modelId: string): boolean;
     static supportsTranscription(modelId: string): boolean;
     static supportsModeration(modelId: string): boolean;
-    static getInputPrice(modelId: string): number;
-    static getCachedInputPrice(modelId: string): number | undefined;
-    static getOutputPrice(modelId: string): number;
-    static getModelType(modelId: string): "embedding" | "audio" | "moderation" | "image" | "chat";
-    static formatDisplayName(modelId: string): string;
-    private static applySpecialFormatting;
-    private static specialPrefixFormat;
-    static normalizeTemperature(temperature: number | undefined, modelId: string): number | undefined | null;
+    static getModelType(modelId: string): "embedding" | "audio" | "moderation" | "image" | "chat" | "audio_transcription" | "audio_speech";
     static getModalities(modelId: string): {
         input: string[];
         output: string[];
     };
     static getCapabilities(modelId: string): string[];
+    static normalizeTemperature(temperature: number | undefined, modelId: string): number | undefined | null;
+    static formatDisplayName(modelId: string): string;
     static getPricing(modelId: string): any;
 }
 //# sourceMappingURL=Capabilities.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Capabilities.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAE7E,qBAAa,YAAY;IACvB,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAUzC,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,qBAAqB;IAK5D,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAIvD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAIzD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAI/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAI9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIzD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIjD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAInD,MAAM,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIxD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAMtD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAInD,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAK7C,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAI/D,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAK9C,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW,GAAG,OAAO,GAAG,YAAY,GAAG,OAAO,GAAG,MAAM;IAI7F,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAKjD,OAAO,CAAC,MAAM,CAAC,sBAAsB;IAarC,OAAO,CAAC,MAAM,CAAC,mBAAmB;IAUlC,MAAM,CAAC,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,SAAS,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,GAAG,IAAI;IAUxG,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE;IAqB5E,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAiBjD,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM;CAyBlC"}
+{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Capabilities.ts"],"names":[],"mappings":"AAEA,qBAAa,YAAY;IACvB,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAUvD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IASzD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAO/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAO9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOzD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIjD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOnD,MAAM,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOxD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOtD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOnD,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW,GAAG,OAAO,GAAG,YAAY,GAAG,OAAO,GAAG,MAAM,GAAG,qBAAqB,GAAG,cAAc;IAUtI,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE;IAiB5E,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAoBjD,MAAM,CAAC,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,SAAS,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,GAAG,IAAI;IAMxG,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAOjD,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,GAAG;CAcxC"}

package/dist/providers/openai/Capabilities.js

@@ -1,160 +1,157 @@
-import { OPENAI_MODELS } from "./ModelDefinitions.js";
+import { ModelRegistry } from "../../models/ModelRegistry.js";
 export class Capabilities {
-    static getFamily(modelId) {
-        for (const [key, def] of Object.entries(OPENAI_MODELS)) {
-            if (key === "other")
-                continue;
-            if (def.pattern.test(modelId)) {
-                return key;
-            }
-        }
-        return "other";
-    }
-    static getDefinition(modelId) {
-        const family = this.getFamily(modelId);
-        return OPENAI_MODELS[family];
-    }
     static getContextWindow(modelId) {
-        return this.getDefinition(modelId).contextWindow;
+        const val = ModelRegistry.getContextWindow(modelId, "openai");
+        if (val)
+            return val;
+        if (/gpt-4.*(preview|turbo|vision|o)/.test(modelId) || /o1|o3/.test(modelId))
+            return 128_000;
+        if (/gpt-4/.test(modelId))
+            return 8_192;
+        if (/gpt-3\.5/.test(modelId))
+            return 16_385;
+        return 128_000;
     }
     static getMaxOutputTokens(modelId) {
-        return this.getDefinition(modelId).maxOutputTokens;
+        const val = ModelRegistry.getMaxOutputTokens(modelId, "openai");
+        if (val)
+            return val;
+        if (/o1.*(pro|mini)|o3/.test(modelId))
+            return 65_536;
+        if (/gpt-4o/.test(modelId))
+            return 16_384;
+        return 4_096;
     }
     static supportsVision(modelId) {
-        return !!this.getDefinition(modelId).features.vision;
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model?.modalities?.input?.includes("image"))
+            return true;
+        return /gpt-4(?!-3)|o1/.test(modelId) && !/audio|realtime|voice/.test(modelId);
     }
     static supportsTools(modelId) {
-        return !!this.getDefinition(modelId).features.tools;
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model?.capabilities?.includes("function_calling"))
+            return true;
+        return !/embedding|moderation|dall-e|tts|whisper/.test(modelId);
     }
     static supportsStructuredOutput(modelId) {
-        return !!this.getDefinition(modelId).features.structuredOutput;
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model?.capabilities?.includes("structured_output"))
+            return true;
+        return /gpt-4|o1|o3/.test(modelId);
     }
     static supportsJsonMode(modelId) {
         return this.supportsStructuredOutput(modelId);
     }
     static supportsEmbeddings(modelId) {
-        return this.getDefinition(modelId).type === "embedding";
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model?.modalities?.output?.includes("embeddings"))
+            return true;
+        return /embedding/.test(modelId);
     }
     static supportsImageGeneration(modelId) {
-        return this.getDefinition(modelId).type === "image";
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model?.capabilities?.includes("image_generation") || model?.modalities?.output?.includes("image"))
+            return true;
+        return /dall-e|image/.test(modelId);
     }
     static supportsTranscription(modelId) {
-        // Transcription is supported by audio models or specific models like gpt-4o-audio
-        const def = this.getDefinition(modelId);
-        return def.type === "audio" || (def.type === "chat" && /audio|transcribe/.test(modelId));
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model?.modalities?.input?.includes("audio"))
+            return true;
+        return /whisper|audio|transcribe/.test(modelId);
     }
     static supportsModeration(modelId) {
-        return this.getDefinition(modelId).type === "moderation";
-    }
-    static getInputPrice(modelId) {
-        const prices = this.getDefinition(modelId).pricing;
-        return prices.input || prices.price || 0.5;
-    }
-    static getCachedInputPrice(modelId) {
-        return this.getDefinition(modelId).pricing.cached_input;
-    }
-    static getOutputPrice(modelId) {
-        const prices = this.getDefinition(modelId).pricing;
-        return prices.output || prices.price || 1.5;
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model?.modalities?.output?.includes("moderation"))
+            return true;
+        return /moderation/.test(modelId);
     }
     static getModelType(modelId) {
-        return this.getDefinition(modelId).type;
+        if (/moderation/.test(modelId))
+            return "moderation";
+        if (/embedding/.test(modelId))
+            return "embedding";
+        if (/dall-e|image/.test(modelId))
+            return "image";
+        if (/whisper|transcribe/.test(modelId))
+            return "audio_transcription";
+        if (/tts|speech/.test(modelId))
+            return "audio_speech";
+        if (/audio/.test(modelId))
+            return "audio";
+        return "chat";
     }
-    static formatDisplayName(modelId) {
-        const humanized = modelId.replace(/-/g, " ").split(" ").map(s => s.charAt(0).toUpperCase() + s.slice(1)).join(" ");
-        return this.applySpecialFormatting(humanized);
-    }
-    static applySpecialFormatting(name) {
-        return name
-            .replace(/(\d{4}) (\d{2}) (\d{2})/, "$1$2$3")
-            .replace(/^(?:Gpt|Chatgpt|Tts|Dall E) /g, (m) => this.specialPrefixFormat(m.trim()))
-            .replace(/^O([13]) /g, "O$1-")
-            .replace(/^O[13] Mini/g, (m) => m.replace(" ", "-"))
-            .replace(/\d\.\d /g, (m) => m.replace(" ", "-"))
-            .replace(/4o (?=Mini|Preview|Turbo|Audio|Realtime|Transcribe|Tts)/g, "4o-")
-            .replace(/\bHd\b/g, "HD")
-            .replace(/(?:Omni|Text) Moderation/g, (m) => m.replace(" ", "-"))
-            .replace("Text Embedding", "text-embedding-");
-    }
-    static specialPrefixFormat(prefix) {
-        switch (prefix) {
-            case "Gpt": return "GPT-";
-            case "Chatgpt": return "ChatGPT-";
-            case "Tts": return "TTS-";
-            case "Dall E": return "DALL-E-";
-            default: return prefix + "-";
+    static getModalities(modelId) {
+        const input = ["text"];
+        const output = ["text"];
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model?.modalities)
+            return model.modalities;
+        if (this.supportsVision(modelId))
+            input.push("image", "pdf");
+        if (this.supportsTranscription(modelId))
+            input.push("audio");
+        if (this.supportsImageGeneration(modelId))
+            output.push("image");
+        if (this.supportsEmbeddings(modelId))
+            output.push("embeddings");
+        if (this.supportsModeration(modelId))
+            output.push("moderation");
+        return { input, output };
+    }
+    static getCapabilities(modelId) {
+        const caps = ["streaming"];
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model) {
+            model.capabilities.forEach(c => { if (!caps.includes(c))
+                caps.push(c); });
+            return caps;
         }
+        if (this.supportsTools(modelId))
+            caps.push("function_calling");
+        if (this.supportsStructuredOutput(modelId))
+            caps.push("structured_output");
+        if (this.supportsEmbeddings(modelId))
+            caps.push("batch");
+        if (/o\d|gpt-5/.test(modelId))
+            caps.push("reasoning");
+        if (this.supportsImageGeneration(modelId))
+            caps.push("image_generation");
+        if (this.supportsTranscription(modelId))
+            caps.push("transcription");
+        return caps;
     }
     static normalizeTemperature(temperature, modelId) {
-        if (/^(o\d|gpt-5)/.test(modelId)) {
+        if (/^(o\d|gpt-5)/.test(modelId))
             return 1.0;
-        }
-        if (/-search/.test(modelId)) {
+        if (/-search/.test(modelId))
             return null;
-        }
         return temperature;
     }
-    static getModalities(modelId) {
-        const type = this.getModelType(modelId);
-        const features = this.getDefinition(modelId).features;
-        const modalities = {
-            input: ["text"],
-            output: ["text"]
-        };
-        if (features.vision)
-            modalities.input.push("image", "pdf");
-        if (type === "audio") {
-            modalities.input.push("audio");
-            modalities.output.push("audio");
-        }
-        if (type === "image")
-            modalities.output.push("image");
-        if (type === "embedding")
-            modalities.output.push("embeddings");
-        if (type === "moderation")
-            modalities.output.push("moderation");
-        return modalities;
-    }
-    static getCapabilities(modelId) {
-        const capabilities = [];
-        const type = this.getModelType(modelId);
-        const features = this.getDefinition(modelId).features;
-        if (type !== "moderation" && type !== "embedding")
-            capabilities.push("streaming");
-        if (features.tools)
-            capabilities.push("function_calling");
-        if (features.structuredOutput)
-            capabilities.push("structured_output");
-        if (type === "embedding")
-            capabilities.push("batch");
-        if (/o\d|gpt-5|codex/.test(modelId))
-            capabilities.push("reasoning");
-        if (type === "image")
-            capabilities.push("image_generation");
-        if (type === "audio")
-            capabilities.push("speech_generation", "transcription");
-        return capabilities;
+    static formatDisplayName(modelId) {
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model?.name && model.name !== modelId)
+            return model.name;
+        return modelId.replace(/-/g, " ").replace(/\b\w/g, c => c.toUpperCase());
     }
     static getPricing(modelId) {
-        const standardPricing = {
-            input_per_million: this.getInputPrice(modelId),
-            output_per_million: this.getOutputPrice(modelId)
-        };
-        const cachedPrice = this.getCachedInputPrice(modelId);
-        const pricing = {
+        const model = ModelRegistry.find(modelId, "openai");
+        if (model?.pricing)
+            return model.pricing;
+        let input = 2.5, output = 10.0;
+        if (/gpt-3/.test(modelId)) {
+            input = 0.5;
+            output = 1.5;
+        }
+        if (/mini/.test(modelId)) {
+            input = 0.15;
+            output = 0.6;
+        }
+        return {
             text_tokens: {
-                standard: {
-                    ...standardPricing,
-                    ...(cachedPrice ? { cached_input_per_million: cachedPrice } : {})
-                }
+                standard: { input_per_million: input, output_per_million: output }
             }
         };
-        if (this.getModelType(modelId) === "embedding") {
-            pricing.text_tokens.batch = {
-                input_per_million: standardPricing.input_per_million * 0.5,
-                output_per_million: standardPricing.output_per_million * 0.5
-            };
-        }
-        return pricing;
     }
 }

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

@@ -1 +1 @@
-{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Chat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAK3D,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;CA0D3D"}
+{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/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;CAsD3D"}