Add Granite 3.1 8B to chatbot

ansible · Jan 13, 2025 · b42bb42 · b42bb42
1 parent 6a65fc3
commit b42bb42
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/ansible_ai_connect_chatbot/src/App.test.tsx b/ansible_ai_connect_chatbot/src/App.test.tsx
@@ -402,7 +402,10 @@ test("Debug mode test", async () => {
   await expect
     .element(view.getByRole("menuitem", { name: "granite3-8b" }))
     .toBeTruthy();
-  await view.getByRole("menuitem", { name: "granite3-8b" }).click();
+  await expect
+    .element(view.getByRole("menuitem", { name: "granite31-8b" }))
+    .toBeTruthy();
+  await view.getByRole("menuitem", { name: "granite31-8b" }).click();
 
   await sendMessage("Hello");
   await expect

diff --git a/ansible_ai_connect_chatbot/src/useChatbot/useChatbot.ts b/ansible_ai_connect_chatbot/src/useChatbot/useChatbot.ts
@@ -25,6 +25,7 @@ const botName =
 
 export const modelsSupported: LLMModel[] = [
   { model: "granite3-8b", provider: "my_rhoai_g3" },
+  { model: "granite31-8b", provider: "my_rhoai_g31" },
 ];
 
 export const readCookie = (name: string): string | null => {