没想到不仅官方文档写的绕,网上的教程也是“全面”得过头——太复杂了。但是功夫不负有心人,找到了大佬写的兼容OpenAI的教程:百度智能云大模型兼容OpenAI的方法_千帆api openai对接openai sdk-CSDN博客(写得这么简洁好用居然才2赞4收藏😭)

本文在上文基础上补充前置步骤,包括ak和sk的生成(ak=ACCESS_KEY, sk=SECRET_KEY)

*是的,百度大模型的调用略显复杂,需要先通过AK和SK来生成api_key.

一、获取AK和SK

首先进入百度智能云千帆大模型平台 (baidu.com),找到左侧导航条的API Key。

进入后点击Access Key进行创建(具体怎么创建不记得了,应该按照指引走就可以),再点击Secret Key下方的显示,就得到了橙色矩形框对应的Access Key(AK)和Secret Key(SK).

二、环境配置

pip install qianfan

三、编写python代码调用

Step1. 调用IAM生成 BearerToken(获取API_key)

        这一步print出来的一长串字符就是下一步要用的api_key,输出后记得复制。

import os
from qianfan import Qianfan
from qianfan.resources.console.iam import IAM

def test_create_token():
    response = IAM.create_bearer_token(
        expire_in_seconds=2592000,
        ak='YOU_ACCESS_KEY', sk='YOU_SECRET_KEY',
    )
    print(response.body['token'])

test_create_token()

Step2. 使用OpenAI SDK 调用

from openai import OpenAI

client = OpenAI(
    api_key="第二步生成的key",
    base_url="https://qianfan.baidubce.com/v2",
)

completion = client.chat.completions.create(
    model="ernie-3.5-8k",
    messages=[{'role': 'user', 'content': '你是谁?'}]
)

print(completion.choices[0].message.content)

ps:model选择可以看 API列表 - ModelBuilder (baidu.com)

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐