From 52dd9f050e2c4277be8cc9298fbaa4afdad59f27 Mon Sep 17 00:00:00 2001 From: huni Date: Fri, 24 Jul 2026 13:48:26 +0800 Subject: [PATCH] zero-dep main.py --- main.py | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/main.py b/main.py index d929b3b..5c2b66c 100644 --- a/main.py +++ b/main.py @@ -66,12 +66,15 @@ def _aes_cbc_b64(pt, key, iv): def _sign(model, cfg, t_ms): - a = _sha_hex("computility:model-verify:browser-submit" + "|v1|" + str(t_ms) + "|" + model) - ii = _sha_hex(cfg) - kf = _sha_bytes("key|" + a + "|" + ii)[:16] - ivf = _sha_bytes("iv|" + model + "|" + str(t_ms) + "|" + str(len(cfg)))[:16] - o = str(t_ms) + "\n" + model + "\n" + cfg - return _sha_hex("v1." + a + "." + _aes_cbc_b64(o, kf, ivf) + "." + ii) + try: + a = _sha_hex("computility:model-verify:browser-submit" + "|v1|" + str(t_ms) + "|" + model) + ii = _sha_hex(cfg) + kf = _sha_bytes("key|" + a + "|" + ii)[:16] + ivf = _sha_bytes("iv|" + model + "|" + str(t_ms) + "|" + str(len(cfg)))[:16] + o = str(t_ms) + "\n" + model + "\n" + cfg + return _sha_hex("v1." + a + "." + _aes_cbc_b64(o, kf, ivf) + "." + ii) + except Exception: + return None # cryptography 不可用(零依赖构建)→ 提交走 token-only def build_config(gpu): @@ -109,7 +112,9 @@ def _post(model, gpu, cfg, signed): t_ms = int(time.time() * 1000) h = {"Authorization": TOKEN, "Content-Type": "application/json"} if signed: - h.update({"X-Request-Time": str(t_ms), "X-Model-Address": model, "X-Request-Sign": _sign(model, cfg, t_ms)}) + sig = _sign(model, cfg, t_ms) + if sig: + h.update({"X-Request-Time": str(t_ms), "X-Model-Address": model, "X-Request-Sign": sig}) body = json.dumps({"modelAddress": model, "taskType": TASK_TYPE, "targetGpu": gpu, "framework": FRAMEWORK, "configParams": cfg}) st, txt = _http("POST", MAIN + "/api/adapt/task/add", h, body)