Merge pull request #2000 from stakwork/feature/unique-question-ai

feat: added uniqnues for ai questions
stakwork · Aug 6, 2024 · aa42d2e · aa42d2e
2 parents 302d9f4 + 3c3685b
commit aa42d2e
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 11 deletions.
diff --git a/src/components/App/SideBar/index.tsx b/src/components/App/SideBar/index.tsx
@@ -207,14 +207,16 @@ const Content = forwardRef<HTMLDivElement, ContentProp>(({ subViewOpen }, ref) =
           </TrendingWrapper>
         )}
         <Flex>
-          {Object.keys(aiSummaryAnswers).map((i: string) => (
-            <AiSummary
-              key={i}
-              question={aiSummaryAnswers[i]?.question || ''}
-              refId={i}
-              response={aiSummaryAnswers[i]}
-            />
-          ))}
+          {Object.keys(aiSummaryAnswers)
+            .filter((key) => aiSummaryAnswers[key].shouldRender)
+            .map((i: string) => (
+              <AiSummary
+                key={i}
+                question={aiSummaryAnswers[i]?.question || ''}
+                refId={i}
+                response={aiSummaryAnswers[i]}
+              />
+            ))}
 
           {isLoading ? <EpisodeSkeleton /> : !hasAiChats && <LatestView isSearchResult={!!searchTerm || hasAiChats} />}
         </Flex>

diff --git a/src/stores/useDataStore/index.ts b/src/stores/useDataStore/index.ts
@@ -185,11 +185,15 @@ export const useDataStore = create<DataStore>()(
         if (data?.query_data?.ref_id) {
           useAiSummaryStore.setState({ aiRefId: data?.query_data?.ref_id })
 
+          const { aiSummaryAnswers } = useAiSummaryStore.getState()
+          const { answer } = aiSummaryAnswers[data?.query_data?.ref_id] || {}
+
           setAiSummaryAnswer(data?.query_data?.ref_id, {
             question: AISearchQuery,
-            answer: '',
-            answerLoading: true,
-            sourcesLoading: true,
+            answer: answer || '',
+            answerLoading: !answer,
+            sourcesLoading: !answer,
+            shouldRender: true,
           })
         }
 

diff --git a/src/types/index.ts b/src/types/index.ts
@@ -313,6 +313,7 @@ export type AIEntity = {
   questionsLoading?: boolean
   hasBeenRendered?: boolean
   entities?: ExtractedEntity[]
+  shouldRender?: boolean
 }
 
 export interface ExtractedEntity {