aimodels 0.2.1 → 0.2.3

package/dist/index.d.mts

@@ -13,11 +13,22 @@ interface ModelPrice {
 type Capability = "chat" | "reason" | "text-in" | "text-out" | "img-in" | "img-out" | "sound-in" | "sound-out" | "json-out" | "function-out" | "vectors-out";
 
 declare class ModelCollection extends Array<Model> {
-    constructor(models: Model[]);
+    /** Create a new ModelCollection from an array of models */
+    constructor(models?: Model[]);
+    /** Filter models by one or more capabilities (all must be present) */
     can(...capabilities: Capability[]): ModelCollection;
+    /** Filter models by one or more languages (all must be supported) */
     know(...languages: string[]): ModelCollection;
+    /** Override array filter to return ModelCollection */
     filter(predicate: (value: Model, index: number, array: Model[]) => boolean): ModelCollection;
+    /** Override array slice to return ModelCollection */
     slice(start?: number, end?: number): ModelCollection;
+    /** Find a model by its ID */
+    id(modelId: string): Model | undefined;
+    /** Get models available from a specific provider */
+    fromProvider(provider: string): ModelCollection;
+    /** Filter models by minimum context window size */
+    withMinContext(tokens: number): ModelCollection;
 }
 interface ModelContext {
     /** Maximum total tokens (input + output) */
@@ -41,26 +52,6 @@ interface Model {
     /** Context window information */
     context: ModelContext;
 }
-interface ModelsAPI {
-    /** All available models */
-    all: ModelCollection;
-    /** List of all creators */
-    creators: string[];
-    /** List of all providers */
-    providers: string[];
-    /** Get models from a specific creator */
-    fromCreator(creator: string): ModelCollection;
-    /** Get models from a specific provider */
-    fromProvider(provider: string): ModelCollection;
-    /** Find a specific model by ID */
-    find(id: string): Model | undefined;
-    /** Filter models by one or more capabilities (all must be present) */
-    can(...capabilities: string[]): ModelCollection;
-    /** Filter models by minimum context window */
-    withMinContext(tokens: number): ModelCollection;
-    /** Get pricing for a model from a specific provider */
-    getPrice(modelId: string, provider: string): ModelPrice | undefined;
-}
 
 var creators = {
 	openai: {
@@ -84,6 +75,12 @@ var creators$1 = {
 	creators: creators
 };
 
-declare const models: ModelsAPI;
+declare class ModelsCollection extends ModelCollection {
+    get creators(): string[];
+    get providers(): string[];
+    fromCreator(creator: string): ModelCollection;
+    getPrice(modelId: string, provider: string): ModelPrice | undefined;
+}
+declare const models: ModelsCollection;
 
-export { creators$1 as creators, models };
+export { type Capability, type Model, ModelCollection, type ModelContext, type ModelPrice, creators$1 as creators, models };

package/dist/index.d.ts

@@ -13,11 +13,22 @@ interface ModelPrice {
 type Capability = "chat" | "reason" | "text-in" | "text-out" | "img-in" | "img-out" | "sound-in" | "sound-out" | "json-out" | "function-out" | "vectors-out";
 
 declare class ModelCollection extends Array<Model> {
-    constructor(models: Model[]);
+    /** Create a new ModelCollection from an array of models */
+    constructor(models?: Model[]);
+    /** Filter models by one or more capabilities (all must be present) */
     can(...capabilities: Capability[]): ModelCollection;
+    /** Filter models by one or more languages (all must be supported) */
     know(...languages: string[]): ModelCollection;
+    /** Override array filter to return ModelCollection */
     filter(predicate: (value: Model, index: number, array: Model[]) => boolean): ModelCollection;
+    /** Override array slice to return ModelCollection */
     slice(start?: number, end?: number): ModelCollection;
+    /** Find a model by its ID */
+    id(modelId: string): Model | undefined;
+    /** Get models available from a specific provider */
+    fromProvider(provider: string): ModelCollection;
+    /** Filter models by minimum context window size */
+    withMinContext(tokens: number): ModelCollection;
 }
 interface ModelContext {
     /** Maximum total tokens (input + output) */
@@ -41,26 +52,6 @@ interface Model {
     /** Context window information */
     context: ModelContext;
 }
-interface ModelsAPI {
-    /** All available models */
-    all: ModelCollection;
-    /** List of all creators */
-    creators: string[];
-    /** List of all providers */
-    providers: string[];
-    /** Get models from a specific creator */
-    fromCreator(creator: string): ModelCollection;
-    /** Get models from a specific provider */
-    fromProvider(provider: string): ModelCollection;
-    /** Find a specific model by ID */
-    find(id: string): Model | undefined;
-    /** Filter models by one or more capabilities (all must be present) */
-    can(...capabilities: string[]): ModelCollection;
-    /** Filter models by minimum context window */
-    withMinContext(tokens: number): ModelCollection;
-    /** Get pricing for a model from a specific provider */
-    getPrice(modelId: string, provider: string): ModelPrice | undefined;
-}
 
 var creators = {
 	openai: {
@@ -84,6 +75,12 @@ var creators$1 = {
 	creators: creators
 };
 
-declare const models: ModelsAPI;
+declare class ModelsCollection extends ModelCollection {
+    get creators(): string[];
+    get providers(): string[];
+    fromCreator(creator: string): ModelCollection;
+    getPrice(modelId: string, provider: string): ModelPrice | undefined;
+}
+declare const models: ModelsCollection;
 
-export { creators$1 as creators, models };
+export { type Capability, type Model, ModelCollection, type ModelContext, type ModelPrice, creators$1 as creators, models };

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, {
+  ModelCollection: () => ModelCollection,
   creators: () => creators_default,
   models: () => models
 });
@@ -27,26 +28,43 @@ module.exports = __toCommonJS(index_exports);
 
 // src/types/models.ts
 var ModelCollection = class _ModelCollection extends Array {
-  constructor(models2) {
-    super(...models2);
+  /** Create a new ModelCollection from an array of models */
+  constructor(models2 = []) {
+    super();
+    if (models2.length > 0) {
+      this.push(...models2);
+    }
     Object.setPrototypeOf(this, _ModelCollection.prototype);
   }
+  /** Filter models by one or more capabilities (all must be present) */
   can(...capabilities) {
-    return new _ModelCollection(
-      this.filter((model) => capabilities.every((cap) => model.can.includes(cap)))
-    );
+    return this.filter((model) => capabilities.every((cap) => model.can.includes(cap)));
   }
+  /** Filter models by one or more languages (all must be supported) */
   know(...languages) {
-    return new _ModelCollection(
-      this.filter((model) => languages.every((lang) => model.languages?.includes(lang)))
-    );
+    return this.filter((model) => languages.every((lang) => model.languages?.includes(lang)));
   }
-  // Override array methods to return ModelCollection
+  /** Override array filter to return ModelCollection */
   filter(predicate) {
-    return new _ModelCollection(super.filter(predicate));
+    const filtered = Array.from(this).filter(predicate);
+    return new _ModelCollection(filtered);
   }
+  /** Override array slice to return ModelCollection */
   slice(start, end) {
-    return new _ModelCollection(super.slice(start, end));
+    const sliced = Array.from(this).slice(start, end);
+    return new _ModelCollection(sliced);
+  }
+  /** Find a model by its ID */
+  id(modelId) {
+    return this.find((model) => model.id === modelId);
+  }
+  /** Get models available from a specific provider */
+  fromProvider(provider) {
+    return this.filter((model) => model.providers.includes(provider));
+  }
+  /** Filter models by minimum context window size */
+  withMinContext(tokens) {
+    return this.filter((model) => model.context.total >= tokens);
   }
 };
 
@@ -54,12 +72,31 @@ var ModelCollection = class _ModelCollection extends Array {
 var openai_models_default = {
   models: [
     {
-      id: "whisper-1",
-      name: "Whisper-1",
-      license: "proprietary",
+      id: "whisper-large-v3",
+      name: "Whisper Large v3",
+      license: "mit",
       providers: [
         "openai",
-        "azure"
+        "azure",
+        "groq"
+      ],
+      can: [
+        "sound-in",
+        "text-out"
+      ],
+      context: {
+        total: null,
+        maxOutput: null
+      }
+    },
+    {
+      id: "whisper-large-v3-turbo",
+      name: "Whisper Large v3 Turbo",
+      license: "mit",
+      providers: [
+        "openai",
+        "azure",
+        "groq"
       ],
       can: [
         "sound-in",
@@ -308,11 +345,55 @@ var anthropic_models_default = {
 var meta_models_default = {
   models: [
     {
-      id: "llama2-70b-4096",
-      name: "Llama 2 70B",
-      license: "llama-2-community",
-      providers: ["groq", "ollama"],
-      can: ["chat", "json-out", "function-out"],
+      id: "llama3-70b-8192",
+      name: "Llama 3 70B",
+      license: "llama-3-community",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      context: {
+        total: 8192,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "llama3-8b-8192",
+      name: "Llama 3 8B",
+      license: "llama-3-community",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      context: {
+        total: 8192,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "llama-3.2-11b-vision-preview",
+      name: "Llama 3.2 11B Vision",
+      license: "llama-3-community",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out", "img-in"],
+      context: {
+        total: 4096,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "llama-3.2-90b-vision-preview",
+      name: "Llama 3.2 90B Vision",
+      license: "llama-3-community",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out", "img-in"],
+      context: {
+        total: 4096,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "llama-guard-3-8b",
+      name: "LlamaGuard 3 8B",
+      license: "llama-3-community",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out"],
       context: {
         total: 4096,
         maxOutput: 4096
@@ -406,13 +487,73 @@ var mistral_models_default = {
   ]
 };
 
+// src/data/models/google-models.json
+var google_models_default = {
+  models: [
+    {
+      id: "gemma2-9b-it",
+      name: "Gemma 2 9B Instruct",
+      license: "apache-2.0",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      context: {
+        total: 8192,
+        maxOutput: 4096
+      }
+    }
+  ]
+};
+
+// src/data/models/deepseek-models.json
+var deepseek_models_default = {
+  models: [
+    {
+      id: "deepseek-chat",
+      name: "DeepSeek V3",
+      license: "apache-2.0",
+      providers: ["deepseek"],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "json-out",
+        "function-out"
+      ],
+      context: {
+        total: 131072,
+        maxOutput: 131072
+      }
+    },
+    {
+      id: "deepseek-reasoner",
+      name: "DeepSeek R1",
+      license: "apache-2.0",
+      providers: ["deepseek"],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "json-out",
+        "function-out",
+        "reason"
+      ],
+      context: {
+        total: 131072,
+        maxOutput: 131072
+      }
+    }
+  ]
+};
+
 // src/builders/models.ts
 function buildAllModels() {
   return [
     ...openai_models_default.models,
     ...anthropic_models_default.models,
     ...meta_models_default.models,
-    ...mistral_models_default.models
+    ...mistral_models_default.models,
+    ...google_models_default.models,
+    ...deepseek_models_default.models
   ];
 }
 
@@ -639,51 +780,32 @@ var creators_default = {
 // src/index.ts
 var allModels = buildAllModels();
 var providersData = buildProvidersData();
-var models = {
-  all: new ModelCollection(allModels),
+var ModelsCollection = class extends ModelCollection {
   get creators() {
     return Object.keys(creators_default.creators);
-  },
+  }
   get providers() {
     return providersData.providers.map((p) => p.id);
-  },
+  }
   fromCreator(creator) {
     return new ModelCollection(
-      allModels.filter(
+      this.filter(
         (model) => model.license.startsWith(creator) || // For open source models
-        providersData.providers.find((p) => p.id === creator)?.models[model.id]
+        !!providersData.providers.find((p) => p.id === creator)?.models[model.id]
         // For proprietary models
       )
     );
-  },
-  fromProvider(provider) {
-    return new ModelCollection(
-      allModels.filter((model) => model.providers.includes(provider))
-    );
-  },
-  find(id) {
-    return allModels.find((model) => model.id === id);
-  },
-  can(...capabilities) {
-    return new ModelCollection(
-      allModels.filter(
-        (model) => capabilities.every((capability) => model.can.includes(capability))
-      )
-    );
-  },
-  withMinContext(tokens) {
-    return new ModelCollection(
-      allModels.filter((model) => model.context.total >= tokens)
-    );
-  },
+  }
   getPrice(modelId, provider) {
     const providerData = providersData.providers.find((p) => p.id === provider);
     const price = providerData?.models[modelId];
     return price?.type === "token" ? price : void 0;
   }
 };
+var models = new ModelsCollection(allModels);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  ModelCollection,
   creators,
   models
 });

package/dist/index.mjs

@@ -1,25 +1,42 @@
 // src/types/models.ts
 var ModelCollection = class _ModelCollection extends Array {
-  constructor(models2) {
-    super(...models2);
+  /** Create a new ModelCollection from an array of models */
+  constructor(models2 = []) {
+    super();
+    if (models2.length > 0) {
+      this.push(...models2);
+    }
     Object.setPrototypeOf(this, _ModelCollection.prototype);
   }
+  /** Filter models by one or more capabilities (all must be present) */
   can(...capabilities) {
-    return new _ModelCollection(
-      this.filter((model) => capabilities.every((cap) => model.can.includes(cap)))
-    );
+    return this.filter((model) => capabilities.every((cap) => model.can.includes(cap)));
   }
+  /** Filter models by one or more languages (all must be supported) */
   know(...languages) {
-    return new _ModelCollection(
-      this.filter((model) => languages.every((lang) => model.languages?.includes(lang)))
-    );
+    return this.filter((model) => languages.every((lang) => model.languages?.includes(lang)));
   }
-  // Override array methods to return ModelCollection
+  /** Override array filter to return ModelCollection */
   filter(predicate) {
-    return new _ModelCollection(super.filter(predicate));
+    const filtered = Array.from(this).filter(predicate);
+    return new _ModelCollection(filtered);
   }
+  /** Override array slice to return ModelCollection */
   slice(start, end) {
-    return new _ModelCollection(super.slice(start, end));
+    const sliced = Array.from(this).slice(start, end);
+    return new _ModelCollection(sliced);
+  }
+  /** Find a model by its ID */
+  id(modelId) {
+    return this.find((model) => model.id === modelId);
+  }
+  /** Get models available from a specific provider */
+  fromProvider(provider) {
+    return this.filter((model) => model.providers.includes(provider));
+  }
+  /** Filter models by minimum context window size */
+  withMinContext(tokens) {
+    return this.filter((model) => model.context.total >= tokens);
   }
 };
 
@@ -27,12 +44,31 @@ var ModelCollection = class _ModelCollection extends Array {
 var openai_models_default = {
   models: [
     {
-      id: "whisper-1",
-      name: "Whisper-1",
-      license: "proprietary",
+      id: "whisper-large-v3",
+      name: "Whisper Large v3",
+      license: "mit",
       providers: [
         "openai",
-        "azure"
+        "azure",
+        "groq"
+      ],
+      can: [
+        "sound-in",
+        "text-out"
+      ],
+      context: {
+        total: null,
+        maxOutput: null
+      }
+    },
+    {
+      id: "whisper-large-v3-turbo",
+      name: "Whisper Large v3 Turbo",
+      license: "mit",
+      providers: [
+        "openai",
+        "azure",
+        "groq"
       ],
       can: [
         "sound-in",
@@ -281,11 +317,55 @@ var anthropic_models_default = {
 var meta_models_default = {
   models: [
     {
-      id: "llama2-70b-4096",
-      name: "Llama 2 70B",
-      license: "llama-2-community",
-      providers: ["groq", "ollama"],
-      can: ["chat", "json-out", "function-out"],
+      id: "llama3-70b-8192",
+      name: "Llama 3 70B",
+      license: "llama-3-community",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      context: {
+        total: 8192,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "llama3-8b-8192",
+      name: "Llama 3 8B",
+      license: "llama-3-community",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      context: {
+        total: 8192,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "llama-3.2-11b-vision-preview",
+      name: "Llama 3.2 11B Vision",
+      license: "llama-3-community",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out", "img-in"],
+      context: {
+        total: 4096,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "llama-3.2-90b-vision-preview",
+      name: "Llama 3.2 90B Vision",
+      license: "llama-3-community",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out", "img-in"],
+      context: {
+        total: 4096,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "llama-guard-3-8b",
+      name: "LlamaGuard 3 8B",
+      license: "llama-3-community",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out"],
       context: {
         total: 4096,
         maxOutput: 4096
@@ -379,13 +459,73 @@ var mistral_models_default = {
   ]
 };
 
+// src/data/models/google-models.json
+var google_models_default = {
+  models: [
+    {
+      id: "gemma2-9b-it",
+      name: "Gemma 2 9B Instruct",
+      license: "apache-2.0",
+      providers: ["groq"],
+      can: ["chat", "text-in", "text-out", "json-out", "function-out"],
+      context: {
+        total: 8192,
+        maxOutput: 4096
+      }
+    }
+  ]
+};
+
+// src/data/models/deepseek-models.json
+var deepseek_models_default = {
+  models: [
+    {
+      id: "deepseek-chat",
+      name: "DeepSeek V3",
+      license: "apache-2.0",
+      providers: ["deepseek"],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "json-out",
+        "function-out"
+      ],
+      context: {
+        total: 131072,
+        maxOutput: 131072
+      }
+    },
+    {
+      id: "deepseek-reasoner",
+      name: "DeepSeek R1",
+      license: "apache-2.0",
+      providers: ["deepseek"],
+      can: [
+        "chat",
+        "text-in",
+        "text-out",
+        "json-out",
+        "function-out",
+        "reason"
+      ],
+      context: {
+        total: 131072,
+        maxOutput: 131072
+      }
+    }
+  ]
+};
+
 // src/builders/models.ts
 function buildAllModels() {
   return [
     ...openai_models_default.models,
     ...anthropic_models_default.models,
     ...meta_models_default.models,
-    ...mistral_models_default.models
+    ...mistral_models_default.models,
+    ...google_models_default.models,
+    ...deepseek_models_default.models
   ];
 }
 
@@ -612,50 +752,31 @@ var creators_default = {
 // src/index.ts
 var allModels = buildAllModels();
 var providersData = buildProvidersData();
-var models = {
-  all: new ModelCollection(allModels),
+var ModelsCollection = class extends ModelCollection {
   get creators() {
     return Object.keys(creators_default.creators);
-  },
+  }
   get providers() {
     return providersData.providers.map((p) => p.id);
-  },
+  }
   fromCreator(creator) {
     return new ModelCollection(
-      allModels.filter(
+      this.filter(
         (model) => model.license.startsWith(creator) || // For open source models
-        providersData.providers.find((p) => p.id === creator)?.models[model.id]
+        !!providersData.providers.find((p) => p.id === creator)?.models[model.id]
         // For proprietary models
       )
     );
-  },
-  fromProvider(provider) {
-    return new ModelCollection(
-      allModels.filter((model) => model.providers.includes(provider))
-    );
-  },
-  find(id) {
-    return allModels.find((model) => model.id === id);
-  },
-  can(...capabilities) {
-    return new ModelCollection(
-      allModels.filter(
-        (model) => capabilities.every((capability) => model.can.includes(capability))
-      )
-    );
-  },
-  withMinContext(tokens) {
-    return new ModelCollection(
-      allModels.filter((model) => model.context.total >= tokens)
-    );
-  },
+  }
   getPrice(modelId, provider) {
     const providerData = providersData.providers.find((p) => p.id === provider);
     const price = providerData?.models[modelId];
     return price?.type === "token" ? price : void 0;
   }
 };
+var models = new ModelsCollection(allModels);
 export {
+  ModelCollection,
   creators_default as creators,
   models
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aimodels",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "A collection of AI model specifications across different providers",
   "main": "dist/index.js",
   "module": "dist/index.mjs",