import { Configuration, OpenAIApi } from "openai";

const params = {
  organization: "org-ew4m4pHcq5yYaTGF4VtbSMGH",
  apiKey: "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",
  // organization: "YOUR OWN ORGANIZATION HERE",
  // apiKey: "YOUR OWN APIKEY HERE",
};

export const OpenAIService = async () => {
  const config = new Configuration(params);
  const openai = new OpenAIApi(config);
  return {
    openai,
    getPrompts: async (keywords: string[]) => {
      const resp = await openai.createChatCompletion({
        model: "gpt-3.5-turbo",
        messages: [
          {
            role: "user",
            content: `我想用NovelAI生成一张${keywords.join(
              ","
            )}画像，请为我生成不少于20个提示词，请使用英文表示。`,
          },
        ],
      });
      const content = resp.data.choices[0].message?.content;
      const prompts = content?.replace(/\d|\./g, "").split("\n");
      return prompts;
    },
  };
};