最近，在使用千帆平台的API模型，于是用本文章记录调用方法。

获取API_KEY和SECRET_KEY后，执行以下代码

import requests
import json
import pandas as pd

API_KEY = ""
SECRET_KEY = ""

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 str(requests.post(url, params=params).json().get("access_token"))


def main():
    url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro?access_token=" + get_access_token()
    result = []
    query = ""
    prompt = "" + '\n' + "query：" + query
    payload = json.dumps({
        "messages": [
            {
                "role": "user",
                "content": prompt
            }
        ]
    })
    headers = {
        'Content-Type': 'application/json'
    }
    response = requests.request("POST", url, headers=headers, data=payload)
    ger = json.loads(response.text)['result']
    return ger

if __name__ == '__main__':
    print(main())

接口地址统一前缀为https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop， 拼接上接口地址后缀就是完整接口地址