Trivial

mlc-ai · Sep 20, 2024 · b398412 · b398412
1 parent 19c9991
commit b398412
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/tests/conversation.test.ts b/tests/conversation.test.ts
@@ -257,12 +257,16 @@ describe("Test getConversationFromChatCompletionRequest with image", () => {
       ],
       `<|assistant|>\n`,
     ]);
+    expect(conv1.getPromptArrayLastRound()).toEqual([
+      [
+        `<|user|>\n`,
+        { url: imageUrl2 } as ImageURL,
+        `\n`,
+        { url: imageUrl3 } as ImageURL,
+        `\n`,
+        `${dummyRequestStr}<|end|>\n`,
+      ],
+      `<|assistant|>\n`,
+    ]);
   });
-
-  // test("Test compareConversationObject with image message", () => {
-  //   const config_json = JSON.parse(phi3_5VisionChatConfigJSONString);
-  //   const config = { ...config_json } as ChatConfig;
-  //   const conversation = getConversation(config.conv_template);
-
-  // });
 });