Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-26 01:30:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix the impersonate response role

Browse Source
This commit is contained in:
Yiximail 2024-08-22 16:26:41 +08:00
parent 34b493b231
commit 38b3daad55
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
extensions/openai/completions.py
View File

@ -371,6 +371,7 @@ def chat_completions_common(body: dict, is_legacy: bool = False, stream=False, p
yield chunk yield chunk
else: else:
role = 'user' if impersonate else 'assistant'
resp = { resp = {
"id": cmpl_id, "id": cmpl_id,
"object": object_type, "object": object_type,
@ -379,7 +380,7 @@ def chat_completions_common(body: dict, is_legacy: bool = False, stream=False, p
resp_list: [{ resp_list: [{
"index": 0, "index": 0,
"finish_reason": stop_reason, "finish_reason": stop_reason,
"message": {"role": "assistant", "content": answer} "message": {"role": role, "content": answer}
}], }],
"usage": { "usage": {
"prompt_tokens": token_count, "prompt_tokens": token_count,