文心一言4.0API接入指南

小小晓晓阳2024-02-06 8:40

概述

文心一言是百度打造出来的人工智能大语言模型，具备跨模态、跨语言的深度语义理解与生成能力，文心一言有五大能力，文学创作、商业文案创作、数理逻辑推算、中文理解、多模态生成，其在搜索问答、内容创作生成、智能办公等众多领域都有更广阔的想象空间。文心一言企业服务由千帆大模型平台提供，包括推理服务及大模型微调等一系列开发和应用工具链。文心一言大模型现已升级至4.0，企业客户可通过百度智能云千帆大模型平台申请接入。

API调用流程

步骤一. 创建千帆应用

（1）登录百度智能云千帆控制台。

注册并登录百度智能云千帆控制台

（2）创建千帆应用

进入控制台创建应用。如果已有应用，此步骤可跳过。

（3）创建应用后，获取AppID、API Key、Secret Key。

步骤二. 开通文心一言

（1）进入模型广场，筛选所需模型。

（2）选择体验模型，并开通付费（文心一言4.0没有免费额度，需开通付费后使用）

代码调用示例

python 复制代码

import json
import requests

API_KEY = '*******'
SECRET_KEY = '*******'

def get_access_token(self):
    """
    使用 API Key，Secret Key 获取access_token，替换下列示例中的应用API Key、应用Secret Key
    """

    url = f"https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={self.api_key}&client_secret={self.secret_key}"

    payload = json.dumps("")
    headers = {
        'Content-Type': 'application/json',
        'Accept': 'application/json'
    }

    response = requests.request("POST", url, headers=headers, data=payload)
    return response.json().get("access_token")


def do_chat():
    url =   "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro?access_token=" + self.get_access_token()

    payload = json.dumps({
        "messages": [
            {
                "role": "user",
                "content": "今天天气怎么样？"
            }
         ],
         "temperature": 1.0,
         "response_format": "json_object"
    })
    headers = {
        'Content-Type': 'application/json'
    }

    response = requests.request("POST", url, headers=headers, data=payload)

if __name__ == '__main__':
    do_chat()

其他方式调用及参数说明请参考官方文档