@auxiora/model-registry 1.10.7 → 1.10.8

package/dist/db.d.ts

@@ -0,0 +1,3 @@
+import { DatabaseSync } from 'node:sqlite';
+export declare function openDatabase(dbPath: string): DatabaseSync;
+//# sourceMappingURL=db.d.ts.map

package/dist/db.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"db.d.ts","sourceRoot":"","sources":["../src/db.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAgC3C,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,YAAY,CAMzD"}

package/{src/db.ts → dist/db.js}

@@ -1,5 +1,4 @@
 import { DatabaseSync } from 'node:sqlite';
-
 const SCHEMA_DDL = `
   CREATE TABLE IF NOT EXISTS discovered_models (
     id TEXT PRIMARY KEY,
@@ -29,11 +28,11 @@ const SCHEMA_DDL = `
   CREATE INDEX IF NOT EXISTS idx_models_source ON discovered_models(provider_source);
   CREATE INDEX IF NOT EXISTS idx_models_enabled ON discovered_models(enabled);
 `;
-
-export function openDatabase(dbPath: string): DatabaseSync {
-  const db = new DatabaseSync(dbPath);
-  db.exec('PRAGMA journal_mode=WAL');
-  db.exec('PRAGMA foreign_keys=ON');
-  db.exec(SCHEMA_DDL);
-  return db;
+export function openDatabase(dbPath) {
+    const db = new DatabaseSync(dbPath);
+    db.exec('PRAGMA journal_mode=WAL');
+    db.exec('PRAGMA foreign_keys=ON');
+    db.exec(SCHEMA_DDL);
+    return db;
 }
+//# sourceMappingURL=db.js.map

package/dist/db.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"db.js","sourceRoot":"","sources":["../src/db.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,MAAM,UAAU,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4BlB,CAAC;AAEF,MAAM,UAAU,YAAY,CAAC,MAAc;IACzC,MAAM,EAAE,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IACpC,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IACnC,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAClC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACpB,OAAO,EAAE,CAAC;AACZ,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export { ModelRegistry } from './registry.js';
+export type { DiscoveredModel, ModelSearchOptions, ModelCapabilitiesLike, } from './types.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,YAAY,EACV,eAAe,EACf,kBAAkB,EAClB,qBAAqB,GACtB,MAAM,YAAY,CAAC"}

package/dist/index.js

@@ -0,0 +1,2 @@
+export { ModelRegistry } from './registry.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC"}

package/dist/registry.d.ts

@@ -0,0 +1,18 @@
+import type { DiscoveredModel, ModelCapabilitiesLike, ModelSearchOptions } from './types.js';
+export declare class ModelRegistry {
+    private db;
+    constructor(dbPath: string);
+    upsertModels(models: DiscoveredModel[]): void;
+    getModels(options?: ModelSearchOptions): DiscoveredModel[];
+    getModel(id: string): DiscoveredModel | undefined;
+    /**
+     * Convert discovered models to the format ModelRouter expects.
+     * Returns Record<modelId, ModelCapabilities>.
+     */
+    toModelCapabilities(source: string): Record<string, ModelCapabilitiesLike>;
+    search(query: string): DiscoveredModel[];
+    getTrending(limit?: number): DiscoveredModel[];
+    pruneStale(maxAgeMs: number): number;
+    setEnabled(id: string, enabled: boolean): void;
+}
+//# sourceMappingURL=registry.d.ts.map

package/dist/registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../src/registry.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,eAAe,EAAE,qBAAqB,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAoC7F,qBAAa,aAAa;IACxB,OAAO,CAAC,EAAE,CAAe;gBAEb,MAAM,EAAE,MAAM;IAI1B,YAAY,CAAC,MAAM,EAAE,eAAe,EAAE,GAAG,IAAI;IAqC7C,SAAS,CAAC,OAAO,CAAC,EAAE,kBAAkB,GAAG,eAAe,EAAE;IAqC1D,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAQjD;;;OAGG;IACH,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,qBAAqB,CAAC;IAqB1E,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,eAAe,EAAE;IAIxC,WAAW,CAAC,KAAK,SAAK,GAAG,eAAe,EAAE;IAW1C,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM;IAQpC,UAAU,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;CAK/C"}

package/dist/registry.js

@@ -0,0 +1,135 @@
+import { openDatabase } from './db.js';
+function rowToModel(row) {
+    return {
+        id: row.id,
+        providerSource: row.provider_source,
+        modelId: row.model_id,
+        displayName: row.display_name,
+        contextLength: row.context_length,
+        supportsVision: row.supports_vision === 1,
+        supportsTools: row.supports_tools === 1,
+        supportsStreaming: row.supports_streaming === 1,
+        supportsImageGen: row.supports_image_gen === 1,
+        costPer1kInput: row.cost_per_1k_input,
+        costPer1kOutput: row.cost_per_1k_output,
+        strengths: JSON.parse(row.strengths || '[]'),
+        rawMetadata: row.raw_metadata,
+        hfModelCard: row.hf_model_card,
+        hfDownloads: row.hf_downloads,
+        hfLikes: row.hf_likes,
+        hfTrendingScore: row.hf_trending_score,
+        hfTags: row.hf_tags ? JSON.parse(row.hf_tags) : undefined,
+        hfBenchmarkScores: row.hf_benchmark_scores
+            ? JSON.parse(row.hf_benchmark_scores)
+            : undefined,
+        hfInferenceProviders: row.hf_inference_providers
+            ? JSON.parse(row.hf_inference_providers)
+            : undefined,
+        lastRefreshedAt: row.last_refreshed_at,
+        createdAt: row.created_at,
+        enabled: row.enabled === 1,
+    };
+}
+export class ModelRegistry {
+    db;
+    constructor(dbPath) {
+        this.db = openDatabase(dbPath);
+    }
+    upsertModels(models) {
+        const stmt = this.db.prepare(`
+      INSERT OR REPLACE INTO discovered_models (
+        id, provider_source, model_id, display_name, context_length,
+        supports_vision, supports_tools, supports_streaming, supports_image_gen,
+        cost_per_1k_input, cost_per_1k_output, strengths, raw_metadata,
+        hf_model_card, hf_downloads, hf_likes, hf_trending_score,
+        hf_tags, hf_benchmark_scores, hf_inference_providers,
+        last_refreshed_at, created_at, enabled
+      ) VALUES (
+        ?, ?, ?, ?, ?,
+        ?, ?, ?, ?,
+        ?, ?, ?, ?,
+        ?, ?, ?, ?,
+        ?, ?, ?,
+        ?, ?, ?
+      )
+    `);
+        for (const m of models) {
+            stmt.run(m.id, m.providerSource, m.modelId, m.displayName, m.contextLength, m.supportsVision ? 1 : 0, m.supportsTools ? 1 : 0, m.supportsStreaming ? 1 : 0, m.supportsImageGen ? 1 : 0, m.costPer1kInput, m.costPer1kOutput, JSON.stringify(m.strengths), m.rawMetadata ?? null, m.hfModelCard ?? null, m.hfDownloads ?? null, m.hfLikes ?? null, m.hfTrendingScore ?? null, m.hfTags ? JSON.stringify(m.hfTags) : null, m.hfBenchmarkScores ? JSON.stringify(m.hfBenchmarkScores) : null, m.hfInferenceProviders ? JSON.stringify(m.hfInferenceProviders) : null, m.lastRefreshedAt, m.createdAt, m.enabled ? 1 : 0);
+        }
+    }
+    getModels(options) {
+        const conditions = [];
+        const params = [];
+        if (options?.source) {
+            conditions.push('provider_source = ?');
+            params.push(options.source);
+        }
+        if (options?.query) {
+            conditions.push('(display_name LIKE ? OR model_id LIKE ?)');
+            const q = `%${options.query}%`;
+            params.push(q, q);
+        }
+        if (options?.supportsVision !== undefined) {
+            conditions.push('supports_vision = ?');
+            params.push(options.supportsVision ? 1 : 0);
+        }
+        if (options?.supportsTools !== undefined) {
+            conditions.push('supports_tools = ?');
+            params.push(options.supportsTools ? 1 : 0);
+        }
+        if (options?.enabled !== undefined) {
+            conditions.push('enabled = ?');
+            params.push(options.enabled ? 1 : 0);
+        }
+        const where = conditions.length > 0 ? `WHERE ${conditions.join(' AND ')}` : '';
+        const limit = options?.limit ?? 200;
+        const offset = options?.offset ?? 0;
+        const rows = this.db.prepare(`SELECT * FROM discovered_models ${where} ORDER BY display_name ASC LIMIT ? OFFSET ?`).all(...params, limit, offset);
+        return rows.map(rowToModel);
+    }
+    getModel(id) {
+        const row = this.db.prepare('SELECT * FROM discovered_models WHERE id = ?').get(id);
+        return row ? rowToModel(row) : undefined;
+    }
+    /**
+     * Convert discovered models to the format ModelRouter expects.
+     * Returns Record<modelId, ModelCapabilities>.
+     */
+    toModelCapabilities(source) {
+        const models = this.getModels({ source, enabled: true });
+        const result = {};
+        for (const m of models) {
+            result[m.modelId] = {
+                maxContextTokens: m.contextLength,
+                supportsVision: m.supportsVision,
+                supportsTools: m.supportsTools,
+                supportsStreaming: m.supportsStreaming,
+                supportsImageGen: m.supportsImageGen,
+                costPer1kInput: m.costPer1kInput,
+                costPer1kOutput: m.costPer1kOutput,
+                strengths: m.strengths,
+                isLocal: false,
+            };
+        }
+        return result;
+    }
+    search(query) {
+        return this.getModels({ query, enabled: true });
+    }
+    getTrending(limit = 20) {
+        const rows = this.db.prepare(`SELECT * FROM discovered_models
+       WHERE hf_trending_score IS NOT NULL AND enabled = 1
+       ORDER BY hf_trending_score DESC
+       LIMIT ?`).all(limit);
+        return rows.map(rowToModel);
+    }
+    pruneStale(maxAgeMs) {
+        const cutoff = Date.now() - maxAgeMs;
+        const result = this.db.prepare('DELETE FROM discovered_models WHERE last_refreshed_at < ?').run(cutoff);
+        return Number(result.changes);
+    }
+    setEnabled(id, enabled) {
+        this.db.prepare('UPDATE discovered_models SET enabled = ? WHERE id = ?').run(enabled ? 1 : 0, id);
+    }
+}
+//# 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,YAAY,EAAE,MAAM,SAAS,CAAC;AAKvC,SAAS,UAAU,CAAC,GAAQ;IAC1B,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAY;QACpB,cAAc,EAAE,GAAG,CAAC,eAAyB;QAC7C,OAAO,EAAE,GAAG,CAAC,QAAkB;QAC/B,WAAW,EAAE,GAAG,CAAC,YAAsB;QACvC,aAAa,EAAE,GAAG,CAAC,cAAwB;QAC3C,cAAc,EAAG,GAAG,CAAC,eAA0B,KAAK,CAAC;QACrD,aAAa,EAAG,GAAG,CAAC,cAAyB,KAAK,CAAC;QACnD,iBAAiB,EAAG,GAAG,CAAC,kBAA6B,KAAK,CAAC;QAC3D,gBAAgB,EAAG,GAAG,CAAC,kBAA6B,KAAK,CAAC;QAC1D,cAAc,EAAE,GAAG,CAAC,iBAA2B;QAC/C,eAAe,EAAE,GAAG,CAAC,kBAA4B;QACjD,SAAS,EAAE,IAAI,CAAC,KAAK,CAAE,GAAG,CAAC,SAAoB,IAAI,IAAI,CAAa;QACpE,WAAW,EAAE,GAAG,CAAC,YAAkC;QACnD,WAAW,EAAE,GAAG,CAAC,aAAmC;QACpD,WAAW,EAAE,GAAG,CAAC,YAAkC;QACnD,OAAO,EAAE,GAAG,CAAC,QAA8B;QAC3C,eAAe,EAAE,GAAG,CAAC,iBAAuC;QAC5D,MAAM,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAiB,CAAa,CAAC,CAAC,CAAC,SAAS;QAC/E,iBAAiB,EAAE,GAAG,CAAC,mBAAmB;YACxC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,mBAA6B,CAA2B;YACzE,CAAC,CAAC,SAAS;QACb,oBAAoB,EAAE,GAAG,CAAC,sBAAsB;YAC9C,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,sBAAgC,CAAa;YAC9D,CAAC,CAAC,SAAS;QACb,eAAe,EAAE,GAAG,CAAC,iBAA2B;QAChD,SAAS,EAAE,GAAG,CAAC,UAAoB;QACnC,OAAO,EAAG,GAAG,CAAC,OAAkB,KAAK,CAAC;KACvC,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,aAAa;IAChB,EAAE,CAAe;IAEzB,YAAY,MAAc;QACxB,IAAI,CAAC,EAAE,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC;IAED,YAAY,CAAC,MAAyB;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;;;;;;;;;;;KAgB5B,CAAC,CAAC;QAEH,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YACvB,IAAI,CAAC,GAAG,CACN,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,aAAa,EACjE,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACjD,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACvD,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,eAAe,EACnC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,WAAW,IAAI,IAAI,EAClD,CAAC,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC,CAAC,WAAW,IAAI,IAAI,EAC5C,CAAC,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC,CAAC,eAAe,IAAI,IAAI,EAC5C,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,EAC1C,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,IAAI,EAChE,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,IAAI,EACtE,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC,SAAS,EAC9B,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAClB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,SAAS,CAAC,OAA4B;QACpC,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,MAAM,GAAoB,EAAE,CAAC;QAEnC,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;YACpB,UAAU,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACvC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC9B,CAAC;QACD,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;YACnB,UAAU,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;YAC5D,MAAM,CAAC,GAAG,IAAI,OAAO,CAAC,KAAK,GAAG,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACpB,CAAC;QACD,IAAI,OAAO,EAAE,cAAc,KAAK,SAAS,EAAE,CAAC;YAC1C,UAAU,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACvC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,OAAO,EAAE,aAAa,KAAK,SAAS,EAAE,CAAC;YACzC,UAAU,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACtC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,OAAO,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;YACnC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/E,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,GAAG,CAAC;QACpC,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC;QAEpC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,mCAAmC,KAAK,6CAA6C,CACtF,CAAC,GAAG,CAAC,GAAG,MAAM,EAAE,KAAK,EAAE,MAAM,CAAU,CAAC;QAEzC,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAC9B,CAAC;IAED,QAAQ,CAAC,EAAU;QACjB,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CACzB,8CAA8C,CAC/C,CAAC,GAAG,CAAC,EAAE,CAAoB,CAAC;QAE7B,OAAO,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3C,CAAC;IAED;;;OAGG;IACH,mBAAmB,CAAC,MAAc;QAChC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACzD,MAAM,MAAM,GAA0C,EAAE,CAAC;QAEzD,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YACvB,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG;gBAClB,gBAAgB,EAAE,CAAC,CAAC,aAAa;gBACjC,cAAc,EAAE,CAAC,CAAC,cAAc;gBAChC,aAAa,EAAE,CAAC,CAAC,aAAa;gBAC9B,iBAAiB,EAAE,CAAC,CAAC,iBAAiB;gBACtC,gBAAgB,EAAE,CAAC,CAAC,gBAAgB;gBACpC,cAAc,EAAE,CAAC,CAAC,cAAc;gBAChC,eAAe,EAAE,CAAC,CAAC,eAAe;gBAClC,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,OAAO,EAAE,KAAK;aACf,CAAC;QACJ,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,MAAM,CAAC,KAAa;QAClB,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,WAAW,CAAC,KAAK,GAAG,EAAE;QACpB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B;;;eAGS,CACV,CAAC,GAAG,CAAC,KAAK,CAAU,CAAC;QAEtB,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAC9B,CAAC;IAED,UAAU,CAAC,QAAgB;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC5B,2DAA2D,CAC5D,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACd,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,UAAU,CAAC,EAAU,EAAE,OAAgB;QACrC,IAAI,CAAC,EAAE,CAAC,OAAO,CACb,uDAAuD,CACxD,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC7B,CAAC;CACF"}

package/dist/types.d.ts

@@ -0,0 +1,57 @@
+/**
+ * A model discovered from an external provider API (OpenRouter, HuggingFace, etc.).
+ */
+export interface DiscoveredModel {
+    /** Composite key: `${providerSource}:${modelId}` */
+    id: string;
+    /** Source provider that discovered this model */
+    providerSource: string;
+    /** Original model ID from the external API */
+    modelId: string;
+    displayName: string;
+    contextLength: number;
+    supportsVision: boolean;
+    supportsTools: boolean;
+    supportsStreaming: boolean;
+    supportsImageGen: boolean;
+    costPer1kInput: number;
+    costPer1kOutput: number;
+    strengths: string[];
+    rawMetadata?: string;
+    /** HuggingFace-specific fields */
+    hfModelCard?: string;
+    hfDownloads?: number;
+    hfLikes?: number;
+    hfTrendingScore?: number;
+    hfTags?: string[];
+    hfBenchmarkScores?: Record<string, number>;
+    hfInferenceProviders?: string[];
+    lastRefreshedAt: number;
+    createdAt: number;
+    enabled: boolean;
+}
+export interface ModelSearchOptions {
+    source?: string;
+    query?: string;
+    supportsVision?: boolean;
+    supportsTools?: boolean;
+    enabled?: boolean;
+    limit?: number;
+    offset?: number;
+}
+/**
+ * Structural type matching ModelCapabilities from @auxiora/providers
+ * to avoid a hard dependency on that package.
+ */
+export interface ModelCapabilitiesLike {
+    maxContextTokens: number;
+    supportsVision: boolean;
+    supportsTools: boolean;
+    supportsStreaming: boolean;
+    supportsImageGen: boolean;
+    costPer1kInput: number;
+    costPer1kOutput: number;
+    strengths: string[];
+    isLocal: boolean;
+}
+//# 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;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,oDAAoD;IACpD,EAAE,EAAE,MAAM,CAAC;IACX,iDAAiD;IACjD,cAAc,EAAE,MAAM,CAAC;IACvB,8CAA8C;IAC9C,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,OAAO,CAAC;IACxB,aAAa,EAAE,OAAO,CAAC;IACvB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,gBAAgB,EAAE,OAAO,CAAC;IAC1B,cAAc,EAAE,MAAM,CAAC;IACvB,eAAe,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,kCAAkC;IAClC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3C,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;IAChC,eAAe,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,kBAAkB;IACjC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,gBAAgB,EAAE,MAAM,CAAC;IACzB,cAAc,EAAE,OAAO,CAAC;IACxB,aAAa,EAAE,OAAO,CAAC;IACvB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,gBAAgB,EAAE,OAAO,CAAC;IAC1B,cAAc,EAAE,MAAM,CAAC;IACvB,eAAe,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,OAAO,EAAE,OAAO,CAAC;CAClB"}

package/dist/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/package.json

@@ -1,10 +1,29 @@
 {
   "name": "@auxiora/model-registry",
-  "version": "1.10.7",
+  "version": "1.10.8",
   "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
   "exports": {
-    ".": "./src/index.ts"
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
   },
+  "engines": {
+    "node": ">=22.0.0"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "files": [
+    "dist/"
+  ],
   "dependencies": {},
-  "devDependencies": {}
+  "devDependencies": {},
+  "scripts": {
+    "build": "tsc",
+    "clean": "rm -rf dist",
+    "typecheck": "tsc --noEmit"
+  }
 }

package/src/index.ts

@@ -1,6 +0,0 @@
-export { ModelRegistry } from './registry.js';
-export type {
-  DiscoveredModel,
-  ModelSearchOptions,
-  ModelCapabilitiesLike,
-} from './types.js';

package/src/registry.ts

@@ -1,181 +0,0 @@
-import type { DatabaseSync } from 'node:sqlite';
-import { openDatabase } from './db.js';
-import type { DiscoveredModel, ModelCapabilitiesLike, ModelSearchOptions } from './types.js';
-
-type Row = Record<string, unknown>;
-
-function rowToModel(row: Row): DiscoveredModel {
-  return {
-    id: row.id as string,
-    providerSource: row.provider_source as string,
-    modelId: row.model_id as string,
-    displayName: row.display_name as string,
-    contextLength: row.context_length as number,
-    supportsVision: (row.supports_vision as number) === 1,
-    supportsTools: (row.supports_tools as number) === 1,
-    supportsStreaming: (row.supports_streaming as number) === 1,
-    supportsImageGen: (row.supports_image_gen as number) === 1,
-    costPer1kInput: row.cost_per_1k_input as number,
-    costPer1kOutput: row.cost_per_1k_output as number,
-    strengths: JSON.parse((row.strengths as string) || '[]') as string[],
-    rawMetadata: row.raw_metadata as string | undefined,
-    hfModelCard: row.hf_model_card as string | undefined,
-    hfDownloads: row.hf_downloads as number | undefined,
-    hfLikes: row.hf_likes as number | undefined,
-    hfTrendingScore: row.hf_trending_score as number | undefined,
-    hfTags: row.hf_tags ? JSON.parse(row.hf_tags as string) as string[] : undefined,
-    hfBenchmarkScores: row.hf_benchmark_scores
-      ? JSON.parse(row.hf_benchmark_scores as string) as Record<string, number>
-      : undefined,
-    hfInferenceProviders: row.hf_inference_providers
-      ? JSON.parse(row.hf_inference_providers as string) as string[]
-      : undefined,
-    lastRefreshedAt: row.last_refreshed_at as number,
-    createdAt: row.created_at as number,
-    enabled: (row.enabled as number) === 1,
-  };
-}
-
-export class ModelRegistry {
-  private db: DatabaseSync;
-
-  constructor(dbPath: string) {
-    this.db = openDatabase(dbPath);
-  }
-
-  upsertModels(models: DiscoveredModel[]): void {
-    const stmt = this.db.prepare(`
-      INSERT OR REPLACE INTO discovered_models (
-        id, provider_source, model_id, display_name, context_length,
-        supports_vision, supports_tools, supports_streaming, supports_image_gen,
-        cost_per_1k_input, cost_per_1k_output, strengths, raw_metadata,
-        hf_model_card, hf_downloads, hf_likes, hf_trending_score,
-        hf_tags, hf_benchmark_scores, hf_inference_providers,
-        last_refreshed_at, created_at, enabled
-      ) VALUES (
-        ?, ?, ?, ?, ?,
-        ?, ?, ?, ?,
-        ?, ?, ?, ?,
-        ?, ?, ?, ?,
-        ?, ?, ?,
-        ?, ?, ?
-      )
-    `);
-
-    for (const m of models) {
-      stmt.run(
-        m.id, m.providerSource, m.modelId, m.displayName, m.contextLength,
-        m.supportsVision ? 1 : 0, m.supportsTools ? 1 : 0,
-        m.supportsStreaming ? 1 : 0, m.supportsImageGen ? 1 : 0,
-        m.costPer1kInput, m.costPer1kOutput,
-        JSON.stringify(m.strengths), m.rawMetadata ?? null,
-        m.hfModelCard ?? null, m.hfDownloads ?? null,
-        m.hfLikes ?? null, m.hfTrendingScore ?? null,
-        m.hfTags ? JSON.stringify(m.hfTags) : null,
-        m.hfBenchmarkScores ? JSON.stringify(m.hfBenchmarkScores) : null,
-        m.hfInferenceProviders ? JSON.stringify(m.hfInferenceProviders) : null,
-        m.lastRefreshedAt, m.createdAt,
-        m.enabled ? 1 : 0,
-      );
-    }
-  }
-
-  getModels(options?: ModelSearchOptions): DiscoveredModel[] {
-    const conditions: string[] = [];
-    const params: unknown[] = [];
-
-    if (options?.source) {
-      conditions.push('provider_source = ?');
-      params.push(options.source);
-    }
-    if (options?.query) {
-      conditions.push('(display_name LIKE ? OR model_id LIKE ?)');
-      const q = `%${options.query}%`;
-      params.push(q, q);
-    }
-    if (options?.supportsVision !== undefined) {
-      conditions.push('supports_vision = ?');
-      params.push(options.supportsVision ? 1 : 0);
-    }
-    if (options?.supportsTools !== undefined) {
-      conditions.push('supports_tools = ?');
-      params.push(options.supportsTools ? 1 : 0);
-    }
-    if (options?.enabled !== undefined) {
-      conditions.push('enabled = ?');
-      params.push(options.enabled ? 1 : 0);
-    }
-
-    const where = conditions.length > 0 ? `WHERE ${conditions.join(' AND ')}` : '';
-    const limit = options?.limit ?? 200;
-    const offset = options?.offset ?? 0;
-
-    const rows = this.db.prepare(
-      `SELECT * FROM discovered_models ${where} ORDER BY display_name ASC LIMIT ? OFFSET ?`,
-    ).all(...params, limit, offset) as Row[];
-
-    return rows.map(rowToModel);
-  }
-
-  getModel(id: string): DiscoveredModel | undefined {
-    const row = this.db.prepare(
-      'SELECT * FROM discovered_models WHERE id = ?',
-    ).get(id) as Row | undefined;
-
-    return row ? rowToModel(row) : undefined;
-  }
-
-  /**
-   * Convert discovered models to the format ModelRouter expects.
-   * Returns Record<modelId, ModelCapabilities>.
-   */
-  toModelCapabilities(source: string): Record<string, ModelCapabilitiesLike> {
-    const models = this.getModels({ source, enabled: true });
-    const result: Record<string, ModelCapabilitiesLike> = {};
-
-    for (const m of models) {
-      result[m.modelId] = {
-        maxContextTokens: m.contextLength,
-        supportsVision: m.supportsVision,
-        supportsTools: m.supportsTools,
-        supportsStreaming: m.supportsStreaming,
-        supportsImageGen: m.supportsImageGen,
-        costPer1kInput: m.costPer1kInput,
-        costPer1kOutput: m.costPer1kOutput,
-        strengths: m.strengths,
-        isLocal: false,
-      };
-    }
-
-    return result;
-  }
-
-  search(query: string): DiscoveredModel[] {
-    return this.getModels({ query, enabled: true });
-  }
-
-  getTrending(limit = 20): DiscoveredModel[] {
-    const rows = this.db.prepare(
-      `SELECT * FROM discovered_models
-       WHERE hf_trending_score IS NOT NULL AND enabled = 1
-       ORDER BY hf_trending_score DESC
-       LIMIT ?`,
-    ).all(limit) as Row[];
-
-    return rows.map(rowToModel);
-  }
-
-  pruneStale(maxAgeMs: number): number {
-    const cutoff = Date.now() - maxAgeMs;
-    const result = this.db.prepare(
-      'DELETE FROM discovered_models WHERE last_refreshed_at < ?',
-    ).run(cutoff);
-    return result.changes;
-  }
-
-  setEnabled(id: string, enabled: boolean): void {
-    this.db.prepare(
-      'UPDATE discovered_models SET enabled = ? WHERE id = ?',
-    ).run(enabled ? 1 : 0, id);
-  }
-}

package/src/types.ts

@@ -1,58 +0,0 @@
-/**
- * A model discovered from an external provider API (OpenRouter, HuggingFace, etc.).
- */
-export interface DiscoveredModel {
-  /** Composite key: `${providerSource}:${modelId}` */
-  id: string;
-  /** Source provider that discovered this model */
-  providerSource: string;
-  /** Original model ID from the external API */
-  modelId: string;
-  displayName: string;
-  contextLength: number;
-  supportsVision: boolean;
-  supportsTools: boolean;
-  supportsStreaming: boolean;
-  supportsImageGen: boolean;
-  costPer1kInput: number;
-  costPer1kOutput: number;
-  strengths: string[];
-  rawMetadata?: string;
-  /** HuggingFace-specific fields */
-  hfModelCard?: string;
-  hfDownloads?: number;
-  hfLikes?: number;
-  hfTrendingScore?: number;
-  hfTags?: string[];
-  hfBenchmarkScores?: Record<string, number>;
-  hfInferenceProviders?: string[];
-  lastRefreshedAt: number;
-  createdAt: number;
-  enabled: boolean;
-}
-
-export interface ModelSearchOptions {
-  source?: string;
-  query?: string;
-  supportsVision?: boolean;
-  supportsTools?: boolean;
-  enabled?: boolean;
-  limit?: number;
-  offset?: number;
-}
-
-/**
- * Structural type matching ModelCapabilities from @auxiora/providers
- * to avoid a hard dependency on that package.
- */
-export interface ModelCapabilitiesLike {
-  maxContextTokens: number;
-  supportsVision: boolean;
-  supportsTools: boolean;
-  supportsStreaming: boolean;
-  supportsImageGen: boolean;
-  costPer1kInput: number;
-  costPer1kOutput: number;
-  strengths: string[];
-  isLocal: boolean;
-}

package/tests/registry.test.ts

@@ -1,239 +0,0 @@
-import { describe, it, expect, beforeEach } from 'vitest';
-import { ModelRegistry } from '../src/registry.js';
-import type { DiscoveredModel } from '../src/types.js';
-
-function makeModel(overrides: Partial<DiscoveredModel> = {}): DiscoveredModel {
-  const now = Date.now();
-  return {
-    id: 'openrouter:test/model-1',
-    providerSource: 'openrouter',
-    modelId: 'test/model-1',
-    displayName: 'Test Model 1',
-    contextLength: 128000,
-    supportsVision: true,
-    supportsTools: true,
-    supportsStreaming: true,
-    supportsImageGen: false,
-    costPer1kInput: 0.003,
-    costPer1kOutput: 0.015,
-    strengths: ['reasoning', 'code'],
-    lastRefreshedAt: now,
-    createdAt: now,
-    enabled: true,
-    ...overrides,
-  };
-}
-
-describe('ModelRegistry', () => {
-  let registry: ModelRegistry;
-
-  beforeEach(() => {
-    registry = new ModelRegistry(':memory:');
-  });
-
-  describe('upsertModels', () => {
-    it('should insert models and retrieve them', () => {
-      const model = makeModel();
-      registry.upsertModels([model]);
-
-      const result = registry.getModel('openrouter:test/model-1');
-      expect(result).toBeDefined();
-      expect(result!.displayName).toBe('Test Model 1');
-      expect(result!.supportsVision).toBe(true);
-      expect(result!.strengths).toEqual(['reasoning', 'code']);
-    });
-
-    it('should upsert (update on conflict)', () => {
-      const model = makeModel();
-      registry.upsertModels([model]);
-
-      const updated = makeModel({ displayName: 'Updated Name', costPer1kInput: 0.005 });
-      registry.upsertModels([updated]);
-
-      const result = registry.getModel('openrouter:test/model-1');
-      expect(result!.displayName).toBe('Updated Name');
-      expect(result!.costPer1kInput).toBe(0.005);
-    });
-
-    it('should insert multiple models at once', () => {
-      const models = [
-        makeModel({ id: 'openrouter:a/model-a', modelId: 'a/model-a', displayName: 'Model A' }),
-        makeModel({ id: 'openrouter:b/model-b', modelId: 'b/model-b', displayName: 'Model B' }),
-        makeModel({ id: 'openrouter:c/model-c', modelId: 'c/model-c', displayName: 'Model C' }),
-      ];
-      registry.upsertModels(models);
-
-      const all = registry.getModels();
-      expect(all).toHaveLength(3);
-    });
-  });
-
-  describe('getModels', () => {
-    beforeEach(() => {
-      registry.upsertModels([
-        makeModel({ id: 'openrouter:a/big', modelId: 'a/big', displayName: 'Big Model', providerSource: 'openrouter', supportsVision: true }),
-        makeModel({ id: 'openrouter:a/small', modelId: 'a/small', displayName: 'Small Model', providerSource: 'openrouter', supportsVision: false }),
-        makeModel({ id: 'huggingface:b/chat', modelId: 'b/chat', displayName: 'Chat Model', providerSource: 'huggingface', supportsVision: false }),
-      ]);
-    });
-
-    it('should filter by source', () => {
-      const result = registry.getModels({ source: 'openrouter' });
-      expect(result).toHaveLength(2);
-      expect(result.every(m => m.providerSource === 'openrouter')).toBe(true);
-    });
-
-    it('should filter by vision support', () => {
-      const result = registry.getModels({ supportsVision: true });
-      expect(result).toHaveLength(1);
-      expect(result[0]!.displayName).toBe('Big Model');
-    });
-
-    it('should search by query', () => {
-      const result = registry.getModels({ query: 'chat' });
-      expect(result).toHaveLength(1);
-      expect(result[0]!.displayName).toBe('Chat Model');
-    });
-
-    it('should support limit and offset', () => {
-      const page1 = registry.getModels({ limit: 2, offset: 0 });
-      expect(page1).toHaveLength(2);
-
-      const page2 = registry.getModels({ limit: 2, offset: 2 });
-      expect(page2).toHaveLength(1);
-    });
-  });
-
-  describe('getModel', () => {
-    it('should return undefined for non-existent model', () => {
-      const result = registry.getModel('nonexistent');
-      expect(result).toBeUndefined();
-    });
-  });
-
-  describe('toModelCapabilities', () => {
-    it('should convert to ModelCapabilities format', () => {
-      registry.upsertModels([
-        makeModel({ id: 'openrouter:a/model', modelId: 'a/model', providerSource: 'openrouter' }),
-      ]);
-
-      const caps = registry.toModelCapabilities('openrouter');
-      expect(caps['a/model']).toBeDefined();
-      expect(caps['a/model']!.maxContextTokens).toBe(128000);
-      expect(caps['a/model']!.supportsVision).toBe(true);
-      expect(caps['a/model']!.isLocal).toBe(false);
-      expect(caps['a/model']!.strengths).toEqual(['reasoning', 'code']);
-    });
-
-    it('should exclude disabled models', () => {
-      registry.upsertModels([
-        makeModel({ id: 'openrouter:a/enabled', modelId: 'a/enabled', enabled: true }),
-        makeModel({ id: 'openrouter:a/disabled', modelId: 'a/disabled', enabled: false }),
-      ]);
-
-      const caps = registry.toModelCapabilities('openrouter');
-      expect(caps['a/enabled']).toBeDefined();
-      expect(caps['a/disabled']).toBeUndefined();
-    });
-  });
-
-  describe('search', () => {
-    it('should search by display name and model ID', () => {
-      registry.upsertModels([
-        makeModel({ id: 'or:anthropic/claude-3-opus', modelId: 'anthropic/claude-3-opus', displayName: 'Claude 3 Opus' }),
-        makeModel({ id: 'or:meta/llama-3', modelId: 'meta/llama-3', displayName: 'Llama 3' }),
-      ]);
-
-      const results = registry.search('claude');
-      expect(results).toHaveLength(1);
-      expect(results[0]!.displayName).toBe('Claude 3 Opus');
-    });
-  });
-
-  describe('getTrending', () => {
-    it('should return models sorted by trending score', () => {
-      registry.upsertModels([
-        makeModel({ id: 'hf:a', modelId: 'a', providerSource: 'huggingface', hfTrendingScore: 0.8 }),
-        makeModel({ id: 'hf:b', modelId: 'b', providerSource: 'huggingface', hfTrendingScore: 0.95 }),
-        makeModel({ id: 'hf:c', modelId: 'c', providerSource: 'huggingface', hfTrendingScore: 0.5 }),
-      ]);
-
-      const trending = registry.getTrending(2);
-      expect(trending).toHaveLength(2);
-      expect(trending[0]!.modelId).toBe('b');
-      expect(trending[1]!.modelId).toBe('a');
-    });
-
-    it('should exclude models without trending score', () => {
-      registry.upsertModels([
-        makeModel({ id: 'or:x', modelId: 'x', providerSource: 'openrouter' }),
-        makeModel({ id: 'hf:y', modelId: 'y', providerSource: 'huggingface', hfTrendingScore: 0.5 }),
-      ]);
-
-      const trending = registry.getTrending();
-      expect(trending).toHaveLength(1);
-      expect(trending[0]!.modelId).toBe('y');
-    });
-  });
-
-  describe('pruneStale', () => {
-    it('should delete models older than maxAge', () => {
-      const old = Date.now() - 10 * 24 * 60 * 60 * 1000; // 10 days ago
-      const recent = Date.now();
-
-      registry.upsertModels([
-        makeModel({ id: 'or:old', modelId: 'old', lastRefreshedAt: old }),
-        makeModel({ id: 'or:new', modelId: 'new', lastRefreshedAt: recent }),
-      ]);
-
-      const pruned = registry.pruneStale(7 * 24 * 60 * 60 * 1000); // 7 days
-      expect(pruned).toBe(1);
-
-      const remaining = registry.getModels();
-      expect(remaining).toHaveLength(1);
-      expect(remaining[0]!.modelId).toBe('new');
-    });
-  });
-
-  describe('setEnabled', () => {
-    it('should disable and re-enable a model', () => {
-      registry.upsertModels([makeModel()]);
-
-      registry.setEnabled('openrouter:test/model-1', false);
-      let model = registry.getModel('openrouter:test/model-1');
-      expect(model!.enabled).toBe(false);
-
-      registry.setEnabled('openrouter:test/model-1', true);
-      model = registry.getModel('openrouter:test/model-1');
-      expect(model!.enabled).toBe(true);
-    });
-  });
-
-  describe('HuggingFace-specific fields', () => {
-    it('should persist and retrieve HF metadata', () => {
-      registry.upsertModels([
-        makeModel({
-          id: 'hf:meta/llama',
-          modelId: 'meta/llama',
-          providerSource: 'huggingface',
-          hfModelCard: '# Llama\n\nA great model.',
-          hfDownloads: 1500000,
-          hfLikes: 5000,
-          hfTrendingScore: 0.92,
-          hfTags: ['transformers', 'pytorch', 'llama'],
-          hfBenchmarkScores: { mmlu: 0.82, humaneval: 0.71 },
-          hfInferenceProviders: ['together', 'fireworks-ai'],
-        }),
-      ]);
-
-      const model = registry.getModel('hf:meta/llama')!;
-      expect(model.hfModelCard).toBe('# Llama\n\nA great model.');
-      expect(model.hfDownloads).toBe(1500000);
-      expect(model.hfLikes).toBe(5000);
-      expect(model.hfTrendingScore).toBe(0.92);
-      expect(model.hfTags).toEqual(['transformers', 'pytorch', 'llama']);
-      expect(model.hfBenchmarkScores).toEqual({ mmlu: 0.82, humaneval: 0.71 });
-      expect(model.hfInferenceProviders).toEqual(['together', 'fireworks-ai']);
-    });
-  });
-});