66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
|
|
|||
|
|
import requests
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_access_token():
|
|||
|
|
"""
|
|||
|
|
使用 API Key，Secret Key 获取access_token，替换下列示例中的应用API Key、应用Secret Key
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=tqsbQSj3DUGlebWsPwrQOCqg&client_secret=A7TV1Xs788sAaKt03xIGPE3AFGmMsxjh"
|
|||
|
|
|
|||
|
|
payload = json.dumps("")
|
|||
|
|
headers = {
|
|||
|
|
'Content-Type': 'application/json',
|
|||
|
|
'Accept': 'application/json'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
response = requests.request("POST", url, headers=headers, data=payload)
|
|||
|
|
print(response.json().get("access_token"))
|
|||
|
|
return response.json().get("access_token")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def baidutext():
|
|||
|
|
token = '24.18fd91a916f1c97f2bc70e9d1e7ea932.2592000.1705819202.282335-45506921'
|
|||
|
|
url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=" + token
|
|||
|
|
|
|||
|
|
payload = json.dumps({
|
|||
|
|
"messages": [
|
|||
|
|
{
|
|||
|
|
"role": "user",
|
|||
|
|
"content": "介绍一下你自己"
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
})
|
|||
|
|
headers = {
|
|||
|
|
'Content-Type': 'application/json'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
response = requests.request("POST", url, headers=headers, data=payload)
|
|||
|
|
return response.json().get("result")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def bdgettext(text):
|
|||
|
|
token = '24.18fd91a916f1c97f2bc70e9d1e7ea932.2592000.1705819202.282335-45506921'
|
|||
|
|
url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=" + token
|
|||
|
|
|
|||
|
|
payload = json.dumps({
|
|||
|
|
"messages": [
|
|||
|
|
{
|
|||
|
|
"role": "user",
|
|||
|
|
"content": text
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
})
|
|||
|
|
headers = {
|
|||
|
|
'Content-Type': 'application/json'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
response = requests.request("POST", url, headers=headers, data=payload)
|
|||
|
|
return response.json().get("result")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
print(baidutext())
|