python调用文心一言大模型代码
要调用文心一言(ERNIE Bot)大模型,你需要使用官方提供的API接口。以下是一个使用Python调用文心一言API的基本示例。请注意,你需要先在文心一言的官方网站注册并获取API密钥(API Key)和API Secret。替换为你从文心一言官方网站获取的实际API Key和API Secret。这段代码首先获取访问令牌(access token),然后使用该令牌调用文心一言API,并打印出
·
要调用文心一言(ERNIE Bot)大模型,你需要使用官方提供的API接口。以下是一个使用Python调用文心一言API的基本示例。请注意,你需要先在文心一言的官方网站注册并获取API密钥(API Key)和API Secret。
import requests
import json
# 你的API Key和API Secret
API_KEY = '你的API_KEY'
API_SECRET = '你的API_SECRET'
# 获取访问令牌的URL
TOKEN_URL = 'https://aip.baidubce.com/oauth/2.0/token'
# 文心一言API的URL
ERNIE_BOT_URL = 'https://aip.baidubce.com/rpc/2.0/ernie-bot/v1/chat'
# 构建获取访问令牌的headers和data
headers = {
'Content-Type': 'application/json'
}
data = {
'grant_type': 'client_credentials',
'client_id': API_KEY,
'client_secret': API_SECRET
}
# 发送POST请求获取访问令牌
response = requests.post(TOKEN_URL, headers=headers, data=json.dumps(data))
access_token = response.json()['access_token']
# 构建调用文心一言API的headers和data
headers = {
'Content-Type': 'application/json',
'Authorization': f'Bearer {access_token}'
}
data = {
'version': 'v1',
'request': {
'query': '你好,文心一言', # 你的问题
'user_id': 'test_user' # 用户ID,用于标识用户,你可以自定义
}
}
# 发送POST请求调用文心一言API
response = requests.post(ERNIE_BOT_URL, headers=headers, data=json.dumps(data))
result = response.json()
# 打印结果
print(json.dumps(result, ensure_ascii=False, indent=4))
请将 '你的API_KEY'
和 '你的API_SECRET'
替换为你从文心一言官方网站获取的实际API Key和API Secret。这段代码首先获取访问令牌(access token),然后使用该令牌调用文心一言API,并打印出返回的结果。
请注意,这个代码只是一个基本的示例,实际使用时可能需要根据文心一言API的具体要求进行调整。
更多推荐
已为社区贡献2条内容
所有评论(0)