From 68c8690132994f7b3c7fd94d78d34e13942963ab Mon Sep 17 00:00:00 2001 From: Maximilian Winter Date: Wed, 12 Jun 2024 15:35:34 +0200 Subject: [PATCH] Update messages_formatter.py --- src/llama_cpp_agent/messages_formatter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llama_cpp_agent/messages_formatter.py b/src/llama_cpp_agent/messages_formatter.py index 3b0b27b..3fadecb 100644 --- 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("", ""), }