From bf7cce52db0c8e438d671b8b867282e0f682a36d Mon Sep 17 00:00:00 2001
From: ManishMadan2882 <manishmadan321@gmail.com>
Date: Fri, 20 Sep 2024 02:59:15 +0530
Subject: [PATCH] fix(conv): /search  after setting id

---
 frontend/src/conversation/Conversation.tsx     |  2 +-
 frontend/src/conversation/conversationSlice.ts | 17 +++++++++++++++++
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/frontend/src/conversation/Conversation.tsx b/frontend/src/conversation/Conversation.tsx
index 6884c991..d5908ca3 100644
--- a/frontend/src/conversation/Conversation.tsx
+++ b/frontend/src/conversation/Conversation.tsx
@@ -246,7 +246,7 @@ export default function Conversation() {
         )}
 
         {queries.length > 0 ? (
-          <div className="w-full md:w-8/12 px-1">
+          <div className="w-full md:w-8/12">
             {queries.map((query, index) => {
               return (
                 <Fragment key={index}>
diff --git a/frontend/src/conversation/conversationSlice.ts b/frontend/src/conversation/conversationSlice.ts
index 23962a28..10070a97 100644
--- a/frontend/src/conversation/conversationSlice.ts
+++ b/frontend/src/conversation/conversationSlice.ts
@@ -72,6 +72,23 @@ export const fetchAnswer = createAsyncThunk<Answer, { question: string }>(
                   query: { conversationId: data.id },
                 }),
               );
+              handleSearch(
+                //search for sources post streaming
+                question,
+                state.preference.selectedDocs!,
+                state.conversation.conversationId,
+                state.conversation.queries,
+                state.preference.chunks,
+                state.preference.token_limit,
+              ).then((sources) => {
+                //dispatch streaming sources
+                dispatch(
+                  updateStreamingSource({
+                    index: state.conversation.queries.length - 1,
+                    query: { sources: sources ?? [] },
+                  }),
+                );
+              });
             } else if (data.type === 'error') {
               // set status to 'failed'
               dispatch(conversationSlice.actions.setStatus('failed'));