From 53ae135f8d7f9d1c5a1f779dfd273add68939a46 Mon Sep 17 00:00:00 2001
From: eastriver <eastriver0720@gmail.com>
Date: Sat, 8 Jun 2024 20:11:53 +0900
Subject: [PATCH] fix test

---
 Tests/LLMTests/LLMTests.swift | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/Tests/LLMTests/LLMTests.swift b/Tests/LLMTests/LLMTests.swift
index 12979e5..b28b889 100644
--- a/Tests/LLMTests/LLMTests.swift
+++ b/Tests/LLMTests/LLMTests.swift
@@ -104,7 +104,7 @@ final class LLMTests: XCTestCase {
     func testLLaMaPreProcessorWithoutSystemMessage() throws {
         let template = Template.llama()
         let expected = """
-        <s>[INST] \(userPrompt) [/INST]
+        [INST] \(userPrompt) [/INST]
         """
         let output = template.preprocess(userPrompt, [])
         #assert(expected == output)
@@ -113,7 +113,7 @@ final class LLMTests: XCTestCase {
     func testLLaMaPreProcessorWithoutHistory() throws {
         let template = Template.llama(systemPrompt)
         let expected = """
-        <s>[INST] <<SYS>>
+        [INST] <<SYS>>
         \(systemPrompt)
         <</SYS>>
 
@@ -126,7 +126,7 @@ final class LLMTests: XCTestCase {
     func testLLaMaPreProcessorWithHistory() throws {
         let template = Template.llama(systemPrompt)
         let expected = """
-        <s>[INST] <<SYS>>
+        [INST] <<SYS>>
         \(systemPrompt)
         <</SYS>>
 
@@ -139,7 +139,7 @@ final class LLMTests: XCTestCase {
     func testMistralPreProcessorWithoutHistory() throws {
         let template = Template.mistral
         let expected = """
-        <s>[INST] \(userPrompt) [/INST]
+        [INST] \(userPrompt) [/INST]
         """
         let output = template.preprocess(userPrompt, [])
         #assert(expected == output)
@@ -148,7 +148,7 @@ final class LLMTests: XCTestCase {
     func testMistralPreProcessorWithHistory() throws {
         let template = Template.mistral
         let expected = """
-        <s>[INST] \(history[0].content) [/INST]\(history[1].content)</s> [INST] \(userPrompt) [/INST]
+        [INST] \(history[0].content) [/INST]\(history[1].content)</s> [INST] \(userPrompt) [/INST]
         """
         let output = template.preprocess(userPrompt, history)
         #assert(expected == output)