aimodels 0.2.4 → 0.3.0

package/dist/index.d.mts

@@ -30,17 +30,28 @@ declare class ModelCollection extends Array<Model> {
     /** Filter models by minimum context window size */
     withMinContext(tokens: number): ModelCollection;
 }
-interface ModelContext {
+interface TokenModelContext {
     /** Maximum total tokens (input + output) */
-    total: number;
+    total: number | null;
     /** Maximum output tokens */
+    maxOutput: number | null;
+}
+interface ImageModelContext {
+    /** Maximum outputs per request */
     maxOutput: number;
+    /** Available image sizes */
+    sizes: string[];
+    /** Available quality settings */
+    qualities: string[];
 }
+type ModelContext = TokenModelContext | ImageModelContext;
 interface Model {
     /** Unique identifier */
     id: string;
     /** Display name */
     name: string;
+    /** Creator of the model */
+    creator: string;
     /** License type (e.g., "proprietary", "apache-2.0", "llama-2-community") */
     license: string;
     /** List of providers that can serve this model */

package/dist/index.d.ts

@@ -30,17 +30,28 @@ declare class ModelCollection extends Array<Model> {
     /** Filter models by minimum context window size */
     withMinContext(tokens: number): ModelCollection;
 }
-interface ModelContext {
+interface TokenModelContext {
     /** Maximum total tokens (input + output) */
-    total: number;
+    total: number | null;
     /** Maximum output tokens */
+    maxOutput: number | null;
+}
+interface ImageModelContext {
+    /** Maximum outputs per request */
     maxOutput: number;
+    /** Available image sizes */
+    sizes: string[];
+    /** Available quality settings */
+    qualities: string[];
 }
+type ModelContext = TokenModelContext | ImageModelContext;
 interface Model {
     /** Unique identifier */
     id: string;
     /** Display name */
     name: string;
+    /** Creator of the model */
+    creator: string;
     /** License type (e.g., "proprietary", "apache-2.0", "llama-2-community") */
     license: string;
     /** List of providers that can serve this model */

package/dist/index.js

@@ -64,12 +64,16 @@ var ModelCollection = class _ModelCollection extends Array {
   }
   /** Filter models by minimum context window size */
   withMinContext(tokens) {
-    return this.filter((model) => model.context.total >= tokens);
+    return this.filter((model) => {
+      const context = model.context;
+      return "total" in context && context.total !== null && context.total >= tokens;
+    });
   }
 };
 
 // src/data/models/openai-models.json
 var openai_models_default = {
+  creator: "openai",
   models: [
     {
       id: "whisper-large-v3",
@@ -143,7 +147,7 @@ var openai_models_default = {
     },
     {
       id: "gpt-4o",
-      name: "GPT-4O",
+      name: "GPT-4o",
       license: "proprietary",
       providers: [
         "openai",
@@ -304,13 +308,25 @@ var openai_models_default = {
 
 // src/data/models/anthropic-models.json
 var anthropic_models_default = {
+  creator: "anthropic",
   models: [
     {
       id: "claude-3-opus",
       name: "Claude 3 Opus",
       license: "proprietary",
-      providers: ["anthropic", "aws", "google"],
-      can: ["chat", "img-in", "json-out", "function-out"],
+      providers: [
+        "anthropic",
+        "aws",
+        "google"
+      ],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "img-in",
+        "json-out",
+        "function-out"
+      ],
       context: {
         total: 2e5,
         maxOutput: 4096
@@ -320,8 +336,19 @@ var anthropic_models_default = {
       id: "claude-3-sonnet",
       name: "Claude 3 Sonnet",
       license: "proprietary",
-      providers: ["anthropic", "aws", "google"],
-      can: ["chat", "img-in", "json-out", "function-out"],
+      providers: [
+        "anthropic",
+        "aws",
+        "google"
+      ],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "img-in",
+        "json-out",
+        "function-out"
+      ],
       context: {
         total: 2e5,
         maxOutput: 4096
@@ -331,8 +358,19 @@ var anthropic_models_default = {
       id: "claude-3-haiku",
       name: "Claude 3 Haiku",
       license: "proprietary",
-      providers: ["anthropic", "aws", "google"],
-      can: ["chat", "img-in", "json-out", "function-out"],
+      providers: [
+        "anthropic",
+        "aws",
+        "google"
+      ],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "img-in",
+        "json-out",
+        "function-out"
+      ],
       context: {
         total: 2e5,
         maxOutput: 4096
@@ -343,6 +381,7 @@ var anthropic_models_default = {
 
 // src/data/models/meta-models.json
 var meta_models_default = {
+  creator: "meta",
   models: [
     {
       id: "llama3-70b-8192",
@@ -404,6 +443,7 @@ var meta_models_default = {
 
 // src/data/models/mistral-models.json
 var mistral_models_default = {
+  creator: "mistral",
   models: [
     {
       id: "mistral-large-2402",
@@ -489,12 +529,93 @@ var mistral_models_default = {
 
 // src/data/models/google-models.json
 var google_models_default = {
+  creator: "google",
   models: [
     {
-      id: "gemma2-9b-it",
-      name: "Gemma 2 9B Instruct",
+      id: "gemini-2.0-flash",
+      name: "Gemini 2.0 Flash",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in", "reason"],
+      context: {
+        total: 1048576,
+        maxOutput: 8192
+      },
+      aliases: ["gemini-2.0-flash-001"]
+    },
+    {
+      id: "gemini-2.0-flash-lite",
+      name: "Gemini 2.0 Flash Lite",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
+      context: {
+        total: 1048576,
+        maxOutput: 8192
+      },
+      aliases: ["gemini-2.0-flash-lite-preview-02-05"]
+    },
+    {
+      id: "gemini-1.5-flash",
+      name: "Gemini 1.5 Flash",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
+      context: {
+        total: 2097152,
+        maxOutput: 8192
+      }
+    },
+    {
+      id: "gemini-1.5-flash-8b",
+      name: "Gemini 1.5 Flash 8B",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
+      context: {
+        total: 2097152,
+        maxOutput: 8192
+      }
+    },
+    {
+      id: "gemini-1.5-pro",
+      name: "Gemini 1.5 Pro",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in", "reason"],
+      context: {
+        total: 2097152,
+        maxOutput: 8192
+      },
+      aliases: ["gemini-1.5-pro-latest", "gemini-1.5-pro-001", "gemini-1.5-pro-002"]
+    },
+    {
+      id: "text-embedding-004",
+      name: "Text Embedding 004",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["text-in", "vectors-out"],
+      context: {
+        total: 2048,
+        maxOutput: 768
+      }
+    },
+    {
+      id: "gemma-7b-it",
+      name: "Gemma 7B Instruct",
       license: "apache-2.0",
-      providers: ["groq"],
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      context: {
+        total: 8192,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "gemma-2b-it",
+      name: "Gemma 2B Instruct",
+      license: "apache-2.0",
+      providers: ["google"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out"],
       context: {
         total: 8192,
@@ -506,6 +627,7 @@ var google_models_default = {
 
 // src/data/models/deepseek-models.json
 var deepseek_models_default = {
+  creator: "deepseek",
   models: [
     {
       id: "deepseek-chat",
@@ -545,16 +667,144 @@ var deepseek_models_default = {
   ]
 };
 
+// src/data/models/xai-models.json
+var xai_models_default = {
+  creator: "xai",
+  models: [
+    {
+      id: "grok-2-vision-1212",
+      name: "Grok 2 Vision",
+      creator: "xai",
+      license: "proprietary",
+      providers: ["xai"],
+      can: ["chat", "text-in", "text-out", "img-in"],
+      context: {
+        total: 32768,
+        maxOutput: 4096
+      },
+      aliases: ["grok-2-vision", "grok-2-vision-latest"]
+    },
+    {
+      id: "grok-2-1212",
+      name: "Grok 2",
+      creator: "xai",
+      license: "proprietary",
+      providers: ["xai"],
+      can: ["chat", "text-in", "text-out"],
+      context: {
+        total: 131072,
+        maxOutput: 4096
+      },
+      aliases: ["grok-2", "grok-2-latest"]
+    },
+    {
+      id: "grok-vision-beta",
+      name: "Grok Vision Beta",
+      creator: "xai",
+      license: "proprietary",
+      providers: ["xai"],
+      can: ["chat", "text-in", "text-out", "img-in"],
+      context: {
+        total: 8192,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "grok-beta",
+      name: "Grok Beta",
+      creator: "xai",
+      license: "proprietary",
+      providers: ["xai"],
+      can: ["chat", "text-in", "text-out"],
+      context: {
+        total: 131072,
+        maxOutput: 4096
+      }
+    }
+  ]
+};
+
 // src/builders/models.ts
+function validateModel(raw) {
+  if (typeof raw !== "object" || raw === null) {
+    throw new Error("Model data must be an object");
+  }
+  const model = raw;
+  const stringFields = ["id", "name", "creator", "license"];
+  for (const field of stringFields) {
+    if (typeof model[field] !== "string") {
+      throw new Error(`Model ${field} must be a string`);
+    }
+  }
+  if (!Array.isArray(model.providers)) {
+    throw new Error("Model providers must be an array");
+  }
+  if (!model.providers.every((p) => typeof p === "string")) {
+    throw new Error("Model providers must be strings");
+  }
+  if (!Array.isArray(model.can)) {
+    throw new Error("Model capabilities must be an array");
+  }
+  const validCapabilities = [
+    "chat",
+    "reason",
+    "text-in",
+    "text-out",
+    "img-in",
+    "img-out",
+    "sound-in",
+    "sound-out",
+    "json-out",
+    "function-out",
+    "vectors-out"
+  ];
+  if (!model.can.every((c) => validCapabilities.includes(c))) {
+    throw new Error(`Model has invalid capabilities: ${model.can.join(", ")}`);
+  }
+  if (typeof model.context !== "object" || model.context === null) {
+    throw new Error("Model context must be an object");
+  }
+  const context = model.context;
+  if (Array.isArray(context.sizes) && Array.isArray(context.qualities)) {
+    if (typeof context.maxOutput !== "number") {
+      throw new Error("Image model context.maxOutput 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");
+    }
+    if (typeof context.maxOutput !== "number" && context.maxOutput !== null) {
+      throw new Error("Token model context.maxOutput must be a number or null");
+    }
+  }
+  return {
+    id: model.id,
+    name: model.name,
+    creator: model.creator,
+    license: model.license,
+    providers: model.providers,
+    can: model.can,
+    context: model.context,
+    ...model.languages ? { languages: model.languages } : {}
+  };
+}
 function buildAllModels() {
-  return [
-    ...openai_models_default.models,
-    ...anthropic_models_default.models,
-    ...meta_models_default.models,
-    ...mistral_models_default.models,
-    ...google_models_default.models,
-    ...deepseek_models_default.models
+  const allModelData = [
+    openai_models_default,
+    anthropic_models_default,
+    meta_models_default,
+    mistral_models_default,
+    google_models_default,
+    deepseek_models_default,
+    xai_models_default
   ];
+  const allModels2 = allModelData.flatMap(
+    (data) => data.models.map((model) => ({
+      ...model,
+      creator: data.creator
+    }))
+  );
+  return allModels2.map((model) => validateModel(model));
 }
 
 // src/data/providers/openai-provider.json

package/dist/index.mjs

@@ -36,12 +36,16 @@ var ModelCollection = class _ModelCollection extends Array {
   }
   /** Filter models by minimum context window size */
   withMinContext(tokens) {
-    return this.filter((model) => model.context.total >= tokens);
+    return this.filter((model) => {
+      const context = model.context;
+      return "total" in context && context.total !== null && context.total >= tokens;
+    });
   }
 };
 
 // src/data/models/openai-models.json
 var openai_models_default = {
+  creator: "openai",
   models: [
     {
       id: "whisper-large-v3",
@@ -115,7 +119,7 @@ var openai_models_default = {
     },
     {
       id: "gpt-4o",
-      name: "GPT-4O",
+      name: "GPT-4o",
       license: "proprietary",
       providers: [
         "openai",
@@ -276,13 +280,25 @@ var openai_models_default = {
 
 // src/data/models/anthropic-models.json
 var anthropic_models_default = {
+  creator: "anthropic",
   models: [
     {
       id: "claude-3-opus",
       name: "Claude 3 Opus",
       license: "proprietary",
-      providers: ["anthropic", "aws", "google"],
-      can: ["chat", "img-in", "json-out", "function-out"],
+      providers: [
+        "anthropic",
+        "aws",
+        "google"
+      ],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "img-in",
+        "json-out",
+        "function-out"
+      ],
       context: {
         total: 2e5,
         maxOutput: 4096
@@ -292,8 +308,19 @@ var anthropic_models_default = {
       id: "claude-3-sonnet",
       name: "Claude 3 Sonnet",
       license: "proprietary",
-      providers: ["anthropic", "aws", "google"],
-      can: ["chat", "img-in", "json-out", "function-out"],
+      providers: [
+        "anthropic",
+        "aws",
+        "google"
+      ],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "img-in",
+        "json-out",
+        "function-out"
+      ],
       context: {
         total: 2e5,
         maxOutput: 4096
@@ -303,8 +330,19 @@ var anthropic_models_default = {
       id: "claude-3-haiku",
       name: "Claude 3 Haiku",
       license: "proprietary",
-      providers: ["anthropic", "aws", "google"],
-      can: ["chat", "img-in", "json-out", "function-out"],
+      providers: [
+        "anthropic",
+        "aws",
+        "google"
+      ],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "img-in",
+        "json-out",
+        "function-out"
+      ],
       context: {
         total: 2e5,
         maxOutput: 4096
@@ -315,6 +353,7 @@ var anthropic_models_default = {
 
 // src/data/models/meta-models.json
 var meta_models_default = {
+  creator: "meta",
   models: [
     {
       id: "llama3-70b-8192",
@@ -376,6 +415,7 @@ var meta_models_default = {
 
 // src/data/models/mistral-models.json
 var mistral_models_default = {
+  creator: "mistral",
   models: [
     {
       id: "mistral-large-2402",
@@ -461,12 +501,93 @@ var mistral_models_default = {
 
 // src/data/models/google-models.json
 var google_models_default = {
+  creator: "google",
   models: [
     {
-      id: "gemma2-9b-it",
-      name: "Gemma 2 9B Instruct",
+      id: "gemini-2.0-flash",
+      name: "Gemini 2.0 Flash",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in", "reason"],
+      context: {
+        total: 1048576,
+        maxOutput: 8192
+      },
+      aliases: ["gemini-2.0-flash-001"]
+    },
+    {
+      id: "gemini-2.0-flash-lite",
+      name: "Gemini 2.0 Flash Lite",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
+      context: {
+        total: 1048576,
+        maxOutput: 8192
+      },
+      aliases: ["gemini-2.0-flash-lite-preview-02-05"]
+    },
+    {
+      id: "gemini-1.5-flash",
+      name: "Gemini 1.5 Flash",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
+      context: {
+        total: 2097152,
+        maxOutput: 8192
+      }
+    },
+    {
+      id: "gemini-1.5-flash-8b",
+      name: "Gemini 1.5 Flash 8B",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in"],
+      context: {
+        total: 2097152,
+        maxOutput: 8192
+      }
+    },
+    {
+      id: "gemini-1.5-pro",
+      name: "Gemini 1.5 Pro",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out", "img-in", "sound-in", "reason"],
+      context: {
+        total: 2097152,
+        maxOutput: 8192
+      },
+      aliases: ["gemini-1.5-pro-latest", "gemini-1.5-pro-001", "gemini-1.5-pro-002"]
+    },
+    {
+      id: "text-embedding-004",
+      name: "Text Embedding 004",
+      license: "proprietary",
+      providers: ["google"],
+      can: ["text-in", "vectors-out"],
+      context: {
+        total: 2048,
+        maxOutput: 768
+      }
+    },
+    {
+      id: "gemma-7b-it",
+      name: "Gemma 7B Instruct",
       license: "apache-2.0",
-      providers: ["groq"],
+      providers: ["google"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      context: {
+        total: 8192,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "gemma-2b-it",
+      name: "Gemma 2B Instruct",
+      license: "apache-2.0",
+      providers: ["google"],
       can: ["chat", "text-in", "text-out", "json-out", "function-out"],
       context: {
         total: 8192,
@@ -478,6 +599,7 @@ var google_models_default = {
 
 // src/data/models/deepseek-models.json
 var deepseek_models_default = {
+  creator: "deepseek",
   models: [
     {
       id: "deepseek-chat",
@@ -517,16 +639,144 @@ var deepseek_models_default = {
   ]
 };
 
+// src/data/models/xai-models.json
+var xai_models_default = {
+  creator: "xai",
+  models: [
+    {
+      id: "grok-2-vision-1212",
+      name: "Grok 2 Vision",
+      creator: "xai",
+      license: "proprietary",
+      providers: ["xai"],
+      can: ["chat", "text-in", "text-out", "img-in"],
+      context: {
+        total: 32768,
+        maxOutput: 4096
+      },
+      aliases: ["grok-2-vision", "grok-2-vision-latest"]
+    },
+    {
+      id: "grok-2-1212",
+      name: "Grok 2",
+      creator: "xai",
+      license: "proprietary",
+      providers: ["xai"],
+      can: ["chat", "text-in", "text-out"],
+      context: {
+        total: 131072,
+        maxOutput: 4096
+      },
+      aliases: ["grok-2", "grok-2-latest"]
+    },
+    {
+      id: "grok-vision-beta",
+      name: "Grok Vision Beta",
+      creator: "xai",
+      license: "proprietary",
+      providers: ["xai"],
+      can: ["chat", "text-in", "text-out", "img-in"],
+      context: {
+        total: 8192,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "grok-beta",
+      name: "Grok Beta",
+      creator: "xai",
+      license: "proprietary",
+      providers: ["xai"],
+      can: ["chat", "text-in", "text-out"],
+      context: {
+        total: 131072,
+        maxOutput: 4096
+      }
+    }
+  ]
+};
+
 // src/builders/models.ts
+function validateModel(raw) {
+  if (typeof raw !== "object" || raw === null) {
+    throw new Error("Model data must be an object");
+  }
+  const model = raw;
+  const stringFields = ["id", "name", "creator", "license"];
+  for (const field of stringFields) {
+    if (typeof model[field] !== "string") {
+      throw new Error(`Model ${field} must be a string`);
+    }
+  }
+  if (!Array.isArray(model.providers)) {
+    throw new Error("Model providers must be an array");
+  }
+  if (!model.providers.every((p) => typeof p === "string")) {
+    throw new Error("Model providers must be strings");
+  }
+  if (!Array.isArray(model.can)) {
+    throw new Error("Model capabilities must be an array");
+  }
+  const validCapabilities = [
+    "chat",
+    "reason",
+    "text-in",
+    "text-out",
+    "img-in",
+    "img-out",
+    "sound-in",
+    "sound-out",
+    "json-out",
+    "function-out",
+    "vectors-out"
+  ];
+  if (!model.can.every((c) => validCapabilities.includes(c))) {
+    throw new Error(`Model has invalid capabilities: ${model.can.join(", ")}`);
+  }
+  if (typeof model.context !== "object" || model.context === null) {
+    throw new Error("Model context must be an object");
+  }
+  const context = model.context;
+  if (Array.isArray(context.sizes) && Array.isArray(context.qualities)) {
+    if (typeof context.maxOutput !== "number") {
+      throw new Error("Image model context.maxOutput 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");
+    }
+    if (typeof context.maxOutput !== "number" && context.maxOutput !== null) {
+      throw new Error("Token model context.maxOutput must be a number or null");
+    }
+  }
+  return {
+    id: model.id,
+    name: model.name,
+    creator: model.creator,
+    license: model.license,
+    providers: model.providers,
+    can: model.can,
+    context: model.context,
+    ...model.languages ? { languages: model.languages } : {}
+  };
+}
 function buildAllModels() {
-  return [
-    ...openai_models_default.models,
-    ...anthropic_models_default.models,
-    ...meta_models_default.models,
-    ...mistral_models_default.models,
-    ...google_models_default.models,
-    ...deepseek_models_default.models
+  const allModelData = [
+    openai_models_default,
+    anthropic_models_default,
+    meta_models_default,
+    mistral_models_default,
+    google_models_default,
+    deepseek_models_default,
+    xai_models_default
   ];
+  const allModels2 = allModelData.flatMap(
+    (data) => data.models.map((model) => ({
+      ...model,
+      creator: data.creator
+    }))
+  );
+  return allModels2.map((model) => validateModel(model));
 }
 
 // src/data/providers/openai-provider.json

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aimodels",
-  "version": "0.2.4",
+  "version": "0.3.0",
   "description": "A collection of AI model specifications across different providers",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -19,7 +19,6 @@
   "scripts": {
     "prebuild": "npm run typecheck && npm run lint && npm test",
     "build": "tsup src/index.ts --format cjs,esm --dts",
-    "gen-ai-rules": "node scripts/gen-ai-rules.js",
     "test": "deno test tests/",
     "test:watch": "deno test --watch tests/",
     "typecheck": "tsc --noEmit",
@@ -30,7 +29,11 @@
     "version": "git add -A",
     "postversion": "git push && git push --tags",
     "prepublishOnly": "npm run clean && npm run typecheck && npm test && npm run lint",
-    "postpublish": "npm run clean"
+    "postpublish": "npm run clean",
+    "rules": "airul generate",
+    "rules:comment": "# Generate AI rules from documentation",
+    "pregenerate": "[ -f .airul.json ] || airul init",
+    "postinstall": "echo \"\nRun 'npm run rules' to generate AI rules from your documentation\""
   },
   "keywords": [
     "ai",
@@ -61,6 +64,7 @@
     "@types/node": "^20.0.0",
     "@typescript-eslint/eslint-plugin": "^6.0.0",
     "@typescript-eslint/parser": "^6.0.0",
+    "airul": "^0.1.10",
     "eslint": "^8.0.0",
     "tsup": "^8.0.0",
     "typescript": "^5.0.0"