百度文心一言API-Python版（完整代码）

原创已于 2024-12-10 07:31:08 修改 · 3.6k 阅读

23 ·

CC 4.0 BY-SA版权

文章标签：

#python #人工智能 #百度 #文心一言

于 2024-11-18 15:34:41 首次发布

Python 专栏收录该内容

16 篇文章

订阅专栏

大家好啊！我是NiJiMingCheng

我的博客：NiJiMingCheng

上一节我们分享了实现AI智能回复微信的内容，这一节我们来探索其中需要的百度文心一言，本文详细介绍了我们从注册账号到实现百度文心一言智能回复，同时多种模型自行选择，加油哦！
Python实现百度文心一言智能回复微信

在当今的软件开发领域，与各类人工智能 API 进行集成以实现丰富多样的功能变得愈发常见。本文将深入剖析一段 Python 代码，该代码主要实现了通过调用百度文心 API 来进行文本交互的功能，为大家详细解读其中的技术细节与实现逻辑。

结果演示：

Python实现文心一言智能回复微信

注册百度千帆平台账户--并注册api

1、文心一言的使用需要注册一个百度账户，可以使用百度网盘的账户进行登录百度智能云-登录 (baidu.com)

2、申请以后，进入文心千帆大模型操作百度智能云控制台 (baidu.com)

3、选择右侧边栏的应用接入，再点击创建应用就可以申请一个api了

4、创建过程中只需要填写当前api的名称以及描述，默认所有服务都是勾选的（对api的使用没有任何影响）

5、创建成功后会返回到应用接入的界面，记录下此时AppID、API Key、Secret Key

配置api，开通服务接口
1、点击控制台右上角的计费管理开通服务

2、注意文心一言的api的调用是需要付费的，并且提供多个大模型可以根据自己的选择进行开通，博主只开通了 ERNIE-Bot-turbo大模型公有云在线调用服务和ERNIE-Bot大模型公有云在线调用服务，不同服务的差别可以参考API列表API列表 - 千帆大模型平台 | 百度智能云文档 (baidu.com)

python装配库
- 下载requsets库
- 下载json库

代码整体介绍

这段 Python 代码的核心目标是与百度文心 API 建立连接，向其发送用户提供的文本内容，并获取 API 返回的相应回复，从而实现一个简单的文本交互系统。代码中主要涉及到了 requests 库用于发送 HTTP 请求，以及对返回结果的处理，同时通过自定义函数来完成获取访问令牌（Access Token）以及与 API 进行交互的完整流程。

代码详细剖析

（一）API 密钥设置

API_KEY = "YOUR_API_KEY"
SECRET_KEY = "YOUR_SECRET_KEY"

这两个密钥分别是百度 AI 平台提供的 API Key 和 Secret Key，它们是与百度文心 API 进行安全认证和交互的关键凭证，用于后续生成访问令牌以及验证请求的合法性。

（二）获取访问令牌函数 `get_access_token`

def get_access_token():
    """
    使用 AK，SK 生成鉴权签名（Access Token）
    :return: access_token，或是None(如果错误)
    """
    url = "https://aip.baidubce.com/oauth/2.0/token"
    params = {"grant_type": "client_credentials", "client_id": API_KEY, "client_secret": SECRET_KEY}
    # print(str(requests.post(url, params=params).json().get("access_token")))
    return str(requests.post(url, params=params).json().get("access_token"))

此函数的主要作用是向百度 AI 平台的认证服务器发送请求，通过提供之前定义的 API Key 和 Secret Key 作为参数，按照 client_credentials 授权类型来获取访问令牌（Access Token）。该访问令牌将用于后续与文心 API 进行具体的文本交互操作，它是验证请求来源合法性以及获取服务权限的重要标识。

（三）与文心 API 交互函数 `main_`

def main_(content):
    url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-char-8k?access_token=" + get_access_token()

    payload = json.dumps({
        "messages": [
            {
                "role": "user",
                "content": content
            }
        ]
    })
    headers = {
        'Content-Type': 'application/json'
    }
    # print(payload)
    response = requests.request("POST", url, headers=headers, data=payload).json()
    print(response['result'])
    return response['result']

构建请求 URL：首先，函数内部会构建与文心 API 进行交互的完整 URL。它通过拼接固定的 API 端点地址以及通过调用 get_access_token 函数获取到的访问令牌来形成一个有效的请求链接，确保请求能够被正确识别和处理。
准备请求数据：接着，将用户提供的文本内容 content 按照 API 要求的格式进行封装。这里使用 json.dumps 函数将包含用户消息角色（user）和具体内容的字典数据结构转换为 JSON 字符串格式，作为请求的有效载荷（payload）。
设置请求头：同时，设置了请求头 headers，明确指定了 Content-Type 为 application/json，以告知服务器请求数据的格式类型。
发送请求并处理结果：最后，使用 requests 库的 request 函数以 POST 方式向构建好的 URL 发送请求，携带准备好的请求头和数据。接收到服务器返回的响应后，将其转换为 JSON 格式，并从中提取出 result 字段的值作为 API 的回复内容进行打印输出，同时返回该值以便在其他地方进一步使用。

（四）主程序入口

在 if __name__ == '__main__': 部分：

if __name__ == '__main__':
    res = main_('你好')
    print(res)

这里作为程序的主入口点，简单地调用了 main_ 函数并传入文本内容 '你好'，然后将获取到的 API 回复进行打印输出，展示了整个代码从发送请求到获取并展示回复的完整流程。

总结

全部代码

# -*- coding:utf-8 -*-
import requests
import json

API_KEY = "YOUR_API_KEY"
SECRET_KEY = "YOUR_SECRET_KEY"


def main_(content):
    url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-char-8k?access_token=" + get_access_token()

    payload = json.dumps({
        "messages": [
            {
                "role": "user",
                "content": content
            }
        ]
    })
    headers = {
        'Content-Type': 'application/json'
    }
    # print(payload)
    response = requests.request("POST", url, headers=headers, data=payload).json()
    print(response['result'])
    return response['result']


def get_access_token():
    """
    使用 AK，SK 生成鉴权签名（Access Token）
    :return: access_token，或是None(如果错误)
    """
    url = "https://aip.baidubce.com/oauth/2.0/token"
    params = {"grant_type": "client_credentials", "client_id": API_KEY, "client_secret": SECRET_KEY}
    # print(str(requests.post(url, params=params).json().get("access_token")))
    return str(requests.post(url, params=params).json().get("access_token"))


if __name__ == '__main__':
    res = main_('你好')
    print(res)

通过对这段 Python 代码的详细解析，我们清晰地看到了如何利用百度文心 API 的相关接口，通过获取访问令牌、构建合适的请求数据和 URL，以及正确处理返回结果等一系列步骤，实现了一个简单而有效的文本交互功能。这为我们在开发中集成第三方 AI 服务提供了一个很好的示例，希望读者能够通过本文的解读，更好地理解此类代码的实现原理与应用场景，在自己的项目中灵活运用类似的技术手段。

请注意，在实际应用中，需要确保 API Key 和 Secret Key 的安全性和正确使用，同时也要关注百度文心 API 的相关使用规范和限制条件，以保证程序的正常运行和合法使用。