python import requests import json

API_KEY = "your_api_key" SECRET_KEY = "your_secret_key"

获取access_token

def get_access_token(): url = "https://aip.baidubce.com/oauth/2.0/token" params = { "grant_type": "client_credentials", "client_id": API_KEY, "client_secret": SECRET_KEY } return requests.post(url, params=params).json().get("access_token")

调用文心一言聊天接口

def chat_with_ernie(query): access_token = get_access_token() url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro?access_token=" + access_token payload = json.dumps({ "messages": [ {"role": "user", "content": query} ] }) headers = {'Content-Type': 'application/json'} response = requests.post(url, headers=headers, data=payload) return response.json().get("result")

print(chat_with_ernie("用一句话介绍人工智能"))