Python使用百度文心一言AI方法
首先,你需要确保你有一个百度开发者账号,并且已经创建了文心一言应用,获取了API Key。
以下是一个使用Python发送请求到百度文心一言API的示例代码:
import requests
# 替换成你的API Key
API_KEY = "your_api_key"
# 文心一言的API地址
API_URL = "https://openapi.baidu.com/oauth/2.0/token"
# 获取access token
def get_access_token(api_key):
params = {
'grant_type': 'client_credentials',
'client_id': api_key
}
response = requests.post(API_URL, params=params)
if response.ok:
return response.json().get('access_token')
else:
raise Exception('Error retrieving access token: ', response.text)
# 调用文心一言
def call_wenxin_ai(access_token, text):
headers = {
'Content-Type': 'application/json',
'Authorization': f'Bearer {access_token}'
}
data = {
'session_id': "wenxin_test",
'query': text,
'user_id': "test_user"
}
response = requests.post('https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxin-bot/chat', headers=headers, json=data)
if response.ok:
return response.json().get('result').get('response')
else:
raise Exception('Error callingwenxin ai: ', response.text)
# 使用示例
access_token = get_access_token(API_KEY)
print(call_wenxin_ai(access_token, "你好,文心一言"))
确保替换your_api_key
为你的API Key,运行代码后,它将向百度文心一言发送一个问题,并打印出回答。
注意:上述代码中的API\_URL可能会变化,请确保你使用的是最新的API地址。
评论已关闭