diff --git a/common/models/user.py b/common/models/user.py index e1caa2e..f806dfe 100644 --- a/common/models/user.py +++ b/common/models/user.py @@ -35,6 +35,9 @@ class Tenant(db.Model): html_end_tags = db.Column(ARRAY(sa.String(10)), nullable=True, default=['p', 'li']) html_included_elements = db.Column(ARRAY(sa.String(50)), nullable=True) html_excluded_elements = db.Column(ARRAY(sa.String(50)), nullable=True) + min_chunk_size = db.Column(db.Integer, nullable=True, default=2000) + max_chunk_size = db.Column(db.Integer, nullable=True, default=3000) + # Embedding search variables es_k = db.Column(db.Integer, nullable=True, default=5) @@ -77,6 +80,9 @@ class Tenant(db.Model): 'html_end_tags': self.html_end_tags, 'html_included_elements': self.html_included_elements, 'html_excluded_elements': self.html_excluded_elements, + 'min_chunk_size': self.min_chunk_size, + 'max_chunk_size': self.max_chunk_size, + 'es_k' 'es_k': self.es_k, 'es_similarity_threshold': self.es_similarity_threshold, 'chat_RAG_temperature': self.chat_RAG_temperature, diff --git a/eveai_app/templates/user/tenant_overview.html b/eveai_app/templates/user/tenant_overview.html index 1145fb2..f3d5cb6 100644 --- a/eveai_app/templates/user/tenant_overview.html +++ b/eveai_app/templates/user/tenant_overview.html @@ -31,8 +31,8 @@