From 22f56f07c3a190a8667a6b75d2699c0255643de9 Mon Sep 17 00:00:00 2001
From: Steven <stevenlgtm@gmail.com>
Date: Thu, 20 Apr 2023 21:49:50 +0800
Subject: [PATCH] fix: prompt generator

---
 src/components/ConversationView/index.tsx | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/components/ConversationView/index.tsx b/src/components/ConversationView/index.tsx
index e8663b1..4828123 100644
--- a/src/components/ConversationView/index.tsx
+++ b/src/components/ConversationView/index.tsx
@@ -33,7 +33,9 @@ const ConversationView = () => {
   const [showHeaderShadow, setShowHeaderShadow] = useState<boolean>(false);
   const conversationViewRef = useRef<HTMLDivElement>(null);
   const currentConversation = conversationStore.getConversationById(conversationStore.currentConversationId);
-  const messageList = messageStore.messageList.filter((message) => message.conversationId === currentConversation?.id);
+  const messageList = currentConversation
+    ? messageStore.messageList.filter((message) => message.conversationId === currentConversation.id)
+    : [];
   const lastMessage = last(messageList);
 
   useEffect(() => {
@@ -111,7 +113,8 @@ const ConversationView = () => {
     }
 
     const messageList = messageStore.getState().messageList.filter((message) => message.conversationId === currentConversation.id);
-    let prompt = "";
+    const promptGenerator = getPromptGeneratorOfAssistant(getAssistantById(currentConversation.assistantId)!);
+    let prompt = promptGenerator();
     let tokens = 0;
 
     const message: Message = {
@@ -138,7 +141,6 @@ const ConversationView = () => {
       } catch (error: any) {
         toast.error(error.message);
       }
-      const promptGenerator = getPromptGeneratorOfAssistant(getAssistantById(currentConversation.assistantId)!);
       prompt = promptGenerator(schema);
     }
     let formatedMessageList = [];