Browse Source

Add Baidu access token validation

master
Kevin Li 1 year ago
parent
commit
24b63bc5bd
1 changed files with 9 additions and 2 deletions
  1. +9
    -2
      bot/baidu/baidu_wenxin.py

+ 9
- 2
bot/baidu/baidu_wenxin.py View File

@@ -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'


Loading…
Cancel
Save