aimodels 0.3.3 → 0.3.6

package/dist/index.js

@@ -20,6 +20,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  AIModels: () => AIModels,
   ModelCollection: () => ModelCollection,
   creators: () => creators_default,
   models: () => models
@@ -64,11 +65,20 @@ var ModelCollection = class _ModelCollection extends Array {
   fromProvider(provider) {
     return this.filter((model) => model.providers.includes(provider));
   }
+  /** Get models available from a specific creator */
+  fromCreator(creator) {
+    return new _ModelCollection(
+      this.filter((model) => model.creator === creator)
+    );
+  }
   /** Filter models by minimum context window size */
   withMinContext(tokens) {
     return this.filter((model) => {
       const context = model.context;
-      if (!("total" in context) || context.total === null) {
+      if (context.type !== "token" && context.type !== "character") {
+        return false;
+      }
+      if (context.total === null) {
         return false;
       }
       return context.total >= tokens;
@@ -81,61 +91,74 @@ var openai_models_default = {
   creator: "openai",
   models: [
     {
-      id: "whisper-large-v3",
-      name: "Whisper Large v3",
-      license: "mit",
+      id: "gpt-4",
+      name: "GPT-4",
+      license: "proprietary",
       providers: [
         "openai",
-        "azure",
-        "groq"
+        "azure"
       ],
       can: [
-        "sound-in",
-        "text-out"
+        "chat",
+        "text-in",
+        "text-out",
+        "json-out",
+        "function-out"
       ],
       context: {
-        total: null,
-        maxOutput: null
+        type: "token",
+        total: 8192,
+        maxOutput: 4096
       }
     },
     {
-      id: "whisper-large-v3-turbo",
-      name: "Whisper Large v3 Turbo",
-      license: "mit",
-      providers: [
-        "openai",
-        "azure",
-        "groq"
-      ],
-      can: [
-        "sound-in",
-        "text-out"
-      ],
-      context: {
-        total: null,
-        maxOutput: null
+      id: "gpt-4-turbo-preview",
+      extends: "gpt-4",
+      overrides: {
+        name: "GPT-4 Turbo",
+        context: {
+          type: "token",
+          total: 128e3,
+          maxOutput: 4096
+        }
       }
     },
     {
-      id: "tts-1",
-      name: "TTS-1",
+      id: "gpt-3.5-turbo",
+      name: "GPT-3.5 Turbo",
       license: "proprietary",
       providers: [
         "openai",
         "azure"
       ],
       can: [
+        "chat",
         "text-in",
-        "sound-out"
+        "text-out",
+        "json-out",
+        "function-out"
       ],
       context: {
-        total: null,
-        maxOutput: null
+        type: "token",
+        total: 16385,
+        maxOutput: 4096
       }
     },
     {
-      id: "tts-1-hd",
-      name: "TTS-1 HD",
+      id: "gpt-3.5-turbo-16k",
+      extends: "gpt-3.5-turbo",
+      overrides: {
+        name: "GPT-3.5 Turbo 16K",
+        context: {
+          type: "token",
+          total: 16385,
+          maxOutput: 4096
+        }
+      }
+    },
+    {
+      id: "gpt-3.5-turbo-instruct",
+      name: "GPT-3.5 Turbo Instruct",
       license: "proprietary",
       providers: [
         "openai",
@@ -143,11 +166,12 @@ var openai_models_default = {
       ],
       can: [
         "text-in",
-        "sound-out"
+        "text-out"
       ],
       context: {
-        total: null,
-        maxOutput: null
+        type: "token",
+        total: 4096,
+        maxOutput: 4096
       }
     },
     {
@@ -160,32 +184,184 @@ var openai_models_default = {
       ],
       can: [
         "chat",
+        "text-in",
+        "text-out",
         "img-in",
         "json-out",
         "function-out"
       ],
       context: {
+        type: "token",
         total: 128e3,
         maxOutput: 16384
       }
     },
+    {
+      id: "chatgpt-4o-latest",
+      extends: "gpt-4o",
+      overrides: {
+        name: "GPT-4o used in ChatGPT"
+      }
+    },
+    {
+      id: "gpt-4o-2024-11-20",
+      extends: "gpt-4o",
+      overrides: {
+        name: "GPT-4o (2024-11-20)"
+      }
+    },
+    {
+      id: "gpt-4o-2024-08-06",
+      extends: "gpt-4o",
+      overrides: {
+        name: "GPT-4o (2024-08-06)"
+      }
+    },
+    {
+      id: "gpt-4o-2024-05-13",
+      extends: "gpt-4o",
+      overrides: {
+        name: "GPT-4o (2024-05-13)",
+        context: {
+          type: "token",
+          total: 128e3,
+          maxOutput: 4096
+        }
+      }
+    },
     {
       id: "gpt-4o-mini",
-      name: "GPT-4O Mini",
-      license: "proprietary",
-      providers: [
-        "openai",
-        "azure"
-      ],
-      can: [
-        "chat",
-        "img-in",
-        "json-out",
-        "function-out"
-      ],
-      context: {
-        total: 128e3,
-        maxOutput: 16384
+      extends: "gpt-4o",
+      overrides: {
+        name: "GPT-4o Mini",
+        context: {
+          type: "token",
+          total: 128e3,
+          maxOutput: 16384
+        }
+      }
+    },
+    {
+      id: "gpt-4o-audio-preview",
+      extends: "gpt-4o",
+      overrides: {
+        name: "GPT-4o Audio",
+        can: [
+          "chat",
+          "text-in",
+          "text-out",
+          "sound-in",
+          "json-out",
+          "function-out"
+        ],
+        context: {
+          type: "token",
+          total: 128e3,
+          maxOutput: 16384
+        }
+      }
+    },
+    {
+      id: "gpt-4o-audio-preview-2024-12-17",
+      extends: "gpt-4o-audio-preview",
+      overrides: {
+        name: "GPT-4o Audio (2024-12-17)"
+      }
+    },
+    {
+      id: "gpt-4o-audio-preview-2024-10-01",
+      extends: "gpt-4o-audio-preview",
+      overrides: {
+        name: "GPT-4o Audio (2024-10-01)"
+      }
+    },
+    {
+      id: "gpt-4o-mini-audio-preview",
+      extends: "gpt-4o-mini",
+      overrides: {
+        name: "GPT-4o Mini Audio",
+        can: [
+          "chat",
+          "text-in",
+          "text-out",
+          "sound-in",
+          "sound-out",
+          "json-out",
+          "function-out"
+        ],
+        context: {
+          type: "token",
+          total: 128e3,
+          maxOutput: 16384
+        }
+      }
+    },
+    {
+      id: "gpt-4o-mini-audio-preview-2024-12-17",
+      extends: "gpt-4o-mini-audio-preview",
+      overrides: {
+        name: "GPT-4o Mini Audio (2024-12-17)"
+      }
+    },
+    {
+      id: "gpt-4o-realtime-preview",
+      extends: "gpt-4o",
+      overrides: {
+        name: "GPT-4o Realtime",
+        can: [
+          "chat",
+          "text-in",
+          "text-out",
+          "sound-in",
+          "json-out",
+          "function-out"
+        ],
+        context: {
+          type: "token",
+          total: 128e3,
+          maxOutput: 4096
+        }
+      }
+    },
+    {
+      id: "gpt-4o-realtime-preview-2024-12-17",
+      extends: "gpt-4o-realtime-preview",
+      overrides: {
+        name: "GPT-4o Realtime (2024-12-17)"
+      }
+    },
+    {
+      id: "gpt-4o-realtime-preview-2024-10-01",
+      extends: "gpt-4o-realtime-preview",
+      overrides: {
+        name: "GPT-4o Realtime (2024-10-01)"
+      }
+    },
+    {
+      id: "gpt-4o-mini-realtime-preview",
+      extends: "gpt-4o-mini",
+      overrides: {
+        name: "GPT-4o Mini Realtime",
+        can: [
+          "chat",
+          "text-in",
+          "text-out",
+          "sound-in",
+          "json-out",
+          "function-out"
+        ],
+        context: {
+          type: "token",
+          total: 128e3,
+          maxOutput: 4096
+        }
+      }
+    },
+    {
+      id: "gpt-4o-mini-realtime-preview-2024-12-17",
+      extends: "gpt-4o-mini-realtime-preview",
+      overrides: {
+        name: "GPT-4o Mini Realtime (2024-12-17)"
       }
     },
     {
@@ -198,19 +374,75 @@ var openai_models_default = {
       ],
       can: [
         "chat",
+        "text-in",
+        "text-out",
         "img-in",
         "json-out",
         "function-out",
         "reason"
       ],
       context: {
+        type: "token",
         total: 2e5,
         maxOutput: 1e5
       }
     },
+    {
+      id: "o1-2024-12-17",
+      extends: "o1",
+      overrides: {
+        name: "OpenAI O1 (2024-12-17)"
+      }
+    },
     {
       id: "o1-mini",
-      name: "OpenAI O1 Mini",
+      extends: "o1",
+      overrides: {
+        name: "OpenAI O1 Mini",
+        can: [
+          "chat",
+          "text-in",
+          "text-out",
+          "json-out",
+          "function-out",
+          "reason"
+        ],
+        context: {
+          type: "token",
+          total: 128e3,
+          maxOutput: 65536
+        }
+      }
+    },
+    {
+      id: "o1-mini-2024-09-12",
+      extends: "o1-mini",
+      overrides: {
+        name: "OpenAI O1 Mini (2024-09-12)"
+      }
+    },
+    {
+      id: "o1-preview",
+      extends: "o1",
+      overrides: {
+        name: "OpenAI O1 Preview",
+        context: {
+          type: "token",
+          total: 128e3,
+          maxOutput: 32768
+        }
+      }
+    },
+    {
+      id: "o1-preview-2024-09-12",
+      extends: "o1-preview",
+      overrides: {
+        name: "OpenAI O1 Preview (2024-09-12)"
+      }
+    },
+    {
+      id: "o3-mini",
+      name: "OpenAI O3 Mini",
       license: "proprietary",
       providers: [
         "openai",
@@ -218,70 +450,89 @@ var openai_models_default = {
       ],
       can: [
         "chat",
+        "text-in",
+        "text-out",
         "json-out",
         "function-out",
         "reason"
       ],
       context: {
-        total: 128e3,
-        maxOutput: 65536
+        type: "token",
+        total: 2e5,
+        maxOutput: 1e5
       }
     },
     {
-      id: "o3-mini",
-      name: "OpenAI O3 Mini",
+      id: "o3-mini-2025-01-31",
+      extends: "o3-mini",
+      overrides: {
+        name: "OpenAI O3 Mini (2025-01-31)"
+      }
+    },
+    {
+      id: "whisper-1",
+      name: "Whisper",
       license: "proprietary",
       providers: [
         "openai",
         "azure"
       ],
       can: [
-        "chat",
-        "json-out",
-        "function-out",
-        "reason"
+        "sound-in",
+        "text-out"
       ],
       context: {
-        total: 2e5,
-        maxOutput: 1e5
+        type: "audio-in",
+        total: null,
+        maxOutput: null
       }
     },
     {
-      id: "gpt-4o-audio-preview",
-      name: "GPT-4O Audio",
+      id: "tts-1",
+      name: "TTS-1",
       license: "proprietary",
       providers: [
         "openai",
         "azure"
       ],
       can: [
-        "chat",
-        "sound-in",
-        "json-out",
-        "function-out"
+        "text-in",
+        "sound-out"
       ],
       context: {
-        total: 128e3,
-        maxOutput: 16384
+        type: "audio-out",
+        total: null,
+        maxOutput: null
       }
     },
     {
-      id: "gpt-4o-realtime-preview",
-      name: "GPT-4O Realtime",
+      id: "tts-1-hd",
+      extends: "tts-1",
+      overrides: {
+        name: "TTS-1 HD"
+      }
+    },
+    {
+      id: "dall-e-2",
+      name: "DALL-E 2",
       license: "proprietary",
       providers: [
         "openai",
         "azure"
       ],
       can: [
-        "chat",
-        "sound-in",
-        "json-out",
-        "function-out"
+        "img-out"
       ],
       context: {
-        total: 128e3,
-        maxOutput: 4096
+        maxOutput: 1,
+        sizes: [
+          "256x256",
+          "512x512",
+          "1024x1024"
+        ],
+        qualities: [
+          "standard"
+        ]
       }
     },
     {
@@ -293,6 +544,7 @@ var openai_models_default = {
         "azure"
       ],
       can: [
+        "text-in",
         "img-out"
       ],
       context: {
@@ -307,6 +559,69 @@ var openai_models_default = {
           "hd"
         ]
       }
+    },
+    {
+      id: "text-embedding-ada-002",
+      name: "Text Embedding Ada 002",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "text-in",
+        "vectors-out"
+      ],
+      context: {
+        type: "embedding",
+        total: 8191,
+        unit: "tokens",
+        dimensions: 1536,
+        embeddingType: "text",
+        normalized: true
+      }
+    },
+    {
+      id: "text-embedding-3-small",
+      name: "Text Embedding 3 Small",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "text-in",
+        "vectors-out"
+      ],
+      context: {
+        type: "embedding",
+        total: 8191,
+        unit: "tokens",
+        dimensions: 1536,
+        embeddingType: "text",
+        normalized: true
+      }
+    },
+    {
+      id: "text-embedding-3-large",
+      name: "Text Embedding 3 Large",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "text-in",
+        "vectors-out"
+      ],
+      context: {
+        type: "embedding",
+        total: 8191,
+        unit: "tokens",
+        dimensions: 3072,
+        embeddingType: "text",
+        normalized: true
+      }
     }
   ]
 };
@@ -316,14 +631,10 @@ var anthropic_models_default = {
   creator: "anthropic",
   models: [
     {
-      id: "claude-3-opus-20240229",
+      id: "claude-3-opus",
       name: "Claude 3 Opus",
       license: "proprietary",
-      providers: [
-        "anthropic",
-        "aws",
-        "google"
-      ],
+      providers: ["anthropic", "aws", "google"],
       can: [
         "chat",
         "text-in",
@@ -333,20 +644,25 @@ var anthropic_models_default = {
         "function-out"
       ],
       context: {
+        type: "token",
         total: 2e5,
         maxOutput: 4096,
         outputIsFixed: 1
+      },
+      aliases: ["claude-3-opus-latest"]
+    },
+    {
+      id: "claude-3-opus-20240229",
+      extends: "claude-3-opus",
+      overrides: {
+        name: "Claude 3 Opus (2024-02-29)"
       }
     },
     {
-      id: "claude-3-5-sonnet-20241022",
+      id: "claude-3-sonnet",
       name: "Claude 3 Sonnet",
       license: "proprietary",
-      providers: [
-        "anthropic",
-        "aws",
-        "google"
-      ],
+      providers: ["anthropic", "aws", "google"],
       can: [
         "chat",
         "text-in",
@@ -356,20 +672,52 @@ var anthropic_models_default = {
         "function-out"
       ],
       context: {
+        type: "token",
         total: 2e5,
-        maxOutput: 8192,
+        maxOutput: 4096,
         outputIsFixed: 1
+      },
+      aliases: ["claude-3-sonnet-latest"]
+    },
+    {
+      id: "claude-3-sonnet-20240229",
+      extends: "claude-3-sonnet",
+      overrides: {
+        name: "Claude 3 Sonnet (2024-02-29)"
       }
     },
     {
-      id: "claude-3-5-haiku-20241022",
+      id: "claude-3-haiku",
       name: "Claude 3 Haiku",
       license: "proprietary",
-      providers: [
-        "anthropic",
-        "aws",
-        "google"
+      providers: ["anthropic", "aws", "google"],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "img-in",
+        "json-out",
+        "function-out"
       ],
+      context: {
+        type: "token",
+        total: 2e5,
+        maxOutput: 4096,
+        outputIsFixed: 1
+      }
+    },
+    {
+      id: "claude-3-haiku-20240307",
+      extends: "claude-3-haiku",
+      overrides: {
+        name: "Claude 3 Haiku (2024-03-07)"
+      }
+    },
+    {
+      id: "claude-3-5-sonnet",
+      name: "Claude 3.5 Sonnet",
+      license: "proprietary",
+      providers: ["anthropic", "aws", "google"],
       can: [
         "chat",
         "text-in",
@@ -379,9 +727,45 @@ var anthropic_models_default = {
         "function-out"
       ],
       context: {
+        type: "token",
+        total: 2e5,
+        maxOutput: 8192,
+        outputIsFixed: 1
+      },
+      aliases: ["claude-3-5-sonnet-latest"]
+    },
+    {
+      id: "claude-3-5-sonnet-20241022",
+      extends: "claude-3-5-sonnet",
+      overrides: {
+        name: "Claude 3.5 Sonnet (2024-10-22)"
+      }
+    },
+    {
+      id: "claude-3-5-haiku",
+      name: "Claude 3.5 Haiku",
+      license: "proprietary",
+      providers: ["anthropic", "aws", "google"],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "json-out",
+        "function-out"
+      ],
+      context: {
+        type: "token",
         total: 2e5,
         maxOutput: 8192,
         outputIsFixed: 1
+      },
+      aliases: ["claude-3-5-haiku-latest"]
+    },
+    {
+      id: "claude-3-5-haiku-20241022",
+      extends: "claude-3-5-haiku",
+      overrides: {
+        name: "Claude 3.5 Haiku (2024-10-22)"
       }
     }
   ]
@@ -398,6 +782,7 @@ var meta_models_default = {
       providers: ["groq"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out"],
       context: {
+        type: "token",
         total: 8192,
         maxOutput: 4096
       }
@@ -409,6 +794,7 @@ var meta_models_default = {
       providers: ["groq"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out"],
       context: {
+        type: "token",
         total: 8192,
         maxOutput: 4096
       }
@@ -420,6 +806,7 @@ var meta_models_default = {
       providers: ["groq"],
       can: ["chat", "text-in", "text-out", "img-in"],
       context: {
+        type: "token",
         total: 4096,
         maxOutput: 4096
       }
@@ -431,6 +818,7 @@ var meta_models_default = {
       providers: ["groq"],
       can: ["chat", "text-in", "text-out", "img-in"],
       context: {
+        type: "token",
         total: 4096,
         maxOutput: 4096
       }
@@ -442,6 +830,7 @@ var meta_models_default = {
       providers: ["groq"],
       can: ["chat", "text-in", "text-out"],
       context: {
+        type: "token",
         total: 4096,
         maxOutput: 4096
       }
@@ -466,6 +855,7 @@ var mistral_models_default = {
         "function-out"
       ],
       context: {
+        type: "token",
         total: 32768,
         maxOutput: 4096
       }
@@ -483,6 +873,7 @@ var mistral_models_default = {
         "function-out"
       ],
       context: {
+        type: "token",
         total: 32768,
         maxOutput: 4096
       }
@@ -498,6 +889,7 @@ var mistral_models_default = {
         "text-out"
       ],
       context: {
+        type: "token",
         total: 32768,
         maxOutput: 4096
       }
@@ -513,6 +905,7 @@ var mistral_models_default = {
         "text-out"
       ],
       context: {
+        type: "token",
         total: 32768,
         maxOutput: 4096
       }
@@ -528,6 +921,7 @@ var mistral_models_default = {
         "text-out"
       ],
       context: {
+        type: "token",
         total: 32768,
         maxOutput: 4096
       }
@@ -546,6 +940,7 @@ var google_models_default = {
       providers: ["google"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in", "reason"],
       context: {
+        type: "token",
         total: 1048576,
         maxOutput: 8192
       },
@@ -558,6 +953,7 @@ var google_models_default = {
       providers: ["google"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
       context: {
+        type: "token",
         total: 1048576,
         maxOutput: 8192
       },
@@ -570,6 +966,7 @@ var google_models_default = {
       providers: ["google"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
       context: {
+        type: "token",
         total: 2097152,
         maxOutput: 8192
       }
@@ -581,6 +978,7 @@ var google_models_default = {
       providers: ["google"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
       context: {
+        type: "token",
         total: 2097152,
         maxOutput: 8192
       }
@@ -592,6 +990,7 @@ var google_models_default = {
       providers: ["google"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in", "reason"],
       context: {
+        type: "token",
         total: 2097152,
         maxOutput: 8192
       },
@@ -604,6 +1003,7 @@ var google_models_default = {
       providers: ["google"],
       can: ["text-in", "vectors-out"],
       context: {
+        type: "token",
         total: 2048,
         maxOutput: 768
       }
@@ -615,6 +1015,7 @@ var google_models_default = {
       providers: ["google"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out"],
       context: {
+        type: "token",
         total: 8192,
         maxOutput: 4096
       }
@@ -626,6 +1027,7 @@ var google_models_default = {
       providers: ["google"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out"],
       context: {
+        type: "token",
         total: 8192,
         maxOutput: 4096
       }
@@ -650,6 +1052,7 @@ var deepseek_models_default = {
         "function-out"
       ],
       context: {
+        type: "token",
         total: 131072,
         maxOutput: 8192
       }
@@ -668,6 +1071,7 @@ var deepseek_models_default = {
         "reason"
       ],
       context: {
+        type: "token",
         total: 131072,
         maxOutput: 8192
       }
@@ -687,6 +1091,7 @@ var xai_models_default = {
       providers: ["xai"],
       can: ["chat", "text-in", "text-out", "img-in"],
       context: {
+        type: "token",
         total: 32768,
         maxOutput: 4096
       },
@@ -700,6 +1105,7 @@ var xai_models_default = {
       providers: ["xai"],
       can: ["chat", "text-in", "text-out"],
       context: {
+        type: "token",
         total: 131072,
         maxOutput: 4096
       },
@@ -713,6 +1119,7 @@ var xai_models_default = {
       providers: ["xai"],
       can: ["chat", "text-in", "text-out", "img-in"],
       context: {
+        type: "token",
         total: 8192,
         maxOutput: 4096
       }
@@ -725,6 +1132,7 @@ var xai_models_default = {
       providers: ["xai"],
       can: ["chat", "text-in", "text-out"],
       context: {
+        type: "token",
         total: 131072,
         maxOutput: 4096
       }
@@ -744,6 +1152,7 @@ var cohere_models_default = {
       providers: ["cohere"],
       can: ["chat", "text-in", "text-out", "reason", "function-out"],
       context: {
+        type: "token",
         total: 128e3,
         maxOutput: 4096
       }
@@ -756,6 +1165,7 @@ var cohere_models_default = {
       providers: ["cohere", "bedrock", "azure", "oracle"],
       can: ["chat", "text-in", "text-out", "reason", "function-out"],
       context: {
+        type: "token",
         total: 128e3,
         maxOutput: 4096
       },
@@ -769,6 +1179,7 @@ var cohere_models_default = {
       providers: ["cohere", "bedrock", "azure", "oracle"],
       can: ["chat", "text-in", "text-out", "reason", "function-out"],
       context: {
+        type: "token",
         total: 128e3,
         maxOutput: 4096
       },
@@ -782,6 +1193,7 @@ var cohere_models_default = {
       providers: ["cohere", "bedrock", "oracle"],
       can: ["chat", "text-in", "text-out"],
       context: {
+        type: "token",
         total: 4096,
         maxOutput: 4096
       }
@@ -794,6 +1206,7 @@ var cohere_models_default = {
       providers: ["cohere", "bedrock", "oracle"],
       can: ["chat", "text-in", "text-out"],
       context: {
+        type: "token",
         total: 4096,
         maxOutput: 4096
       }
@@ -806,6 +1219,7 @@ var cohere_models_default = {
       providers: ["cohere", "bedrock", "azure", "oracle"],
       can: ["text-in", "img-in", "vectors-out"],
       context: {
+        type: "token",
         total: 512,
         maxOutput: 1024
       }
@@ -818,6 +1232,7 @@ var cohere_models_default = {
       providers: ["cohere", "oracle"],
       can: ["text-in", "img-in", "vectors-out"],
       context: {
+        type: "token",
         total: 512,
         maxOutput: 384
       }
@@ -830,6 +1245,7 @@ var cohere_models_default = {
       providers: ["cohere", "bedrock", "azure", "oracle"],
       can: ["text-in", "img-in", "vectors-out"],
       context: {
+        type: "token",
         total: 512,
         maxOutput: 1024
       }
@@ -842,6 +1258,7 @@ var cohere_models_default = {
       providers: ["cohere", "oracle"],
       can: ["text-in", "img-in", "vectors-out"],
       context: {
+        type: "token",
         total: 512,
         maxOutput: 384
       }
@@ -854,6 +1271,7 @@ var cohere_models_default = {
       providers: ["cohere", "bedrock", "azure"],
       can: ["text-in", "text-out"],
       context: {
+        type: "token",
         total: 4096,
         maxOutput: null
       }
@@ -866,6 +1284,7 @@ var cohere_models_default = {
       providers: ["cohere", "azure"],
       can: ["text-in", "text-out"],
       context: {
+        type: "token",
         total: 4096,
         maxOutput: null
       }
@@ -878,6 +1297,7 @@ var cohere_models_default = {
       providers: ["cohere", "azure"],
       can: ["text-in", "text-out"],
       context: {
+        type: "token",
         total: 4096,
         maxOutput: null
       }
@@ -897,6 +1317,19 @@ function validateModel(raw) {
       throw new Error(`Model ${field} must be a string`);
     }
   }
+  if (model.extends !== void 0 && typeof model.extends !== "string") {
+    throw new Error("Model extends must be a string");
+  }
+  if (model.overrides !== void 0) {
+    if (typeof model.overrides !== "object" || model.overrides === null) {
+      throw new Error("Model overrides must be an object");
+    }
+    const validOverrideFields = ["name", "creator", "license", "providers", "can", "languages", "aliases", "context"];
+    const invalidFields = Object.keys(model.overrides).filter((field) => !validOverrideFields.includes(field));
+    if (invalidFields.length > 0) {
+      throw new Error(`Invalid override fields: ${invalidFields.join(", ")}`);
+    }
+  }
   if (!Array.isArray(model.providers)) {
     throw new Error("Model providers must be an array");
   }
@@ -938,6 +1371,16 @@ function validateModel(raw) {
     if (typeof context.maxOutput !== "number") {
       throw new Error("Image model context.maxOutput must be a number");
     }
+  } else if (context.type === "embedding") {
+    if (typeof context.total !== "number") {
+      throw new Error("Embedding model context.total must be a number");
+    }
+    if (typeof context.unit !== "string") {
+      throw new Error("Embedding model context.unit must be a string");
+    }
+    if (typeof context.dimensions !== "number") {
+      throw new Error("Embedding model context.dimensions must be a number");
+    }
   } else {
     if (typeof context.total !== "number" && context.total !== null) {
       throw new Error("Token model context.total must be a number or null");
@@ -955,8 +1398,41 @@ function validateModel(raw) {
     can: model.can,
     context: model.context,
     ...model.languages ? { languages: model.languages } : {},
-    ...model.aliases ? { aliases: model.aliases } : {}
+    ...model.aliases ? { aliases: model.aliases } : {},
+    ...model.extends ? { extends: model.extends } : {},
+    ...model.overrides ? { overrides: model.overrides } : {}
+  };
+}
+function resolveModel(model, allModels2, visited = /* @__PURE__ */ new Set()) {
+  if (!model.extends) {
+    return model;
+  }
+  if (visited.has(model.id)) {
+    throw new Error(`Circular dependency detected for model ${model.id}`);
+  }
+  visited.add(model.id);
+  const baseModel = allModels2[model.extends];
+  if (!baseModel) {
+    throw new Error(`Base model ${model.extends} not found for ${model.id}`);
+  }
+  const resolvedBase = resolveModel(baseModel, allModels2, visited);
+  if (!model.overrides) {
+    return {
+      ...resolvedBase,
+      id: model.id,
+      extends: model.extends
+    };
+  }
+  const resolved = {
+    ...resolvedBase,
+    ...model.overrides,
+    id: model.id,
+    extends: model.extends
   };
+  if (!resolved.name || !resolved.creator || !resolved.license || !resolved.providers || !resolved.can || !resolved.context) {
+    throw new Error(`Missing required fields in resolved model ${model.id}`);
+  }
+  return resolved;
 }
 function buildAllModels() {
   const allModelData = [
@@ -970,12 +1446,29 @@ function buildAllModels() {
     cohere_models_default
   ];
   const allModels2 = allModelData.flatMap(
-    (data) => data.models.map((model) => ({
-      ...model,
-      creator: data.creator
-    }))
+    (data) => data.models.map((model) => {
+      if (typeof model.id !== "string") {
+        throw new Error(`Model id must be a string`);
+      }
+      if (model.extends !== void 0 && typeof model.extends !== "string") {
+        throw new Error("Model extends must be a string");
+      }
+      if (model.overrides !== void 0 && (typeof model.overrides !== "object" || model.overrides === null)) {
+        throw new Error("Model overrides must be an object");
+      }
+      return {
+        ...model,
+        creator: data.creator
+      };
+    })
+  );
+  const modelMap = Object.fromEntries(
+    allModels2.map((model) => [model.id, model])
+  );
+  const resolvedModels = allModels2.map(
+    (model) => resolveModel(model, modelMap)
   );
-  return allModels2.map((model) => validateModel(model));
+  return resolvedModels.map((model) => validateModel(model));
 }
 
 // src/data/providers/openai-provider.json
@@ -1201,31 +1694,42 @@ var creators_default = {
 // src/index.ts
 var allModels = buildAllModels();
 var providersData = buildProvidersData();
-var ModelsCollection = class extends ModelCollection {
+var AIModels = class _AIModels extends ModelCollection {
+  constructor(models2 = []) {
+    super(models2);
+    Object.setPrototypeOf(this, _AIModels.prototype);
+  }
   get creators() {
     return Object.keys(creators_default.creators);
   }
   get providers() {
     return providersData.providers.map((p) => p.id);
   }
-  fromCreator(creator) {
-    return new ModelCollection(
-      this.filter(
-        (model) => model.license.startsWith(creator) || // For open source models
-        !!providersData.providers.find((p) => p.id === creator)?.models[model.id]
-        // For proprietary models
-      )
-    );
-  }
   getPrice(modelId, provider) {
     const providerData = providersData.providers.find((p) => p.id === provider);
     const price = providerData?.models[modelId];
     return price?.type === "token" ? price : void 0;
   }
+  /** Get provider information by ID */
+  getProvider(providerId) {
+    return providersData.providers.find((p) => p.id === providerId);
+  }
+  /** Get all providers that can serve a specific model */
+  getProvidersForModel(modelId) {
+    let model = this.id(modelId);
+    if (!model) {
+      model = this.find((m) => m.aliases?.includes(modelId));
+    }
+    if (!model) return [];
+    return providersData.providers.filter(
+      (p) => model.providers.includes(p.id)
+    );
+  }
 };
-var models = new ModelsCollection(allModels);
+var models = new AIModels(allModels);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AIModels,
   ModelCollection,
   creators,
   models