@cogitator-ai/models 0.1.0

package/dist/registry.js

@@ -0,0 +1,258 @@
+import { ModelCache } from './cache';
+import { fetchLiteLLMData, transformLiteLLMData } from './fetcher';
+import { BUILTIN_MODELS, BUILTIN_PROVIDERS } from './providers/index';
+export class ModelRegistry {
+    models = new Map();
+    aliases = new Map();
+    providers = new Map();
+    cache;
+    options;
+    initialized = false;
+    refreshTimer = null;
+    constructor(options = {}) {
+        this.options = {
+            cache: options.cache ?? { ttl: 24 * 60 * 60 * 1000, storage: 'memory' },
+            autoRefresh: options.autoRefresh ?? false,
+            refreshInterval: options.refreshInterval ?? 24 * 60 * 60 * 1000,
+            fallbackToBuiltin: options.fallbackToBuiltin ?? true,
+        };
+        this.cache = new ModelCache(this.options.cache);
+    }
+    async initialize() {
+        if (this.initialized)
+            return;
+        try {
+            const cached = await this.cache.get();
+            if (cached && cached.length > 0) {
+                this.loadModels(cached);
+                this.initialized = true;
+                this.refreshInBackground();
+                return;
+            }
+            await this.refresh();
+        }
+        catch (error) {
+            if (this.options.fallbackToBuiltin) {
+                this.loadModels(BUILTIN_MODELS);
+                this.initialized = true;
+            }
+            else {
+                throw error;
+            }
+        }
+        if (this.options.autoRefresh) {
+            this.startAutoRefresh();
+        }
+    }
+    async refresh() {
+        try {
+            const data = await fetchLiteLLMData();
+            const models = transformLiteLLMData(data);
+            const allModels = this.mergeWithBuiltin(models);
+            await this.cache.set(allModels);
+            this.loadModels(allModels);
+            this.initialized = true;
+        }
+        catch (error) {
+            const stale = await this.cache.getStale();
+            if (stale && stale.length > 0) {
+                this.loadModels(stale);
+                this.initialized = true;
+                return;
+            }
+            if (this.options.fallbackToBuiltin) {
+                this.loadModels(BUILTIN_MODELS);
+                this.initialized = true;
+                return;
+            }
+            throw error;
+        }
+    }
+    getModel(id) {
+        this.ensureInitialized();
+        const normalized = this.normalizeModelId(id);
+        const direct = this.models.get(normalized);
+        if (direct)
+            return direct;
+        const aliasTarget = this.aliases.get(normalized);
+        if (aliasTarget) {
+            return this.models.get(aliasTarget) ?? null;
+        }
+        for (const [modelId, model] of this.models) {
+            if (modelId.includes(normalized) || normalized.includes(modelId)) {
+                return model;
+            }
+        }
+        return null;
+    }
+    getPrice(id) {
+        const model = this.getModel(id);
+        if (!model?.pricing)
+            return null;
+        return { input: model.pricing.input, output: model.pricing.output };
+    }
+    listModels(filter) {
+        this.ensureInitialized();
+        let models = Array.from(this.models.values());
+        if (filter) {
+            models = models.filter((model) => {
+                if (filter.provider && model.provider !== filter.provider) {
+                    return false;
+                }
+                if (filter.supportsTools && !model.capabilities?.supportsTools) {
+                    return false;
+                }
+                if (filter.supportsVision && !model.capabilities?.supportsVision) {
+                    return false;
+                }
+                if (filter.minContextWindow && model.contextWindow < filter.minContextWindow) {
+                    return false;
+                }
+                if (filter.maxPricePerMillion) {
+                    const avgPrice = (model.pricing.input + model.pricing.output) / 2;
+                    if (avgPrice > filter.maxPricePerMillion) {
+                        return false;
+                    }
+                }
+                if (filter.excludeDeprecated && model.deprecated) {
+                    return false;
+                }
+                return true;
+            });
+        }
+        return models;
+    }
+    listProviders() {
+        this.ensureInitialized();
+        return Array.from(this.providers.values());
+    }
+    getProvider(id) {
+        this.ensureInitialized();
+        return this.providers.get(id) ?? null;
+    }
+    isInitialized() {
+        return this.initialized;
+    }
+    getModelCount() {
+        return this.models.size;
+    }
+    shutdown() {
+        if (this.refreshTimer) {
+            clearInterval(this.refreshTimer);
+            this.refreshTimer = null;
+        }
+    }
+    loadModels(models) {
+        this.models.clear();
+        this.aliases.clear();
+        this.providers.clear();
+        const providerModels = new Map();
+        for (const model of models) {
+            this.models.set(model.id, model);
+            if (model.aliases) {
+                for (const alias of model.aliases) {
+                    this.aliases.set(alias.toLowerCase(), model.id);
+                }
+            }
+            const existing = providerModels.get(model.provider) ?? [];
+            existing.push(model.id);
+            providerModels.set(model.provider, existing);
+        }
+        for (const provider of BUILTIN_PROVIDERS) {
+            const models = providerModels.get(provider.id) ?? [];
+            this.providers.set(provider.id, { ...provider, models });
+        }
+        for (const [providerId, modelIds] of providerModels) {
+            if (!this.providers.has(providerId)) {
+                this.providers.set(providerId, {
+                    id: providerId,
+                    name: this.formatProviderName(providerId),
+                    models: modelIds,
+                });
+            }
+        }
+    }
+    mergeWithBuiltin(fetched) {
+        const modelMap = new Map();
+        for (const model of BUILTIN_MODELS) {
+            modelMap.set(model.id, model);
+        }
+        for (const model of fetched) {
+            const existing = modelMap.get(model.id);
+            if (existing) {
+                modelMap.set(model.id, {
+                    ...existing,
+                    pricing: model.pricing,
+                    contextWindow: model.contextWindow || existing.contextWindow,
+                    maxOutputTokens: model.maxOutputTokens || existing.maxOutputTokens,
+                    deprecated: model.deprecated,
+                });
+            }
+            else {
+                modelMap.set(model.id, model);
+            }
+        }
+        return Array.from(modelMap.values());
+    }
+    normalizeModelId(id) {
+        let normalized = id.toLowerCase();
+        if (normalized.includes('/')) {
+            normalized = normalized.split('/').pop() ?? normalized;
+        }
+        return normalized;
+    }
+    formatProviderName(id) {
+        return id
+            .replace(/_/g, ' ')
+            .replace(/-/g, ' ')
+            .replace(/\b\w/g, (c) => c.toUpperCase());
+    }
+    ensureInitialized() {
+        if (!this.initialized) {
+            this.loadModels(BUILTIN_MODELS);
+            this.initialized = true;
+        }
+    }
+    refreshInBackground() {
+        fetchLiteLLMData()
+            .then((data) => {
+            const models = transformLiteLLMData(data);
+            const allModels = this.mergeWithBuiltin(models);
+            this.cache.set(allModels);
+            this.loadModels(allModels);
+        })
+            .catch(() => {
+        });
+    }
+    startAutoRefresh() {
+        this.refreshTimer = setInterval(() => {
+            this.refresh().catch(() => {
+            });
+        }, this.options.refreshInterval);
+    }
+}
+let defaultRegistry = null;
+export function getModelRegistry() {
+    if (!defaultRegistry) {
+        defaultRegistry = new ModelRegistry({
+            cache: { ttl: 24 * 60 * 60 * 1000, storage: 'file' },
+            fallbackToBuiltin: true,
+        });
+    }
+    return defaultRegistry;
+}
+export async function initializeModels() {
+    const registry = getModelRegistry();
+    await registry.initialize();
+    return registry;
+}
+export function getPrice(modelId) {
+    return getModelRegistry().getPrice(modelId);
+}
+export function getModel(modelId) {
+    return getModelRegistry().getModel(modelId);
+}
+export function listModels(filter) {
+    return getModelRegistry().listModels(filter);
+}
+//# sourceMappingURL=registry.js.map

package/dist/registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.js","sourceRoot":"","sources":["../src/registry.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAEtE,MAAM,OAAO,aAAa;IAChB,MAAM,GAAG,IAAI,GAAG,EAAqB,CAAC;IACtC,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;IACpC,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC5C,KAAK,CAAa;IAClB,OAAO,CAA4B;IACnC,WAAW,GAAG,KAAK,CAAC;IACpB,YAAY,GAA0C,IAAI,CAAC;IAEnE,YAAY,UAA2B,EAAE;QACvC,IAAI,CAAC,OAAO,GAAG;YACb,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,EAAE,GAAG,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE;YACvE,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,KAAK;YACzC,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI;YAC/D,iBAAiB,EAAE,OAAO,CAAC,iBAAiB,IAAI,IAAI;SACrD,CAAC;QAEF,IAAI,CAAC,KAAK,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAE7B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;YAEtC,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;gBACxB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;gBAExB,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC3B,OAAO;YACT,CAAC;YAED,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;gBACnC,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;gBAChC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YAC1B,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YAC7B,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,gBAAgB,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;YAE1C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAEhD,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YAC3B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC1C,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBACvB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;gBACxB,OAAO;YACT,CAAC;YAED,IAAI,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;gBACnC,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;gBAChC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;gBACxB,OAAO;YACT,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,EAAU;QACjB,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;QAE7C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC;QAE1B,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACjD,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC;QAC9C,CAAC;QAED,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAC3C,IAAI,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBACjE,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,QAAQ,CAAC,EAAU;QACjB,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK,EAAE,OAAO;YAAE,OAAO,IAAI,CAAC;QACjC,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;IACtE,CAAC;IAED,UAAU,CAAC,MAAoB;QAC7B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,IAAI,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAE9C,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC/B,IAAI,MAAM,CAAC,QAAQ,IAAI,KAAK,CAAC,QAAQ,KAAK,MAAM,CAAC,QAAQ,EAAE,CAAC;oBAC1D,OAAO,KAAK,CAAC;gBACf,CAAC;gBAED,IAAI,MAAM,CAAC,aAAa,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,aAAa,EAAE,CAAC;oBAC/D,OAAO,KAAK,CAAC;gBACf,CAAC;gBAED,IAAI,MAAM,CAAC,cAAc,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,cAAc,EAAE,CAAC;oBACjE,OAAO,KAAK,CAAC;gBACf,CAAC;gBAED,IAAI,MAAM,CAAC,gBAAgB,IAAI,KAAK,CAAC,aAAa,GAAG,MAAM,CAAC,gBAAgB,EAAE,CAAC;oBAC7E,OAAO,KAAK,CAAC;gBACf,CAAC;gBAED,IAAI,MAAM,CAAC,kBAAkB,EAAE,CAAC;oBAC9B,MAAM,QAAQ,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;oBAClE,IAAI,QAAQ,GAAG,MAAM,CAAC,kBAAkB,EAAE,CAAC;wBACzC,OAAO,KAAK,CAAC;oBACf,CAAC;gBACH,CAAC;gBAED,IAAI,MAAM,CAAC,iBAAiB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACjD,OAAO,KAAK,CAAC;gBACf,CAAC;gBAED,OAAO,IAAI,CAAC;YACd,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,aAAa;QACX,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED,WAAW,CAAC,EAAU;QACpB,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC;IACxC,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;IAC1B,CAAC;IAED,QAAQ;QACN,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACjC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC3B,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,MAAmB;QACpC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QAEvB,MAAM,cAAc,GAAG,IAAI,GAAG,EAAoB,CAAC;QAEnD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;YAEjC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClB,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBAClC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC;YAED,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YAC1D,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACxB,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC/C,CAAC;QAED,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACrD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,GAAG,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QAC3D,CAAC;QAED,KAAK,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,IAAI,cAAc,EAAE,CAAC;YACpD,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;gBACpC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE;oBAC7B,EAAE,EAAE,UAAU;oBACd,IAAI,EAAE,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC;oBACzC,MAAM,EAAE,QAAQ;iBACjB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,OAAoB;QAC3C,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAqB,CAAC;QAE9C,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;YACnC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAChC,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACxC,IAAI,QAAQ,EAAE,CAAC;gBACb,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE;oBACrB,GAAG,QAAQ;oBACX,OAAO,EAAE,KAAK,CAAC,OAAO;oBACtB,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa;oBAC5D,eAAe,EAAE,KAAK,CAAC,eAAe,IAAI,QAAQ,CAAC,eAAe;oBAClE,UAAU,EAAE,KAAK,CAAC,UAAU;iBAC7B,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IACvC,CAAC;IAEO,gBAAgB,CAAC,EAAU;QACjC,IAAI,UAAU,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC;QAElC,IAAI,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC7B,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,UAAU,CAAC;QACzD,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,kBAAkB,CAAC,EAAU;QACnC,OAAO,EAAE;aACN,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC;aAClB,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC;aAClB,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;IAC9C,CAAC;IAEO,iBAAiB;QACvB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAChC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;IACH,CAAC;IAEO,mBAAmB;QACzB,gBAAgB,EAAE;aACf,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YACb,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;YAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC1B,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAC7B,CAAC,CAAC;aACD,KAAK,CAAC,GAAG,EAAE;QACZ,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,gBAAgB;QACtB,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE;YACnC,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;YAC1B,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IACnC,CAAC;CACF;AAED,IAAI,eAAe,GAAyB,IAAI,CAAC;AAEjD,MAAM,UAAU,gBAAgB;IAC9B,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,eAAe,GAAG,IAAI,aAAa,CAAC;YAClC,KAAK,EAAE,EAAE,GAAG,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE;YACpD,iBAAiB,EAAE,IAAI;SACxB,CAAC,CAAC;IACL,CAAC;IACD,OAAO,eAAe,CAAC;AACzB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB;IACpC,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;IACpC,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;IAC5B,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,OAAe;IACtC,OAAO,gBAAgB,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,OAAe;IACtC,OAAO,gBAAgB,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,MAAoB;IAC7C,OAAO,gBAAgB,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;AAC/C,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,183 @@
+import { z } from 'zod';
+export declare const ModelCapabilitiesSchema: z.ZodObject<{
+    supportsVision: z.ZodOptional<z.ZodBoolean>;
+    supportsTools: z.ZodOptional<z.ZodBoolean>;
+    supportsFunctions: z.ZodOptional<z.ZodBoolean>;
+    supportsStreaming: z.ZodOptional<z.ZodBoolean>;
+    supportsJson: z.ZodOptional<z.ZodBoolean>;
+}, "strip", z.ZodTypeAny, {
+    supportsVision?: boolean | undefined;
+    supportsTools?: boolean | undefined;
+    supportsFunctions?: boolean | undefined;
+    supportsStreaming?: boolean | undefined;
+    supportsJson?: boolean | undefined;
+}, {
+    supportsVision?: boolean | undefined;
+    supportsTools?: boolean | undefined;
+    supportsFunctions?: boolean | undefined;
+    supportsStreaming?: boolean | undefined;
+    supportsJson?: boolean | undefined;
+}>;
+export type ModelCapabilities = z.infer<typeof ModelCapabilitiesSchema>;
+export declare const ModelPricingSchema: z.ZodObject<{
+    input: z.ZodNumber;
+    output: z.ZodNumber;
+    inputCached: z.ZodOptional<z.ZodNumber>;
+    outputCached: z.ZodOptional<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    input: number;
+    output: number;
+    inputCached?: number | undefined;
+    outputCached?: number | undefined;
+}, {
+    input: number;
+    output: number;
+    inputCached?: number | undefined;
+    outputCached?: number | undefined;
+}>;
+export type ModelPricing = z.infer<typeof ModelPricingSchema>;
+export declare const ModelInfoSchema: z.ZodObject<{
+    id: z.ZodString;
+    provider: z.ZodString;
+    displayName: z.ZodString;
+    pricing: z.ZodObject<{
+        input: z.ZodNumber;
+        output: z.ZodNumber;
+        inputCached: z.ZodOptional<z.ZodNumber>;
+        outputCached: z.ZodOptional<z.ZodNumber>;
+    }, "strip", z.ZodTypeAny, {
+        input: number;
+        output: number;
+        inputCached?: number | undefined;
+        outputCached?: number | undefined;
+    }, {
+        input: number;
+        output: number;
+        inputCached?: number | undefined;
+        outputCached?: number | undefined;
+    }>;
+    contextWindow: z.ZodNumber;
+    maxOutputTokens: z.ZodOptional<z.ZodNumber>;
+    capabilities: z.ZodOptional<z.ZodObject<{
+        supportsVision: z.ZodOptional<z.ZodBoolean>;
+        supportsTools: z.ZodOptional<z.ZodBoolean>;
+        supportsFunctions: z.ZodOptional<z.ZodBoolean>;
+        supportsStreaming: z.ZodOptional<z.ZodBoolean>;
+        supportsJson: z.ZodOptional<z.ZodBoolean>;
+    }, "strip", z.ZodTypeAny, {
+        supportsVision?: boolean | undefined;
+        supportsTools?: boolean | undefined;
+        supportsFunctions?: boolean | undefined;
+        supportsStreaming?: boolean | undefined;
+        supportsJson?: boolean | undefined;
+    }, {
+        supportsVision?: boolean | undefined;
+        supportsTools?: boolean | undefined;
+        supportsFunctions?: boolean | undefined;
+        supportsStreaming?: boolean | undefined;
+        supportsJson?: boolean | undefined;
+    }>>;
+    deprecated: z.ZodOptional<z.ZodBoolean>;
+    aliases: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+}, "strip", z.ZodTypeAny, {
+    id: string;
+    provider: string;
+    displayName: string;
+    pricing: {
+        input: number;
+        output: number;
+        inputCached?: number | undefined;
+        outputCached?: number | undefined;
+    };
+    contextWindow: number;
+    maxOutputTokens?: number | undefined;
+    capabilities?: {
+        supportsVision?: boolean | undefined;
+        supportsTools?: boolean | undefined;
+        supportsFunctions?: boolean | undefined;
+        supportsStreaming?: boolean | undefined;
+        supportsJson?: boolean | undefined;
+    } | undefined;
+    deprecated?: boolean | undefined;
+    aliases?: string[] | undefined;
+}, {
+    id: string;
+    provider: string;
+    displayName: string;
+    pricing: {
+        input: number;
+        output: number;
+        inputCached?: number | undefined;
+        outputCached?: number | undefined;
+    };
+    contextWindow: number;
+    maxOutputTokens?: number | undefined;
+    capabilities?: {
+        supportsVision?: boolean | undefined;
+        supportsTools?: boolean | undefined;
+        supportsFunctions?: boolean | undefined;
+        supportsStreaming?: boolean | undefined;
+        supportsJson?: boolean | undefined;
+    } | undefined;
+    deprecated?: boolean | undefined;
+    aliases?: string[] | undefined;
+}>;
+export type ModelInfo = z.infer<typeof ModelInfoSchema>;
+export declare const ProviderInfoSchema: z.ZodObject<{
+    id: z.ZodString;
+    name: z.ZodString;
+    website: z.ZodOptional<z.ZodString>;
+    models: z.ZodArray<z.ZodString, "many">;
+}, "strip", z.ZodTypeAny, {
+    id: string;
+    name: string;
+    models: string[];
+    website?: string | undefined;
+}, {
+    id: string;
+    name: string;
+    models: string[];
+    website?: string | undefined;
+}>;
+export type ProviderInfo = z.infer<typeof ProviderInfoSchema>;
+export interface ModelFilter {
+    provider?: string;
+    supportsTools?: boolean;
+    supportsVision?: boolean;
+    minContextWindow?: number;
+    maxPricePerMillion?: number;
+    excludeDeprecated?: boolean;
+}
+export interface CacheOptions {
+    ttl: number;
+    storage: 'memory' | 'file';
+    filePath?: string;
+}
+export interface RegistryOptions {
+    cache?: CacheOptions;
+    autoRefresh?: boolean;
+    refreshInterval?: number;
+    fallbackToBuiltin?: boolean;
+}
+export interface LiteLLMModelEntry {
+    max_tokens?: number;
+    max_input_tokens?: number;
+    max_output_tokens?: number;
+    input_cost_per_token?: number;
+    output_cost_per_token?: number;
+    input_cost_per_character?: number;
+    output_cost_per_character?: number;
+    litellm_provider?: string;
+    mode?: string;
+    supports_function_calling?: boolean;
+    supports_parallel_function_calling?: boolean;
+    supports_vision?: boolean;
+    supports_response_schema?: boolean;
+    supports_tool_choice?: boolean;
+    tool_use_system_prompt_tokens?: number;
+    supports_assistant_prefill?: boolean;
+    deprecation_date?: string;
+    source?: string;
+}
+export type LiteLLMModelData = Record<string, LiteLLMModelEntry>;
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;EAMlC,CAAC;AAEH,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAExE,eAAO,MAAM,kBAAkB;;;;;;;;;;;;;;;EAK7B,CAAC;AAEH,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAE9D,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAU1B,CAAC;AAEH,MAAM,MAAM,SAAS,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,eAAe,CAAC,CAAC;AAExD,eAAO,MAAM,kBAAkB;;;;;;;;;;;;;;;EAK7B,CAAC;AAEH,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAE9D,MAAM,WAAW,WAAW;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,MAAM,WAAW,YAAY;IAC3B,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,QAAQ,GAAG,MAAM,CAAC;IAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,CAAC,EAAE,YAAY,CAAC;IACrB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,MAAM,WAAW,iBAAiB;IAChC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,yBAAyB,CAAC,EAAE,OAAO,CAAC;IACpC,kCAAkC,CAAC,EAAE,OAAO,CAAC;IAC7C,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,wBAAwB,CAAC,EAAE,OAAO,CAAC;IACnC,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,6BAA6B,CAAC,EAAE,MAAM,CAAC;IACvC,0BAA0B,CAAC,EAAE,OAAO,CAAC;IACrC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC"}

package/dist/types.js

@@ -0,0 +1,32 @@
+import { z } from 'zod';
+export const ModelCapabilitiesSchema = z.object({
+    supportsVision: z.boolean().optional(),
+    supportsTools: z.boolean().optional(),
+    supportsFunctions: z.boolean().optional(),
+    supportsStreaming: z.boolean().optional(),
+    supportsJson: z.boolean().optional(),
+});
+export const ModelPricingSchema = z.object({
+    input: z.number(),
+    output: z.number(),
+    inputCached: z.number().optional(),
+    outputCached: z.number().optional(),
+});
+export const ModelInfoSchema = z.object({
+    id: z.string(),
+    provider: z.string(),
+    displayName: z.string(),
+    pricing: ModelPricingSchema,
+    contextWindow: z.number(),
+    maxOutputTokens: z.number().optional(),
+    capabilities: ModelCapabilitiesSchema.optional(),
+    deprecated: z.boolean().optional(),
+    aliases: z.array(z.string()).optional(),
+});
+export const ProviderInfoSchema = z.object({
+    id: z.string(),
+    name: z.string(),
+    website: z.string().optional(),
+    models: z.array(z.string()),
+});
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,cAAc,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACtC,aAAa,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACrC,iBAAiB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACzC,iBAAiB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACzC,YAAY,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;CACrC,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;IACjB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACpC,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;IACvB,OAAO,EAAE,kBAAkB;IAC3B,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;IACzB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACtC,YAAY,EAAE,uBAAuB,CAAC,QAAQ,EAAE;IAChD,UAAU,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IAClC,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CACxC,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CAC5B,CAAC,CAAC"}

package/package.json

@@ -0,0 +1,43 @@
+{
+  "name": "@cogitator-ai/models",
+  "version": "0.1.0",
+  "description": "Dynamic model registry with pricing for Cogitator",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "dependencies": {
+    "zod": "^3.24.0"
+  },
+  "devDependencies": {
+    "typescript": "^5.7.2",
+    "vitest": "^2.1.8"
+  },
+  "peerDependencies": {
+    "@cogitator-ai/types": "0.1.0"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/eL1fe/cogitator.git",
+    "directory": "packages/models"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "license": "MIT",
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "clean": "rm -rf dist"
+  }
+}