aimodels 0.3.10 → 0.3.12

package/dist/index.js

@@ -41,6 +41,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)));
@@ -88,6 +130,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
@@ -104,10 +149,10 @@ var openai_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -137,10 +182,10 @@ var openai_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out"
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -169,8 +214,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -188,11 +233,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",
@@ -252,11 +297,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",
@@ -286,12 +331,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",
@@ -314,11 +359,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",
@@ -348,11 +393,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",
@@ -378,11 +423,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: {
@@ -405,10 +450,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: {
@@ -454,10 +499,10 @@ var openai_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
+        "txt-in",
+        "txt-out",
         "json-out",
-        "function-out",
+        "fn-out",
         "reason"
       ],
       context: {
@@ -482,8 +527,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "sound-in",
-        "text-out"
+        "audio-in",
+        "txt-out"
       ],
       context: {
         type: "audio-in",
@@ -500,8 +545,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "sound-out"
+        "txt-in",
+        "audio-out"
       ],
       context: {
         type: "audio-out",
@@ -548,7 +593,7 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
+        "txt-in",
         "img-out"
       ],
       context: {
@@ -573,8 +618,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "vectors-out"
+        "txt-in",
+        "vec-out"
       ],
       context: {
         type: "embedding",
@@ -594,8 +639,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "vectors-out"
+        "txt-in",
+        "vec-out"
       ],
       context: {
         type: "embedding",
@@ -615,8 +660,8 @@ var openai_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "vectors-out"
+        "txt-in",
+        "vec-out"
       ],
       context: {
         type: "embedding",
@@ -641,11 +686,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",
@@ -662,11 +707,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",
@@ -683,11 +728,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",
@@ -704,11 +749,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",
@@ -725,11 +770,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",
@@ -745,10 +790,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",
@@ -765,11 +810,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: {
@@ -800,7 +845,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,
@@ -812,7 +857,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,
@@ -824,7 +869,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,
@@ -836,7 +881,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,
@@ -848,7 +893,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,
@@ -869,10 +914,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",
@@ -887,10 +932,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",
@@ -905,8 +950,8 @@ var mistral_models_default = {
       providers: ["mistral"],
       can: [
         "chat",
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -921,8 +966,8 @@ var mistral_models_default = {
       providers: ["mistral", "groq"],
       can: [
         "chat",
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -937,8 +982,8 @@ var mistral_models_default = {
       providers: ["mistral", "groq"],
       can: [
         "chat",
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -958,7 +1003,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,
@@ -971,7 +1016,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,
@@ -984,7 +1029,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,
@@ -996,7 +1041,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,
@@ -1008,7 +1053,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,
@@ -1021,7 +1066,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,
@@ -1033,7 +1078,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,
@@ -1045,7 +1090,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,
@@ -1066,10 +1111,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",
@@ -1084,10 +1129,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: {
@@ -1109,7 +1154,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,
@@ -1123,7 +1168,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,
@@ -1137,7 +1182,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,
@@ -1150,7 +1195,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,
@@ -1174,9 +1219,10 @@ var cohere_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
-        "function-out"
+        "txt-in",
+        "txt-out",
+        "json-out",
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -1197,9 +1243,10 @@ var cohere_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out",
-        "function-out"
+        "txt-in",
+        "txt-out",
+        "json-out",
+        "fn-out"
       ],
       context: {
         type: "token",
@@ -1223,10 +1270,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",
@@ -1249,8 +1297,8 @@ var cohere_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -1270,8 +1318,8 @@ var cohere_models_default = {
       ],
       can: [
         "chat",
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -1291,9 +1339,9 @@ var cohere_models_default = {
         "oracle"
       ],
       can: [
-        "text-in",
+        "txt-in",
         "img-in",
-        "vectors-out"
+        "vec-out"
       ],
       context: {
         type: "token",
@@ -1311,9 +1359,9 @@ var cohere_models_default = {
         "oracle"
       ],
       can: [
-        "text-in",
+        "txt-in",
         "img-in",
-        "vectors-out"
+        "vec-out"
       ],
       context: {
         type: "token",
@@ -1333,9 +1381,9 @@ var cohere_models_default = {
         "oracle"
       ],
       can: [
-        "text-in",
+        "txt-in",
         "img-in",
-        "vectors-out"
+        "vec-out"
       ],
       context: {
         type: "token",
@@ -1353,9 +1401,9 @@ var cohere_models_default = {
         "oracle"
       ],
       can: [
-        "text-in",
+        "txt-in",
         "img-in",
-        "vectors-out"
+        "vec-out"
       ],
       context: {
         type: "token",
@@ -1374,8 +1422,8 @@ var cohere_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -1393,8 +1441,8 @@ var cohere_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -1412,8 +1460,8 @@ var cohere_models_default = {
         "azure"
       ],
       can: [
-        "text-in",
-        "text-out"
+        "txt-in",
+        "txt-out"
       ],
       context: {
         type: "token",
@@ -1461,15 +1509,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(", ")}`);
@@ -2069,6 +2117,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";
@@ -2128,7 +2232,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() {