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

aws-amplify · Dec 9, 2024 · 5945ba9 · 5945ba9
1 parent 2a9282b
commit 5945ba9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
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',