Update messages_formatter.py

Maximilian-Winter · Jun 12, 2024 · 68c8690 · 68c8690
1 parent a45f852
commit 68c8690
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/llama_cpp_agent/messages_formatter.py b/src/llama_cpp_agent/messages_formatter.py
@@ -145,7 +145,7 @@ def _format_response(
 vicuna_prompt_markers = {
     Roles.system: PromptMarkers("", """\n\n"""),
     Roles.user: PromptMarkers("""USER: """, """\n"""),
-    Roles.assistant: PromptMarkers("""ASSISTANT:""", ""),
+    Roles.assistant: PromptMarkers("""ASSISTANT:""", "\n"),
     Roles.tool: PromptMarkers("", ""),
 }