19 lines
745 B
Python
19 lines
745 B
Python
from flask import session, current_app
|
|
from common.models.user import Tenant
|
|
|
|
|
|
# Definition of Trigger Handlers
|
|
def set_tenant_session_data(sender, user, **kwargs):
|
|
current_app.logger.debug(f"Setting tenant session data for user {user.id}")
|
|
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.embedding_model
|
|
session['default_llm_model'] = tenant.llm_model
|
|
|
|
|
|
def clear_tenant_session_data(sender, user, **kwargs):
|
|
session.pop('tenant', None)
|
|
session.pop('default_language', None)
|
|
session.pop('default_embedding_model', None)
|
|
session.pop('default_llm_model', None) |