add colon two bug fix

vtuber-plan · Dec 27, 2023 · 00281b8 · 00281b8
1 parent 851046f
commit 00281b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chatproto/conversation/history.py b/chatproto/conversation/history.py
@@ -37,7 +37,7 @@ def create_add_colon_two(settings: ConversationSettings, system: Optional[str],
     for i, (role, message) in enumerate(messages):
         if message:
             section = role + ": " + message + seps[i % 2]
-            prefix = ret + ": "
+            prefix = ret + role + ": "
             indices.append((len(prefix), len(prefix) + len(message)))
         else:
             section = role + ":"