bugfix

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)

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