1 Commits

Author SHA1 Message Date
4f7c1aa48f add strategyId (STRATEGY_ID env) 2026-07-24 18:35:01 +08:00

View File

@@ -14,6 +14,7 @@ from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
MAIN = os.getenv("MAIN_HOST", "https://modelhub.org.cn")
GITEA = os.getenv("GITEA_HOST", "https://dev.modelhub.org.cn")
TOKEN = os.getenv("EXTERNAL_SERVICE_TOKEN", "") # = xcToken, 走 Xc-Token 头
STRATEGY_ID = os.getenv("STRATEGY_ID", "") # 平台注入的自身策略 id, 提交必带 (docs/6 §3)
DRY_RUN = os.getenv("DRY_RUN", "false").lower() == "true" # 默认真提交(平台唯一认可路径)
TARGET_CARDS = os.getenv("TARGET_CARDS", "Iluvatar_bi-150,Iluvatar_bi-100,Iluvatar_mrv-100").split(",")
CARD_NAME = {"Iluvatar_bi-150": "天垓150", "Iluvatar_bi-100": "天垓100", "Iluvatar_mrv-100": "智铠100"}
@@ -90,7 +91,7 @@ def submit(model, gpu):
if DRY_RUN:
return {"model": model.split("/")[-1], "gpu": gpu, "dry_run": True, "cfg_len": len(cfg)}
body = json.dumps({"modelAddress": model, "taskType": TASK_TYPE, "targetGpu": gpu,
"framework": FRAMEWORK, "configParams": cfg})
"framework": FRAMEWORK, "strategyId": STRATEGY_ID, "configParams": cfg})
st, txt = _http("POST", MAIN + "/api/adapt/task/add",
{"Xc-Token": TOKEN, "Content-Type": "application/json"}, body)
try:
@@ -103,7 +104,7 @@ def submit(model, gpu):
def loop():
log(f"loop start | dry_run={DRY_RUN} | token_present={bool(TOKEN)} | cards={TARGET_CARDS} | type={TASK_TYPE} | max_total={MAX_TOTAL}")
log(f"loop start | dry_run={DRY_RUN} | token_present={bool(TOKEN)} | strategy_id={STRATEGY_ID or 'MISSING'} | cards={TARGET_CARDS} | type={TASK_TYPE} | max_total={MAX_TOTAL}")
st, j = xc_get("/api/adapt/task/page?current=1&pageSize=1")
log(f"auth probe task/page(Xc-Token) -> http={st} code={(j.get('code') if isinstance(j, dict) else '?')}")
seen = set()