diff --git a/api/fields/hit_testing_fields.py b/api/fields/hit_testing_fields.py index b9f7e78c17..d9758474f7 100644 --- 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, diff --git a/api/fields/segment_fields.py b/api/fields/segment_fields.py index 52f89859c9..aaac31cf40 100644 --- 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, diff --git a/api/models/dataset.py b/api/models/dataset.py index 567f7db432..7c755fd271 100644 --- 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 = []