feat: build_test_script + mode=test
This commit is contained in:
+7
-1
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user