add sign-content (#13050)

1 year ago · a16c729d5a
parent 4765728cef
commit a16c729d5a
3 changed files with 6 additions and 0 deletions
--- a/api/fields/hit_testing_fields.py
+++ b/api/fields/hit_testing_fields.py
@ -14,6 +14,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/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
@ -582,6 +582,10 @@ class DocumentSegment(db.Model):  # type: ignore[name-defined]
                return []
        else:
            return []
+    
+    @property
+    def sign_content(self):
+        return self.get_sign_content()

    def get_sign_content(self):
        signed_urls = []