add porperties to message response model (deleted, synthetic, ...)

LAION-AI · Feb 8, 2023 · 8bad8c3 · 8bad8c3
1 parent a272ec2
commit 8bad8c3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git 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
@@ -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):