百度大模型API调用方法(兼容OpenAI)
没想到不仅官方文档写的绕,网上的教程也是“全面”得过头——太复杂了。本文主要补充前置步骤,包括ak和sk的生成(ak='YOU_ACCESS_KEY', sk='YOU_SECRET_KEY')
·
没想到不仅官方文档写的绕,网上的教程也是“全面”得过头——太复杂了。但是功夫不负有心人,找到了大佬写的兼容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)
更多推荐
所有评论(0)