starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 16:35:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

add exception

Browse Source
This commit is contained in:
Junyang Lin
2023-08-13 12:54:53 +08:00
committed by GitHub
parent f91e7b3c87
commit c84d2f685a
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
openai_api.py
View File

@@ -113,6 +113,10 @@ async def create_chat_completion(request: ChatCompletionRequest):
for i in range(0, len(prev_messages), 2): for i in range(0, len(prev_messages), 2):
if prev_messages[i].role == "user" and prev_messages[i+1].role == "assistant": if prev_messages[i].role == "user" and prev_messages[i+1].role == "assistant":
history.append([prev_messages[i].content, prev_messages[i+1].content]) history.append([prev_messages[i].content, prev_messages[i+1].content])
else:
raise HTTPException(status_code=400, detail="Invalid request.")
else:
raise HTTPException(status_code=400, detail="Invalid request.")
if request.stream: if request.stream:
generate = predict(query, history, request.model) generate = predict(query, history, request.model)