Просмотр исходного кода

fix: prevent title update on invalid message response

Dogtiti 1 год назад
Родитель
Сommit
e8581c8f3c
1 измененных файлов с 5 добавлено и 0 удалено
  1. 5 0
      app/store/chat.ts

+ 5 - 0
app/store/chat.ts

@@ -615,6 +615,7 @@ export const useChatStore = createPersistStore(
               providerName,
               providerName,
             },
             },
             onFinish(message) {
             onFinish(message) {
+              if (!isValidMessage(message)) return;
               get().updateCurrentSession(
               get().updateCurrentSession(
                 (session) =>
                 (session) =>
                   (session.topic =
                   (session.topic =
@@ -690,6 +691,10 @@ export const useChatStore = createPersistStore(
             },
             },
           });
           });
         }
         }
+
+        function isValidMessage(message: any): boolean {
+          return typeof message === "string" && !message.startsWith("```json");
+        }
       },
       },
 
 
       updateStat(message: ChatMessage) {
       updateStat(message: ChatMessage) {