文心一言4.0（ERNIE-Bot-4）申请方法及简单调用代码示例

10月17日过后，估计很多人会看到类似的新闻，如图：

我看到这则新闻也是觉得非常感兴趣，于是本着“百闻不如一见”的实事求是的态度检索如何申请，没想到还真找到了ERNIE-Bot-4（俗称：文心一言4.0）的申请入口，如下：

• 体验地址

企业用户/个人开发者：填写表单申请开通：百度智能云千帆大模型平台邀您测试 审核通过后即可免费调用千帆文心4.0API

• API文档

ERNIE-Bot-4 - 千帆大模型平台 | 百度智能云文档


 

我本人是填写表单申请一天左右收到了“百度智能云”的短信通知我通过了申请，老样子，先上图：

通过之后即使不是很擅长调用接口来测试模型质量，百度智能云也会提供在线测试的入口方便我们测试，如图：

也是可以外接知识库或者调整各种参数的，还算便利，也提供了prompt模板作为参考。

如果我们不想每次都打开网站去测试，也可以在本地进行调用来测试，下面以python为例：

先在百度智能云控制台“应用接入”里创建应用，内容自己发挥，主要是我们需要的AppID，API Key以及Secret Key。

当然，我们需要开通一下对应接口的付费，我这里是只开通了ERNIE-Bot-4的付费，0.12¥/千tokens，跟其他模型比可能有点小贵，自费开通，为爱发电哈哈。

好了，万事具备，下面就是一个简单的多轮对话代码示例：

import requests
import json
import os
import pickleclass WenXinYiYanChat:def __init__(self, api_key, secret_key, user_id="这里输入你的APPID", file_name="history.pkl"):# 初始化方法，用于设置API密钥、用户ID、文件名等self.api_key = api_keyself.secret_key = secret_keyself.user_id = user_idself.file_name = file_nameself.access_token = self.get_access_token()self.messages = []self.is_paused = Falsedef get_access_token(self):# 获取access_token，用于后续的API调用url = "https://aip.baidubce.com/oauth/2.0/token"params = {'grant_type': 'client_credentials','client_id': self.api_key,'client_secret': self.secret_key}headers = {'Content-Type': 'application/json','Accept': 'application/json'}response = requests.post(url, headers=headers, params=params)return response.json().get("access_token")def chat(self, user_message):# 进行对话的主要方法if self.is_paused:return "对话目前已暂停。请先恢复对话再继续。"self.messages.append({"role": "user", "content": user_message})payload = {"messages": self.messages,"user_id": self.user_id,"temperature": 0.95,"top_p": 0.8,"penalty_score": 1.0}url = f"https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro?access_token={self.access_token}"headers = {'Content-Type': 'application/json'}response = requests.post(url, headers=headers, data=json.dumps(payload))assistant_message = response.json().get("result")self.messages.append({"role": "assistant", "content": assistant_message})return assistant_messagedef clear_history(self):# 清除对话历史记录self.messages = []def get_chat_history(self):# 获取对话历史记录return self.messagesdef save_history(self):# 将对话历史记录保存到文件中with open(self.file_name, "wb") as f:pickle.dump(self.messages, f)def load_history(self):# 从文件中读取对话历史记录if os.path.exists(self.file_name):with open(self.file_name, "rb") as f:self.messages = pickle.load(f)def pause_chat(self):# 暂停对话，并保存对话历史记录self.is_paused = Trueself.save_history()def resume_chat(self):# 恢复对话，并加载对话历史记录self.is_paused = Falseself.load_history()if __name__ == "__main__":api_key = input("请输入您的API Key: ")secret_key = input("请输入您的Secret Key: ")'''如果不想每次都输入这两个KEY就把上面两行注释掉，去除下面两行的注释'''# api_key = "在这里输入你的API Key"# secret_key = "在这里输入你的Secret Key"chat_instance = WenXinYiYanChat(api_key, secret_key)while True:user_message = input("靓仔: ")# 添加控制语句，用于实现功能if user_message.lower() == "暂停":chat_instance.pause_chat()print("对话已暂停。")elif user_message.lower() == "恢复":chat_instance.resume_chat()print("对话已恢复。")elif user_message.lower() == "清除":chat_instance.clear_history()print("对话历史记录已清除。")elif user_message.lower() == "查看":history = chat_instance.get_chat_history()print("对话历史记录如下：")for message in history:print(message["role"] + ": " + message["content"])elif user_message.lower() == "载入":chat_instance.load_history()print("对话历史记录已载入。")elif user_message.lower() in ["exit", "退出"]:breakelse:response = chat_instance.chat(user_message)print("文心一言4.0: ", response)

以上是一个非常简陋的多轮对话代码示例，仅供测试接口，佬轻喷，虽然国外的模型质量很高，价格也便宜，国内模型开发的开源项目确实少之又少，但还是希望有更多人能够关注国内模型，众人拾柴火焰高，一同开发出更多更优秀的项目，最后，上效果图：

如果有友友开发出了更好的项目或者发现了更好的项目，记得踢我，一起学习！