Changes Documents - llm and languagefields on tenant, processing on documents
first version of Adding Documents (excl. embeddings)
This commit is contained in:
12
eveai_app/utils/app_hooks.py
Normal file
12
eveai_app/utils/app_hooks.py
Normal file
@@ -0,0 +1,12 @@
|
||||
from flask import session
|
||||
|
||||
|
||||
def make_session_permanent():
|
||||
session.permanent = True # Refresh the session timeout on every request
|
||||
# if 'user_id' in session and session.get('was_authenticated'):
|
||||
# if session.modified: # Check if the session was modified
|
||||
# session['was_authenticated'] = True
|
||||
# else:
|
||||
# session.pop('user_id', None) # Clear session
|
||||
# session.pop('was_authenticated', None)
|
||||
# return redirect(url_for('login')) # Redirect to login page if session expired
|
||||
@@ -8,7 +8,6 @@ from flask import session
|
||||
|
||||
from ..models.user import User, Tenant
|
||||
from .database import Database
|
||||
from ..views.document_views import document_bp
|
||||
|
||||
|
||||
def mw_before_request():
|
||||
@@ -21,8 +20,8 @@ def mw_before_request():
|
||||
if not tenant_id:
|
||||
return {"message": "You are not logged into any tenant"}, 403
|
||||
|
||||
user = User.get_by_id(current_user.id)
|
||||
if user.has_roles(['Super User']) or user.tenant_id == tenant_id:
|
||||
# user = User.query.get(current_user.id)
|
||||
if current_user.has_roles(['Super User']) or current_user.tenant_id == tenant_id:
|
||||
Database(tenant_id).switch_schema()
|
||||
else:
|
||||
return {"message": "You are not a member of this tenant"}, 403
|
||||
|
||||
@@ -6,3 +6,7 @@ from ..models.user import User, Tenant
|
||||
def set_tenant_session_data(sender, user, **kwargs):
|
||||
tenant = Tenant.query.filter_by(id=user.tenant_id).first()
|
||||
session['tenant'] = tenant.to_dict()
|
||||
session['default_language'] = tenant.default_language
|
||||
session['default_embedding_model'] = tenant.default_embedding_model
|
||||
session['default_llm_model'] = tenant.default_llm_model
|
||||
|
||||
|
||||
Reference in New Issue
Block a user