from flask import session
from common.models import User, Tenant


# Definition of Trigger Handlers
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


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)