import type { LLMModelsConfig } from "./types";
import DeepSeekIcon from "@/assets/deepseek-color.svg";
import QwenIcon from "@/assets/qwen.svg";

export const MODEL_CONFIGS = {
  stream: true,
  enableMock: true,
};

export enum LLMProviders {
  SILICON_FLOW = "siliconflow",
  DEEP_SEEK = "deepseek",
  QWEN = "qwen",
}

export enum LLMClientKey {
  openai = "openai",
}

export const LLM_MODELS: LLMModelsConfig[] = [
  {
    // 硅基流动
    providerKey: LLMProviders.SILICON_FLOW,
    apiPath: "https://api.siliconflow.cn/v1",
    apiKey: "",
    models: [
      { name: "deepseek-ai/DeepSeek-R1", iconPath: DeepSeekIcon },
      { name: "Qwen/Qwen3-8B1", iconPath: QwenIcon },
    ],
    available: true,
    clientKey: LLMClientKey.openai,
  },
  {
    // deepseek
    providerKey: LLMProviders.DEEP_SEEK,
    apiPath: "https://api.deepseek.com",
    apiKey: "",
    models: [
      { name: "deepseek-chat", iconPath: DeepSeekIcon },
      { name: "deepseek-reasoner", iconPath: DeepSeekIcon },
    ],
    available: true,
    clientKey: LLMClientKey.openai,
  },
];

export const ESSAY_CN_GRADE = [
  "三年级",
  "四年级",
  "五年级",
  "六年级",
  "七年级",
  "八年级",
  "九年级",
];

export const ESSAY_EN_GRADE = [
  "Grade 3",
  "Grade 4",
  "Grade 5",
  "Grade 6",
  "Grade 7",
  "Grade 8",
  "Grade 9",
];
