当前位置:AIGC资讯 > AIGC > 正文

文心一言API调用(python)

        创建一个百度账号,登录百度智能云,最好先实名认证一下,因为后面开通服务会需要实名认证

一、点击左上角产品服务,搜索千帆

二、点击搜索结果百度智能云千帆大模型平台进入如下界面

三、然后点击开通付费进入如下界面

四、开通所有服务,包括预制服务和大模型训练服务

五、返回千帆大模型界面点击应用接入,创建一个应用

创建应用该应用会生成API key和Secret Key,这两个参数就是用来连接文心一言的API需要的参数

六、使用API访问文心一言,下面以python为例

#让ai介绍一下自己
import requests
import json


API_KEY = "你自己的api key"
SECRET_KEY = "你自己的secret key"

def main():
        
    url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro?access_token=" + get_access_token()
    
    payload = json.dumps({
        "messages": [
            {
                "role": "user",
                "content": "介绍一下自己"
            },
        ]
    })
    headers = {
        'Content-Type': 'application/json'
    }
    
    response = requests.request("POST", url, headers=headers, data=payload)
    result = json.loads(response.text)
    print(result['result'])
    
    

def get_access_token():
    """
    使用 AK,SK 生成鉴权签名(Access Token)
    :return: access_token,或是None(如果错误)
    """
    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"))

if __name__ == '__main__':
    main()

下面是运行效果

更新时间 2023-12-05