Kaynağa Gözat

fix ts error

lloydzhou 1 yıl önce
ebeveyn
işleme
93ca303b6c
2 değiştirilmiş dosya ile 14 ekleme ve 7 silme
  1. 12 5
      app/store/chat.ts
  2. 2 2
      app/store/config.ts

+ 12 - 5
app/store/chat.ts

@@ -18,6 +18,7 @@ import {
   StoreKey,
   SUMMARIZE_MODEL,
   GEMINI_SUMMARIZE_MODEL,
+  ServiceProvider,
 } from "../constant";
 import Locale, { getLang } from "../locales";
 import { isDalle3, safeLocalStorage } from "../utils";
@@ -107,7 +108,10 @@ function createEmptySession(): ChatSession {
   };
 }
 
-function getSummarizeModel(currentModel: string, providerName: string) {
+function getSummarizeModel(
+  currentModel: string,
+  providerName: string,
+): string[] {
   // if it is using gpt-* models, force to use 4o-mini to summarize
   if (currentModel.startsWith("gpt") || currentModel.startsWith("chatgpt")) {
     const configStore = useAppConfig.getState();
@@ -121,7 +125,10 @@ function getSummarizeModel(currentModel: string, providerName: string) {
       (m) => m.name === SUMMARIZE_MODEL && m.available,
     );
     if (summarizeModel) {
-      return [summarizeModel.name, summarizeModel.providerName];
+      return [
+        summarizeModel.name,
+        summarizeModel.provider?.providerName as string,
+      ];
     }
   }
   if (currentModel.startsWith("gemini")) {
@@ -613,7 +620,7 @@ export const useChatStore = createPersistStore(
               session.mask.modelConfig.model,
               session.mask.modelConfig.providerName,
             );
-        const api: ClientApi = getClientApi(providerName);
+        const api: ClientApi = getClientApi(providerName as ServiceProvider);
 
         // remove error messages if any
         const messages = session.messages;
@@ -822,8 +829,8 @@ export const useChatStore = createPersistStore(
       if (version < 3.3) {
         newState.sessions.forEach((s) => {
           const config = useAppConfig.getState();
-          s.mask.modelConfig.compressModel = undefined;
-          s.mask.modelConfig.compressProviderName = undefined;
+          s.mask.modelConfig.compressModel = "";
+          s.mask.modelConfig.compressProviderName = "";
         });
       }
 

+ 2 - 2
app/store/config.ts

@@ -71,8 +71,8 @@ export const DEFAULT_CONFIG = {
     sendMemory: true,
     historyMessageCount: 4,
     compressMessageLengthThreshold: 1000,
-    compressModel: undefined,
-    compressProviderName: undefined,
+    compressModel: "",
+    compressProviderName: "",
     enableInjectSystemPrompts: true,
     template: config?.template ?? DEFAULT_INPUT_TEMPLATE,
     size: "1024x1024" as DalleSize,