From 4f7c1aa48f2da5d2a25f33f8d9fcd88c2fa15aaa Mon Sep 17 00:00:00 2001 From: huni Date: Fri, 24 Jul 2026 18:35:01 +0800 Subject: [PATCH] add strategyId (STRATEGY_ID env) --- main.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index a0fd5bf..871de75 100644 --- a/main.py +++ b/main.py @@ -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()