# llm_token_revoke.dspy — 吊销短期 LLM 代理 token

user_id = await get_user()
if not user_id:
    return json.dumps({"success": False, "error": "未登录"}, ensure_ascii=False)

p = params_kw or {}
token_id = (p.get('token_id', '') or '').strip()
task_id = (p.get('task_id', '') or '').strip()

if task_id:
    ok, msg = await revoke_task_tokens(task_id)
elif token_id:
    ok, msg = await revoke_llm_token(token_id=token_id)
else:
    return json.dumps({"success": False, "error": "须指定 token_id 或 task_id"},
                      ensure_ascii=False)

if ok:
    return json.dumps({"success": True, "message": msg}, ensure_ascii=False)
return json.dumps({"success": False, "error": msg}, ensure_ascii=False)