File tree Expand file tree Collapse file tree 1 file changed +18
-5
lines changed
sources/gc-qa-rag-server/ragapp/common Expand file tree Collapse file tree 1 file changed +18
-5
lines changed Original file line number Diff line number Diff line change 11from typing import Generator , List , Dict , Any
2- from datetime import datetime
2+ from datetime import datetime , timedelta
33from sqlalchemy import (
44 create_engine ,
55 Column ,
@@ -145,18 +145,31 @@ def add_qa_feedback(
145145
146146 def get_search_history_by_date (self , date : str ) -> List [Dict [str , Any ]]:
147147 """Get search history records for a specific date.
148-
148+
149149 Args:
150- date: The date to query for
151-
150+ date: The date to query for in format 'YYYY/MM/DD' or 'YYYY-MM-DD'
151+
152152 Returns:
153153 List[Dict[str, Any]]: List of search history records
154154 """
155155 try :
156+ # Convert input date to datetime object
157+ try :
158+ # Try with / format first
159+ date_obj = datetime .strptime (date , "%Y/%m/%d" ).date ()
160+ except ValueError :
161+ # Try with - format if / fails
162+ date_obj = datetime .strptime (date , "%Y-%m-%d" ).date ()
163+
164+ next_day = date_obj + timedelta (days = 1 )
165+
156166 with self .get_session () as session :
157167 results = (
158168 session .query (SearchHistory )
159- .filter (SearchHistory .create_time .cast (String ).like (f"{ date } %" ))
169+ .filter (
170+ SearchHistory .create_time >= date_obj ,
171+ SearchHistory .create_time < next_day
172+ )
160173 .all ()
161174 )
162175 return [
You can’t perform that action at this time.
0 commit comments