From bd09160a4a76c5b8570b0391eb5880dbf10c2d77 Mon Sep 17 00:00:00 2001 From: Taha Yassine Kraiem Date: Wed, 12 Mar 2025 13:42:30 +0100 Subject: [PATCH] fix(chalice): fixed search usability-test's sessions --- ee/api/chalicelib/core/sessions/__init__.py | 3 +-- ee/api/chalicelib/core/sessions/sessions_search_exp.py | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/ee/api/chalicelib/core/sessions/__init__.py b/ee/api/chalicelib/core/sessions/__init__.py index 4b9a40bc7..d03144e4f 100644 --- a/ee/api/chalicelib/core/sessions/__init__.py +++ b/ee/api/chalicelib/core/sessions/__init__.py @@ -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 diff --git a/ee/api/chalicelib/core/sessions/sessions_search_exp.py b/ee/api/chalicelib/core/sessions/sessions_search_exp.py index e53de4f6f..2c875bf57 100644 --- a/ee/api/chalicelib/core/sessions/sessions_search_exp.py +++ b/ee/api/chalicelib/core/sessions/sessions_search_exp.py @@ -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)