浏览代码

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年前
父节点
当前提交
eb20b60b13
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 24 次插入1 次删除
  1. 24
    1
      api/apps/conversation_app.py

+ 24
- 1
api/apps/conversation_app.py 查看文件

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


ConversationService.update_by_id(conv["id"], conv) ConversationService.update_by_id(conv["id"], conv)
return get_json_result(data=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)

正在加载...
取消
保存