Enable model variables & start working on RAG task

common/models/user.py

@@ -2,6 +2,7 @@ from common.extensions import db
 from flask_security import UserMixin, RoleMixin
 from sqlalchemy.dialects.postgresql import ARRAY
 import sqlalchemy as sa
+from sqlalchemy import CheckConstraint
 
 
 class Tenant(db.Model):
@@ -33,6 +34,14 @@ class Tenant(db.Model):
     html_included_elements = db.Column(ARRAY(sa.String(50)), nullable=True)
     html_excluded_elements = db.Column(ARRAY(sa.String(50)), nullable=True)
 
+    # Embedding search variables
+    es_k = db.Column(db.Integer, nullable=True, default=5)
+    es_similarity_threshold = db.Column(db.Float, nullable=True, default=0.7)
+
+    # Chat variables
+    chat_RAG_temperature = db.Column(db.Float, nullable=True, default=0.3)
+    chat_no_RAG_temperature = db.Column(db.Float, nullable=True, default=0.5)
+
     # Licensing Information
     license_start_date = db.Column(db.Date, nullable=True)
     license_end_date = db.Column(db.Date, nullable=True)

common/utils/celery_utils.py

@@ -7,6 +7,8 @@ celery_app = Celery()
 
 def init_celery(celery, app):
     celery_app.main = app.name
+    app.logger.debug(f'CELERY_BROKER_URL: {app.config["CELERY_BROKER_URL"]}')
+    app.logger.debug(f'CELERY_RESULT_BACKEND: {app.config["CELERY_RESULT_BACKEND"]}')
     celery_config = {
         'broker_url': app.config.get('CELERY_BROKER_URL', 'redis://localhost:6379/0'),
         'result_backend': app.config.get('CELERY_RESULT_BACKEND', 'redis://localhost:6379/0'),