feat: add manifest model catalog planner

shakkernerd · shakkernerd · commit dabdc779be8b · 2026-04-25T04:05:30.000+01:00
diff --git a/src/model-catalog/index.ts b/src/model-catalog/index.ts
@@ -8,6 +8,13 @@ export {
   normalizeModelCatalogProviderRows,
   normalizeModelCatalogRows,
 } from "./normalize.js";
+export { planManifestModelCatalogRows } from "./manifest-planner.js";
+export type {
+  ManifestModelCatalogPlan,
+  ManifestModelCatalogPlanEntry,
+  ManifestModelCatalogPlugin,
+  ManifestModelCatalogRegistry,
+} from "./manifest-planner.js";
 export type {
   ModelCatalog,
   ModelCatalogAlias,
diff --git a/src/model-catalog/manifest-planner.ts b/src/model-catalog/manifest-planner.ts
@@ -0,0 +1,94 @@
+import { normalizeModelCatalogProviderRows } from "./normalize.js";
+import { normalizeModelCatalogProviderId } from "./refs.js";
+import type { ModelCatalog, NormalizedModelCatalogRow } from "./types.js";
+
+export type ManifestModelCatalogPlugin = {
+  id: string;
+  modelCatalog?: Pick<ModelCatalog, "providers">;
+};
+
+export type ManifestModelCatalogRegistry = {
+  plugins: readonly ManifestModelCatalogPlugin[];
+};
+
+export type ManifestModelCatalogPlanEntry = {
+  pluginId: string;
+  provider: string;
+  rows: readonly NormalizedModelCatalogRow[];
+};
+
+export type ManifestModelCatalogPlan = {
+  rows: readonly NormalizedModelCatalogRow[];
+  entries: readonly ManifestModelCatalogPlanEntry[];
+};
+
+export function planManifestModelCatalogRows(params: {
+  registry: ManifestModelCatalogRegistry;
+  providerFilter?: string;
+}): ManifestModelCatalogPlan {
+  const providerFilter = params.providerFilter
+    ? normalizeModelCatalogProviderId(params.providerFilter)
+    : undefined;
+  const entries: ManifestModelCatalogPlanEntry[] = [];
+
+  for (const plugin of params.registry.plugins) {
+    for (const entry of planManifestModelCatalogPluginEntries({ plugin, providerFilter })) {
+      entries.push(entry);
+    }
+  }
+
+  const rows: NormalizedModelCatalogRow[] = [];
+  const seenMergeKeys = new Set<string>();
+  for (const entry of entries) {
+    for (const row of entry.rows) {
+      if (seenMergeKeys.has(row.mergeKey)) {
+        continue;
+      }
+      seenMergeKeys.add(row.mergeKey);
+      rows.push(row);
+    }
+  }
+
+  return {
+    entries,
+    rows: rows.toSorted(
+      (left, right) =>
+        left.provider.localeCompare(right.provider) || left.id.localeCompare(right.id),
+    ),
+  };
+}
+
+function planManifestModelCatalogPluginEntries(params: {
+  plugin: ManifestModelCatalogPlugin;
+  providerFilter: string | undefined;
+}): ManifestModelCatalogPlanEntry[] {
+  const providers = params.plugin.modelCatalog?.providers;
+  if (!providers) {
+    return [];
+  }
+
+  return Object.entries(providers).flatMap(([provider, providerCatalog]) => {
+    const normalizedProvider = normalizeModelCatalogProviderId(provider);
+    if (
+      !normalizedProvider ||
+      (params.providerFilter && normalizedProvider !== params.providerFilter)
+    ) {
+      return [];
+    }
+    const rows = normalizeModelCatalogProviderRows({
+      provider: normalizedProvider,
+      providerCatalog,
+      source: "manifest",
+    });
+    if (rows.length === 0) {
+      return [];
+    }
+    return [
+      {
+        pluginId: params.plugin.id,
+        provider: normalizedProvider,
+        rows,
+      },
+    ];
+  });
+}
