Merge pull request #280 from amosproj/function-chat-fix

refactor: Update Chat component to include chat history in
amosproj · Jul 17, 2024 · d44ef01 · d44ef01
2 parents a78bafa + 15249c7
commit d44ef01
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/frontend/screens/Chat/index.tsx b/src/frontend/screens/Chat/index.tsx
@@ -87,13 +87,16 @@ export function Chat() {
       if (!queryText) return;
       if (activeChatId === 'new') {
         const chatId = await createNewChat(text);
-        const data = await fetchResponse(queryText);
+        const { data } = await getResponse({
+          query: queryText,
+          llms: chat?.model || ['gpt-4']
+        });
         await updateChat(chatId, { conversation: arrayUnion({ type: 'AI', message: data }) });
       } else {
         await updateChat(activeChatId, {
           conversation: arrayUnion({ type: 'USER', message: queryText })
         });
-        const data = await fetchResponse(queryText);
+        const { data } = await getResponse({ query: queryText, llms: chat?.model || [] });
         await updateChat(activeChatId, { conversation: arrayUnion({ type: 'AI', message: data }) });
       }
     } catch (error) {