1

video-gen-api/app/services/generation_ai_service.py

@@ -770,6 +770,7 @@ async def list_generation_record_history_grouped_days(
     gen_type: str,
     page: int,
     page_size: int,
+    keyword: str | None = None,
 ):
     """
     按生成日期倒序返回旧 generation_records 历史记录分组。
@@ -783,6 +784,8 @@ async def list_generation_record_history_grouped_days(
     page_size = min(max(page_size, 1), HISTORY_DAY_PAGE_SIZE_MAX)
 
     filters = _generation_record_history_base_filters(user_id, gen_type)
+    if keyword and keyword.strip():
+        filters.append(GenerationRecord.original_prompt.ilike(f"%{keyword.strip()}%"))
     day_expr = func.date(GenerationRecord.generated_at).label("generated_date")
 
     days_subquery = (
@@ -868,6 +871,7 @@ async def list_generation_record_history_day_items(
     generated_date: str,
     page: int,
     page_size: int,
+    keyword: str | None = None,
 ):
     """
     获取旧 generation_records 指定生成日期下的历史记录分页。
@@ -878,6 +882,8 @@ async def list_generation_record_history_day_items(
     page_size = min(max(page_size, 1), 100)
 
     filters = _generation_record_history_base_filters(user_id, gen_type)
+    if keyword and keyword.strip():
+        filters.append(GenerationRecord.original_prompt.ilike(f"%{keyword.strip()}%"))
     day_expr = func.date(GenerationRecord.generated_at)
 
     total = (
@@ -935,6 +941,7 @@ async def list_generation_history_grouped_days(
     page: int,
     page_size: int,
     history_source: str | None = None,
+    keyword: str | None = None,
 ):
     """
     按生成日期倒序返回历史记录分组。
@@ -952,6 +959,7 @@ async def list_generation_history_grouped_days(
             gen_type=gen_type,
             page=page,
             page_size=page_size,
+            keyword=keyword,
         )
 
     gen_type = _normalize_history_gen_type(gen_type)
@@ -959,6 +967,8 @@ async def list_generation_history_grouped_days(
     page_size = min(max(page_size, 1), HISTORY_DAY_PAGE_SIZE_MAX)
 
     filters = _history_base_filters(user_id, gen_type, source)
+    if keyword and keyword.strip():
+        filters.append(ChatGenerationTask.original_prompt.ilike(f"%{keyword.strip()}%"))
     day_expr = func.date(ChatGenerationTask.generated_at).label("generated_date")
 
     days_subquery = (
@@ -1049,6 +1059,7 @@ async def list_generation_history_day_items(
     page: int,
     page_size: int,
     history_source: str | None = None,
+    keyword: str | None = None,
 ):
     """
     获取指定生成日期下的历史记录分页。
@@ -1065,6 +1076,7 @@ async def list_generation_history_day_items(
             generated_date=generated_date,
             page=page,
             page_size=page_size,
+            keyword=keyword,
         )
 
     gen_type = _normalize_history_gen_type(gen_type)
@@ -1073,6 +1085,8 @@ async def list_generation_history_day_items(
     page_size = min(max(page_size, 1), 100)
 
     filters = _history_base_filters(user_id, gen_type, source)
+    if keyword and keyword.strip():
+        filters.append(ChatGenerationTask.original_prompt.ilike(f"%{keyword.strip()}%"))
     day_expr = func.date(ChatGenerationTask.generated_at)
 
     total = (