feat: 增加限时公益模型，可以免费体验更强的模型

Author: codexu

messages/en.json

@@ -935,7 +935,8 @@
       "header": {
         "configApiKey": "Configure API KEY",
         "clearChat": "Clear Chat",
-        "selectPrompt": "Select Prompt"
+        "selectPrompt": "Select Prompt",
+        "noModel": "AI model not selected"
       },
       "clipboard": {
         "image": {

messages/ja.json

@@ -899,7 +899,8 @@
         "configApiKey": "API KEY の設定",
         "clearChat": "チャットをクリア",
         "configPrompt": "マスクの設定",
-        "selectPrompt": "マスクを選択"
+        "selectPrompt": "マスクを選択",
+        "noModel": "AI モデルが選択されていません"
       },
       "clipboard": {
         "image": {

messages/zh.json

@@ -913,7 +913,8 @@
         "configApiKey": "配置 API KEY",
         "clearChat": "清空聊天",
         "configPrompt": "配置 Prompt",
-        "selectPrompt": "选择 Prompt"
+        "selectPrompt": "选择 Prompt",
+        "noModel": "未选择 AI 模型"
       },
       "clipboard": {
         "image": {

src/app/core/record/chat/chat-header.tsx

@@ -1,12 +1,14 @@
 "use client"
 
-import { BotMessageSquare, Drama } from "lucide-react"
+import { BotMessageSquare, BotOff, Drama } from "lucide-react"
 import usePromptStore from "@/stores/prompt"
 import useSettingStore from "@/stores/setting"
 import { NewChat } from "./new-chat"
 import { RemoveChat } from "./remove-chat"
+import { useTranslations } from "next-intl"
 
 export function ChatHeader() {
+  const t = useTranslations('record.chat.header')
   const { currentPrompt } = usePromptStore()
   const { primaryModel, aiModelList } = useSettingStore()
 
@@ -18,13 +20,17 @@ export function ChatHeader() {
       </div>
       <div className="flex items-center justify-center gap-1">
         {
-          primaryModel &&
+          aiModelList?.find(model => model.key === primaryModel) ?
           <>
             <BotMessageSquare className="!size-4" />
             <span className="line-clamp-1 flex-1 md:flex-none">
               {aiModelList?.find(model => model.key === primaryModel)?.model}
               ({aiModelList?.find(model => model.key === primaryModel)?.title})
             </span>
+          </> :
+          <>
+            <BotOff className="!size-4" />
+            <span>{t('noModel')}</span>
           </>
         }
       </div>

src/app/core/record/chat/model-select.tsx

@@ -26,16 +26,22 @@ import { TooltipButton } from "@/components/tooltip-button"
 
 export function ModelSelect() {
   const [list, setList] = useState<AiConfig[]>([])
-  const { primaryModel, setPrimaryModel, aiModelList } = useSettingStore()
+  const { primaryModel, setPrimaryModel, aiModelList, initSettingData } = useSettingStore()
   const [open, setOpen] = React.useState(false)
   const t = useTranslations('record.chat.input.modelSelect')
 
+
   async function modelSelectChangeHandler(key: string) {
     setPrimaryModel(key)
     const store = await Store.load('store.json');
     store.set('primaryModel', key)
   }
 
+  function handleSetOpen(isOpen: boolean) {
+    setOpen(isOpen)
+    initSettingData()
+  }
+
   // 监听 aiModelList 变化，实时更新本地列表
   useEffect(() => {
     if (aiModelList && aiModelList.length > 0) {
@@ -47,7 +53,7 @@ export function ModelSelect() {
   }, [aiModelList])
 
   return (
-    <Popover open={open} onOpenChange={setOpen}>
+    <Popover open={open} onOpenChange={handleSetOpen}>
       <PopoverTrigger asChild>
         <div className="hidden md:block">
           <TooltipButton

src/app/core/setting/ai/page.tsx

@@ -40,7 +40,7 @@ export default function AiPage() {
   } = useSettingStore()
 
   // 过滤掉默认模型，只显示用户自定义模型
-  const userCustomModels = aiModelList.filter(model => !noteGenModelKeys.includes(model.key))
+  const userCustomModels = aiModelList.filter(model => !noteGenModelKeys.includes(model.key) && model.title !== 'NoteGen Limited')
   const [apiKey, setApiKey] = useState<string>('')
   const [baseURL, setBaseURL] = useState<string>('')
   const [model, setModel] = useState<string>('')

src/stores/setting.ts

@@ -4,6 +4,8 @@ import { getVersion } from '@tauri-apps/api/app'
 import { AiConfig } from '@/app/core/setting/config'
 import { GitlabInstanceType } from '@/lib/gitlab.types'
 import { noteGenDefaultModels, noteGenModelKeys } from '@/app/model-config'
+import { fetch } from '@tauri-apps/plugin-http'
+import { v4 as uuid } from 'uuid';
 
 export enum GenTemplateRange {
   All = 'all',
@@ -210,6 +212,38 @@ const useSettingStore = create<SettingState>((set, get) => ({
       set({ imageMethodModel: 'note-gen-vlm' })
     }
 
+    // 获取 NoteGen 限时免费模型
+    const apiKey = noteGenDefaultModels[0].apiKey
+    const headers = {
+      'Content-Type': 'application/json',
+      'Authorization': `Bearer ${apiKey}`
+    }
+    const res = await fetch('http://api.notegen.top/v1/models', {
+      method: 'GET',
+      headers
+    })
+
+    const resModels = await res.json()
+
+    if (resModels.data && resModels.data.length > 0) {
+      finalAiModelList = finalAiModelList.filter(model => model.title !== 'NoteGen Limited')
+      const limitFreeModels = resModels.data.filter((model: any) => {
+        return noteGenDefaultModels.every(item => item.model !== model.id)
+      }).map((model: any) => ({
+        apiKey,
+        baseURL: "http://api.notegen.top/v1",
+        "key": uuid(),
+        "model": model.id,
+        "modelType": "chat",
+        "temperature": 0.7,
+        "title": "NoteGen Limited",
+        "topP": 1
+      }))
+      finalAiModelList.unshift(...limitFreeModels)
+      await store.set('aiModelList', finalAiModelList)
+      set({ aiModelList: finalAiModelList })
+    }
+
     Object.entries(get()).forEach(async ([key, value]) => {
       const res = await store.get(key)