From b398412ad04d6305404c729b5182da3a5792448c Mon Sep 17 00:00:00 2001
From: Charlie Ruan <53290280+CharlieFRuan@users.noreply.github.com>
Date: Fri, 20 Sep 2024 10:41:42 -0400
Subject: [PATCH] Trivial

---
 tests/conversation.test.ts | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/tests/conversation.test.ts b/tests/conversation.test.ts
index 14dbf815..bdb42bf7 100644
--- 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);
-
-  // });
 });