Move tags to BaseContent

aleph-im · Dec 15, 2023 · 11ee30c · 11ee30c
1 parent f6498f4
commit 11ee30c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/aleph_message/models/__init__.py b/aleph_message/models/__init__.py
@@ -216,9 +216,6 @@ class BaseMessage(BaseModel):
     )
     item_hash: ItemHash = Field(description="Hash of the content (sha256 by default)")
     content: BaseContent = Field(description="Content of the message, ready to be used")
-    tags: Optional[List[str]] = Field(
-        default=None, description="Tags associated with the message"
-    )
 
     forgotten_by: Optional[List[str]]
 

diff --git a/aleph_message/models/abstract.py b/aleph_message/models/abstract.py
@@ -1,4 +1,6 @@
-from pydantic import BaseModel, Extra
+from typing import List, Optional
+
+from pydantic import BaseModel, Extra, Field
 
 
 def hashable(obj):
@@ -23,6 +25,9 @@ class BaseContent(BaseModel):
 
     address: str
     time: float
+    tags: Optional[List[str]] = Field(
+        default=None, description="Tags associated with the message"
+    )
 
     class Config:
         extra = Extra.forbid