手把手教你调用文心一言API,含py调用示例代码
大模型开发/技术交流
- LLM
- API
3月6日9616看过
获取API密钥
注册或登录账号
选择应用接入
创建应用
随便起个名字
点击显示即可。
这个
API Key
和Secret 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 jsonimport pandas as pdimport requestsfrom tqdm import tqdmfilename = "一列问题.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)
上面是个比较简单的例子,大家可以自由拓展。
————————————————
版权声明:本文为CSDN博主「Dream of Grass」的原创文章
原文链接:https://blog.csdn.net/dream_of_grass/article/details/135535369
如有侵权,请联系千帆社区进行删除
评论