diff --git a/packages/catalog/src/compat/bedrock.ts b/packages/catalog/src/compat/bedrock.ts index 4d61496e8..bee4f14d7 100644 --- a/packages/catalog/src/compat/bedrock.ts +++ b/packages/catalog/src/compat/bedrock.ts @@ -52,8 +52,11 @@ function detectedBedrockCompat(modelId: string): ResolvedBedrockCompat { const id = modelId.toLowerCase(); if ( + id === "amazon.nova-lite-v1:0" || id === "us.amazon.nova-lite-v1:0" || + id === "amazon.nova-micro-v1:0" || id === "us.amazon.nova-micro-v1:0" || + id === "amazon.nova-pro-v1:0" || id === "us.amazon.nova-pro-v1:0" || id === "amazon.nova-premier-v1:0" || id === "us.amazon.nova-premier-v1:0" diff --git a/packages/catalog/test/bedrock-prompt-cache.test.ts b/packages/catalog/test/bedrock-prompt-cache.test.ts index 2e65a318d..d9e1b5942 100644 --- a/packages/catalog/test/bedrock-prompt-cache.test.ts +++ b/packages/catalog/test/bedrock-prompt-cache.test.ts @@ -111,8 +111,14 @@ describe("Bedrock prompt-cache compat", () => { expect(getBundledModel<"bedrock-converse-stream">("amazon-bedrock", id)?.compat).toEqual(expected); } - // AWS documents these as Nova Premier's in-region model and US geo inference IDs. - for (const id of ["amazon.nova-premier-v1:0", "us.amazon.nova-premier-v1:0"] as const) { + // AWS documents both the in-region model IDs and the US geo inference IDs. + for (const id of [ + "amazon.nova-lite-v1:0", + "amazon.nova-micro-v1:0", + "amazon.nova-pro-v1:0", + "amazon.nova-premier-v1:0", + "us.amazon.nova-premier-v1:0", + ] as const) { expect(buildModel(bedrockSpec({ id })).compat).toEqual(expected); } });