fix(chalice): fixed search usability-test's sessions

This commit is contained in:
Taha Yassine Kraiem 2025-03-12 13:42:30 +01:00 committed by Kraiem Taha Yassine
parent 136a5b2bfb
commit bd09160a4a
2 changed files with 3 additions and 4 deletions

View file

@ -6,13 +6,12 @@ logger = logging.getLogger(__name__)
from . import sessions_pg
from . import sessions_pg as sessions_legacy
from . import sessions_ch
from . import sessions_search as sessions_search_legacy
if config("EXP_SESSIONS_SEARCH", cast=bool, default=False):
logger.info(">>> Using experimental sessions search")
from . import sessions_ch as sessions
from . import sessions_search as sessions_search_legacy
from . import sessions_search_exp as sessions_search
else:
from . import sessions as sessions
from . import sessions_search as sessions_search
from . import sessions_search as sessions_search_legacy

View file

@ -3,7 +3,7 @@ import logging
import schemas
from chalicelib.core import metadata, projects
from . import sessions_favorite, sessions as sessions_legacy, sessions_ch as sessions, sessions_legacy_mobil
from . import sessions_favorite, sessions_search_legacy, sessions_ch as sessions, sessions_legacy_mobil
from chalicelib.utils import pg_client, helper, ch_client, exp_ch_helper
logger = logging.getLogger(__name__)
@ -279,4 +279,4 @@ def search_by_metadata(tenant_id, user_id, m_key, m_value, project_id=None):
# TODO: rewrite this function to use ClickHouse
def search_sessions_by_ids(project_id: int, session_ids: list, sort_by: str = 'session_id',
ascending: bool = False) -> dict:
return sessions_legacy.search_sessions_by_ids(project_id, session_ids, sort_by, ascending)
return sessions_search_legacy.search_sessions_by_ids(project_id, session_ids, sort_by, ascending)