feat(Assistant): Show sources when present

cozy · Oct 23, 2024 · 5c820ae · 5c820ae
1 parent 9f68cc1
commit 5c820ae
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 9 deletions.
diff --git a/src/assistant/AssistantProvider.jsx b/src/assistant/AssistantProvider.jsx
@@ -64,7 +64,9 @@ const AssistantProvider = ({ children }) => {
               }))
             }, 250)
           }
-        } else {
+        }
+
+        if (res.object === 'delta') {
           setAssistantState(v => ({
             ...v,
             message: v.message + res.content,

diff --git a/src/assistant/Conversations/ChatAssistantItem.jsx b/src/assistant/Conversations/ChatAssistantItem.jsx
@@ -5,21 +5,25 @@ import Icon from 'cozy-ui/transpiled/react/Icon'
 
 import AssistantIcon from 'assets/images/icon-assistant.png'
 
+import Sources from './Sources/Sources'
 import ChatItem from './ChatItem'
 
-const ChatAssistantItem = ({ className, label, ...props }) => {
+const ChatAssistantItem = ({ className, label, sources, ...props }) => {
   const { t } = useI18n()
   // need memo to avoid rendering it everytime
   const icon = useMemo(() => <Icon icon={AssistantIcon} size={32} />, [])
 
   return (
-    <ChatItem
-      {...props}
-      className={className}
-      icon={icon}
-      name={t('assistant.name')}
-      label={label}
-    />
+    <>
+      <ChatItem
+        {...props}
+        className={className}
+        icon={icon}
+        name={t('assistant.name')}
+        label={label}
+      />
+      {sources?.length > 0 && <Sources sources={sources} />}
+    </>
   )
 }
 

diff --git a/src/assistant/Conversations/ChatConversation.jsx b/src/assistant/Conversations/ChatConversation.jsx
@@ -43,6 +43,7 @@ const ChatConversation = ({ conversation, myself }) => {
               key={conversation._id + '--' + idx}
               className="u-mt-1-half"
               label={message.content}
+              sources={message.sources}
             />
           )
         }
@@ -56,6 +57,7 @@ const ChatConversation = ({ conversation, myself }) => {
             key={conversation._id + '--' + idx}
             className="u-mt-1-half"
             label={message.content}
+            sources={message.sources}
           />
         )
       })}