使用Python调用百度千帆免费大模型接口

百度的千帆大模型免费了。

一、开通免费模型接口

在 在线服务中，找到模型列表，在右侧有 免费开通 按钮，点击开通后 如图所示。

二、获取API Key和Secret Key

进入百度智能云 千帆大模型平台， https://console.bce.baidu.com/qianfan/overview

点击应用接入，进入应用列表

点击创建应用，

这样就有了

三、开发

开发文档 地址：https://cloud.baidu.com/doc/WENXINWORKSHOP/s/klqx7b1xf

响应说明

基本信息

请求地址： https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie_speed

请求方式： POST

Header参数

根据不同鉴权方式，查看对应Header参数。

• 访问凭证access_token鉴权

• 基于安全认证AK/SK进行签名计算鉴权

Query参数

只有访问凭证access_token鉴权方式，需使用Query参数。

• 访问凭证access_token鉴权

Body参数

message说明

响应头Header参数

部分参数如下。

响应体参数

usage说明

注意 ：同步模式和流式模式，响应参数返回不同，详细内容参考示例描述。

代码如下：

调用方法如下：

四、查看统计

在调用统计中可以看到统计信息

 1 import requests
 2 import json
 3 import datetime
 4
 5 class QIANFAN:
 6
 7     _api_url = "https://aip.baidubce.com"
 8
 9     def __init__(self, api_key, secret_key):
10         self.API_KEY = api_key
11         self.SECRET_KEY = secret_key
12
13         url = self._api_url + "/oauth/2.0/token"
14         params = {"grant_type": "client_credentials", "client_id": self.API_KEY, "client_secret": self.SECRET_KEY}
15         result = self.http_request_v2(url, method="POST", params=params)
16         if 'access_token' in result:
17             self.access_token = result["access_token"]
18         else:
19             print(result)
20             exit()
21
22
23     def chat(self, model="ernie-lite-8k", message=None, **kwargs):
24         url = f"{self._api_url}/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/{model}?access_token={self.access_token}"
25
26         payload = json.dumps({
27             "messages": [{"role": "user", "content": message}],
28             "temperature": 0.95,
29             "penalty_score": 1
30         })
31
32         for key, value in kwargs.items():
33             payload[key] = value
34
35         print(payload)
36         response = self.http_request_v2(url, method="POST", params=payload)
37         return response
38
39     # 生成headers头
40     def headers(self, params=None):
41         headers = {}
42         headers['Content-Type'] = 'application/json'
43         return headers
44
45     def http_request_v2(self, url, method="GET", headers={}, params=None):
46         headers['User-Agent'] = 'Mozilla/5.0 \(Windows NT 6.1; WOW64\) AppleWebKit/537.36 \(KHTML, like Gecko\) Chrome/39.0.2171.71 Safari/537.36'
47         if method == "GET":
48             response = requests.get(url)
49         elif method == "POST":
50             # data = bytes(json.dumps(params), 'utf-8')
51             response = requests.post(url, data= params)
52         elif method == "DELETE":
53             response = requests.delete(url, data= data)
54
55         result = response.json()
56         return result

# 示例用法
API_KEY = "dYsBvZdiownWCyVvgdBzWIzo"
SECRET_KEY = "******************"

chat_client = QIANFAN(API_KEY, SECRET_KEY)
print(vars(chat_client))

result = chat_client.chat(model='ernie_speed', message="1加1为什么等于2？")
print(result["result"])