import os
import qianfan

# 通过环境变量初始化认证信息
# 方式一：【推荐】使用安全认证AK/SK鉴权
# 替换下列示例中参数，安全认证Access Key替换your_iam_ak，Secret Key替换your_iam_sk，如何获取请查看https://cloud.baidu.com/doc/Reference/s/9jwvz2egb
os.environ["QIANFAN_ACCESS_KEY"] = "your_iam_ak"
os.environ["QIANFAN_SECRET_KEY"] = "your_iam_sk"

# 方式二：【不推荐】使用应用AK/SK鉴权
# 替换下列示例中参数，将应用API_Key、应用Secret key值替换为真实值
#os.environ["QIANFAN_AK"] = "应用API_Key"
#os.environ["QIANFAN_SK"] = "应用Secret_Key"

chat_comp = qianfan.ChatCompletion()

# 指定特定模型
resp = chat_comp.do(model="ERNIE-Novel-8K", messages=[{
    "role": "user",
    "content": "第二天下午，羽林卿卿来到了A市。“你怎么在这儿？”，浑厚的声线从卿卿的身后传来。卿卿转身：“真是踏破铁鞋无觅处，得来全不费工夫。我就是过来找你！”叶风白林听后一怔，还没来得及反应，“啪！”卿卿将委任状摔到了他的脸上，“这就是你做的好事？！"
}],system="你是一位优秀的网文小说作家，你需要根据【上文】内容进行续写，请保持上文的语言风格、结构特色，生成的内容和上文衔接紧密。\n*写作要求：\n**1.**不要写结尾：**一定不要有结尾！确保故事留下悬念\n2.*增加人物互动：*侧重描写人物之间的对话、动作等互动细节，使小说更加生动有趣，吸引读者的兴趣。\n3.**情节丰富：**情节要有合理反转，出人意料的发展，或者能够带给读者较高期待感的情节。\n**注意事项*：\n1.不要涉及暴力黄色\n2.不要涉及政治人物")

print(resp["body"])

{
    'id': 'as-75q7q2i2tm', 
    'object': 'chat.completion', 
    'created': 1719051683, 
    'result': '叶风白林被羽林卿卿的突然出现和激烈的反应弄得一愣，他低头捡起那张委任状，看了一眼，脸色微微一变。\n\n“你听我解释，卿卿。”他深吸一口气，尽量让自己保持冷静，“这个委任状的事情，并不是你想的那样的。”\n\n“不是我想的这样？那是怎样？”羽林卿卿冷笑一声，双手环胸，一脸嘲讽地看着他，“你背叛了我们的约定，还想要怎么解释？”\n\n“我没有背叛我们的约定，卿卿。”叶风白林看着她，眼神坚定，“这个委任状，是上面直接下来的命令，我根本没有选择的余地。”\n\n“没有选择的余地？”羽林卿卿眉头一挑，“那你为什么不告诉我？为什么要瞒着我？”\n\n“我……”叶风白林张了张嘴，却发现自己无言以对。他确实没有告诉羽林卿卿，因为他知道，如果告诉她，她一定会像现在这样生气。\n\n“你无话可说了吧？”羽林卿卿冷笑一声，“叶风白林，我真的对你很失望。”\n\n说完，她转身就要走。叶风白林见状，连忙拉住她的手腕：“卿卿，你听我说，我真的不是故意的。我知道我错了，我不应该瞒着你，但是请你给我一个解释的机会。”\n\n羽林卿卿看着他，眼中闪过一丝犹豫。她深吸一口气，然后转身看向他：“好，我给你一个解释的机会。但是，如果你不能给我一个满意的解释，那我们之间，就真的完了。”\n\n叶风白林看着她，眼中闪过一丝坚定：“谢谢你，卿卿。我一定会给你一个满意的解释。”\n\n两人相视而坐，叶风白林开始解释起整个事件的经过。他讲述了自己是如何接到这个委任状的，以及自己为什么没有告诉羽林卿卿的原因。他的话语中充满了真诚和歉意，让羽林卿卿不禁开始重新思考整个事件。\n\n听着叶风白林的解释，羽林卿卿心中的怒火渐渐平息了下来。她看着眼前的男人，心中五味杂陈。她知道，叶风白林并没有背叛她，也没有背叛他们的约定。他只是被迫接受了这个委任状，而没有告诉她，也只是因为不想让她担心。\n\n“对不起，卿卿。”叶风白林看着她，眼中满是歉意，“我知道我错了，我不应该瞒着你。但是请你相信我，我真的没有背叛你。”\n\n羽林卿卿看着他，眼中闪过一丝泪光。她深吸一口气，然后轻轻地点了点头：“我相信你，叶风白林。”\n\n两人相视而笑，心中的隔阂在这一刻烟消云散。他们知道，无论未来会遇到什么困难和挑战，只要彼此信任、相互扶持，就一定能够度过难关。', 
    'is_truncated': False, 
    'need_clear_history': False, 
    'finish_reason': 'normal', 
    'usage': {'prompt_tokens': 197, 'completion_tokens': 540, 'total_tokens': 737}
}