chore(ai): add nova models (#408)

atierian · web-flow · commit 5945ba924581 · 2024-12-09T14:02:55.000-05:00
diff --git a/.changeset/giant-tables-battle.md 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
@@ -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',

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +---
 +"@aws-amplify/data-schema": patch
 +---
++
 +add Amazon Nova Micro, Lite, and Pro models