山东大学创新项目智研雅思先锋队第三周

AAA_Roof_Mending

于 2024-04-29 19:02:40 发布

阅读量144

点赞数 2

文章标签：数据库

本文链接：https://blog.csdn.net/AAA_Roof_Mending/article/details/138322343

版权

主要任务：调试模型与链接文心一言API接口

搭建知识库

设置模型参数以及关键词
接入外部API

import requests

def query(payload):
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.json()

API_URL = "https://tcj443h498i2p5p7.aistudio-hub.baidu.com/chat/completions"
headers = {
    # 请前往 https://aistudio.baidu.com/index/accessToken 查看 访问令牌
    "Authorization": "06ccc8bc7d13dfc3b6ce4854734a238b63b9a088 ",
    "Content-Type": "application/json"
}
output = query({
    "messages": [
        {
            "role": "user",
            "content": "开始考试"
        }
    ]
})
print(output.get("result"))