手把手教你调用文心一言API,含py调用示例代码

请添加图片描述

获取API密钥

打开网址:百度智能云千帆大模型 (baidu.com)

注册或登录账号

选择应用接入

创建应用

随便起个名字

点击显示即可。

这个API KeySecret Key就是我们需要的。

Python调用示例

下面的例子是用文心一言批量回答问题。

输入长这样就行:

下面就是代码啦!

"""
@author: HeroZhang(池塘春草梦)
@contact:herozhang101@gmail.com
@version: 1.0.0
@file: law_ques.py
@time: 2024/1/11 16:25
@description: 调用文心一言api,实现批量回答问题
"""

import json

import pandas as pd
import requests
from tqdm import tqdm

filename = "一列问题.CSV"
# 格式:一列问题
filepath = "D:/gun/data/"

API_KEY = "换成你的"
SECRET_KEY = "换成你的"

def ask_Q(question):
    url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=" + get_access_token()

    payload = json.dumps({
        "messages": [
            {
                "role": "user",
                "content": question
            }
        ]
    })
    headers = {
        'Content-Type': 'application/json'
    }

    response = requests.request("POST", url, headers=headers, data=payload)
    return response
    # print(response.text)


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}
    return str(requests.post(url, params=params).json().get("access_token"))


questions = pd.read_csv(filepath + filename, encoding="gbk", header=None, names=['questions'])
questions['answer'] = ""
# %%
for i in tqdm(range(len(questions))):
    question = questions.iloc[i, 0]
    Input = question + '并给出具体的法律条文'
    ans = ask_Q(Input)
    ans = json.loads(ans.text)
    questions.loc[i, 'answer'] = ans['result']

questions.to_csv(filepath + '输出文件_文心一言.csv', encoding="gbk", index=False)

上面是个比较简单的例子,大家可以自由拓展。

可以看看我写的用gpt辅助读文献的代码,可以实现总结+分段总结,并且用简介好看的格式输出到markdown文件里。

【文献copilot】调用文心一言api对论文逐段总结_总结文章内容api-优快云博客

如果觉得还不错,记得点赞+收藏哟!谢谢大家的阅读!( ̄︶ ̄)↗

最后给大家分享一首词。

鹧鸪天·西都作

宋 · 朱敦儒

我是清都山水郎。天教分付与疏狂。曾批给雨支风券,累上留云借月章。
)

如果觉得还不错,记得点赞+收藏哟!谢谢大家的阅读!( ̄︶ ̄)↗

最后给大家分享一首词。

鹧鸪天·西都作

宋 · 朱敦儒

我是清都山水郎。天教分付与疏狂。曾批给雨支风券,累上留云借月章。
诗万首,酒千觞。几曾著眼看侯王。玉楼金阙慵归去,且插梅花醉洛阳。

### 文心一言 API 调用示例代码 为了成功调用文心一言 API,需先访问幂简集成平台完成注册并获得对应的 API 密钥。此过程确保开发者能够合法地请求服务[^1]。 下面展示的是 Python 中调用文心一言大模型的具体实例: ```python import requests import json def get_access_token(api_key, api_secret): url = "https://aip.baidubce.com/oauth/2.0/token" params = { "grant_type": "client_credentials", "client_id": api_key, "client_secret": api_secret } response = requests.post(url, data=params) result = json.loads(response.text) if 'access_token' in result.keys(): return result['access_token'] else: raise Exception("Failed to obtain access token") def call_wenxin_api(access_token, message): url = f"https://api.wenxin.baidu.com/v1/eden/chat?access_token={access_token}" headers = {'Content-Type': 'application/json'} payload = {"message": message} response = requests.post(url, headers=headers, data=json.dumps(payload)) answer = json.loads(response.text) return answer.get('result', '') if __name__ == "__main__": # 用户应替换为自己申请得到的API密钥和秘钥 API_KEY = '你的API_KEY' API_SECRET = '你的API_SECRET' try: access_token = get_access_token(API_KEY, API_SECRET) reply = call_wenxin_api(access_token, "你好") print(f"收到回复: {reply}") except Exception as e: print(e) ``` 上述脚本实现了两个主要功能:一是通过给定的 `API_KEY` 和 `API_SECRET` 获取访问令牌;二是利用获取到的访问令牌向文心一言发送消息,并接收其响应[^2]。
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值