fix completion_text

This commit is contained in:
josc146
2023-05-21 23:25:58 +08:00
parent c40e57aafd
commit 74ceffb32c
2 changed files with 27 additions and 2 deletions

View File

@@ -42,9 +42,23 @@ async def completions(body: CompletionBody, request: Request):
completion_text = ""
for message in body.messages:
if message.role == "user":
completion_text += "Bob: " + message.content + "\n\n"
completion_text += (
"Bob: "
+ message.content.replace("\\n", "\n")
.replace("\r\n", "\n")
.replace("\n\n", "\n")
.strip()
+ "\n\n"
)
elif message.role == "assistant":
completion_text += "Alice: " + message.content + "\n\n"
completion_text += (
"Alice: "
+ message.content.replace("\\n", "\n")
.replace("\r\n", "\n")
.replace("\n\n", "\n")
.strip()
+ "\n\n"
)
completion_text += "Alice:"
async def eval_rwkv():