fix: fix tests

llm-modules/prompt-script/src/test/kotlin/cc/unitmesh/prompt/executor/TemplateRoleSplitterTest.kt

@@ -22,6 +22,38 @@ class TemplateRoleSplitterTest {
  }
 
 
+ @Test
+ fun should_splitInputIntoSections() {
+ // given
+ val input = "```system```\nYou are a helpful assistant.\n\n```user```\n${'$'}{question}\n"
+ val expectedSections = mapOf(
+ "system" to "You are a helpful assistant.\n\n",
+ "user" to "${'$'}{question}\n\n"
+ )
+
+ // when
+ val sections = TemplateRoleSplitter().split(input)
+
+ // then
+ assertEquals(expectedSections, sections)
+ }
+
+ @Test
+ fun should_handleRemainingContentAfterLastSection() {
+ // given
+ val input = "```system```\nYou are a helpful assistant.\n\n```user```\n${'$'}{question}\nRemaining content"
+ val expectedSections = mapOf(
+ "system" to "You are a helpful assistant.\n\n",
+ "user" to "${'$'}{question}\nRemaining content\n"
+ )
+
+ // when
+ val sections = TemplateRoleSplitter().split(input)
+
+ // then
+ assertEquals(expectedSections, sections)
+ }
+
  @Test
  fun `should split input into sections`() {
  // given