drpy/utils/ai/Kimi.js

import axios from 'axios';

// https://platform.moonshot.cn/console/api-keys
// https://platform.moonshot.cn/console/limits
class Kimi {
    constructor({apiKey, baseURL}) {
        if (!apiKey) {
            throw new Error('Missing required configuration parameters.');
        }
        this.apiKey = apiKey;
        this.baseURL = baseURL || 'https://api.moonshot.cn/v1';
        this.userContexts = {}; // 存储每个用户的上下文
    }

    // 初始化用户上下文
    initUserContext(userId) {
        if (!this.userContexts[userId]) {
            this.userContexts[userId] = [
                {role: "system", content: "你是一名优秀的AI助手，知道最新的互联网内容，善用搜索引擎和github并总结最贴切的结论来回答我提出的每一个问题"}
            ];
        }
    }

    // 更新用户上下文
    updateUserContext(userId, message) {
        this.userContexts[userId].push(message);
        // 控制上下文长度，保留最新的20条消息
        if (this.userContexts[userId].length > 20) {
            this.userContexts[userId] = this.userContexts[userId].slice(-20);
        }
    }

    async ask(userId, prompt, options = {}) {
        this.initUserContext(userId); // 初始化用户上下文

        const payload = {
            model: 'moonshot-v1-8k', // 使用的模型名称
            messages: this.userContexts[userId].concat([{
                role: 'user',
                content: prompt
            }]),
            ...options, // 其他选项，如 temperature、max_tokens 等
        };

        console.log(payload);

        try {
            const response = await axios.post(`${this.baseURL}/chat/completions`, payload, {
                headers: {
                    'Content-Type': 'application/json',
                    Authorization: `Bearer ${this.apiKey}`, // 使用鉴权信息
                },
            });

            if (response.data && response.data.choices) {
                const assistantMessage = response.data.choices[0].message;
                this.updateUserContext(userId, assistantMessage); // 更新用户上下文
                return assistantMessage.content;
            } else {
                throw new Error(
                    `Error from Kimi AI: ${response.data.error || 'Unknown error'}`
                );
            }
        } catch (error) {
            console.error('Error while communicating with Kimi AI:', error.message);
            throw error;
        }
    }
}

export default Kimi;