Преглед изворни кода

Add Baidu access token validation

Kevin Li пре 2 година
родитељ
комит
24b63bc5bd
1 измењених фајлова са 9 додато и 2 уклоњено
  1. 9 2
      bot/baidu/baidu_wenxin.py

+ 9 - 2
bot/baidu/baidu_wenxin.py

@@ -1,7 +1,6 @@
 # encoding:utf-8
 
 import requests, json
-import pdb
 from bot.bot import Bot
 from bridge.reply import Reply, ReplyType
 from bot.session_manager import SessionManager
@@ -12,7 +11,7 @@ from config import conf
 from bot.baidu.baidu_wenxin_session import BaiduWenxinSession
 
 BAIDU_API_KEY = conf().get("baidu_wenxin_api_key")
-BAIDU_SECRET_KEY = conf().get("baidu_wenxin_api_key")
+BAIDU_SECRET_KEY = conf().get("baidu_wenxin_secret_key")
 
 class BaiduWenxinBot(Bot):
 
@@ -62,7 +61,15 @@ class BaiduWenxinBot(Bot):
 
     def reply_text(self, session: BaiduWenxinSession, retry_count=0):
         try:
+            logger.info("[BAIDU] model={}".format(session.model))
             access_token = self.get_access_token()
+            if access_token == 'None':
+                logger.warn("[BAIDU] access token 获取失败")
+                return {
+                    "total_tokens": 0,
+                    "completion_tokens": 0,
+                    "content": 0,
+                    }
             url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/" + session.model + "?access_token=" + access_token
             headers = {
                 'Content-Type': 'application/json'