feat: build_test_script + mode=test

This commit is contained in:
Repinoid
2026-06-13 20:57:24 +04:00
parent 71200709ab
commit 6e59fef093
2 changed files with 46 additions and 1 deletions
+7 -1
View File
@@ -16,7 +16,7 @@ from flask import jsonify, request
from api.config import load
from api.db import Database
from api.parser import format_no_llm, parse_batch
from api.scripts import build_default_script, build_full_script, build_script_for_level, build_dynamic_script
from api.scripts import build_default_script, build_full_script, build_script_for_level, build_dynamic_script, build_test_script
from brain.client import Diagnoser, LLMError
from brain.prompts import SYSTEM_PROMPT, DYNAMIC_PROMPT
@@ -93,6 +93,12 @@ def register(app):
script = build_default_script()
elif mode == "dynamic":
script = build_dynamic_script()
elif mode == "test":
wait = int(request.args.get("wait", 400))
pid_str = request.args.get("pids", "")
pids = pid_str.split(",") if pid_str else None
repeat = int(request.args.get("repeat", 10))
script = build_test_script(wait_ms=wait, pids=pids, repeat=repeat)
elif mode == "full":
script = build_full_script()
else: