Selaa lähdekoodia

add inferface for message thumbup (#2091)

### What problem does this PR solve?

#2088

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
tags/v0.11.0
Kevin Hu 1 vuosi sitten
vanhempi
commit
eb20b60b13
No account linked to committer's email address
1 muutettua tiedostoa jossa 24 lisäystä ja 1 poistoa
  1. 24
    1
      api/apps/conversation_app.py

+ 24
- 1
api/apps/conversation_app.py Näytä tiedosto

@@ -178,7 +178,7 @@ def completion():
@manager.route('/delete_msg', methods=['POST'])
@login_required
@validate_request("conversation_id", "message_id")
def completion():
def delete_msg():
req = request.json
e, conv = ConversationService.get_by_id(req["conversation_id"])
if not e:
@@ -196,3 +196,26 @@ def completion():

ConversationService.update_by_id(conv["id"], conv)
return get_json_result(data=conv)


@manager.route('/thumbup', methods=['POST'])
@login_required
@validate_request("conversation_id", "message_id")
def thumbup():
req = request.json
e, conv = ConversationService.get_by_id(req["conversation_id"])
if not e:
return get_data_error_result(retmsg="Conversation not found!")
up_down = req.get("set")
feedback = req.get("feedback", "")
conv = conv.to_dict()
for i, msg in enumerate(conv["message"]):
if req["message_id"] == msg.get("id", "") and msg.get("role", "") == "assistant":
if up_down: msg["thumbup"] = True
else:
msg["thumbup"] = False
msg["feedback"] = feedback
break

ConversationService.update_by_id(conv["id"], conv)
return get_json_result(data=conv)

Loading…
Peruuta
Tallenna