Update messages_formatter.py

Maximilian-Winter · Maximilian-Winter · commit 68c869013299 · 2024-06-12T15:35:34.000+02:00
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("", ""),
 }
 

Original file line number	Diff line number	Diff line change
`@@ -145,7 +145,7 @@ def _format_response(`
`145`	`145`	`vicuna_prompt_markers = {`
`146`	`146`	`Roles.system: PromptMarkers("", """\n\n"""),`
`147`	`147`	`Roles.user: PromptMarkers("""USER: """, """\n"""),`
`148`		`- Roles.assistant: PromptMarkers("""ASSISTANT:""", ""),`
	`148`	`+ Roles.assistant: PromptMarkers("""ASSISTANT:""", "\n"),`
`149`	`149`	`Roles.tool: PromptMarkers("", ""),`
`150`	`150`	`}`
`151`	`151`