add mistral-large 2407

aws-samples · Jul 24, 2024 · 783131d · 783131d
1 parent 0fb949e
commit 783131d
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/docs/DEPLOY_OPTION.md b/docs/DEPLOY_OPTION.md
@@ -269,15 +269,16 @@ Knowledge base プロンプト例: キーワードで検索し情報を取得し
 "anthropic.claude-3-sonnet-20240229-v1:0",
 "anthropic.claude-3-haiku-20240307-v1:0",
 "amazon.titan-text-premier-v1:0",
-"meta.llama3-1-8b-instruct-v1:0",
-"meta.llama3-1-70b-instruct-v1:0",
 "meta.llama3-1-405b-instruct-v1:0",
+"meta.llama3-1-70b-instruct-v1:0",
+"meta.llama3-1-8b-instruct-v1:0",
 "meta.llama3-70b-instruct-v1:0",
 "meta.llama3-8b-instruct-v1:0",
 "cohere.command-r-plus-v1:0",
 "cohere.command-r-v1:0",
-"mistral.mistral-small-2402-v1:0",
+"mistral.mistral-large-2407-v1:0",
 "mistral.mistral-large-2402-v1:0",
+"mistral.mistral-small-2402-v1:0",
 "anthropic.claude-v2:1",
 "anthropic.claude-v2",
 "anthropic.claude-instant-v1",
@@ -330,7 +331,7 @@ Knowledge base プロンプト例: キーワードで検索し情報を取得し
     "meta.llama3-1-8b-instruct-v1:0",
     "cohere.command-r-plus-v1:0",
     "cohere.command-r-v1:0",
-    "mistral.mistral-large-2402-v1:0"
+    "mistral.mistral-large-2407-v1:0"
   ],
   "imageGenerationModelIds": [
     "amazon.titan-image-generator-v1",

diff --git a/packages/cdk/lambda/utils/models.ts b/packages/cdk/lambda/utils/models.ts
@@ -603,6 +603,14 @@ export const BEDROCK_TEXT_GEN_MODELS: {
     extractConverseOutputText: extractConverseOutputText,
     extractConverseStreamOutputText: extractConverseStreamOutputText,
   },
+  'mistral.mistral-large-2407-v1:0': {
+    defaultParams: MISTRAL_DEFAULT_PARAMS,
+    usecaseParams: USECASE_DEFAULT_PARAMS,
+    createConverseCommandInput: createConverseCommandInput,
+    createConverseStreamCommandInput: createConverseStreamCommandInput,
+    extractConverseOutputText: extractConverseOutputText,
+    extractConverseStreamOutputText: extractConverseStreamOutputText,
+  },
   'cohere.command-r-v1:0': {
     defaultParams: COMMANDR_DEFAULT_PARAMS,
     usecaseParams: USECASE_DEFAULT_PARAMS,

diff --git a/packages/cdk/lib/construct/api.ts b/packages/cdk/lib/construct/api.ts
@@ -86,6 +86,7 @@ export class Api extends Construct {
       'mistral.mixtral-8x7b-instruct-v0:1',
       'mistral.mistral-small-2402-v1:0',
       'mistral.mistral-large-2402-v1:0',
+      'mistral.mistral-large-2407-v1:0',
       'cohere.command-r-v1:0',
       'cohere.command-r-plus-v1:0',
     ];