lm-lib refactoring;

git-svn-id: svn://home0/lm-lib@3610 71dcc7fd-b1f2-4e5d-bb8e-274146e81173

src/main/java/de/extio/lmlib/prompt/ChatMLPromptStrategy.java

@@ -0,0 +1,49 @@
+package de.extio.lmlib.prompt;
+
+import org.springframework.stereotype.Component;
+
+@Component
+public class ChatMLPromptStrategy implements PromptStrategy {
+	
+	@Override
+	public StringBuilder start(final String system, final String question, final String text) {
+		final StringBuilder prompt = new StringBuilder();
+		prompt.append("<|im_start|>");
+		if (!system.isEmpty()) {
+			prompt.append("system\n");
+			prompt.append(system);
+			prompt.append("<|im_end|>\n");
+		}
+		prompt.append("<|im_start|>user\n");
+		prompt.append(question);
+		if (!question.isEmpty() && !text.isEmpty()) {
+			prompt.append("\n");
+		}
+		prompt.append(text);
+		prompt.append("<|im_end|>\n<|im_start|>assistant\n");
+		return prompt;
+	}
+	
+	@Override
+	public void continue_(final StringBuilder prompt, final String assistant) {
+		this.next(prompt, assistant, "Continue");
+	}
+	
+	@Override
+	public void next(final StringBuilder prompt, final String assistant, final String user) {
+		prompt.append(assistant);
+		prompt.append("<|im_end|>\n<|im_start|>user\n");
+		prompt.append(user);
+		prompt.append("<|im_end|>\n<|im_start|>assistant\n");
+	}
+	
+	@Override
+	public String removeEOT(final String prompt) {
+		return prompt.strip().replace("<|im_end|>", "");
+	}
+	
+	@Override
+	public String getPromptName() {
+		return "chatml";
+	}
+}