From 8bad8c32cd1db10eca4e36e3d75491dae88f2b88 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20K=C3=B6pf?= Date: Wed, 8 Feb 2023 20:24:06 +0100 Subject: [PATCH] add porperties to message response model (deleted, synthetic, ...) --- backend/oasst_backend/api/v1/utils.py | 6 ++++++ oasst-shared/oasst_shared/schemas/protocol.py | 6 ++++++ 2 files changed, 12 insertions(+) diff --git a/backend/oasst_backend/api/v1/utils.py b/backend/oasst_backend/api/v1/utils.py index 0b7749e2..6f66482e 100644 --- a/backend/oasst_backend/api/v1/utils.py +++ b/backend/oasst_backend/api/v1/utils.py @@ -20,6 +20,12 @@ def prepare_message(m: Message) -> protocol.Message: user_is_author=m.user_is_author, review_result=m.review_result, review_count=m.review_count, + ranking_count=m.ranking_count, + deleted=m.deleted, + synthetic=m.synthetic, + model_name=m.model_name, + message_tree_id=m.message_tree_id, + rank=m.rank, ) diff --git a/oasst-shared/oasst_shared/schemas/protocol.py b/oasst-shared/oasst_shared/schemas/protocol.py index 7d1d256a..908df9df 100644 --- a/oasst-shared/oasst_shared/schemas/protocol.py +++ b/oasst-shared/oasst_shared/schemas/protocol.py @@ -102,6 +102,12 @@ class Message(ConversationMessage): created_date: Optional[datetime] review_result: Optional[bool] review_count: Optional[int] + deleted: Optional[bool] + synthetic: Optional[bool] + model_name: Optional[str] + message_tree_id: Optional[UUID] + ranking_count: Optional[int] + rank: Optional[int] class MessagePage(PageResult):