From 2aabd661e32f0ed688ef893431f8fd438ad50b96 Mon Sep 17 00:00:00 2001
From: William <arenabpo@gmail.com>
Date: Fri, 6 Sep 2024 11:43:54 +0800
Subject: [PATCH] add summarize model setting

---
 app/components/settings.tsx | 29 +++++++++++++++++++++++++++++
 app/locales/ar.ts           |  1 +
 app/locales/bn.ts           |  1 +
 app/locales/cn.ts           |  1 +
 app/locales/cs.ts           |  1 +
 app/locales/de.ts           |  1 +
 app/locales/en.ts           |  1 +
 app/locales/es.ts           |  1 +
 app/locales/fr.ts           |  1 +
 app/locales/id.ts           |  1 +
 app/locales/it.ts           |  1 +
 app/locales/jp.ts           |  1 +
 app/locales/ko.ts           |  1 +
 app/locales/no.ts           |  1 +
 app/locales/pt.ts           |  1 +
 app/locales/sk.ts           |  1 +
 app/locales/tr.ts           |  1 +
 app/locales/tw.ts           |  1 +
 app/locales/vi.ts           |  1 +
 app/store/chat.ts           |  5 +++++
 app/store/config.ts         |  1 +
 21 files changed, 53 insertions(+)

diff --git a/app/components/settings.tsx b/app/components/settings.tsx
index ca0a5a187..26f8fcdae 100644
--- a/app/components/settings.tsx
+++ b/app/components/settings.tsx
@@ -81,6 +81,8 @@ import { nanoid } from "nanoid";
 import { useMaskStore } from "../store/mask";
 import { ProviderType } from "../utils/cloud";
 
+import { useAllModels } from "../utils/hooks";
+
 function EditPromptModal(props: { id: string; onClose: () => void }) {
   const promptStore = usePromptStore();
   const prompt = promptStore.get(props.id);
@@ -578,6 +580,8 @@ export function Settings() {
   const config = useAppConfig();
   const updateConfig = config.update;
 
+  const allModels = useAllModels();
+
   const updateStore = useUpdateStore();
   const [checkingUpdate, setCheckingUpdate] = useState(false);
   const currentVersion = updateStore.formatVersion(updateStore.version);
@@ -1448,6 +1452,31 @@ export function Settings() {
             ></input>
           </ListItem>
 
+          {config.enableAutoGenerateTitle && (
+            <ListItem title={Locale.Settings.SummarizeModel}>
+              <Select
+                aria-label={Locale.Settings.SummarizeModel}
+                value={config.summarizeModel}
+                onChange={(e) => {
+                  const [model, _providerName] =
+                    e.currentTarget.value.split("@");
+                  updateConfig((config) => {
+                    console.log("config", config);
+                    config.summarizeModel = model;
+                  });
+                }}
+              >
+                {allModels
+                  .filter((v) => v.available)
+                  .map((v, i) => (
+                    <option value={v.name} key={i}>
+                      {v.displayName}({v.provider?.providerName})
+                    </option>
+                  ))}
+              </Select>
+            </ListItem>
+          )}
+
           <ListItem
             title={Locale.Settings.SendPreviewBubble.Title}
             subTitle={Locale.Settings.SendPreviewBubble.SubTitle}
diff --git a/app/locales/ar.ts b/app/locales/ar.ts
index 9bd491083..4410c2274 100644
--- a/app/locales/ar.ts
+++ b/app/locales/ar.ts
@@ -187,6 +187,7 @@ const ar: PartialLocaleType = {
       Title: "توليد العنوان تلقائيًا",
       SubTitle: "توليد عنوان مناسب بناءً على محتوى الدردشة",
     },
+    SummarizeModel: "النموذج لتوليد العنوان تلقائيًا",
     Sync: {
       CloudState: "بيانات السحابة",
       NotSyncYet: "لم يتم التزامن بعد",
diff --git a/app/locales/bn.ts b/app/locales/bn.ts
index acabc8e2a..81f9b8cf1 100644
--- a/app/locales/bn.ts
+++ b/app/locales/bn.ts
@@ -188,6 +188,7 @@ const bn: PartialLocaleType = {
       Title: "স্বয়ংক্রিয় শিরোনাম জেনারেশন",
       SubTitle: "চ্যাট কনটেন্টের ভিত্তিতে উপযুক্ত শিরোনাম তৈরি করুন",
     },
+    SummarizeModel: "শিরোনাম স্বয়ংক্রিয় জেনারেশনের জন্য নির্দেশনা",
     Sync: {
       CloudState: "ক্লাউড ডেটা",
       NotSyncYet: "এখনো সিঙ্ক করা হয়নি",
diff --git a/app/locales/cn.ts b/app/locales/cn.ts
index 9a3227d68..ce98b3563 100644
--- a/app/locales/cn.ts
+++ b/app/locales/cn.ts
@@ -191,6 +191,7 @@ const cn = {
       Title: "自动生成标题",
       SubTitle: "根据对话内容生成合适的标题",
     },
+    SummarizeModel: "自动生成标题使用模型",
     Sync: {
       CloudState: "云端数据",
       NotSyncYet: "还没有进行过同步",
diff --git a/app/locales/cs.ts b/app/locales/cs.ts
index d16c474e8..35da4f23a 100644
--- a/app/locales/cs.ts
+++ b/app/locales/cs.ts
@@ -188,6 +188,7 @@ const cs: PartialLocaleType = {
       Title: "Automatické generování názvu",
       SubTitle: "Generovat vhodný název na základě obsahu konverzace",
     },
+    SummarizeModel: "Model pro automatické generování názvu",
     Sync: {
       CloudState: "Data na cloudu",
       NotSyncYet: "Ještě nebylo synchronizováno",
diff --git a/app/locales/de.ts b/app/locales/de.ts
index a1f817047..dac8808a4 100644
--- a/app/locales/de.ts
+++ b/app/locales/de.ts
@@ -192,6 +192,7 @@ const de: PartialLocaleType = {
       SubTitle:
         "Basierend auf dem Chat-Inhalt einen passenden Titel generieren",
     },
+    SummarizeModel: "Model für automatische Titelgenerierung",
     Sync: {
       CloudState: "Cloud-Daten",
       NotSyncYet: "Noch nicht synchronisiert",
diff --git a/app/locales/en.ts b/app/locales/en.ts
index 77f3a700a..eee54342a 100644
--- a/app/locales/en.ts
+++ b/app/locales/en.ts
@@ -194,6 +194,7 @@ const en: LocaleType = {
       Title: "Auto Generate Title",
       SubTitle: "Generate a suitable title based on the conversation content",
     },
+    SummarizeModel: "Model For Auto Generate Title",
     Sync: {
       CloudState: "Last Update",
       NotSyncYet: "Not sync yet",
diff --git a/app/locales/es.ts b/app/locales/es.ts
index 5e4f900b7..86ee38263 100644
--- a/app/locales/es.ts
+++ b/app/locales/es.ts
@@ -195,6 +195,7 @@ const es: PartialLocaleType = {
       Title: "Generar título automáticamente",
       SubTitle: "Generar un título adecuado basado en el contenido del chat",
     },
+    SummarizeModel: "Modelo para generar título automáticamente",
     Sync: {
       CloudState: "Datos en la nube",
       NotSyncYet: "Aún no se ha sincronizado",
diff --git a/app/locales/fr.ts b/app/locales/fr.ts
index 65efc32b8..62871e379 100644
--- a/app/locales/fr.ts
+++ b/app/locales/fr.ts
@@ -194,6 +194,7 @@ const fr: PartialLocaleType = {
       SubTitle:
         "Générer un titre approprié en fonction du contenu de la discussion",
     },
+    SummarizeModel: "Modèle pour générer automatiquement le titre",
     Sync: {
       CloudState: "Données cloud",
       NotSyncYet: "Pas encore synchronisé",
diff --git a/app/locales/id.ts b/app/locales/id.ts
index 3ac7af490..453bd254d 100644
--- a/app/locales/id.ts
+++ b/app/locales/id.ts
@@ -189,6 +189,7 @@ const id: PartialLocaleType = {
       Title: "Otomatis Membuat Judul",
       SubTitle: "Membuat judul yang sesuai berdasarkan konten obrolan",
     },
+    SummarizeModel: "Model untuk otomatis membuat judul",
     Sync: {
       CloudState: "Data Cloud",
       NotSyncYet: "Belum disinkronkan",
diff --git a/app/locales/it.ts b/app/locales/it.ts
index 1a54cfa43..c67cb56d4 100644
--- a/app/locales/it.ts
+++ b/app/locales/it.ts
@@ -196,6 +196,7 @@ const it: PartialLocaleType = {
       SubTitle:
         "Genera un titolo appropriato in base al contenuto della conversazione",
     },
+    SummarizeModel: "Modello per generare automaticamente il titolo",
     Sync: {
       CloudState: "Dati cloud",
       NotSyncYet: "Non è ancora avvenuta alcuna sincronizzazione",
diff --git a/app/locales/jp.ts b/app/locales/jp.ts
index 6aaf0ba67..ac78e8ca5 100644
--- a/app/locales/jp.ts
+++ b/app/locales/jp.ts
@@ -188,6 +188,7 @@ const jp: PartialLocaleType = {
       Title: "自動タイトル生成",
       SubTitle: "チャット内容に基づいて適切なタイトルを生成",
     },
+    SummarizeModel: "自動タイトル生成モデル",
     Sync: {
       CloudState: "クラウドデータ",
       NotSyncYet: "まだ同期されていません",
diff --git a/app/locales/ko.ts b/app/locales/ko.ts
index 563827fb9..5e820bf6b 100644
--- a/app/locales/ko.ts
+++ b/app/locales/ko.ts
@@ -187,6 +187,7 @@ const ko: PartialLocaleType = {
       Title: "제목 자동 생성",
       SubTitle: "대화 내용에 따라 적절한 제목 생성",
     },
+    SummarizeModel: "제목 자동 생성 모델",
     Sync: {
       CloudState: "클라우드 데이터",
       NotSyncYet: "아직 동기화되지 않았습니다.",
diff --git a/app/locales/no.ts b/app/locales/no.ts
index d7dc16b3f..46c33a400 100644
--- a/app/locales/no.ts
+++ b/app/locales/no.ts
@@ -193,6 +193,7 @@ const no: PartialLocaleType = {
       Title: "Automatisk generere tittel",
       SubTitle: "Generer en passende tittel basert på samtaleinnholdet",
     },
+    SummarizeModel: "Modell for automatisk generering av tittel",
     Sync: {
       CloudState: "Skydatasynkronisering",
       NotSyncYet: "Har ikke blitt synkronisert ennå",
diff --git a/app/locales/pt.ts b/app/locales/pt.ts
index 9fd13ba1c..bf8f078e1 100644
--- a/app/locales/pt.ts
+++ b/app/locales/pt.ts
@@ -187,6 +187,7 @@ const pt: PartialLocaleType = {
       Title: "Gerar Título Automaticamente",
       SubTitle: "Gerar um título adequado baseado no conteúdo da conversa",
     },
+    SummarizeModel: "Modelo para gerar automaticamente o título",
     Sync: {
       CloudState: "Última Atualização",
       NotSyncYet: "Ainda não sincronizado",
diff --git a/app/locales/sk.ts b/app/locales/sk.ts
index 2586aaaa7..7e2728119 100644
--- a/app/locales/sk.ts
+++ b/app/locales/sk.ts
@@ -189,6 +189,7 @@ const sk: PartialLocaleType = {
       Title: "Automaticky generovať názov",
       SubTitle: "Generovať vhodný názov na základe obsahu konverzácie",
     },
+    SummarizeModel: "Model na automatické generovanie názvu",
     Sync: {
       CloudState: "Posledná aktualizácia",
       NotSyncYet: "Zatiaľ nesynchronizované",
diff --git a/app/locales/tr.ts b/app/locales/tr.ts
index ac410615e..af3fcc2bb 100644
--- a/app/locales/tr.ts
+++ b/app/locales/tr.ts
@@ -188,6 +188,7 @@ const tr: PartialLocaleType = {
       Title: "Başlığı Otomatik Oluştur",
       SubTitle: "Sohbet içeriğine göre uygun başlık oluştur",
     },
+    SummarizeModel: "Başlık Otomatik Oluşturma Modeli",
     Sync: {
       CloudState: "Bulut Verisi",
       NotSyncYet: "Henüz senkronize edilmedi",
diff --git a/app/locales/tw.ts b/app/locales/tw.ts
index 6b2c0fd65..db0095f67 100644
--- a/app/locales/tw.ts
+++ b/app/locales/tw.ts
@@ -186,6 +186,7 @@ const tw = {
       Title: "自動產生標題",
       SubTitle: "根據對話內容產生合適的標題",
     },
+    SummarizeModel: "自動產生標題模型",
     Sync: {
       CloudState: "雲端資料",
       NotSyncYet: "還沒有進行過同步",
diff --git a/app/locales/vi.ts b/app/locales/vi.ts
index 9a21ee406..f57d3a978 100644
--- a/app/locales/vi.ts
+++ b/app/locales/vi.ts
@@ -188,6 +188,7 @@ const vi: PartialLocaleType = {
       Title: "Tự động tạo tiêu đề",
       SubTitle: "Tạo tiêu đề phù hợp dựa trên nội dung cuộc trò chuyện",
     },
+    SummarizeModel: "Mô hình tự động tạo tiêu đề",
     Sync: {
       CloudState: "Dữ liệu đám mây",
       NotSyncYet: "Chưa thực hiện đồng bộ",
diff --git a/app/store/chat.ts b/app/store/chat.ts
index de2a63078..d8dff2173 100644
--- a/app/store/chat.ts
+++ b/app/store/chat.ts
@@ -92,6 +92,11 @@ function createEmptySession(): ChatSession {
 }
 
 function getSummarizeModel(currentModel: string) {
+  const config = useAppConfig.getState();
+  if (config.summarizeModel) {
+    return config.summarizeModel;
+  }
+
   // if it is using gpt-* models, force to use 4o-mini to summarize
   if (currentModel.startsWith("gpt")) {
     const configStore = useAppConfig.getState();
diff --git a/app/store/config.ts b/app/store/config.ts
index e8e3c9863..f9079e09b 100644
--- a/app/store/config.ts
+++ b/app/store/config.ts
@@ -39,6 +39,7 @@ export const DEFAULT_CONFIG = {
   tightBorder: !!config?.isApp,
   sendPreviewBubble: true,
   enableAutoGenerateTitle: true,
+  summarizeModel: "gpt-3.5-turbo" as ModelType,
   sidebarWidth: DEFAULT_SIDEBAR_WIDTH,
 
   disablePromptHint: false,