怎么调用文心一言的api接口生成一个简单的聊天机器人(python代码)
要调用文心一言的API接口生成一个简单的聊天机器人,你需要使用Python的requests库来发送HTTP请求。以下是一个简单的例子:
首先,安装requests库(如果你还没有安装的话):
pip install requests
然后,使用以下Python代码创建一个简单的聊天机器人:
import requests
def send_message(message):
# 文心一言的API接口地址
api_url = "https://openapi.baidu.com/oauth/2.0/token"
# 替换为你的API Key和Secret Key
api_key = "YOUR_API_KEY"
secret_key = "YOUR_SECRET_KEY"
# 获取access_token
response = requests.post(api_url, data={
'grant_type': 'client_credentials',
'client_id': api_key,
'client_secret': secret_key
})
access_token = response.json()['access_token']
# 设置文心一言的对话API接口
text_generate_url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/chat/completions"
# 发送消息
response = requests.post(text_generate_url, data={
"session_id": "chatbot", # 可以自定义,表示会话ID
"log_id": "123", # 可以自定义,表示日志ID
"request": {
"query": message, # 用户输入的消息
"user_id": "test_user" # 用户ID
}
}, headers={
'Content-Type': 'application/json',
'Authorization': 'Bearer ' + access_token
})
# 解析返回的消息
response_json = response.json()
if 'results' in response_json and len(response_json['results']) > 0:
return response_json['results'][0]['values']['text']
else:
return "对不起,我无法理解你的问题。"
# 用户与机器人交互的示例
while True:
message = input("你: ")
if message.strip() != '':
reply = send_message(message)
print("机器人: ", reply)
在使用这段代码之前,请确保你已经从百度AI开放平台申请了文心一言的API Key和Secret Key,并且替换了代码中的YOUR_API_KEY
和YOUR_SECRET_KEY
。
这个简易的聊天机器人会源源不断地接收用户输入的消息,并返回文心一言预测的回复。你可以根据需要扩展这个简单的聊天机器人,比如添加更复杂的会话处理逻辑、上下文管理、多轮对话等功能。
评论已关闭