aimodels 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Dmitry Kury
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,78 @@
+# AIModels
+
+A collection of AI model specifications across different providers. This package provides normalized data about AI models, including their capabilities, context windows, and pricing information.
+
+## Installation
+
+```bash
+npm install aimodels
+```
+
+## Usage
+
+```typescript
+import { models } from 'aimodels';
+
+// Get all available models
+console.log(models.all);
+
+// Get list of all providers
+console.log(models.providers); // ['openai', 'anthropic', 'meta', ...]
+
+// Get models from a specific provider
+const openAiModels = models.from('openai');
+
+// Find a specific model
+const model = models.find('deepseek-r1');
+console.log(model.contextWindow);
+console.log(model.providers); 
+console.log(model.pricing.input);
+
+// Filter models by capability
+const chatModels = models.withCapability('chat');
+
+// Filter by context window
+const largeContextModels = models.withMinContext(32768);
+
+// Filter by maximum price
+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
+- Normalized data structure for easy comparison
+- Regular updates with new models
+- TypeScript support
+- 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
+
+MIT

package/dist/index.d.mts

@@ -0,0 +1,36 @@
+interface ModelPricing {
+    input: number;
+    output: number;
+}
+interface ModelSpec {
+    id: string;
+    name: string;
+    provider: string;
+    contextWindow: number;
+    capabilities: string[];
+    pricing: ModelPricing;
+    released: string;
+    license?: string;
+    trainingData?: string[];
+    parameters?: number;
+}
+interface ModelsCollection {
+    /** All available models */
+    all: ModelSpec[];
+    /** List of all unique providers */
+    providers: string[];
+    /** Get models from a specific provider */
+    from(provider: string): ModelSpec[];
+    /** Find a specific model by ID */
+    find(id: string): ModelSpec | undefined;
+    /** Filter models by capability */
+    withCapability(capability: string): ModelSpec[];
+    /** Filter models by minimum context window */
+    withMinContext(tokens: number): ModelSpec[];
+    /** Filter models by maximum price (per 1K tokens) */
+    withMaxPrice(price: number): ModelSpec[];
+}
+
+declare const models: ModelsCollection;
+
+export { type ModelPricing, type ModelSpec, type ModelsCollection, models };

package/dist/index.d.ts

@@ -0,0 +1,36 @@
+interface ModelPricing {
+    input: number;
+    output: number;
+}
+interface ModelSpec {
+    id: string;
+    name: string;
+    provider: string;
+    contextWindow: number;
+    capabilities: string[];
+    pricing: ModelPricing;
+    released: string;
+    license?: string;
+    trainingData?: string[];
+    parameters?: number;
+}
+interface ModelsCollection {
+    /** All available models */
+    all: ModelSpec[];
+    /** List of all unique providers */
+    providers: string[];
+    /** Get models from a specific provider */
+    from(provider: string): ModelSpec[];
+    /** Find a specific model by ID */
+    find(id: string): ModelSpec | undefined;
+    /** Filter models by capability */
+    withCapability(capability: string): ModelSpec[];
+    /** Filter models by minimum context window */
+    withMinContext(tokens: number): ModelSpec[];
+    /** Filter models by maximum price (per 1K tokens) */
+    withMaxPrice(price: number): ModelSpec[];
+}
+
+declare const models: ModelsCollection;
+
+export { type ModelPricing, type ModelSpec, type ModelsCollection, models };

package/dist/index.js

@@ -0,0 +1,110 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  models: () => models
+});
+module.exports = __toCommonJS(index_exports);
+
+// src/data/models.json
+var models_default = [
+  {
+    id: "gpt-4",
+    name: "GPT-4",
+    provider: "openai",
+    contextWindow: 8192,
+    capabilities: ["chat", "completion"],
+    pricing: {
+      input: 0.03,
+      output: 0.06
+    },
+    released: "2023-03-14",
+    parameters: 11e11
+  },
+  {
+    id: "gpt-3.5-turbo",
+    name: "GPT-3.5 Turbo",
+    provider: "openai",
+    contextWindow: 4096,
+    capabilities: ["chat", "completion"],
+    pricing: {
+      input: 15e-4,
+      output: 2e-3
+    },
+    released: "2022-11-30"
+  },
+  {
+    id: "claude-2.1",
+    name: "Claude 2.1",
+    provider: "anthropic",
+    contextWindow: 2e5,
+    capabilities: ["chat", "completion"],
+    pricing: {
+      input: 8e-3,
+      output: 0.024
+    },
+    released: "2023-11-21"
+  },
+  {
+    id: "llama-2-70b",
+    name: "Llama 2 70B",
+    provider: "meta",
+    contextWindow: 4096,
+    capabilities: ["chat", "completion"],
+    pricing: {
+      input: 0,
+      output: 0
+    },
+    released: "2023-07-18",
+    license: "Llama 2 Community License",
+    parameters: 7e10
+  }
+];
+
+// src/index.ts
+var aimodels = models_default;
+var models = {
+  all: aimodels,
+  get providers() {
+    return [...new Set(aimodels.map((model) => model.provider))];
+  },
+  from(provider) {
+    return aimodels.filter((model) => model.provider === provider);
+  },
+  find(id) {
+    return aimodels.find((model) => model.id === id);
+  },
+  withCapability(capability) {
+    return aimodels.filter((model) => model.capabilities.includes(capability));
+  },
+  withMinContext(tokens) {
+    return aimodels.filter((model) => model.contextWindow >= tokens);
+  },
+  withMaxPrice(price) {
+    return aimodels.filter(
+      (model) => model.pricing.input <= price && model.pricing.output <= price
+    );
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  models
+});

package/dist/index.mjs

@@ -0,0 +1,83 @@
+// src/data/models.json
+var models_default = [
+  {
+    id: "gpt-4",
+    name: "GPT-4",
+    provider: "openai",
+    contextWindow: 8192,
+    capabilities: ["chat", "completion"],
+    pricing: {
+      input: 0.03,
+      output: 0.06
+    },
+    released: "2023-03-14",
+    parameters: 11e11
+  },
+  {
+    id: "gpt-3.5-turbo",
+    name: "GPT-3.5 Turbo",
+    provider: "openai",
+    contextWindow: 4096,
+    capabilities: ["chat", "completion"],
+    pricing: {
+      input: 15e-4,
+      output: 2e-3
+    },
+    released: "2022-11-30"
+  },
+  {
+    id: "claude-2.1",
+    name: "Claude 2.1",
+    provider: "anthropic",
+    contextWindow: 2e5,
+    capabilities: ["chat", "completion"],
+    pricing: {
+      input: 8e-3,
+      output: 0.024
+    },
+    released: "2023-11-21"
+  },
+  {
+    id: "llama-2-70b",
+    name: "Llama 2 70B",
+    provider: "meta",
+    contextWindow: 4096,
+    capabilities: ["chat", "completion"],
+    pricing: {
+      input: 0,
+      output: 0
+    },
+    released: "2023-07-18",
+    license: "Llama 2 Community License",
+    parameters: 7e10
+  }
+];
+
+// src/index.ts
+var aimodels = models_default;
+var models = {
+  all: aimodels,
+  get providers() {
+    return [...new Set(aimodels.map((model) => model.provider))];
+  },
+  from(provider) {
+    return aimodels.filter((model) => model.provider === provider);
+  },
+  find(id) {
+    return aimodels.find((model) => model.id === id);
+  },
+  withCapability(capability) {
+    return aimodels.filter((model) => model.capabilities.includes(capability));
+  },
+  withMinContext(tokens) {
+    return aimodels.filter((model) => model.contextWindow >= tokens);
+  },
+  withMaxPrice(price) {
+    return aimodels.filter(
+      (model) => model.pricing.input <= price && model.pricing.output <= price
+    );
+  }
+};
+export {
+  models
+};

package/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "aimodels",
+  "version": "0.1.0",
+  "description": "A collection of AI model specifications across different providers",
+  "main": "dist/index.js",
+  "module": "dist/index.mjs",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsup src/index.ts --format cjs,esm --dts",
+    "test": "jest",
+    "typecheck": "tsc --noEmit",
+    "lint": "eslint src --ext .ts",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "ai",
+    "models",
+    "llm",
+    "machine-learning",
+    "specifications",
+    "openai",
+    "anthropic",
+    "open-source"
+  ],
+  "author": "Dmitry Kury (d@dkury.com)",
+  "license": "MIT",
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "@typescript-eslint/eslint-plugin": "^6.0.0",
+    "@typescript-eslint/parser": "^6.0.0",
+    "eslint": "^8.0.0",
+    "jest": "^29.0.0",
+    "ts-jest": "^29.0.0",
+    "tsup": "^8.0.0",
+    "typescript": "^5.0.0"
+  }
+}