From d551d2508c4c2da7b8affaec913915019fa548f7 Mon Sep 17 00:00:00 2001
From: Ian Saultz <52051793+atierian@users.noreply.github.com>
Date: Wed, 4 Dec 2024 09:44:33 -0800
Subject: [PATCH] chore(ai): add nova models

---
 .changeset/giant-tables-battle.md        | 5 +++++
 packages/data-schema/src/ai/ModelType.ts | 4 ++++
 2 files changed, 9 insertions(+)
 create mode 100644 .changeset/giant-tables-battle.md

diff --git a/.changeset/giant-tables-battle.md b/.changeset/giant-tables-battle.md
new file mode 100644
index 000000000..ef0c32e88
--- /dev/null
+++ b/.changeset/giant-tables-battle.md
@@ -0,0 +1,5 @@
+---
+"@aws-amplify/data-schema": patch
+---
+
+add Amazon Nova Micro, Lite, and Pro models
diff --git a/packages/data-schema/src/ai/ModelType.ts b/packages/data-schema/src/ai/ModelType.ts
index 6819d4133..e96c66f58 100644
--- a/packages/data-schema/src/ai/ModelType.ts
+++ b/packages/data-schema/src/ai/ModelType.ts
@@ -4,6 +4,10 @@
 import { AiModel } from '@aws-amplify/data-schema-types';
 
 const supportedModelsLookup = {
+  // Amazon models
+  'Amazon Nova Pro': 'amazon.nova-pro-v1:0',
+  'Amazon Nova Lite': 'amazon.nova-lite-v1:0',
+  'Amazon Nova Micro': 'amazon.nova-micro-v1:0',
   // Anthropic models
   'Claude 3 Haiku': 'anthropic.claude-3-haiku-20240307-v1:0',
   'Claude 3 Opus': 'anthropic.claude-3-opus-20240229-v1:0',