Explorar o código

fix: config name optimize

zhayujie %!s(int64=2) %!d(string=hai) anos
pai
achega
3590babd8b
Modificáronse 3 ficheiros con 7 adicións e 7 borrados
  1. 4 4
      bot/linkai/link_ai_bot.py
  2. 1 1
      bridge/bridge.py
  3. 2 2
      config.py

+ 4 - 4
bot/linkai/link_ai_bot.py

@@ -37,22 +37,22 @@ class LinkAIBot(Bot):
                 session.messages.pop(0)
                 session.messages.pop(0)
 
 
             # load config
             # load config
-            app_code = conf().get("app_code")
-            app_market_api_key = conf().get("app_market_api_key")
+            app_code = conf().get("linkai_app_code")
+            linkai_api_key = conf().get("linkai_api_key")
             logger.info(f"[LINKAI] query={query}, app_code={app_code}")
             logger.info(f"[LINKAI] query={query}, app_code={app_code}")
 
 
             body = {
             body = {
                 "appCode": app_code,
                 "appCode": app_code,
                 "messages": session.messages
                 "messages": session.messages
             }
             }
-            headers = {"Authorization": "Bearer " + app_market_api_key}
+            headers = {"Authorization": "Bearer " + linkai_api_key}
 
 
             # do http request
             # do http request
             res = requests.post(url=self.base_url + "/chat/completion", json=body, headers=headers).json()
             res = requests.post(url=self.base_url + "/chat/completion", json=body, headers=headers).json()
 
 
             if not res or not res["success"]:
             if not res or not res["success"]:
                 if res.get("code") == self.AUTH_FAILED_CODE:
                 if res.get("code") == self.AUTH_FAILED_CODE:
-                    logger.exception(f"[LINKAI] please check your app_market_api_key, res={res}")
+                    logger.exception(f"[LINKAI] please check your linkai_api_key, res={res}")
                     return Reply(ReplyType.ERROR, "请再问我一次吧")
                     return Reply(ReplyType.ERROR, "请再问我一次吧")
                 else:
                 else:
                     # retry
                     # retry

+ 1 - 1
bridge/bridge.py

@@ -23,7 +23,7 @@ class Bridge(object):
             self.btype["chat"] = const.OPEN_AI
             self.btype["chat"] = const.OPEN_AI
         if conf().get("use_azure_chatgpt", False):
         if conf().get("use_azure_chatgpt", False):
             self.btype["chat"] = const.CHATGPTONAZURE
             self.btype["chat"] = const.CHATGPTONAZURE
-        if conf().get("app_market_api_key") and conf().get("app_code"):
+        if conf().get("linkai_api_key") and conf().get("linkai_app_code"):
             self.btype["chat"] = const.LINKAI
             self.btype["chat"] = const.LINKAI
         self.bots = {}
         self.bots = {}
 
 

+ 2 - 2
config.py

@@ -100,8 +100,8 @@ available_setting = {
     # 插件配置
     # 插件配置
     "plugin_trigger_prefix": "$",  # 规范插件提供聊天相关指令的前缀,建议不要和管理员指令前缀"#"冲突
     "plugin_trigger_prefix": "$",  # 规范插件提供聊天相关指令的前缀,建议不要和管理员指令前缀"#"冲突
     # 应用市场配置
     # 应用市场配置
-    "app_market_api_key": "",
-    "app_code": ""
+    "linkai_api_key": "",
+    "linkai_app_code": ""
 }
 }