diff --git a/src/model-catalog/manifest-planner.test.ts b/src/model-catalog/manifest-planner.test.ts index 1535c2a484b..906c3b77d94 100644 --- a/src/model-catalog/manifest-planner.test.ts +++ b/src/model-catalog/manifest-planner.test.ts @@ -9,6 +9,9 @@ describe("manifest model catalog planner", () => { { id: "moonshot", modelCatalog: { + discovery: { + moonshot: "static", + }, providers: { Moonshot: { api: "openai-responses", @@ -32,6 +35,7 @@ describe("manifest model catalog planner", () => { { pluginId: "moonshot", provider: "moonshot", + discovery: "static", rows: [ { provider: "moonshot", diff --git a/src/model-catalog/manifest-planner.ts b/src/model-catalog/manifest-planner.ts index e0f0f42843e..f14b6d9b61d 100644 --- a/src/model-catalog/manifest-planner.ts +++ b/src/model-catalog/manifest-planner.ts @@ -1,10 +1,10 @@ import { normalizeModelCatalogProviderRows } from "./normalize.js"; import { normalizeModelCatalogProviderId } from "./refs.js"; -import type { ModelCatalog, NormalizedModelCatalogRow } from "./types.js"; +import type { ModelCatalog, ModelCatalogDiscovery, NormalizedModelCatalogRow } from "./types.js"; export type ManifestModelCatalogPlugin = { id: string; - modelCatalog?: Pick; + modelCatalog?: Pick; }; export type ManifestModelCatalogRegistry = { @@ -14,6 +14,7 @@ export type ManifestModelCatalogRegistry = { export type ManifestModelCatalogPlanEntry = { pluginId: string; provider: string; + discovery?: ModelCatalogDiscovery; rows: readonly NormalizedModelCatalogRow[]; }; @@ -113,6 +114,7 @@ function planManifestModelCatalogPluginEntries(params: { { pluginId: params.plugin.id, provider: normalizedProvider, + discovery: params.plugin.modelCatalog?.discovery?.[normalizedProvider], rows, }, ];