add sign-content

1 year ago · 9977b8683e
parent d4be5ef9de
commit 9977b8683e
2 changed files with 5 additions and 0 deletions
--- a/api/fields/segment_fields.py
+++ b/api/fields/segment_fields.py
@ -18,6 +18,7 @@ segment_fields = {
    "position": fields.Integer,
    "document_id": fields.String,
    "content": fields.String,
+    "sign_content": fields.String,
    "answer": fields.String,
    "word_count": fields.Integer,
    "tokens": fields.Integer,
--- a/api/models/dataset.py
+++ b/api/models/dataset.py
@ -584,6 +584,10 @@ class DocumentSegment(db.Model):  # type: ignore[name-defined]
        else:
            return []

+    @property
+    def sign_content(self):
+        return self.get_sign_content()
+
    def get_sign_content(self):
        signed_urls = []
        text = self.content