Closing file context manager if upload is success

awslabs · Dec 3, 2024 · 6dcb60b · 6dcb60b
2 parents c32c2f6 + 9394eeb
commit 6dcb60b
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/lib/user-interface/react/package-lock.json b/lib/user-interface/react/package-lock.json
diff --git a/lib/user-interface/react/src/components/chatbot/Chat.tsx b/lib/user-interface/react/src/components/chatbot/Chat.tsx
@@ -434,7 +434,7 @@ export default function Chat ({ sessionId }) {
                         new LisaChatMessage({
                             type: 'ai',
                             content: result,
-                            metadata: metadata,
+                            metadata: useRag ? { ...metadata, ...prev.history[prev.history.length - 1].metadata } : metadata,
                         }),
                     ),
                 }));

diff --git a/lib/user-interface/react/src/components/chatbot/FileUploadModals.tsx b/lib/user-interface/react/src/components/chatbot/FileUploadModals.tsx
@@ -110,7 +110,11 @@ export function ContextUploadModal ({
                     <SpaceBetween direction='horizontal' size='xs'>
                         <Button
                             onClick={async () => {
-                                await handleUpload(selectedFiles, handleError, processFile, [FileTypes.TEXT], 10240);
+                                const successfulUploads = await handleUpload(selectedFiles, handleError, processFile, [FileTypes.TEXT], 10240);
+                                if (successfulUploads.length > 0) {
+                                    notificationService.generateNotification(`Successfully added file(s) to context ${successfulUploads.join(', ')}`, StatusTypes.SUCCESS);
+                                    setShowContextUploadModal(false);
+                                }
                             }}
                             disabled={selectedFiles.length === 0}
                         >