@rinkuto/koishi-plugin-chatgpt/lib/keyword/ChatGptKeyWord.js

使用open ai的gpt-3.5-turbo模型

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.ChatGptKeyWord = void 0;
const KeyWord_1 = require("./KeyWord");
const AskChatGPT_1 = require("../util/AskChatGPT");
class ChatGptKeyWord extends KeyWord_1.KeyWord {
    async getKeyWords(input) {
        const chatCompletionRequestMessage = [];
        chatCompletionRequestMessage.push({
            role: 'system', content: `请找出最少一个关键词，最多四个关键词，请直接回复关键词，不要加任何东西
      如果没有关键词，请直接回复无，不要加任何句子。
      例如：学习书法，就选唐颜真卿《颜勤礼碑》原碑与对临「第1节」。回复 颜勤礼碑 颜勤礼碑 书法
      例如：你好 回复 无`
        });
        chatCompletionRequestMessage.push(AskChatGPT_1.AskChatGPT.setUserPrompt(input));
        const completion = await AskChatGPT_1.AskChatGPT.getOpenAI().createChatCompletion(AskChatGPT_1.AskChatGPT.getCompletion(chatCompletionRequestMessage), AskChatGPT_1.AskChatGPT.getAxiosConfig());
        const keywords = completion.data.choices[0].message.content.split(' ');
        return keywords.slice(0, this.config.keyWordLength).join(' ');
    }
}
exports.ChatGptKeyWord = ChatGptKeyWord;