# 导库
# pip install OpenAI
from openai import OpenAI
from dotenv import load_dotenv
# 读取.env这个文件添加环境变量里面
load_dotenv()
# 配置环境
client = OpenAI(
base_url='https://api.moonshot.cn/v1'
)
e = 10001
# 设置提问次数
# for i in range(1, e):
# 无限次提问
while True:
# 得向他提问 用ask接受我们向ai的提问
ask = input('请输入你的问题:')
ack = input('请输入你的ai角色:')
# 把这个提问传给ai
completion = client.chat.completions.create(
# 指定ai模型
model="moonshot-v1-32k",
messages=[
# 定制我们的ai
{'role': 'system', "content": ack},
# 提交提问
{"role": "user", "content": ask}
]
)
# 把ai的回答提取再给他打印出来
print(completion.choices[0].message.content)
# print("你剩余的提问次数还有:", e - 1 - int(i), "次")
print("你剩余的提问次数:无限次")