mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-09-09 11:15:08 +08:00
Add dupe checks to store_text_reply() & store_text_labels() in PromptRepository (#1018)
* add dupe checks to store_text_reply() & store_text_labels * remove test export file * add user_id to protocol.ConversationMessage * add show_on_leaderboard ot protocol.FrontEndUser
This commit is contained in:
@@ -58,7 +58,7 @@ def get_children_by_frontend_id(
|
||||
"""
|
||||
pr = PromptRepository(db, api_client)
|
||||
message = pr.fetch_message_by_frontend_message_id(message_id)
|
||||
messages = pr.fetch_message_children(message.id)
|
||||
messages = pr.fetch_message_children(message.id, review_result=None)
|
||||
return utils.prepare_message_list(messages)
|
||||
|
||||
|
||||
|
||||
@@ -201,7 +201,7 @@ def get_children(
|
||||
Get all messages belonging to the same message tree.
|
||||
"""
|
||||
pr = PromptRepository(db, api_client, frontend_user=frontend_user)
|
||||
messages = pr.fetch_message_children(message_id)
|
||||
messages = pr.fetch_message_children(message_id, review_result=None)
|
||||
return utils.prepare_message_list(messages)
|
||||
|
||||
|
||||
|
||||
@@ -10,12 +10,15 @@ def prepare_message(m: Message) -> protocol.Message:
|
||||
id=m.id,
|
||||
frontend_message_id=m.frontend_message_id,
|
||||
parent_id=m.parent_id,
|
||||
user_id=m.user_id,
|
||||
text=m.text,
|
||||
lang=m.lang,
|
||||
is_assistant=(m.role == "assistant"),
|
||||
created_date=m.created_date,
|
||||
emojis=m.emojis or {},
|
||||
user_emojis=m.user_emojis or [],
|
||||
review_result=m.review_result,
|
||||
review_count=m.review_count,
|
||||
)
|
||||
|
||||
|
||||
@@ -26,6 +29,7 @@ def prepare_message_list(messages: list[Message]) -> list[protocol.Message]:
|
||||
def prepare_conversation_message(message: Message) -> protocol.ConversationMessage:
|
||||
return protocol.ConversationMessage(
|
||||
id=message.id,
|
||||
user_id=message.user_id,
|
||||
frontend_message_id=message.frontend_message_id,
|
||||
text=message.text,
|
||||
lang=message.lang,
|
||||
|
||||
Reference in New Issue
Block a user