ai-model-directory 0.1.0

package/README.md

@@ -0,0 +1,11 @@
+# ai-model-directory
+
+Bundled version of [the AI Model Directory's](https://github.com/The-Best-Codes/ai-model-directory) `data/all.min.json` with a compact typed payload.
+
+## Usage
+
+```ts
+import { openai } from "ai-model-directory";
+
+console.log(openai.models["gpt-5.5"]?.pricing?.input);
+```

package/dist/src/codec.d.ts

@@ -0,0 +1,3 @@
+import type { CompactProviderEntry } from "./compact-types.js";
+import type { ModelDirectory } from "./types.js";
+export declare function decodeModelDirectory(data: readonly CompactProviderEntry[]): ModelDirectory;

package/dist/src/codec.js

@@ -0,0 +1,196 @@
+const modelNameSameAsId = "";
+const modalityOrder = [
+    "audio",
+    "file",
+    "image",
+    "text",
+    "video",
+];
+const featureKeys = [
+    "attachment",
+    "reasoning",
+    "tool_call",
+    "structured_output",
+    "temperature",
+];
+const pricingKeys = [
+    "input",
+    "output",
+    "reasoning",
+    "cache_read",
+    "cache_write",
+    "input_audio",
+    "output_audio",
+];
+const limitKeys = [
+    "context",
+    "input",
+    "output",
+];
+function toTimestamp(value) {
+    return value === null || value === undefined ? undefined : String(value);
+}
+function decodeAiSdk(value) {
+    if (!value) {
+        return undefined;
+    }
+    const [npmPackage, defaultApiKeyEnv] = value;
+    const result = {};
+    if (npmPackage !== null && npmPackage !== undefined) {
+        result.npmPackage = npmPackage;
+    }
+    if (defaultApiKeyEnv !== null && defaultApiKeyEnv !== undefined) {
+        result.defaultApiKeyEnv = [...defaultApiKeyEnv];
+    }
+    return Object.keys(result).length > 0 ? result : undefined;
+}
+function decodeFeatures(value) {
+    if (!value) {
+        return undefined;
+    }
+    const [presentMask, truthyMask] = value;
+    const result = {};
+    for (let index = 0; index < featureKeys.length; index += 1) {
+        const key = featureKeys[index];
+        if (!key) {
+            continue;
+        }
+        const bit = 1 << index;
+        if ((presentMask & bit) === 0) {
+            continue;
+        }
+        result[key] = (truthyMask & bit) !== 0;
+    }
+    return Object.keys(result).length > 0 ? result : undefined;
+}
+function decodePricing(value) {
+    if (!value) {
+        return undefined;
+    }
+    const result = {};
+    for (let index = 0; index < pricingKeys.length; index += 1) {
+        const key = pricingKeys[index];
+        const entry = value[index];
+        if (key && entry !== null && entry !== undefined) {
+            result[key] = entry;
+        }
+    }
+    return Object.keys(result).length > 0 ? result : undefined;
+}
+function decodeLimit(value) {
+    if (!value) {
+        return undefined;
+    }
+    const result = {};
+    for (let index = 0; index < limitKeys.length; index += 1) {
+        const key = limitKeys[index];
+        const entry = value[index];
+        if (key && entry !== null && entry !== undefined) {
+            result[key] = entry;
+        }
+    }
+    return Object.keys(result).length > 0 ? result : undefined;
+}
+function decodeModalityMask(mask) {
+    if (mask === null || mask === undefined || mask === 0) {
+        return undefined;
+    }
+    const result = [];
+    for (let index = 0; index < modalityOrder.length; index += 1) {
+        const modality = modalityOrder[index];
+        if (modality && (mask & (1 << index)) !== 0) {
+            result.push(modality);
+        }
+    }
+    return result.length > 0 ? result : undefined;
+}
+function decodeModalities(value) {
+    if (!value) {
+        return undefined;
+    }
+    const [inputMask, outputMask] = value;
+    const result = {};
+    const input = decodeModalityMask(inputMask);
+    const output = decodeModalityMask(outputMask);
+    if (input) {
+        result.input = input;
+    }
+    if (output) {
+        result.output = output;
+    }
+    return Object.keys(result).length > 0 ? result : undefined;
+}
+function decodeModel(entry) {
+    const [id, compactName, releaseDate, knowledgeCutoff, lastUpdated, openWeights, features, pricing, limit, modalities,] = entry;
+    const model = { id };
+    const releaseDateValue = toTimestamp(releaseDate);
+    const knowledgeCutoffValue = toTimestamp(knowledgeCutoff);
+    const lastUpdatedValue = toTimestamp(lastUpdated);
+    const decodedFeatures = decodeFeatures(features);
+    const decodedPricing = decodePricing(pricing);
+    const decodedLimit = decodeLimit(limit);
+    const decodedModalities = decodeModalities(modalities);
+    if (compactName === modelNameSameAsId) {
+        model.name = id;
+    }
+    else if (compactName !== null && compactName !== undefined) {
+        model.name = compactName;
+    }
+    if (knowledgeCutoffValue !== undefined) {
+        model.knowledge_cutoff = knowledgeCutoffValue;
+    }
+    if (releaseDateValue !== undefined) {
+        model.release_date = releaseDateValue;
+    }
+    if (lastUpdatedValue !== undefined) {
+        model.last_updated = lastUpdatedValue;
+    }
+    if (openWeights !== null && openWeights !== undefined) {
+        model.open_weights = openWeights;
+    }
+    if (decodedFeatures) {
+        model.features = decodedFeatures;
+    }
+    if (decodedPricing) {
+        model.pricing = decodedPricing;
+    }
+    if (decodedLimit) {
+        model.limit = decodedLimit;
+    }
+    if (decodedModalities) {
+        model.modalities = decodedModalities;
+    }
+    return model;
+}
+function decodeProvider(entry) {
+    const [id, name, website, apiBaseUrl, aiSdk, models = []] = entry;
+    const compactModels = models ?? [];
+    const provider = {
+        id,
+        name,
+    };
+    if (website !== null && website !== undefined) {
+        provider.website = website;
+    }
+    if (apiBaseUrl !== null && apiBaseUrl !== undefined) {
+        provider.apiBaseUrl = apiBaseUrl;
+    }
+    const decodedAiSdk = decodeAiSdk(aiSdk);
+    if (decodedAiSdk) {
+        provider.aiSdk = decodedAiSdk;
+    }
+    provider.models = {};
+    for (const modelEntry of compactModels) {
+        const model = decodeModel(modelEntry);
+        provider.models[model.id] = model;
+    }
+    return provider;
+}
+export function decodeModelDirectory(data) {
+    const result = {};
+    for (const providerEntry of data) {
+        const provider = decodeProvider(providerEntry);
+        result[provider.id] = provider;
+    }
+    return result;
+}

package/dist/src/compact-types.d.ts

@@ -0,0 +1,46 @@
+export type CompactAiSdk = readonly [
+    npmPackage?: string | null,
+    defaultApiKeyEnv?: readonly string[] | null
+];
+export type CompactFeatures = readonly [
+    presentMask: number,
+    truthyMask: number
+];
+export type CompactPricing = readonly [
+    input?: number | null,
+    output?: number | null,
+    reasoning?: number | null,
+    cacheRead?: number | null,
+    cacheWrite?: number | null,
+    inputAudio?: number | null,
+    outputAudio?: number | null
+];
+export type CompactLimit = readonly [
+    context?: number | null,
+    input?: number | null,
+    output?: number | null
+];
+export type CompactModalities = readonly [
+    inputMask?: number | null,
+    outputMask?: number | null
+];
+export type CompactModelEntry = readonly [
+    id: string,
+    name?: string | null,
+    releaseDate?: number | null,
+    knowledgeCutoff?: number | null,
+    lastUpdated?: number | null,
+    openWeights?: boolean | null,
+    features?: CompactFeatures | null,
+    pricing?: CompactPricing | null,
+    limit?: CompactLimit | null,
+    modalities?: CompactModalities | null
+];
+export type CompactProviderEntry = readonly [
+    id: string,
+    name: string,
+    website?: string | null,
+    apiBaseUrl?: string | null,
+    aiSdk?: CompactAiSdk | null,
+    models?: readonly CompactModelEntry[] | null
+];

package/dist/src/compact-types.js

@@ -0,0 +1 @@
+export {};

package/dist/src/data.d.ts

@@ -0,0 +1,2 @@
+import type { CompactProviderEntry } from "./compact-types.js";
+export declare const compactModelDirectoryData: readonly CompactProviderEntry[];

package/dist/src/data.js

@@ -0,0 +1,2 @@
+import { rawCompactModelDirectoryData } from "./generated-data.js";
+export const compactModelDirectoryData = rawCompactModelDirectoryData;

package/dist/src/generated-data.d.ts

@@ -0,0 +1 @@
+export declare const rawCompactModelDirectoryData: (string | (string | string[])[] | (string | number | boolean | (number | null)[] | null)[][])[][];