mirror of
https://github.com/langgenius/dify.git
synced 2025-07-23 17:40:22 +00:00
fix: admin feedback uses the same method create_feedback (#22580)
Co-authored-by: 刘江波 <jiangbo721@163.com>
This commit is contained in:
parent
460a825ef1
commit
ffee6f3288
@ -5,6 +5,7 @@ from flask_restful import Resource, fields, marshal_with, reqparse
|
|||||||
from flask_restful.inputs import int_range
|
from flask_restful.inputs import int_range
|
||||||
from werkzeug.exceptions import Forbidden, InternalServerError, NotFound
|
from werkzeug.exceptions import Forbidden, InternalServerError, NotFound
|
||||||
|
|
||||||
|
import services
|
||||||
from controllers.console import api
|
from controllers.console import api
|
||||||
from controllers.console.app.error import (
|
from controllers.console.app.error import (
|
||||||
CompletionRequestError,
|
CompletionRequestError,
|
||||||
@ -27,7 +28,7 @@ from fields.conversation_fields import annotation_fields, message_detail_fields
|
|||||||
from libs.helper import uuid_value
|
from libs.helper import uuid_value
|
||||||
from libs.infinite_scroll_pagination import InfiniteScrollPagination
|
from libs.infinite_scroll_pagination import InfiniteScrollPagination
|
||||||
from libs.login import login_required
|
from libs.login import login_required
|
||||||
from models.model import AppMode, Conversation, Message, MessageAnnotation, MessageFeedback
|
from models.model import AppMode, Conversation, Message, MessageAnnotation
|
||||||
from services.annotation_service import AppAnnotationService
|
from services.annotation_service import AppAnnotationService
|
||||||
from services.errors.conversation import ConversationNotExistsError
|
from services.errors.conversation import ConversationNotExistsError
|
||||||
from services.errors.message import MessageNotExistsError, SuggestedQuestionsAfterAnswerDisabledError
|
from services.errors.message import MessageNotExistsError, SuggestedQuestionsAfterAnswerDisabledError
|
||||||
@ -124,33 +125,16 @@ class MessageFeedbackApi(Resource):
|
|||||||
parser.add_argument("rating", type=str, choices=["like", "dislike", None], location="json")
|
parser.add_argument("rating", type=str, choices=["like", "dislike", None], location="json")
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
message_id = str(args["message_id"])
|
try:
|
||||||
|
MessageService.create_feedback(
|
||||||
message = db.session.query(Message).filter(Message.id == message_id, Message.app_id == app_model.id).first()
|
app_model=app_model,
|
||||||
|
message_id=str(args["message_id"]),
|
||||||
if not message:
|
user=current_user,
|
||||||
raise NotFound("Message Not Exists.")
|
rating=args.get("rating"),
|
||||||
|
content=None,
|
||||||
feedback = message.admin_feedback
|
|
||||||
|
|
||||||
if not args["rating"] and feedback:
|
|
||||||
db.session.delete(feedback)
|
|
||||||
elif args["rating"] and feedback:
|
|
||||||
feedback.rating = args["rating"]
|
|
||||||
elif not args["rating"] and not feedback:
|
|
||||||
raise ValueError("rating cannot be None when feedback not exists")
|
|
||||||
else:
|
|
||||||
feedback = MessageFeedback(
|
|
||||||
app_id=app_model.id,
|
|
||||||
conversation_id=message.conversation_id,
|
|
||||||
message_id=message.id,
|
|
||||||
rating=args["rating"],
|
|
||||||
from_source="admin",
|
|
||||||
from_account_id=current_user.id,
|
|
||||||
)
|
)
|
||||||
db.session.add(feedback)
|
except services.errors.message.MessageNotExistsError:
|
||||||
|
raise NotFound("Message Not Exists.")
|
||||||
db.session.commit()
|
|
||||||
|
|
||||||
return {"result": "success"}
|
return {"result": "success"}
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user