aimodels 0.2.2 → 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);
   }
 };
 
@@ -762,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);
   }
 };
 
@@ -735,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.2",
+  "version": "0.2.3",
   "description": "A collection of AI model specifications across different providers",
   "main": "dist/index.js",
   "module": "dist/index.mjs",