aimodels 0.3.3 → 0.3.5

package/dist/index.d.mts

@@ -1,4 +1,4 @@
-interface ModelPrice {
+interface TokenBasedPricePerMillionTokens {
     /** Price per million input tokens */
     input: number;
     /** Price per million output tokens */
@@ -6,6 +6,16 @@ interface ModelPrice {
     /** Price type */
     type: 'token';
 }
+interface ImagePrice {
+    /** Price per image */
+    price: number;
+    /** Image size */
+    size: string;
+    /** Price type */
+    type: 'image';
+    /** Price unit */
+    unit: 'per_image';
+}
 
 /**
  * Defines all possible model capabilities
@@ -27,6 +37,8 @@ declare class ModelCollection extends Array<Model> {
     id(modelId: string): Model | undefined;
     /** Get models available from a specific provider */
     fromProvider(provider: string): ModelCollection;
+    /** Get models available from a specific creator */
+    fromCreator(creator: string): ModelCollection;
     /** Filter models by minimum context window size */
     withMinContext(tokens: number): ModelCollection;
 }
@@ -72,6 +84,23 @@ interface Model {
     context: ModelContext;
 }
 
+interface Provider {
+    /** Provider identifier */
+    id: string;
+    /** Display name */
+    name: string;
+    /** Website URL */
+    websiteUrl: string;
+    /** API endpoint */
+    apiUrl: string;
+    /** Default model */
+    defaultModel?: string;
+    /** Whether this is a local provider */
+    isLocal?: number;
+    /** Model pricing */
+    models: Record<string, TokenBasedPricePerMillionTokens | ImagePrice>;
+}
+
 var creators = {
 	openai: {
 		name: "OpenAI",
@@ -94,12 +123,16 @@ var creators$1 = {
 	creators: creators
 };
 
-declare class ModelsCollection extends ModelCollection {
+declare class AIModels extends ModelCollection {
+    constructor(models?: Model[]);
     get creators(): string[];
     get providers(): string[];
-    fromCreator(creator: string): ModelCollection;
-    getPrice(modelId: string, provider: string): ModelPrice | undefined;
+    getPrice(modelId: string, provider: string): TokenBasedPricePerMillionTokens | undefined;
+    /** Get provider information by ID */
+    getProvider(providerId: string): Provider | undefined;
+    /** Get all providers that can serve a specific model */
+    getProvidersForModel(modelId: string): Provider[];
 }
-declare const models: ModelsCollection;
+declare const models: AIModels;
 
-export { type Capability, type Model, ModelCollection, type ModelContext, type ModelPrice, creators$1 as creators, models };
+export { AIModels, type Capability, type Model, ModelCollection, type ModelContext, type Provider, type TokenBasedPricePerMillionTokens, creators$1 as creators, models };

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-interface ModelPrice {
+interface TokenBasedPricePerMillionTokens {
     /** Price per million input tokens */
     input: number;
     /** Price per million output tokens */
@@ -6,6 +6,16 @@ interface ModelPrice {
     /** Price type */
     type: 'token';
 }
+interface ImagePrice {
+    /** Price per image */
+    price: number;
+    /** Image size */
+    size: string;
+    /** Price type */
+    type: 'image';
+    /** Price unit */
+    unit: 'per_image';
+}
 
 /**
  * Defines all possible model capabilities
@@ -27,6 +37,8 @@ declare class ModelCollection extends Array<Model> {
     id(modelId: string): Model | undefined;
     /** Get models available from a specific provider */
     fromProvider(provider: string): ModelCollection;
+    /** Get models available from a specific creator */
+    fromCreator(creator: string): ModelCollection;
     /** Filter models by minimum context window size */
     withMinContext(tokens: number): ModelCollection;
 }
@@ -72,6 +84,23 @@ interface Model {
     context: ModelContext;
 }
 
+interface Provider {
+    /** Provider identifier */
+    id: string;
+    /** Display name */
+    name: string;
+    /** Website URL */
+    websiteUrl: string;
+    /** API endpoint */
+    apiUrl: string;
+    /** Default model */
+    defaultModel?: string;
+    /** Whether this is a local provider */
+    isLocal?: number;
+    /** Model pricing */
+    models: Record<string, TokenBasedPricePerMillionTokens | ImagePrice>;
+}
+
 var creators = {
 	openai: {
 		name: "OpenAI",
@@ -94,12 +123,16 @@ var creators$1 = {
 	creators: creators
 };
 
-declare class ModelsCollection extends ModelCollection {
+declare class AIModels extends ModelCollection {
+    constructor(models?: Model[]);
     get creators(): string[];
     get providers(): string[];
-    fromCreator(creator: string): ModelCollection;
-    getPrice(modelId: string, provider: string): ModelPrice | undefined;
+    getPrice(modelId: string, provider: string): TokenBasedPricePerMillionTokens | undefined;
+    /** Get provider information by ID */
+    getProvider(providerId: string): Provider | undefined;
+    /** Get all providers that can serve a specific model */
+    getProvidersForModel(modelId: string): Provider[];
 }
-declare const models: ModelsCollection;
+declare const models: AIModels;
 
-export { type Capability, type Model, ModelCollection, type ModelContext, type ModelPrice, creators$1 as creators, models };
+export { AIModels, type Capability, type Model, ModelCollection, type ModelContext, type Provider, type TokenBasedPricePerMillionTokens, 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, {
+  AIModels: () => AIModels,
   ModelCollection: () => ModelCollection,
   creators: () => creators_default,
   models: () => models
@@ -64,6 +65,12 @@ var ModelCollection = class _ModelCollection extends Array {
   fromProvider(provider) {
     return this.filter((model) => model.providers.includes(provider));
   }
+  /** Get models available from a specific creator */
+  fromCreator(creator) {
+    return new _ModelCollection(
+      this.filter((model) => model.creator === creator)
+    );
+  }
   /** Filter models by minimum context window size */
   withMinContext(tokens) {
     return this.filter((model) => {
@@ -336,7 +343,8 @@ var anthropic_models_default = {
         total: 2e5,
         maxOutput: 4096,
         outputIsFixed: 1
-      }
+      },
+      aliases: ["claude-3-opus-latest"]
     },
     {
       id: "claude-3-5-sonnet-20241022",
@@ -359,7 +367,8 @@ var anthropic_models_default = {
         total: 2e5,
         maxOutput: 8192,
         outputIsFixed: 1
-      }
+      },
+      aliases: ["claude-3-sonnet-latest"]
     },
     {
       id: "claude-3-5-haiku-20241022",
@@ -1201,31 +1210,39 @@ var creators_default = {
 // src/index.ts
 var allModels = buildAllModels();
 var providersData = buildProvidersData();
-var ModelsCollection = class extends ModelCollection {
+var AIModels = class _AIModels extends ModelCollection {
+  constructor(models2 = []) {
+    super(models2);
+    Object.setPrototypeOf(this, _AIModels.prototype);
+  }
   get creators() {
     return Object.keys(creators_default.creators);
   }
   get providers() {
     return providersData.providers.map((p) => p.id);
   }
-  fromCreator(creator) {
-    return new ModelCollection(
-      this.filter(
-        (model) => model.license.startsWith(creator) || // For open source models
-        !!providersData.providers.find((p) => p.id === creator)?.models[model.id]
-        // For proprietary models
-      )
-    );
-  }
   getPrice(modelId, provider) {
     const providerData = providersData.providers.find((p) => p.id === provider);
     const price = providerData?.models[modelId];
     return price?.type === "token" ? price : void 0;
   }
+  /** Get provider information by ID */
+  getProvider(providerId) {
+    return providersData.providers.find((p) => p.id === providerId);
+  }
+  /** Get all providers that can serve a specific model */
+  getProvidersForModel(modelId) {
+    const model = this.id(modelId);
+    if (!model) return [];
+    return providersData.providers.filter(
+      (p) => model.providers.includes(p.id)
+    );
+  }
 };
-var models = new ModelsCollection(allModels);
+var models = new AIModels(allModels);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AIModels,
   ModelCollection,
   creators,
   models

package/dist/index.mjs

@@ -36,6 +36,12 @@ var ModelCollection = class _ModelCollection extends Array {
   fromProvider(provider) {
     return this.filter((model) => model.providers.includes(provider));
   }
+  /** Get models available from a specific creator */
+  fromCreator(creator) {
+    return new _ModelCollection(
+      this.filter((model) => model.creator === creator)
+    );
+  }
   /** Filter models by minimum context window size */
   withMinContext(tokens) {
     return this.filter((model) => {
@@ -308,7 +314,8 @@ var anthropic_models_default = {
         total: 2e5,
         maxOutput: 4096,
         outputIsFixed: 1
-      }
+      },
+      aliases: ["claude-3-opus-latest"]
     },
     {
       id: "claude-3-5-sonnet-20241022",
@@ -331,7 +338,8 @@ var anthropic_models_default = {
         total: 2e5,
         maxOutput: 8192,
         outputIsFixed: 1
-      }
+      },
+      aliases: ["claude-3-sonnet-latest"]
     },
     {
       id: "claude-3-5-haiku-20241022",
@@ -1173,30 +1181,38 @@ var creators_default = {
 // src/index.ts
 var allModels = buildAllModels();
 var providersData = buildProvidersData();
-var ModelsCollection = class extends ModelCollection {
+var AIModels = class _AIModels extends ModelCollection {
+  constructor(models2 = []) {
+    super(models2);
+    Object.setPrototypeOf(this, _AIModels.prototype);
+  }
   get creators() {
     return Object.keys(creators_default.creators);
   }
   get providers() {
     return providersData.providers.map((p) => p.id);
   }
-  fromCreator(creator) {
-    return new ModelCollection(
-      this.filter(
-        (model) => model.license.startsWith(creator) || // For open source models
-        !!providersData.providers.find((p) => p.id === creator)?.models[model.id]
-        // For proprietary models
-      )
-    );
-  }
   getPrice(modelId, provider) {
     const providerData = providersData.providers.find((p) => p.id === provider);
     const price = providerData?.models[modelId];
     return price?.type === "token" ? price : void 0;
   }
+  /** Get provider information by ID */
+  getProvider(providerId) {
+    return providersData.providers.find((p) => p.id === providerId);
+  }
+  /** Get all providers that can serve a specific model */
+  getProvidersForModel(modelId) {
+    const model = this.id(modelId);
+    if (!model) return [];
+    return providersData.providers.filter(
+      (p) => model.providers.includes(p.id)
+    );
+  }
 };
-var models = new ModelsCollection(allModels);
+var models = new AIModels(allModels);
 export {
+  AIModels,
   ModelCollection,
   creators_default as creators,
   models

package/package.json

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