@node-llm/core 0.3.0 → 0.4.0

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

@@ -0,0 +1,118 @@
+export interface GeminiPart {
+    text?: string;
+    inlineData?: {
+        mimeType: string;
+        data: string;
+    };
+    functionCall?: {
+        name: string;
+        args: Record<string, any>;
+    };
+    functionResponse?: {
+        name: string;
+        response: Record<string, any>;
+    };
+}
+export interface GeminiContent {
+    role?: "user" | "model";
+    parts: GeminiPart[];
+}
+export interface GeminiGenerateContentRequest {
+    contents: GeminiContent[];
+    systemInstruction?: {
+        parts: GeminiPart[];
+    };
+    tools?: {
+        functionDeclarations: Array<{
+            name: string;
+            description?: string;
+            parameters?: Record<string, any>;
+        }>;
+    }[];
+    generationConfig?: {
+        temperature?: number;
+        maxOutputTokens?: number;
+        stopSequences?: string[];
+        topP?: number;
+        topK?: number;
+        responseMimeType?: string;
+        responseSchema?: Record<string, any>;
+    };
+    safetySettings?: Array<{
+        category: string;
+        threshold: string;
+    }>;
+}
+export interface GeminiGenerateContentResponse {
+    candidates?: Array<{
+        content: GeminiContent;
+        finishReason?: string;
+        index?: number;
+        safetyRatings?: Array<{
+            category: string;
+            probability: string;
+        }>;
+    }>;
+    usageMetadata?: {
+        promptTokenCount: number;
+        candidatesTokenCount: number;
+        totalTokenCount: number;
+    };
+    modelVersion?: string;
+}
+export interface GeminiModel {
+    name: string;
+    version: string;
+    displayName: string;
+    description: string;
+    inputTokenLimit: number;
+    outputTokenLimit: number;
+    supportedGenerationMethods: string[];
+    temperature?: number;
+    topP?: number;
+    topK?: number;
+}
+export interface GeminiListModelsResponse {
+    models: GeminiModel[];
+    nextPageToken?: string;
+}
+export interface GeminiImageRequest {
+    prompt: string;
+    numberOfImages?: number;
+    aspectRatio?: string;
+    safetySetting?: Array<{
+        category: string;
+        threshold: string;
+    }>;
+}
+export interface GeminiImageResponse {
+    images: Array<{
+        image: {
+            mimeType: string;
+            data: string;
+        };
+    }>;
+}
+export interface GeminiEmbedRequest {
+    model: string;
+    content: {
+        parts: Array<{
+            text: string;
+        }>;
+    };
+    outputDimensionality?: number;
+}
+export interface GeminiEmbedResponse {
+    embedding: {
+        values: number[];
+    };
+}
+export interface GeminiBatchEmbedRequest {
+    requests: GeminiEmbedRequest[];
+}
+export interface GeminiBatchEmbedResponse {
+    embeddings: Array<{
+        values: number[];
+    }>;
+}
+//# sourceMappingURL=types.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/providers/gemini/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,UAAU;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE;QACX,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;IACF,YAAY,CAAC,EAAE;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC3B,CAAC;IACF,gBAAgB,CAAC,EAAE;QACjB,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC/B,CAAC;CACH;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;IACxB,KAAK,EAAE,UAAU,EAAE,CAAC;CACrB;AAED,MAAM,WAAW,4BAA4B;IAC3C,QAAQ,EAAE,aAAa,EAAE,CAAC;IAC1B,iBAAiB,CAAC,EAAE;QAClB,KAAK,EAAE,UAAU,EAAE,CAAC;KACrB,CAAC;IACF,KAAK,CAAC,EAAE;QACN,oBAAoB,EAAE,KAAK,CAAC;YAC1B,IAAI,EAAE,MAAM,CAAC;YACb,WAAW,CAAC,EAAE,MAAM,CAAC;YACrB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;SAClC,CAAC,CAAC;KACJ,EAAE,CAAC;IACJ,gBAAgB,CAAC,EAAE;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;QACzB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KACtC,CAAC;IACF,cAAc,CAAC,EAAE,KAAK,CAAC;QACrB,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,6BAA6B;IAC5C,UAAU,CAAC,EAAE,KAAK,CAAC;QACjB,OAAO,EAAE,aAAa,CAAC;QACvB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,aAAa,CAAC,EAAE,KAAK,CAAC;YACpB,QAAQ,EAAE,MAAM,CAAC;YACjB,WAAW,EAAE,MAAM,CAAC;SACrB,CAAC,CAAC;KACJ,CAAC,CAAC;IACH,aAAa,CAAC,EAAE;QACd,gBAAgB,EAAE,MAAM,CAAC;QACzB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,eAAe,EAAE,MAAM,CAAC;KACzB,CAAC;IACF,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,EAAE,MAAM,CAAC;IACzB,0BAA0B,EAAE,MAAM,EAAE,CAAC;IACrC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,wBAAwB;IACvC,MAAM,EAAE,WAAW,EAAE,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AACD,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,KAAK,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,mBAAmB;IAClC,MAAM,EAAE,KAAK,CAAC;QACZ,KAAK,EAAE;YACL,QAAQ,EAAE,MAAM,CAAC;YACjB,IAAI,EAAE,MAAM,CAAC;SACd,CAAC;KACH,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE;QACP,KAAK,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KAChC,CAAC;IACF,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED,MAAM,WAAW,mBAAmB;IAClC,SAAS,EAAE;QACT,MAAM,EAAE,MAAM,EAAE,CAAC;KAClB,CAAC;CACH;AAED,MAAM,WAAW,uBAAuB;IACtC,QAAQ,EAAE,kBAAkB,EAAE,CAAC;CAChC;AAED,MAAM,WAAW,wBAAwB;IACvC,UAAU,EAAE,KAAK,CAAC;QAChB,MAAM,EAAE,MAAM,EAAE,CAAC;KAClB,CAAC,CAAC;CACJ"}

package/dist/providers/gemini/types.js

@@ -0,0 +1 @@
+export {};

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

@@ -1,12 +1,17 @@
-export type ModelFamily = "dall_e" | "chatgpt4o" | "gpt41" | "gpt41_mini" | "gpt41_nano" | "gpt4" | "gpt4_turbo" | "gpt45" | "gpt35_turbo" | "gpt4o" | "gpt4o_audio" | "gpt4o_mini" | "gpt4o_mini_audio" | "gpt4o_mini_realtime" | "gpt4o_mini_transcribe" | "gpt4o_mini_tts" | "gpt4o_realtime" | "gpt4o_search" | "gpt4o_transcribe" | "gpt5" | "gpt5_mini" | "gpt5_nano" | "o1" | "o1_mini" | "o1_pro" | "o3_mini" | "babbage" | "davinci" | "embedding3_large" | "embedding3_small" | "embedding_ada" | "tts1" | "tts1_hd" | "whisper" | "moderation" | "other";
+import { ModelFamilyDefinition } from "./ModelDefinitions.js";
 export declare class Capabilities {
-    static getFamily(modelId: string): ModelFamily;
+    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;
     static supportsTools(modelId: string): boolean;
     static supportsStructuredOutput(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 getInputPrice(modelId: string): number;
     static getCachedInputPrice(modelId: string): number | undefined;
     static getOutputPrice(modelId: string): number;

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,MAAM,MAAM,WAAW,GACnB,QAAQ,GACR,WAAW,GACX,OAAO,GACP,YAAY,GACZ,YAAY,GACZ,MAAM,GACN,YAAY,GACZ,OAAO,GACP,aAAa,GACb,OAAO,GACP,aAAa,GACb,YAAY,GACZ,kBAAkB,GAClB,qBAAqB,GACrB,uBAAuB,GACvB,gBAAgB,GAChB,gBAAgB,GAChB,cAAc,GACd,kBAAkB,GAClB,MAAM,GACN,WAAW,GACX,WAAW,GACX,IAAI,GACJ,SAAS,GACT,QAAQ,GACR,SAAS,GACT,SAAS,GACT,SAAS,GACT,kBAAkB,GAClB,kBAAkB,GAClB,eAAe,GACf,MAAM,GACN,SAAS,GACT,SAAS,GACT,YAAY,GACZ,OAAO,CAAC;AA4EZ,qBAAa,YAAY;IACvB,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW;IAS9C,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IA8CvD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAgDzD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAS/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,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAM7C,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAM/D,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAM9C,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW,GAAG,OAAO,GAAG,YAAY,GAAG,OAAO,GAAG,MAAM;IAS7F,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;IAgB5E,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAmBjD,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM;CAyBlC"}
+{"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"}

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;CAsD3D"}
+{"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,6 +24,9 @@ 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: {

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