feat(ai): add PixelRAG VLM generate preflight

This commit is contained in:
ogt
2026-07-10 00:17:02 +08:00
parent d1ca067054
commit bed3a8a8ee
11 changed files with 384 additions and 14 deletions

View File

@@ -785,7 +785,16 @@ def ai_automation_pixelrag_vlm_replay_worker_api():
'false',
'no',
}
probe_generate_before_execute = (
str(request.args.get('probe_generate_before_execute', 'true')).strip().lower()
not in {
'0',
'false',
'no',
}
)
route_readiness_timeout = request.args.get('route_readiness_timeout', 3, type=int)
route_generate_probe_timeout = request.args.get('route_generate_probe_timeout', 8, type=int)
return jsonify(run_pixelrag_ollama_vlm_replay_worker(
platform=platforms,
max_age_hours=max(1, min(max_age_hours or 168, 720)),
@@ -797,6 +806,8 @@ def ai_automation_pixelrag_vlm_replay_worker_api():
write_receipt=bool(execute and write_receipt),
auto_select_model=auto_select_model,
route_readiness_timeout=max(1, min(route_readiness_timeout or 3, 20)),
probe_generate_before_execute=probe_generate_before_execute,
route_generate_probe_timeout=max(1, min(route_generate_probe_timeout or 8, 30)),
))
@@ -813,11 +824,19 @@ def ai_automation_pixelrag_vlm_route_readiness_api():
'true',
'yes',
}
probe_generate = str(request.args.get('probe_generate') or '').strip().lower() in {
'1',
'true',
'yes',
}
timeout = request.args.get('timeout', 3, type=int)
probe_timeout = request.args.get('probe_timeout', 8, type=int)
return jsonify(build_pixelrag_vlm_route_readiness(
model=str(request.args.get('model') or '').strip() or None,
timeout_seconds=max(1, min(timeout or 3, 20)),
include_models=include_models,
probe_generate=probe_generate,
probe_timeout_seconds=max(1, min(probe_timeout or 8, 30)),
))