Python如何对接文心一言

import requests
import json
import redis
# 文心一言配置
api_key = "你的api_key"
secret_key = "你的secret_key"
# redis配置
redis_host = "127.0.0.1"
redis_port = 6379
redis_db = 0
class ChatBot:
    def __init__(self, api_key, secret_key):
        self.api_key = api_key
        self.secret_key = secret_key
        self.message_history = []
        self.redis_client = redis.Redis(host=redis_host, port=redis_port, db=redis_db)
        self.chat_url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token={}"
    def get_token(self):
        if self.redis_client.exists('access_token'):
            return self.redis_client.get('access_token').decode()
        else:
            get_access_token_url = ("https://aip.baidubce.com/oauth/2.0/token?"
                                    "client_id={}"
                                    "&client_secret={}"
                                    "&grant_type=client_credentials").format(
                self.api_key, self.secret_key)
            response = requests.get(get_access_token_url)
            self.redis_client.setex('access_token', response.json()['expires_in'], response.json()['access_token'])
            return response.json()['access_token']
    def check_tokens(self, total_tokens):
        if total_tokens > 4800:
            self.message_history = self.message_history[len(self.message_history) / 2:]
    def add_chat_history(self, message):
        self.message_history.append(message)
        payload = json.dumps({
            "messages": self.message_history
        })
        return payload
    def send_message(self, message):
        payload = self.add_chat_history({
            "role": "user",
            "content": message
        })
        headers = {'Content-Type': 'application/json'}
        response = requests.post(self.chat_url.format(self.get_token()), headers=headers, data=payload)
        self.add_chat_history({
            "role": "assistant",
            "content": response.json()['result']
        })
        return response.json()['result']
if __name__ == '__main__':
    chatbot = ChatBot(api_key, secret_key)
    while True:
        message = input("you: ")
        if message.strip() != "":
            reply = chatbot.send_message(message)
            print("bot: ", reply)