logo
1

【AI大模型应用开发】0.1 百度文心一言 API接入详细步骤和简单应用

0. 接入流程

整体流程图如下:

1. 注册百度千帆账号,创建应用

(1)首先注册千帆账号并登录千帆平台
(2)根据实际需求创建千帆应用,创建成功后,获取AppID、API Key、Secret Key 等信息。如果已有千帆应用,可以查看已有应用的API Key、Secret Key 等信息。

2. 获取Access Token

(1)将百度千帆的API Key和Secret Key放到环境变量中,加载
  
  
  
  
  
  
BAIDU_QIANFAN_CLINE_AK = "xxxxx"
BAIDU_QIANFAN_CLINE_SK = "xxxxx"
(2)获取Access Token的Python代码
  
  
  
  
  
  
import requests
import json
import os
# 加载 .env 到环境变量
from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv())
client_id = os.getenv('BAIDU_QIANFAN_CLINE_AK') # 设置百度千帆AK: API Key
client_secret = os.getenv('BAIDU_QIANFAN_CLINE_SK') # 设置百度千帆SK: Secret Key
def get_access_token():
url = f"https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={client_id}&client_secret={client_secret}"
payload = ""
headers = {
'Content-Type': 'application/json',
'Accept': 'application/json'
}
response = requests.request("POST", url, headers=headers, data=payload)
print(response.json().get("access_token"))
if __name__ == '__main__':
get_access_token()

3. 调用文心大模型接口

  • url决定了调用哪个大模型以及哪个接口,同时url中需要填入Acess Token的值
  • payload为POST请求数据体
  • headers为请求头
  • 发送POST请求
  
  
  
  
  
  
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)
print(response.text)
if __name__ == '__main__':
main()
运行结果:
  • result为返回结果
  • usage字段为输入和输出以及总的消耗的token数
4. API在线调试
百度千帆平台还提供了API在线调试服务,这是一个可视化、提升接口调试效率的调试平台,集成快速检索、开发文档、调试、复制和下载示例代码等功能,简单易用。通过API在线调试平台,开发者无需编写代码,即可快捷完成服务端的接口调用,同时还可以快速地获取渲染好的示例代码等。
(1)在messages中更改对话内容,调试,可视化查看结果
(2)直接在右侧底部对话框输入对话,查看返回结果,会自动更新对话内容到左侧的messages中。
(3)调试好之后,可以点击示例代码,直接将代码拷贝粘贴到程序中,自己不用再一行一行写了,并且保证正确。以下就是上面多轮对话的示例代码。

5. 其它

  • 绘画接口,需开通付费才能用…以后再说。
好了,本文就到这了。百度千帆还有很多方便的功能,感兴趣的可以去探索下。因为我目前的目的只是看看百度文心一言的API怎么接入以及怎么替换掉OpenAI的接口,所以没有太过深入。后面有机会继续探索。

————————————————
版权声明:本文为CSDN博主「同学小张」的原创文章
原文链接:https://blog.csdn.net/Attitude93/article/details/136056304
如有侵权,请联系千帆社区进行删除
评论
用户头像