@node-llm/core 0.2.2 → 0.4.0

package/dist/providers/openai/Capabilities.js

@@ -1,227 +1,64 @@
-const MODEL_PATTERNS = {
-    dall_e: /^dall-e/,
-    chatgpt4o: /^chatgpt-4o/,
-    gpt41: /^gpt-4\.1(?!-(?:mini|nano))/,
-    gpt41_mini: /^gpt-4\.1-mini/,
-    gpt41_nano: /^gpt-4\.1-nano/,
-    gpt4: /^gpt-4(?:-\d{6})?$/,
-    gpt4_turbo: /^gpt-4(?:\.5)?-(?:\d{6}-)?(preview|turbo)/,
-    gpt35_turbo: /^gpt-3\.5-turbo/,
-    gpt4o: /^gpt-4o(?!-(?:mini|audio|realtime|transcribe|tts|search))/,
-    gpt4o_audio: /^gpt-4o-(?:audio)/,
-    gpt4o_mini: /^gpt-4o-mini(?!-(?:audio|realtime|transcribe|tts|search))/,
-    gpt4o_mini_audio: /^gpt-4o-mini-audio/,
-    gpt4o_mini_realtime: /^gpt-4o-mini-realtime/,
-    gpt4o_mini_transcribe: /^gpt-4o-mini-transcribe/,
-    gpt4o_mini_tts: /^gpt-4o-mini-tts/,
-    gpt4o_realtime: /^gpt-4o-realtime/,
-    gpt4o_search: /^gpt-4o-search/,
-    gpt4o_transcribe: /^gpt-4o-transcribe/,
-    gpt5: /^gpt-5/,
-    gpt5_mini: /^gpt-5-mini/,
-    gpt5_nano: /^gpt-5-nano/,
-    o1: /^o1(?!-(?:mini|pro))/,
-    o1_mini: /^o1-mini/,
-    o1_pro: /^o1-pro/,
-    o3_mini: /^o3-mini/,
-    babbage: /^babbage/,
-    davinci: /^davinci/,
-    embedding3_large: /^text-embedding-3-large/,
-    embedding3_small: /^text-embedding-3-small/,
-    embedding_ada: /^text-embedding-ada/,
-    tts1: /^tts-1(?!-hd)/,
-    tts1_hd: /^tts-1-hd/,
-    whisper: /^whisper/,
-    moderation: /^(?:omni|text)-moderation/,
-};
-const PRICES = {
-    gpt5: { input: 1.25, output: 10.0, cached_input: 0.125 },
-    gpt5_mini: { input: 0.25, output: 2.0, cached_input: 0.025 },
-    gpt5_nano: { input: 0.05, output: 0.4, cached_input: 0.005 },
-    gpt41: { input: 2.0, output: 8.0, cached_input: 0.5 },
-    gpt41_mini: { input: 0.4, output: 1.6, cached_input: 0.1 },
-    gpt41_nano: { input: 0.1, output: 0.4 },
-    chatgpt4o: { input: 5.0, output: 15.0 },
-    gpt4: { input: 10.0, output: 30.0 },
-    gpt4_turbo: { input: 10.0, output: 30.0 },
-    gpt45: { input: 75.0, output: 150.0 },
-    gpt35_turbo: { input: 0.5, output: 1.5 },
-    gpt4o: { input: 2.5, output: 10.0 },
-    gpt4o_audio: { input: 2.5, output: 10.0, audio_input: 40.0, audio_output: 80.0 },
-    gpt4o_mini: { input: 0.15, output: 0.6 },
-    gpt4o_mini_audio: { input: 0.15, output: 0.6, audio_input: 10.0, audio_output: 20.0 },
-    gpt4o_mini_realtime: { input: 0.6, output: 2.4 },
-    gpt4o_mini_transcribe: { input: 1.25, output: 5.0, audio_input: 3.0 },
-    gpt4o_mini_tts: { input: 0.6, output: 12.0 },
-    gpt4o_realtime: { input: 5.0, output: 20.0 },
-    gpt4o_search: { input: 2.5, output: 10.0 },
-    gpt4o_transcribe: { input: 2.5, output: 10.0, audio_input: 6.0 },
-    o1: { input: 15.0, output: 60.0 },
-    o1_mini: { input: 1.1, output: 4.4 },
-    o1_pro: { input: 150.0, output: 600.0 },
-    o3_mini: { input: 1.1, output: 4.4 },
-    babbage: { input: 0.4, output: 0.4 },
-    davinci: { input: 2.0, output: 2.0 },
-    embedding3_large: { price: 0.13 },
-    embedding3_small: { price: 0.02 },
-    embedding_ada: { price: 0.10 },
-    tts1: { price: 15.0 },
-    tts1_hd: { price: 30.0 },
-    whisper: { price: 0.006 },
-    moderation: { price: 0.0 }
-};
+import { OPENAI_MODELS } from "./ModelDefinitions.js";
 export class Capabilities {
     static getFamily(modelId) {
-        for (const [family, pattern] of Object.entries(MODEL_PATTERNS)) {
-            if (pattern.test(modelId)) {
-                return family;
+        for (const [key, def] of Object.entries(OPENAI_MODELS)) {
+            if (key === "other")
+                continue;
+            if (def.pattern.test(modelId)) {
+                return key;
             }
         }
         return "other";
     }
-    static getContextWindow(modelId) {
+    static getDefinition(modelId) {
         const family = this.getFamily(modelId);
-        switch (family) {
-            case "gpt41":
-            case "gpt41_mini":
-            case "gpt41_nano":
-                return 1_047_576;
-            case "gpt5":
-            case "gpt5_mini":
-            case "gpt5_nano":
-            case "chatgpt4o":
-            case "gpt4_turbo":
-            case "gpt4o":
-            case "gpt4o_audio":
-            case "gpt4o_mini":
-            case "gpt4o_mini_audio":
-            case "gpt4o_mini_realtime":
-            case "gpt4o_realtime":
-            case "gpt4o_search":
-            case "gpt4o_transcribe":
-            case "o1_mini":
-                return 128_000;
-            case "gpt4":
-                return 8_192;
-            case "gpt4o_mini_transcribe":
-                return 16_000;
-            case "o1":
-            case "o1_pro":
-            case "o3_mini":
-                return 200_000;
-            case "gpt35_turbo":
-                return 16_385;
-            case "gpt4o_mini_tts":
-            case "tts1":
-            case "tts1_hd":
-            case "whisper":
-            case "moderation":
-            case "embedding3_large":
-            case "embedding3_small":
-            case "embedding_ada":
-                return null;
-            default:
-                return 4_096;
-        }
+        return OPENAI_MODELS[family];
+    }
+    static getContextWindow(modelId) {
+        return this.getDefinition(modelId).contextWindow;
     }
     static getMaxOutputTokens(modelId) {
-        const family = this.getFamily(modelId);
-        switch (family) {
-            case "gpt5":
-            case "gpt5_mini":
-            case "gpt5_nano":
-                return 400_000;
-            case "gpt41":
-            case "gpt41_mini":
-            case "gpt41_nano":
-                return 32_768;
-            case "chatgpt4o":
-            case "gpt4o":
-            case "gpt4o_mini":
-                return 16_384;
-            case "babbage":
-            case "davinci":
-                return 16_384;
-            case "gpt4":
-                return 8_192;
-            case "gpt35_turbo":
-                return 4_096;
-            case "gpt4_turbo":
-            case "gpt4o_realtime":
-            case "gpt4o_mini_realtime":
-                return 4_096;
-            case "gpt4o_mini_transcribe":
-                return 2_000;
-            case "o1":
-            case "o1_pro":
-            case "o3_mini":
-                return 100_000;
-            case "o1_mini":
-                return 65_536;
-            case "gpt4o_mini_tts":
-            case "tts1":
-            case "tts1_hd":
-            case "whisper":
-            case "moderation":
-            case "embedding3_large":
-            case "embedding3_small":
-            case "embedding_ada":
-                return null;
-            default:
-                return 16_384;
-        }
+        return this.getDefinition(modelId).maxOutputTokens;
     }
     static supportsVision(modelId) {
-        const family = this.getFamily(modelId);
-        return [
-            "gpt5", "gpt5_mini", "gpt5_nano", "gpt41", "gpt41_mini", "gpt41_nano",
-            "chatgpt4o", "gpt4", "gpt4_turbo", "gpt4o", "gpt4o_mini", "o1", "o1_pro",
-            "moderation", "gpt4o_search", "gpt4o_mini_search"
-        ].includes(family);
+        return !!this.getDefinition(modelId).features.vision;
     }
     static supportsTools(modelId) {
-        const family = this.getFamily(modelId);
-        return [
-            "gpt5", "gpt5_mini", "gpt5_nano", "gpt41", "gpt41_mini", "gpt41_nano",
-            "gpt4", "gpt4_turbo", "gpt4o", "gpt4o_mini", "o1", "o1_pro", "o3_mini"
-        ].includes(family);
+        return !!this.getDefinition(modelId).features.tools;
     }
     static supportsStructuredOutput(modelId) {
-        const family = this.getFamily(modelId);
-        return [
-            "gpt5", "gpt5_mini", "gpt5_nano", "gpt41", "gpt41_mini", "gpt41_nano",
-            "chatgpt4o", "gpt4o", "gpt4o_mini", "o1", "o1_pro", "o3_mini"
-        ].includes(family);
+        return !!this.getDefinition(modelId).features.structuredOutput;
     }
     static supportsJsonMode(modelId) {
         return this.supportsStructuredOutput(modelId);
     }
+    static supportsEmbeddings(modelId) {
+        return this.getDefinition(modelId).type === "embedding";
+    }
+    static supportsImageGeneration(modelId) {
+        return this.getDefinition(modelId).type === "image";
+    }
+    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));
+    }
+    static supportsModeration(modelId) {
+        return this.getDefinition(modelId).type === "moderation";
+    }
     static getInputPrice(modelId) {
-        const family = this.getFamily(modelId);
-        const prices = PRICES[family] || { input: 0.5 };
+        const prices = this.getDefinition(modelId).pricing;
         return prices.input || prices.price || 0.5;
     }
     static getCachedInputPrice(modelId) {
-        const family = this.getFamily(modelId);
-        const prices = PRICES[family] || {};
-        return prices.cached_input;
+        return this.getDefinition(modelId).pricing.cached_input;
     }
     static getOutputPrice(modelId) {
-        const family = this.getFamily(modelId);
-        const prices = PRICES[family] || { output: 1.5 };
+        const prices = this.getDefinition(modelId).pricing;
         return prices.output || prices.price || 1.5;
     }
     static getModelType(modelId) {
-        const family = this.getFamily(modelId);
-        if (/embedding/.test(family))
-            return "embedding";
-        if (/^tts|whisper|gpt4o_(?:mini_)?(?:transcribe|tts)$/.test(family))
-            return "audio";
-        if (family === "moderation")
-            return "moderation";
-        if (/dall/.test(family))
-            return "image";
-        return "chat";
+        return this.getDefinition(modelId).type;
     }
     static formatDisplayName(modelId) {
         const humanized = modelId.replace(/-/g, " ").split(" ").map(s => s.charAt(0).toUpperCase() + s.slice(1)).join(" ");
@@ -258,43 +95,44 @@ export class Capabilities {
         return temperature;
     }
     static getModalities(modelId) {
+        const type = this.getModelType(modelId);
+        const features = this.getDefinition(modelId).features;
         const modalities = {
             input: ["text"],
             output: ["text"]
         };
-        if (this.supportsVision(modelId))
+        if (features.vision)
             modalities.input.push("image", "pdf");
-        if (/whisper|audio|tts|transcribe/.test(modelId))
+        if (type === "audio") {
             modalities.input.push("audio");
-        if (/tts|audio/.test(modelId))
             modalities.output.push("audio");
-        if (/dall-e|image/.test(modelId))
+        }
+        if (type === "image")
             modalities.output.push("image");
-        if (/embedding/.test(modelId))
+        if (type === "embedding")
             modalities.output.push("embeddings");
-        if (/moderation/.test(modelId))
+        if (type === "moderation")
             modalities.output.push("moderation");
         return modalities;
     }
     static getCapabilities(modelId) {
         const capabilities = [];
-        if (!/moderation|embedding/.test(modelId))
+        const type = this.getModelType(modelId);
+        const features = this.getDefinition(modelId).features;
+        if (type !== "moderation" && type !== "embedding")
             capabilities.push("streaming");
-        if (this.supportsTools(modelId))
+        if (features.tools)
             capabilities.push("function_calling");
-        if (this.supportsStructuredOutput(modelId))
+        if (features.structuredOutput)
             capabilities.push("structured_output");
-        if (/embedding|batch/.test(modelId))
+        if (type === "embedding")
             capabilities.push("batch");
         if (/o\d|gpt-5|codex/.test(modelId))
             capabilities.push("reasoning");
-        if (/gpt-4-turbo|gpt-4o/.test(modelId)) {
-            if (/vision/.test(modelId))
-                capabilities.push("image_generation");
-            if (/audio/.test(modelId)) {
-                capabilities.push("speech_generation", "transcription");
-            }
-        }
+        if (type === "image")
+            capabilities.push("image_generation");
+        if (type === "audio")
+            capabilities.push("speech_generation", "transcription");
         return capabilities;
     }
     static getPricing(modelId) {
@@ -311,7 +149,7 @@ export class Capabilities {
                 }
             }
         };
-        if (/embedding|batch/.test(modelId)) {
+        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

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;CAqD3D"}
+{"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"}

package/dist/providers/openai/Chat.js

@@ -24,11 +24,15 @@ export class OpenAIChat {
         if (request.tools) {
             body.tools = request.tools;
         }
+        if (request.response_format) {
+            body.response_format = request.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),
         });

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

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"Embedding.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Embedding.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAKtE,qBAAa,eAAe;IAExB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADN,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM;IAG3B,OAAO,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,iBAAiB,CAAC;CA8CrE"}

package/dist/providers/openai/Embedding.js

@@ -0,0 +1,48 @@
+import { handleOpenAIError } from "./Errors.js";
+import { Capabilities } from "./Capabilities.js";
+import { DEFAULT_MODELS } from "../../constants.js";
+export class OpenAIEmbedding {
+    baseUrl;
+    apiKey;
+    constructor(baseUrl, apiKey) {
+        this.baseUrl = baseUrl;
+        this.apiKey = apiKey;
+    }
+    async execute(request) {
+        const model = request.model || DEFAULT_MODELS.EMBEDDING;
+        // Validate that the model is an embedding model
+        if (Capabilities.getModelType(model) !== "embedding") {
+            throw new Error(`Model ${model} does not support embeddings.`);
+        }
+        const body = {
+            input: request.input,
+            model,
+        };
+        if (request.dimensions) {
+            body.dimensions = request.dimensions;
+        }
+        if (request.user) {
+            body.user = request.user;
+        }
+        const response = await fetch(`${this.baseUrl}/embeddings`, {
+            method: "POST",
+            headers: {
+                "Authorization": `Bearer ${this.apiKey}`,
+                "Content-Type": "application/json",
+            },
+            body: JSON.stringify(body),
+        });
+        if (!response.ok) {
+            await handleOpenAIError(response, request.model || DEFAULT_MODELS.EMBEDDING);
+        }
+        const json = await response.json();
+        // Extract vectors from the response
+        const vectors = json.data.map((item) => item.embedding);
+        return {
+            vectors,
+            model: json.model,
+            input_tokens: json.usage.prompt_tokens,
+            dimensions: vectors[0]?.length || 0,
+        };
+    }
+}

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

@@ -0,0 +1,25 @@
+export interface ModelPricing {
+    input?: number;
+    output?: number;
+    price?: number;
+    cached_input?: number;
+    audio_input?: number;
+    audio_output?: number;
+}
+export interface ModelFeatures {
+    vision?: boolean;
+    tools?: boolean;
+    structuredOutput?: boolean;
+    jsonMode?: boolean;
+}
+export type ModelType = "chat" | "embedding" | "audio" | "image" | "moderation";
+export interface ModelFamilyDefinition {
+    pattern: RegExp;
+    contextWindow: number | null;
+    maxOutputTokens: number | null;
+    pricing: ModelPricing;
+    features: ModelFeatures;
+    type: ModelType;
+}
+export declare const OPENAI_MODELS: Record<string, ModelFamilyDefinition>;
+//# sourceMappingURL=ModelDefinitions.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelDefinitions.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/ModelDefinitions.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,YAAY;IAC3B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,aAAa;IAC5B,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,MAAM,SAAS,GAAG,MAAM,GAAG,WAAW,GAAG,OAAO,GAAG,OAAO,GAAG,YAAY,CAAC;AAEhF,MAAM,WAAW,qBAAqB;IACpC,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,OAAO,EAAE,YAAY,CAAC;IACtB,QAAQ,EAAE,aAAa,CAAC;IACxB,IAAI,EAAE,SAAS,CAAC;CACjB;AAED,eAAO,MAAM,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,qBAAqB,CAkN/D,CAAC"}

package/dist/providers/openai/ModelDefinitions.js

@@ -0,0 +1,211 @@
+export const OPENAI_MODELS = {
+    gpt5: {
+        pattern: /^gpt-5/,
+        contextWindow: 128_000,
+        maxOutputTokens: 400_000,
+        pricing: { input: 1.25, output: 10.0, cached_input: 0.125 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    gpt5_mini: {
+        pattern: /^gpt-5-mini/,
+        contextWindow: 128_000,
+        maxOutputTokens: 400_000,
+        pricing: { input: 0.25, output: 2.0, cached_input: 0.025 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    gpt5_nano: {
+        pattern: /^gpt-5-nano/,
+        contextWindow: 128_000,
+        maxOutputTokens: 400_000,
+        pricing: { input: 0.05, output: 0.4, cached_input: 0.005 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    gpt41: {
+        pattern: /^gpt-4\.1(?!-(?:mini|nano))/,
+        contextWindow: 1_047_576,
+        maxOutputTokens: 32_768,
+        pricing: { input: 2.0, output: 8.0, cached_input: 0.5 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    gpt41_mini: {
+        pattern: /^gpt-4\.1-mini/,
+        contextWindow: 1_047_576,
+        maxOutputTokens: 32_768,
+        pricing: { input: 0.4, output: 1.6, cached_input: 0.1 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    gpt41_nano: {
+        pattern: /^gpt-4\.1-nano/,
+        contextWindow: 1_047_576,
+        maxOutputTokens: 32_768,
+        pricing: { input: 0.1, output: 0.4 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    chatgpt4o: {
+        pattern: /^chatgpt-4o/,
+        contextWindow: 128_000,
+        maxOutputTokens: 16_384,
+        pricing: { input: 5.0, output: 15.0 },
+        features: { vision: true, tools: false, structuredOutput: true },
+        type: "chat"
+    },
+    gpt4: {
+        pattern: /^gpt-4(?:-\d{6})?$/,
+        contextWindow: 8_192,
+        maxOutputTokens: 8_192,
+        pricing: { input: 10.0, output: 30.0 },
+        features: { vision: true, tools: true, structuredOutput: false },
+        type: "chat"
+    },
+    gpt4_turbo: {
+        pattern: /^gpt-4(?:\.5)?-(?:\d{6}-)?(preview|turbo)/,
+        contextWindow: 128_000,
+        maxOutputTokens: 4_096,
+        pricing: { input: 10.0, output: 30.0 },
+        features: { vision: true, tools: true, structuredOutput: false },
+        type: "chat"
+    },
+    gpt45: {
+        pattern: /^gpt-4\.5/, // Assuming pattern based on name, wasn't explicit in MODEL_PATTERNS but listed in type
+        contextWindow: 128_000, // Guessing based on gpt-4-turbo
+        maxOutputTokens: 4_096,
+        pricing: { input: 75.0, output: 150.0 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    gpt35_turbo: {
+        pattern: /^gpt-3\.5-turbo/,
+        contextWindow: 16_385,
+        maxOutputTokens: 4_096,
+        pricing: { input: 0.5, output: 1.5 },
+        features: { vision: false, tools: false, structuredOutput: false },
+        type: "chat"
+    },
+    gpt4o: {
+        pattern: /^gpt-4o(?!-(?:mini|audio|realtime|transcribe|tts|search))/,
+        contextWindow: 128_000,
+        maxOutputTokens: 16_384,
+        pricing: { input: 2.5, output: 10.0 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    gpt4o_audio: {
+        pattern: /^gpt-4o-(?:audio)/,
+        contextWindow: 128_000,
+        maxOutputTokens: 16_384, // Assuming same as gpt4o
+        pricing: { input: 2.5, output: 10.0, audio_input: 40.0, audio_output: 80.0 },
+        features: { vision: false, tools: false, structuredOutput: false }, // Check features
+        type: "audio"
+    },
+    gpt4o_mini: {
+        pattern: /^gpt-4o-mini(?!-(?:audio|realtime|transcribe|tts|search))/,
+        contextWindow: 128_000,
+        maxOutputTokens: 16_384,
+        pricing: { input: 0.15, output: 0.6 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    o1: {
+        pattern: /^o1(?!-(?:mini|pro))/,
+        contextWindow: 200_000,
+        maxOutputTokens: 100_000,
+        pricing: { input: 15.0, output: 60.0 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    o1_mini: {
+        pattern: /^o1-mini/,
+        contextWindow: 128_000,
+        maxOutputTokens: 65_536,
+        pricing: { input: 1.1, output: 4.4 },
+        features: { vision: false, tools: false, structuredOutput: false }, // Check features
+        type: "chat"
+    },
+    o1_pro: {
+        pattern: /^o1-pro/,
+        contextWindow: 200_000,
+        maxOutputTokens: 100_000,
+        pricing: { input: 150.0, output: 600.0 },
+        features: { vision: true, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    o3_mini: {
+        pattern: /^o3-mini/,
+        contextWindow: 200_000,
+        maxOutputTokens: 100_000,
+        pricing: { input: 1.1, output: 4.4 },
+        features: { vision: false, tools: true, structuredOutput: true },
+        type: "chat"
+    },
+    embedding3_small: {
+        pattern: /^text-embedding-3-small/,
+        contextWindow: null,
+        maxOutputTokens: null,
+        pricing: { price: 0.02 },
+        features: { vision: false, tools: false, structuredOutput: false },
+        type: "embedding"
+    },
+    embedding3_large: {
+        pattern: /^text-embedding-3-large/,
+        contextWindow: null,
+        maxOutputTokens: null,
+        pricing: { price: 0.13 },
+        features: { vision: false, tools: false, structuredOutput: false },
+        type: "embedding"
+    },
+    embedding_ada: {
+        pattern: /^text-embedding-ada/,
+        contextWindow: null,
+        maxOutputTokens: null,
+        pricing: { price: 0.10 },
+        features: { vision: false, tools: false, structuredOutput: false },
+        type: "embedding"
+    },
+    moderation: {
+        pattern: /^(?:omni|text)-moderation/,
+        contextWindow: null,
+        maxOutputTokens: null,
+        pricing: { price: 0.0 },
+        features: { vision: true, tools: false, structuredOutput: false },
+        type: "moderation"
+    },
+    dall_e: {
+        pattern: /^dall-e/,
+        contextWindow: null,
+        maxOutputTokens: null,
+        pricing: {}, // Variable
+        features: { vision: false, tools: false, structuredOutput: false },
+        type: "image"
+    },
+    whisper: {
+        pattern: /^whisper/,
+        contextWindow: null,
+        maxOutputTokens: null,
+        pricing: { price: 0.006 },
+        features: { vision: false, tools: false, structuredOutput: false },
+        type: "audio"
+    },
+    tts1: {
+        pattern: /^tts-1(?!-hd)/,
+        contextWindow: null,
+        maxOutputTokens: null,
+        pricing: { price: 15.0 },
+        features: { vision: false, tools: false, structuredOutput: false },
+        type: "audio"
+    },
+    // Default fallback
+    other: {
+        pattern: /.*/,
+        contextWindow: 4_096,
+        maxOutputTokens: 16_384,
+        pricing: { input: 0.5, output: 1.5 },
+        features: { vision: false, tools: false, structuredOutput: false },
+        type: "chat"
+    }
+};

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

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"Moderation.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Moderation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAIvE,qBAAa,gBAAgB;IACf,OAAO,CAAC,QAAQ,CAAC,OAAO;IAAU,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAAxC,OAAO,EAAE,MAAM,EAAmB,MAAM,EAAE,MAAM;IAEvE,OAAO,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;CAmBvE"}