diff --git a/llmage/init.py b/llmage/init.py index a4574b3..62de315 100644 --- a/llmage/init.py +++ b/llmage/init.py @@ -1,6 +1,7 @@ from appPublic.base64_to_file import hex2base64 from appPublic.registerfunction import RegisterFunction from llmage.jimeng import jimeng_auth_headers +from llmage.keling import keling_token from llmage.llmclient import ( b64media2url, @@ -31,6 +32,7 @@ def load_llmage(): env.default_llmmessage = default_llmmessage env.SessageMessages = SessionMessages env.BaseMessages = BaseMessages + env.keling_token = keling_token rf = RegisterFunction() rf.register('jimeng_auth_headers', jimeng_auth_headers) diff --git a/llmage/keling.py b/llmage/keling.py new file mode 100644 index 0000000..e7e5eeb --- /dev/null +++ b/llmage/keling.py @@ -0,0 +1,18 @@ +import time +import jwt + +ak = "" # 填写access key +sk = "" # 填写secret key + +def keling_token(ak, sk): + headers = { + "alg": "HS256", + "typ": "JWT" + } + payload = { + "iss": ak, + "exp": int(time.time()) + 1800, # 有效时间,此处示例代表当前时间+1800s(30min) + "nbf": int(time.time()) - 5 # 开始生效的时间,此处示例代表当前时间-5秒 + } + token = jwt.encode(payload, sk, headers=headers) + return token