|
|
|
@@ -90,17 +90,18 @@ def completion(tenant_id, chat_id, question, name="New session", session_id=None |
|
|
|
"user_id": kwargs.get("user_id", "") |
|
|
|
} |
|
|
|
ConversationService.save(**conv) |
|
|
|
yield "data:" + json.dumps({"code": 0, "message": "", |
|
|
|
"data": { |
|
|
|
"answer": conv["message"][0]["content"], |
|
|
|
"reference": {}, |
|
|
|
"audio_binary": None, |
|
|
|
"id": None, |
|
|
|
"session_id": session_id |
|
|
|
}}, |
|
|
|
ensure_ascii=False) + "\n\n" |
|
|
|
yield "data:" + json.dumps({"code": 0, "message": "", "data": True}, ensure_ascii=False) + "\n\n" |
|
|
|
return |
|
|
|
if stream: |
|
|
|
yield "data:" + json.dumps({"code": 0, "message": "", |
|
|
|
"data": { |
|
|
|
"answer": conv["message"][0]["content"], |
|
|
|
"reference": {}, |
|
|
|
"audio_binary": None, |
|
|
|
"id": None, |
|
|
|
"session_id": session_id |
|
|
|
}}, |
|
|
|
ensure_ascii=False) + "\n\n" |
|
|
|
yield "data:" + json.dumps({"code": 0, "message": "", "data": True}, ensure_ascii=False) + "\n\n" |
|
|
|
return |
|
|
|
|
|
|
|
conv = ConversationService.query(id=session_id, dialog_id=chat_id) |
|
|
|
if not conv: |