aimodels 0.1.0 → 0.1.2

package/README.md

@@ -38,28 +38,6 @@ const largeContextModels = models.withMinContext(32768);
 const affordableModels = models.withMaxPrice(0.01); // Max $0.01 per 1K tokens
 ```
 
-## Data Structure
-
-Each model entry contains the following information:
-
-```typescript
-interface ModelSpec {
-  id: string;                 // Unique identifier for the model
-  name: string;              // Display name
-  provider: string;          // Provider (e.g., 'openai', 'anthropic', etc.)
-  contextWindow: number;     // Maximum context window size in tokens
-  capabilities: string[];    // Array of capabilities (e.g., ['chat', 'completion'])
-  pricing: {
-    input: number;          // Cost per 1K input tokens in USD
-    output: number;         // Cost per 1K output tokens in USD
-  };
-  released: string;         // Release date
-  license?: string;         // License information for open-source models
-  trainingData?: string[];  // Known training data sources
-  parameters?: number;      // Number of parameters (if known)
-}
-```
-
 ## Features
 
 - Comprehensive database of AI models
@@ -69,9 +47,6 @@ interface ModelSpec {
 - Zero dependencies
 - Universal JavaScript support (Node.js, browsers, Deno)
 
-## Contributing
-
-Contributions are welcome! Please check our contributing guidelines for details on how to submit new models or updates.
 
 ## License
 

package/dist/index.d.mts

@@ -1,36 +1,75 @@
-interface ModelPricing {
+interface ModelPrice {
+    /** Price per million input tokens */
     input: number;
+    /** Price per million output tokens */
     output: number;
+    /** Price type */
+    type: 'token';
 }
-interface ModelSpec {
+
+interface ModelContext {
+    /** Maximum total tokens (input + output) */
+    total: number;
+    /** Maximum output tokens */
+    maxOutput: number;
+}
+interface Model {
+    /** Unique identifier */
     id: string;
+    /** Display name */
     name: string;
-    provider: string;
-    contextWindow: number;
-    capabilities: string[];
-    pricing: ModelPricing;
-    released: string;
-    license?: string;
-    trainingData?: string[];
-    parameters?: number;
+    /** License type (e.g., "proprietary", "apache-2.0", "llama-2-community") */
+    license: string;
+    /** List of providers that can serve this model */
+    providers: string[];
+    /** Model capabilities */
+    can: string[];
+    /** Context window information */
+    context: ModelContext;
 }
-interface ModelsCollection {
+interface ModelCollection {
     /** All available models */
-    all: ModelSpec[];
-    /** List of all unique providers */
+    all: Model[];
+    /** List of all creators */
+    creators: string[];
+    /** List of all providers */
     providers: string[];
+    /** Get models from a specific creator */
+    fromCreator(creator: string): Model[];
     /** Get models from a specific provider */
-    from(provider: string): ModelSpec[];
+    fromProvider(provider: string): Model[];
     /** Find a specific model by ID */
-    find(id: string): ModelSpec | undefined;
-    /** Filter models by capability */
-    withCapability(capability: string): ModelSpec[];
+    find(id: string): Model | undefined;
+    /** Filter models by one or more capabilities (all must be present) */
+    can(...capabilities: string[]): Model[];
     /** Filter models by minimum context window */
-    withMinContext(tokens: number): ModelSpec[];
-    /** Filter models by maximum price (per 1K tokens) */
-    withMaxPrice(price: number): ModelSpec[];
+    withMinContext(tokens: number): Model[];
+    /** Get pricing for a model from a specific provider */
+    getPrice(modelId: string, provider: string): ModelPrice | undefined;
 }
 
-declare const models: ModelsCollection;
+var creators = {
+	openai: {
+		name: "OpenAI",
+		website: "https://openai.com"
+	},
+	anthropic: {
+		name: "Anthropic",
+		website: "https://anthropic.com"
+	},
+	meta: {
+		name: "Meta",
+		website: "https://ai.meta.com"
+	},
+	mistral: {
+		name: "Mistral AI",
+		website: "https://mistral.ai"
+	}
+};
+var creators$1 = {
+	creators: creators
+};
+
+declare const models: ModelCollection;
 
-export { type ModelPricing, type ModelSpec, type ModelsCollection, models };
+export { creators$1 as creators, models };

package/dist/index.d.ts

@@ -1,36 +1,75 @@
-interface ModelPricing {
+interface ModelPrice {
+    /** Price per million input tokens */
     input: number;
+    /** Price per million output tokens */
     output: number;
+    /** Price type */
+    type: 'token';
 }
-interface ModelSpec {
+
+interface ModelContext {
+    /** Maximum total tokens (input + output) */
+    total: number;
+    /** Maximum output tokens */
+    maxOutput: number;
+}
+interface Model {
+    /** Unique identifier */
     id: string;
+    /** Display name */
     name: string;
-    provider: string;
-    contextWindow: number;
-    capabilities: string[];
-    pricing: ModelPricing;
-    released: string;
-    license?: string;
-    trainingData?: string[];
-    parameters?: number;
+    /** License type (e.g., "proprietary", "apache-2.0", "llama-2-community") */
+    license: string;
+    /** List of providers that can serve this model */
+    providers: string[];
+    /** Model capabilities */
+    can: string[];
+    /** Context window information */
+    context: ModelContext;
 }
-interface ModelsCollection {
+interface ModelCollection {
     /** All available models */
-    all: ModelSpec[];
-    /** List of all unique providers */
+    all: Model[];
+    /** List of all creators */
+    creators: string[];
+    /** List of all providers */
     providers: string[];
+    /** Get models from a specific creator */
+    fromCreator(creator: string): Model[];
     /** Get models from a specific provider */
-    from(provider: string): ModelSpec[];
+    fromProvider(provider: string): Model[];
     /** Find a specific model by ID */
-    find(id: string): ModelSpec | undefined;
-    /** Filter models by capability */
-    withCapability(capability: string): ModelSpec[];
+    find(id: string): Model | undefined;
+    /** Filter models by one or more capabilities (all must be present) */
+    can(...capabilities: string[]): Model[];
     /** Filter models by minimum context window */
-    withMinContext(tokens: number): ModelSpec[];
-    /** Filter models by maximum price (per 1K tokens) */
-    withMaxPrice(price: number): ModelSpec[];
+    withMinContext(tokens: number): Model[];
+    /** Get pricing for a model from a specific provider */
+    getPrice(modelId: string, provider: string): ModelPrice | undefined;
 }
 
-declare const models: ModelsCollection;
+var creators = {
+	openai: {
+		name: "OpenAI",
+		website: "https://openai.com"
+	},
+	anthropic: {
+		name: "Anthropic",
+		website: "https://anthropic.com"
+	},
+	meta: {
+		name: "Meta",
+		website: "https://ai.meta.com"
+	},
+	mistral: {
+		name: "Mistral AI",
+		website: "https://mistral.ai"
+	}
+};
+var creators$1 = {
+	creators: creators
+};
+
+declare const models: ModelCollection;
 
-export { type ModelPricing, type ModelSpec, type ModelsCollection, models };
+export { creators$1 as creators, models };