Parcourir la source

修改视频上限

xueyiming il y a 6 jours
Parent
commit
86133c2976

+ 2 - 2
supply_infra/scheduler/jobs/discover_videos_from_demands.py

@@ -1,5 +1,5 @@
 """
-从全部 S/A 级需求及其拓展点位触发 find_agent 视频发现;有效视频满 300 提前结束。
+从全部 S/A 级需求及其拓展点位触发 find_agent 视频发现;有效视频满 450 提前结束。
 
 任务层负责查库与组装上下文;Agent 负责搜索、画像与分池落库。
 
@@ -42,7 +42,7 @@ from supply_infra.services.video_discovery_service import get_video_discovery_se
 logger = logging.getLogger(__name__)
 
 _DEFAULT_WORKERS = 5
-_DAILY_PASSED_VIDEO_LIMIT = 300
+_DAILY_PASSED_VIDEO_LIMIT = 450
 _SUPERVISOR_POLL_SECONDS = 0.5
 _QUEUE_GET_TIMEOUT_SECONDS = 1.0
 

+ 10 - 3
tests/supply_infra/scheduler/test_discover_videos_from_demands.py

@@ -1435,12 +1435,16 @@ def test_evaluate_find_agent_run_fails_for_technical_failure(
 @patch(
     "supply_infra.scheduler.jobs.discover_videos_from_demands.list_find_demand_contexts"
 )
-def test_stops_discovery_after_300_passed_videos(
+def test_stops_discovery_after_daily_passed_video_limit(
     mock_list_contexts,
     mock_filter_contexts,
     mock_count_passed,
     mock_process,
 ) -> None:
+    from supply_infra.scheduler.jobs.discover_videos_from_demands import (
+        _DAILY_PASSED_VIDEO_LIMIT,
+    )
+
     contexts = [
         FindDemandContext(
             biz_dt="20260727",
@@ -1460,8 +1464,10 @@ def test_stops_discovery_after_300_passed_videos(
         contexts,
         {"total_loaded": 2, "skipped_already_done": 0},
     )
+    almost_full = _DAILY_PASSED_VIDEO_LIMIT - 1
+    full = _DAILY_PASSED_VIDEO_LIMIT
     # 初始检查 / 执行前检查 / 执行后检查 / 收尾检查
-    mock_count_passed.side_effect = [299, 299, 300, 300]
+    mock_count_passed.side_effect = [almost_full, almost_full, full, full]
     mock_process.return_value = {
         "success": True,
         "skipped": False,
@@ -1474,8 +1480,9 @@ def test_stops_discovery_after_300_passed_videos(
 
     assert mock_process.call_count == 1
     assert result["processed"] == 1
-    assert result["passed_videos"] == 300
+    assert result["passed_videos"] == full
     assert result["stopped_by_passed_video_limit"] is True
+    assert result["passed_video_limit"] == _DAILY_PASSED_VIDEO_LIMIT
 
 
 @patch(