From 8c763cf3e569cad9b2afcab70fbdf181957bbf7c Mon Sep 17 00:00:00 2001
From: Ben Perlmutter <mongodben@mongodb.com>
Date: Wed, 21 May 2025 11:40:55 -0400
Subject: [PATCH] new relevant bedrock models

---
 .../src/models/models.test.ts                 |  2 +-
 .../mongodb-rag-core/src/models/models.ts     | 32 +++++++++++++++++++
 2 files changed, 33 insertions(+), 1 deletion(-)

diff --git a/packages/mongodb-rag-core/src/models/models.test.ts b/packages/mongodb-rag-core/src/models/models.test.ts
index 57c1baf5d..678c963af 100644
--- a/packages/mongodb-rag-core/src/models/models.test.ts
+++ b/packages/mongodb-rag-core/src/models/models.test.ts
@@ -15,7 +15,7 @@ jest.setTimeout(60000);
 // NOTE: due to this issue https://github.com/nodejs/node/issues/39964,
 // you must run the tests with a Node version >= 20.0.0
 
-describe.skip("Braintrust models", () => {
+describe("Braintrust models", () => {
   test.each(models.filter((m) => m.provider === "braintrust"))(
     "'$label' model should generate data",
     async (model) => {
diff --git a/packages/mongodb-rag-core/src/models/models.ts b/packages/mongodb-rag-core/src/models/models.ts
index 275a6fe03..790ab277d 100644
--- a/packages/mongodb-rag-core/src/models/models.ts
+++ b/packages/mongodb-rag-core/src/models/models.ts
@@ -213,6 +213,14 @@ const allModels = [
     provider: "braintrust",
     authorized: true,
   },
+  {
+    label: "pixtral-large-25-02",
+    deployment: "us.mistral.pixtral-large-2502-v1:0",
+    developer: "Mistral",
+    maxConcurrency: 3,
+    provider: "braintrust",
+    authorized: true,
+  },
   {
     label: "llama-3.1-70b",
     deployment: "us.meta.llama3-1-70b-instruct-v1:0",
@@ -237,6 +245,22 @@ const allModels = [
     provider: "braintrust",
     authorized: true,
   },
+  {
+    label: "llama-4-scout-17b",
+    deployment: "us.meta.llama4-scout-17b-instruct-v1:0",
+    developer: "Meta",
+    maxConcurrency: 5,
+    provider: "braintrust",
+    authorized: true,
+  },
+  {
+    label: "llama-4-maverick-17b",
+    deployment: "us.meta.llama4-maverick-17b-instruct-v1:0",
+    developer: "Meta",
+    maxConcurrency: 5,
+    provider: "braintrust",
+    authorized: true,
+  },
   {
     label: "nova-lite-v1:0",
     deployment: "amazon.nova-lite-v1:0",
@@ -261,6 +285,14 @@ const allModels = [
     authorized: true,
     maxConcurrency: 30,
   },
+  {
+    label: "nova-premier-v1:0",
+    deployment: "us.amazon.nova-premier-v1:0",
+    developer: "Amazon",
+    provider: "braintrust",
+    maxConcurrency: 10,
+    authorized: true,
+  },
   {
     label: "gemini-1.5-flash-002",
     deployment: "google/gemini-1.5-flash-002",