Correct Granite 3.1 model name (#1491)

ansible · Jan 13, 2025 · 845ffef · 845ffef
1 parent 8e9a67c
commit 845ffef
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ansible_ai_connect_chatbot/src/App.test.tsx b/ansible_ai_connect_chatbot/src/App.test.tsx
@@ -403,9 +403,9 @@ test("Debug mode test", async () => {
     .element(view.getByRole("menuitem", { name: "granite3-8b" }))
     .toBeTruthy();
   await expect
-    .element(view.getByRole("menuitem", { name: "granite31-8b" }))
+    .element(view.getByRole("menuitem", { name: "granite3-1-8b" }))
     .toBeTruthy();
-  await view.getByRole("menuitem", { name: "granite31-8b" }).click();
+  await view.getByRole("menuitem", { name: "granite3-1-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,7 +25,7 @@ const botName =
 
 export const modelsSupported: LLMModel[] = [
   { model: "granite3-8b", provider: "my_rhoai_g3" },
-  { model: "granite31-8b", provider: "my_rhoai_g31" },
+  { model: "granite3-1-8b", provider: "my_rhoai_g31" },
 ];
 
 export const readCookie = (name: string): string | null => {