登录管理后台
首先,先去千帆大模型管理中心(https://console.bce.baidu.com/qianfan/overview),然后就会来到这个界面(没账号的要注册):
建议实名了再使用
新建应用
然后点击应用接入
然后点击创建应用
跟着图片操作一下
选择模型
在这里选一下自己的模型
在线测试
这里已经讲的很详细了,但是如果要对话就必须要自己加代码,很麻烦,所以我小改了一下代码,让他更好用
对话代码
代码如下(应该没Bug,有Bug说一下):
import requests
import json
API_KEY = "填上自己的"
SECRET_KEY = "填上自己的"
# 这里都可以按照官方文档修改,改炸了不管
payload = {
"user_id": "python",
"messages": [],
"system": "这里是AI设定,不用可以删掉",
"disable_search": False,
"enable_citation": False
}
def add_message(role, content):
message = {
"role": role,
"content": content
}
payload["messages"].append(message)
def main():
url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=" + get_access_token()
headers = {
'Content-Type': 'application/json'
}
while True:
# 获取用户输入的话
user_input = input("你:")
# 存起来,让AI可以联系上下文
add_message("user", user_input)
# 合成JSON
json_payload = json.dumps(payload)
# 发送请求
response = requests.request("POST", url, headers=headers, data=json_payload)
result = response.json().get("result") # 提取并打印AI的回复
print("AI:"+result)
# 这里是检测到“终止”自动关闭程序,不要可以把整个if删掉
if "TERMINATE" in result:
print("正在终止程序..")
exit()
# 把AI说的话也存成上下文
add_message("assistant", result)
def get_access_token(): # 这里是获取token一般不用管
"""
使用 AK,SK 生成鉴权签名(Access Token)
:return: access_token,或是None(如果错误)
"""
url = "https://aip.baidubce.com/oauth/2.0/token"
params = {"grant_type": "client_credentials", "client_id": API_KEY, "client_secret": SECRET_KEY}
return str(requests.post(url, params=params).json().get("access_token"))
if __name__ == '__main__':
main()