# 使用提醒:

# 1. xbot包提供软件自动化、数据表格、Excel、日志、AI等功能
# 2. package包提供访问当前应用数据的功能，如获取元素、访问全局变量、获取资源文件等功能
# 3. 当此模块作为流程独立运行时执行main函数
# 4. 可视化流程中可以通过"调用模块"的指令使用此模块

import xbot
from xbot import print, sleep
from .import package
from .package import variables as glv

from openai import OpenAI

# 在此处填入您的 API 密钥
api_key = "xxxxxxxxxxxxxxxxx"

def ask_kimi(question):
    client = OpenAI(
        api_key=api_key,
        base_url="https://api.moonshot.cn/v1"
    )
    messages = [
        {"role": "system", "content": "你是 Kimi，由 Moonshot AI 提供的人工智能助手，你更擅长中文和英文的对话。你会为用户提供安全，有帮助，准确的回答。"},
        {"role": "user", "content": question}
    ]

    response = client.chat.completions.create(
        model="moonshot-v1-8k",  # 假设这是 Kimi 模型的名称
        messages=messages,
        temperature=0.3,
    )

    # 返回 Kimi 的回答
    return response.choices[0].message.content

# # 定义一个函数，该函数接受一个问题作为参数
# def kimi_chat_flow(question):
#     # 调用 ask_kimi 函数并传入问题
#     answer = ask_kimi(question)
#     # 打印 Kimi 的回答
#     print(answer)