aimodels 0.3.11 → 0.4.0

package/package.json

@@ -1,26 +1,30 @@
 {
   "name": "aimodels",
-  "version": "0.3.11",
+  "version": "0.4.0",
   "description": "A collection of AI model specifications across different providers",
+  "type": "module",
   "main": "dist/index.js",
-  "module": "dist/index.mjs",
-  "types": "dist/index.d.ts",
+  "module": "dist/index.js",
+  "types": "dist/aimodels.d.ts",
   "exports": {
     ".": {
-      "types": "./dist/index.d.ts",
-      "import": "./dist/index.mjs",
-      "require": "./dist/index.js"
+      "types": "./dist/aimodels.d.ts",
+      "import": "./dist/index.js"
     }
   },
   "files": [
     "dist",
-    "LICENSE"
+    "../LICENSE",
+    "README.md"
   ],
   "scripts": {
-    "prebuild": "npm run typecheck && npm run lint && npm test",
-    "build": "tsup src/index.ts --format cjs,esm --dts",
-    "test": "deno test tests/",
-    "test:watch": "deno test --watch tests/",
+    "gen-data-js-file": "node genDataFile.js",
+    "prebuild": "npm run clean && npm run gen-data-js-file",
+    "build:ts": "tsup",
+    "build": "npm run build:ts && npm test",
+    "test": "vitest run",
+    "test:watch": "vitest watch",
+    "test:coverage": "vitest run --coverage",
     "typecheck": "tsc --noEmit",
     "lint": "eslint src --ext .ts",
     "clean": "rm -rf dist",
@@ -28,7 +32,7 @@
     "preversion": "npm run typecheck && npm run lint && npm test",
     "version": "git add -A",
     "postversion": "git push && git push --tags",
-    "prepublishOnly": "npm run clean && npm run typecheck && npm test && npm run lint",
+    "prepublishOnly": "npm run typecheck && npm run lint && npm run build",
     "postpublish": "npm run clean",
     "rules": "airul generate",
     "rules:comment": "# Generate AI rules from documentation",
@@ -64,9 +68,12 @@
     "@types/node": "^20.0.0",
     "@typescript-eslint/eslint-plugin": "^6.0.0",
     "@typescript-eslint/parser": "^6.0.0",
+    "@vitest/coverage-v8": "^1.0.0",
     "airul": "^0.1.10",
     "eslint": "^8.0.0",
+    "ts-node": "^10.9.2",
     "tsup": "^8.0.0",
-    "typescript": "^5.0.0"
+    "typescript": "^5.0.0",
+    "vitest": "^1.0.0"
   }
 }

package/LICENSE

@@ -1,21 +0,0 @@
-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/dist/index.d.mts

@@ -1,212 +0,0 @@
-/**
- * Defines all possible model capabilities
- */
-type Capability = "chat" | "reason" | "txt-in" | "txt-out" | "img-in" | "img-out" | "audio-in" | "audio-out" | "json-out" | "fn-out" | "vec-out";
-
-declare class ModelCollection extends Array<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;
-    /**
-     * Fluent capability filters for better readability
-     * Each method filters models by a specific capability
-     */
-    canChat(): ModelCollection;
-    canReason(): ModelCollection;
-    canRead(): ModelCollection;
-    canWrite(): ModelCollection;
-    canSee(): ModelCollection;
-    canGenerateImages(): ModelCollection;
-    canHear(): ModelCollection;
-    canSpeak(): ModelCollection;
-    canOutputJSON(): ModelCollection;
-    canCallFunctions(): ModelCollection;
-    canGenerateEmbeddings(): 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 or alias */
-    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;
-    /** Get all providers from all models in the collection deduplicated */
-    getProviders(): string[];
-    getCreators(): string[];
-}
-interface BaseContext {
-    /** The type discriminator */
-    type: string;
-}
-interface TokenContext extends BaseContext {
-    type: "token";
-    /** Maximum input tokens the model can accept */
-    total: number | null;
-    /** Maximum tokens the model can generate in response */
-    maxOutput: number | null;
-    /**
-     * When set to 1, indicates the model can generate up to maxOutput tokens
-     * regardless of input size (as long as input is within total limit).
-     * When not set, available output tokens may be reduced based on input size.
-     */
-    outputIsFixed?: 1;
-    /**
-     * Extended capabilities beyond the standard model behavior.
-     * This is a flexible object that can contain any properties or nested objects
-     * related to model-specific extensions (e.g., reasoning, experimental features).
-     */
-    extended?: Record<string, any>;
-}
-interface CharacterContext extends BaseContext {
-    type: "character";
-    /** Maximum input characters the model can accept */
-    total: number | null;
-    /** Maximum characters the model can generate in response */
-    maxOutput: number | null;
-}
-interface ImageContext extends BaseContext {
-    type: "image";
-    /** Maximum outputs per request */
-    maxOutput: number;
-    /** Available image sizes (e.g. "1024x1024") */
-    sizes: string[];
-    /** Available quality settings (e.g. "standard", "hd") */
-    qualities: string[];
-}
-interface AudioInputContext extends BaseContext {
-    type: "audio-in";
-    /** Maximum duration in seconds, null if unlimited */
-    maxDuration?: number | null;
-    /** Supported input formats */
-    formats?: string[];
-    /** Maximum file size in bytes */
-    maxSize?: number | null;
-}
-interface AudioOutputContext extends BaseContext {
-    type: "audio-out";
-    /** Maximum text length that can be converted to speech */
-    maxInput?: number | null;
-    /** Supported output formats */
-    formats?: string[];
-    /** Available voices */
-    voices?: string[];
-    /** Available quality settings */
-    qualities?: string[];
-}
-interface EmbeddingContext extends BaseContext {
-    type: "embedding";
-    /** Maximum input size */
-    total: number;
-    /** Unit of measurement for input */
-    unit: "tokens" | "characters";
-    /** Size of output embedding vectors */
-    dimensions: number;
-    /** Type of embeddings produced */
-    embeddingType?: "text" | "image" | "audio" | "multimodal";
-    /** Normalization of output vectors */
-    normalized?: boolean;
-}
-type ModelContext = TokenContext | CharacterContext | ImageContext | AudioInputContext | AudioOutputContext | EmbeddingContext;
-interface Model {
-    /** Unique identifier */
-    id: string;
-    /** Display name */
-    name: string;
-    /** Creator of the model */
-    creator: string;
-    /** 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: Capability[];
-    /** Languages the model knows */
-    languages?: string[];
-    /** Alternative identifiers for this model */
-    aliases?: string[];
-    /** Context window information */
-    context: ModelContext;
-    /** Base model ID this model extends */
-    extends?: string;
-    /** Properties that override the base model */
-    overrides?: Partial<Omit<Model, 'id' | 'extends' | 'overrides'>>;
-}
-
-interface TokenBasedPricePerMillionTokens {
-    /** Price per million input tokens */
-    input: number;
-    /** Price per million output tokens */
-    output: number;
-    /** Price type */
-    type: 'token';
-}
-interface ImagePrice {
-    /** Price per image */
-    price: number;
-    /** Image size */
-    size: string;
-    /** Price type */
-    type: 'image';
-    /** Price unit */
-    unit: 'per_image';
-}
-
-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",
-		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 class AIModels extends ModelCollection {
-    constructor(models?: Model[]);
-    get creators(): string[];
-    get providers(): string[];
-    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: AIModels;
-
-export { AIModels, type Capability, type Model, ModelCollection, type ModelContext, type Provider, type TokenBasedPricePerMillionTokens, creators$1 as creators, models };