From f58b8dbbb7fbc08ffe8655d18bb32dda40c1b54f Mon Sep 17 00:00:00 2001
From: can1357 <me@can.ac>
Date: Thu, 23 Jul 2026 22:12:32 +0200
Subject: [PATCH] fix(catalog): include base Nova Lite/Micro/Pro IDs in Bedrock
 cache compat

AWS documents prompt caching for Nova Micro/Lite/Pro under their in-region
model IDs as well; the allowlist covered only the us. geo profiles (Premier
already had both forms). Regional/provisioned-throughput configs using base
IDs now emit the documented default-5m checkpoints.
---
 packages/catalog/src/compat/bedrock.ts             |  3 +++
 packages/catalog/test/bedrock-prompt-cache.test.ts | 10 ++++++++--
 2 files changed, 11 insertions(+), 2 deletions(-)

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);
 		}
 	});