aimodels 0.3.10 → 0.3.12

package/dist/index.mjs

@@ -12,6 +12,48 @@ var ModelCollection = class _ModelCollection extends Array {
   can(...capabilities) {
     return this.filter((model) => capabilities.every((cap) => model.can.includes(cap)));
   }
+  /** 
+   * Fluent capability filters for better readability 
+   * Each method filters models by a specific capability
+   */
+  // Basic capabilities
+  canChat() {
+    return this.can("chat");
+  }
+  canReason() {
+    return this.can("reason");
+  }
+  // Text capabilities
+  canRead() {
+    return this.can("txt-in");
+  }
+  canWrite() {
+    return this.can("txt-out");
+  }
+  // Image capabilities
+  canSee() {
+    return this.can("img-in");
+  }
+  canGenerateImages() {
+    return this.can("img-out");
+  }
+  // Audio capabilities
+  canHear() {
+    return this.can("audio-in");
+  }
+  canSpeak() {
+    return this.can("audio-out");
+  }
+  // Output capabilities
+  canOutputJSON() {
+    return this.can("json-out");
+  }
+  canCallFunctions() {
+    return this.can("fn-out");
+  }
+  canGenerateEmbeddings() {
+    return this.can("vec-out");
+  }
   /** Filter models by one or more languages (all must be supported) */
   know(...languages) {
     return this.filter((model) => languages.every((lang) => model.languages?.includes(lang)));
@@ -59,6 +101,9 @@ var ModelCollection = class _ModelCollection extends Array {
   getProviders() {
     return [...new Set(this.flatMap((model) => model.providers))];
   }
+  getCreators() {
+    return [...new Set(this.map((model) => model.creator))];
+  }
 };
 
 // src/data/models/openai-models.json
@@ -75,10 +120,10 @@ var openai_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -108,10 +153,10 @@ var openai_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -140,8 +185,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -159,11 +204,11 @@ var openai_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "img-in",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -223,11 +268,11 @@ var openai_models_default = {
         name: "GPT-4o Audio",
         can: [
           "chat",
-          "text-in",
-          "text-out",
-          "sound-in",
+          "txt-in",
+          "txt-out",
+          "audio-in",
           "json-out",
-          "function-out"
+          "fn-out"
         ],
         context: {
           type: "token",
@@ -257,12 +302,12 @@ var openai_models_default = {
         name: "GPT-4o Mini Audio",
         can: [
           "chat",
-          "text-in",
-          "text-out",
-          "sound-in",
-          "sound-out",
+          "txt-in",
+          "txt-out",
+          "audio-in",
+          "audio-out",
           "json-out",
-          "function-out"
+          "fn-out"
         ],
         context: {
           type: "token",
@@ -285,11 +330,11 @@ var openai_models_default = {
         name: "GPT-4o Realtime",
         can: [
           "chat",
-          "text-in",
-          "text-out",
-          "sound-in",
+          "txt-in",
+          "txt-out",
+          "audio-in",
           "json-out",
-          "function-out"
+          "fn-out"
         ],
         context: {
           type: "token",
@@ -319,11 +364,11 @@ var openai_models_default = {
         name: "GPT-4o Mini Realtime",
         can: [
           "chat",
-          "text-in",
-          "text-out",
-          "sound-in",
+          "txt-in",
+          "txt-out",
+          "audio-in",
           "json-out",
-          "function-out"
+          "fn-out"
         ],
         context: {
           type: "token",
@@ -349,11 +394,11 @@ var openai_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "img-in",
         "json-out",
-        "function-out",
+        "fn-out",
         "reason"
       ],
       context: {
@@ -376,10 +421,10 @@ var openai_models_default = {
         name: "OpenAI O1 Mini",
         can: [
           "chat",
-          "text-in",
-          "text-out",
+          "txt-in",
+          "txt-out",
           "json-out",
-          "function-out",
+          "fn-out",
           "reason"
         ],
         context: {
@@ -425,10 +470,10 @@ var openai_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out",
+        "fn-out",
         "reason"
       ],
       context: {
@@ -453,8 +498,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "sound-in",
-        "text-out"
+        "audio-in",
+        "txt-out"
       ],
       context: {
         type: "audio-in",
@@ -471,8 +516,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "sound-out"
+        "txt-in",
+        "audio-out"
       ],
       context: {
         type: "audio-out",
@@ -519,7 +564,7 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
+        "txt-in",
         "img-out"
       ],
       context: {
@@ -544,8 +589,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "vectors-out"
+        "txt-in",
+        "vec-out"
       ],
       context: {
         type: "embedding",
@@ -565,8 +610,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "vectors-out"
+        "txt-in",
+        "vec-out"
       ],
       context: {
         type: "embedding",
@@ -586,8 +631,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "vectors-out"
+        "txt-in",
+        "vec-out"
       ],
       context: {
         type: "embedding",
@@ -612,11 +657,11 @@ var anthropic_models_default = {
       providers: ["anthropic", "aws", "google"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "img-in",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -633,11 +678,11 @@ var anthropic_models_default = {
       providers: ["anthropic", "aws", "google"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "img-in",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -654,11 +699,11 @@ var anthropic_models_default = {
       providers: ["anthropic", "aws", "google"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "img-in",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -675,11 +720,11 @@ var anthropic_models_default = {
       providers: ["anthropic", "aws", "google"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "img-in",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -696,11 +741,11 @@ var anthropic_models_default = {
       providers: ["anthropic", "aws", "google"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "img-in",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -716,10 +761,10 @@ var anthropic_models_default = {
       providers: ["anthropic", "aws", "google"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -736,11 +781,11 @@ var anthropic_models_default = {
       providers: ["anthropic", "aws", "google"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "img-in",
         "json-out",
-        "function-out",
+        "fn-out",
         "reason"
       ],
       context: {
@@ -771,7 +816,7 @@ var meta_models_default = {
       name: "Llama 3 70B",
       license: "llama-3-community",
       providers: ["groq"],
-      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      can: ["chat", "txt-in", "txt-out", "json-out", "fn-out"],
       context: {
         type: "token",
         total: 8192,
@@ -783,7 +828,7 @@ var meta_models_default = {
       name: "Llama 3 8B",
       license: "llama-3-community",
       providers: ["groq"],
-      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      can: ["chat", "txt-in", "txt-out", "json-out", "fn-out"],
       context: {
         type: "token",
         total: 8192,
@@ -795,7 +840,7 @@ var meta_models_default = {
       name: "Llama 3.2 11B Vision",
       license: "llama-3-community",
       providers: ["groq"],
-      can: ["chat", "text-in", "text-out", "img-in"],
+      can: ["chat", "txt-in", "txt-out", "img-in"],
       context: {
         type: "token",
         total: 4096,
@@ -807,7 +852,7 @@ var meta_models_default = {
       name: "Llama 3.2 90B Vision",
       license: "llama-3-community",
       providers: ["groq"],
-      can: ["chat", "text-in", "text-out", "img-in"],
+      can: ["chat", "txt-in", "txt-out", "img-in"],
       context: {
         type: "token",
         total: 4096,
@@ -819,7 +864,7 @@ var meta_models_default = {
       name: "LlamaGuard 3 8B",
       license: "llama-3-community",
       providers: ["groq"],
-      can: ["chat", "text-in", "text-out"],
+      can: ["chat", "txt-in", "txt-out"],
       context: {
         type: "token",
         total: 4096,
@@ -840,10 +885,10 @@ var mistral_models_default = {
       providers: ["mistral", "azure"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -858,10 +903,10 @@ var mistral_models_default = {
       providers: ["mistral", "azure"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -876,8 +921,8 @@ var mistral_models_default = {
       providers: ["mistral"],
       can: [
         "chat",
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -892,8 +937,8 @@ var mistral_models_default = {
       providers: ["mistral", "groq"],
       can: [
         "chat",
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -908,8 +953,8 @@ var mistral_models_default = {
       providers: ["mistral", "groq"],
       can: [
         "chat",
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -929,7 +974,7 @@ var google_models_default = {
       name: "Gemini 2.0 Flash",
       license: "proprietary",
       providers: ["google"],
-      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in", "reason"],
+      can: ["chat", "txt-in", "txt-out", "json-out", "fn-out", "img-in", "audio-in", "reason"],
       context: {
         type: "token",
         total: 1048576,
@@ -942,7 +987,7 @@ var google_models_default = {
       name: "Gemini 2.0 Flash Lite",
       license: "proprietary",
       providers: ["google"],
-      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
+      can: ["chat", "txt-in", "txt-out", "json-out", "fn-out", "img-in", "audio-in"],
       context: {
         type: "token",
         total: 1048576,
@@ -955,7 +1000,7 @@ var google_models_default = {
       name: "Gemini 1.5 Flash",
       license: "proprietary",
       providers: ["google"],
-      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
+      can: ["chat", "txt-in", "txt-out", "json-out", "fn-out", "img-in", "audio-in"],
       context: {
         type: "token",
         total: 2097152,
@@ -967,7 +1012,7 @@ var google_models_default = {
       name: "Gemini 1.5 Flash 8B",
       license: "proprietary",
       providers: ["google"],
-      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
+      can: ["chat", "txt-in", "txt-out", "json-out", "fn-out", "img-in", "audio-in"],
       context: {
         type: "token",
         total: 2097152,
@@ -979,7 +1024,7 @@ var google_models_default = {
       name: "Gemini 1.5 Pro",
       license: "proprietary",
       providers: ["google"],
-      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in", "reason"],
+      can: ["chat", "txt-in", "txt-out", "json-out", "fn-out", "img-in", "audio-in", "reason"],
       context: {
         type: "token",
         total: 2097152,
@@ -992,7 +1037,7 @@ var google_models_default = {
       name: "Text Embedding 004",
       license: "proprietary",
       providers: ["google"],
-      can: ["text-in", "vectors-out"],
+      can: ["txt-in", "vec-out"],
       context: {
         type: "token",
         total: 2048,
@@ -1004,7 +1049,7 @@ var google_models_default = {
       name: "Gemma 7B Instruct",
       license: "apache-2.0",
       providers: ["google"],
-      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      can: ["chat", "txt-in", "txt-out", "json-out", "fn-out"],
       context: {
         type: "token",
         total: 8192,
@@ -1016,7 +1061,7 @@ var google_models_default = {
       name: "Gemma 2B Instruct",
       license: "apache-2.0",
       providers: ["google"],
-      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      can: ["chat", "txt-in", "txt-out", "json-out", "fn-out"],
       context: {
         type: "token",
         total: 8192,
@@ -1037,10 +1082,10 @@ var deepseek_models_default = {
       providers: ["deepseek"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -1055,10 +1100,10 @@ var deepseek_models_default = {
       providers: ["deepseek"],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out",
+        "fn-out",
         "reason"
       ],
       context: {
@@ -1080,7 +1125,7 @@ var xai_models_default = {
       creator: "xai",
       license: "proprietary",
       providers: ["xai"],
-      can: ["chat", "text-in", "text-out", "img-in"],
+      can: ["chat", "txt-in", "txt-out", "img-in"],
       context: {
         type: "token",
         total: 32768,
@@ -1094,7 +1139,7 @@ var xai_models_default = {
       creator: "xai",
       license: "proprietary",
       providers: ["xai"],
-      can: ["chat", "text-in", "text-out"],
+      can: ["chat", "txt-in", "txt-out"],
       context: {
         type: "token",
         total: 131072,
@@ -1108,7 +1153,7 @@ var xai_models_default = {
       creator: "xai",
       license: "proprietary",
       providers: ["xai"],
-      can: ["chat", "text-in", "text-out", "img-in"],
+      can: ["chat", "txt-in", "txt-out", "img-in"],
       context: {
         type: "token",
         total: 8192,
@@ -1121,7 +1166,7 @@ var xai_models_default = {
       creator: "xai",
       license: "proprietary",
       providers: ["xai"],
-      can: ["chat", "text-in", "text-out"],
+      can: ["chat", "txt-in", "txt-out"],
       context: {
         type: "token",
         total: 131072,
@@ -1145,9 +1190,10 @@ var cohere_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
-        "function-out"
+        "txt-in",
+        "txt-out",
+        "json-out",
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -1168,9 +1214,10 @@ var cohere_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
-        "function-out"
+        "txt-in",
+        "txt-out",
+        "json-out",
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -1194,10 +1241,11 @@ var cohere_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "reason",
-        "function-out"
+        "json-out",
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -1220,8 +1268,8 @@ var cohere_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -1241,8 +1289,8 @@ var cohere_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -1262,9 +1310,9 @@ var cohere_models_default = {
         "oracle"
       ],
       can: [
-        "text-in",
+        "txt-in",
         "img-in",
-        "vectors-out"
+        "vec-out"
       ],
       context: {
         type: "token",
@@ -1282,9 +1330,9 @@ var cohere_models_default = {
         "oracle"
       ],
       can: [
-        "text-in",
+        "txt-in",
         "img-in",
-        "vectors-out"
+        "vec-out"
       ],
       context: {
         type: "token",
@@ -1304,9 +1352,9 @@ var cohere_models_default = {
         "oracle"
       ],
       can: [
-        "text-in",
+        "txt-in",
         "img-in",
-        "vectors-out"
+        "vec-out"
       ],
       context: {
         type: "token",
@@ -1324,9 +1372,9 @@ var cohere_models_default = {
         "oracle"
       ],
       can: [
-        "text-in",
+        "txt-in",
         "img-in",
-        "vectors-out"
+        "vec-out"
       ],
       context: {
         type: "token",
@@ -1345,8 +1393,8 @@ var cohere_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -1364,8 +1412,8 @@ var cohere_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -1383,8 +1431,8 @@ var cohere_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -1432,15 +1480,15 @@ function validateModel(raw) {
   const validCapabilities = [
     "chat",
     "reason",
-    "text-in",
-    "text-out",
+    "txt-in",
+    "txt-out",
     "img-in",
     "img-out",
-    "sound-in",
-    "sound-out",
+    "audio-in",
+    "audio-out",
     "json-out",
-    "function-out",
-    "vectors-out"
+    "fn-out",
+    "vec-out"
   ];
   if (!model.can.every((c) => validCapabilities.includes(c))) {
     throw new Error(`Model has invalid capabilities: ${model.can.join(", ")}`);
@@ -2040,6 +2088,62 @@ var groq_provider_default = {
   }
 };
 
+// src/data/providers/azure-provider.json
+var azure_provider_default = {
+  id: "azure",
+  name: "Azure OpenAI",
+  websiteUrl: "https://azure.microsoft.com/en-us/products/ai-services/openai-service",
+  apiUrl: "https://learn.microsoft.com/en-us/azure/ai-services/openai/reference",
+  defaultModel: "gpt-4",
+  models: {
+    "gpt-4": {
+      type: "token",
+      input: 0,
+      output: 0
+    },
+    "gpt-4-32k": {
+      type: "token",
+      input: 0,
+      output: 0
+    },
+    "gpt-4-vision": {
+      type: "token",
+      input: 0,
+      output: 0
+    },
+    "gpt-4o": {
+      type: "token",
+      input: 0,
+      output: 0
+    },
+    "gpt-3.5-turbo": {
+      type: "token",
+      input: 0,
+      output: 0
+    },
+    "gpt-3.5-turbo-16k": {
+      type: "token",
+      input: 0,
+      output: 0
+    },
+    "text-embedding-ada-002": {
+      type: "token",
+      input: 0,
+      output: 0
+    },
+    whisper: {
+      type: "minute",
+      price: 0
+    },
+    "dall-e-3": {
+      type: "image",
+      price: 0,
+      size: "1024x1024",
+      unit: "per_image"
+    }
+  }
+};
+
 // src/builders/providers.ts
 function isTokenPrice(price) {
   return typeof price === "object" && price !== null && "type" in price && price.type === "token" && "input" in price && typeof price.input === "number" && "output" in price && typeof price.output === "number";
@@ -2099,7 +2203,8 @@ function buildAllProviders() {
     validateProvider(xai_provider_default),
     validateProvider(google_provider_default),
     validateProvider(deepseek_provider_default),
-    validateProvider(groq_provider_default)
+    validateProvider(groq_provider_default),
+    validateProvider(azure_provider_default)
   ];
 }
 function buildProvidersData() {