From 783131d89b0687b9a36f6c30d5212a6b860c12dc Mon Sep 17 00:00:00 2001
From: shintaok <shintaok@amazon.co.jp>
Date: Wed, 24 Jul 2024 23:54:23 +0000
Subject: [PATCH] add mistral-large 2407

---
 docs/DEPLOY_OPTION.md               | 9 +++++----
 packages/cdk/lambda/utils/models.ts | 8 ++++++++
 packages/cdk/lib/construct/api.ts   | 1 +
 3 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/docs/DEPLOY_OPTION.md b/docs/DEPLOY_OPTION.md
index 47c53fe1..3447b341 100644
--- 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
index 3969fdea..b669c825 100644
--- 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
index 809d1745..1e76611d 100644
--- 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',
     ];