Python如何對(duì)接文心一言

更新時(shí)間：2024年01月10日 11:46:05 作者：bystart?青檬小棧

這篇文章主要為大家介紹了Python如何對(duì)接文心一言的操作實(shí)例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

引言

文心一言是百度研發(fā)的Ai機(jī)器，能夠與人對(duì)話互動(dòng)，回答問(wèn)題，協(xié)助創(chuàng)作，高效便捷地幫助人們獲取信息、知識(shí)和靈感。

申請(qǐng)Api Key

前往百度智能云

https://console.bce.baidu.com/qianfan/ais/console/applicationConsole/application

登錄并創(chuàng)建應(yīng)用，拿到需要的API Key和Secret Key:

編輯源代碼

修改參數(shù)：

api_key：替換成自己的

secret_key：替換自己的

redis的配置是用于保存access_token，該token通過(guò)接口獲取默認(rèn)有效期為30天。可自行決定是否需要redis的配合。

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)