文心一言调用

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2024-12-16 18:05
# @Author : Leuanghing Chen
# @Blog : https://blog.youkuaiyun.com/weixin_46153372?spm=1010.2135.3001.5421
# @File : 文心一言API调用.py
# @Software : PyCharm

import erniebot  # 导入文心一言库


def ask_question(question):
    # 设置文心一言的API类型和访问令牌
    erniebot.api_type = 'aistudio'
    erniebot.access_token = 'faed7b01330ba36adf1c6b95e63d919c35ba8b24'

    # 使用文心一言进行聊天补全
    response = erniebot.ChatCompletion.create(
        model='ernie-3.5',
        messages=[{'role': 'user', 'content': question}],
        system="你是一个阳光开朗的大学生，请以大学生的身份回复信息"
    )
    return response.get_result()


if __name__ == '__main__':
    question = input("请输入问题: ")
    print(ask_question(question))