fix(vector-store): unable to add new files (#181)

Signed-off-by: Petr Kadlec <[email protected]>
i-am-bee · Jan 15, 2025 · c2a7bc2 · c2a7bc2
1 parent c931479
commit c2a7bc2
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 3 deletions.
diff --git a/src/modules/chat/layout/InputBar.tsx b/src/modules/chat/layout/InputBar.tsx
@@ -60,6 +60,7 @@ export const InputBar = memo(function InputBar({
   const {
     files,
     isPending: isFilesPending,
+    hasFilesToUpload,
     dropzone,
     removeFile,
   } = useFilesUpload();
@@ -117,7 +118,10 @@ export const InputBar = memo(function InputBar({
   });
 
   const isSubmitDisabled =
-    isFilesPending || !inputValue || (builderState && !builderState.isSaved);
+    isFilesPending ||
+    hasFilesToUpload ||
+    !inputValue ||
+    (builderState && !builderState.isSaved);
 
   const placeholder = inputPlaceholder
     ? messages.length

diff --git a/src/modules/chat/providers/FilesUploadProvider.tsx b/src/modules/chat/providers/FilesUploadProvider.tsx
@@ -55,6 +55,7 @@ const FilesUploadContext = createContext<{
   files: VectoreStoreFileUpload[];
   attachments: MessageAttachments | null;
   isPending: boolean;
+  hasFilesToUpload: boolean;
   dropzone?: DropzoneState;
   vectorStoreId: string | null;
   removeFile: (id: string) => void;
@@ -66,6 +67,7 @@ const FilesUploadContext = createContext<{
   files: [],
   attachments: null,
   isPending: false,
+  hasFilesToUpload: false,
   vectorStoreId: null,
   removeFile: noop,
   reset: noop,
@@ -87,6 +89,7 @@ export const FilesUploadProvider = ({ children }: PropsWithChildren) => {
     setFiles,
     onFileSubmit,
     isPending,
+    hasFilesToUpload,
     clearFiles,
     setVectorStoreId,
     vectorStoreId,
@@ -244,6 +247,7 @@ export const FilesUploadProvider = ({ children }: PropsWithChildren) => {
       files,
       attachments,
       isPending,
+      hasFilesToUpload,
       dropzone,
       vectorStoreId,
       removeFile,
@@ -256,6 +260,7 @@ export const FilesUploadProvider = ({ children }: PropsWithChildren) => {
     files,
     attachments,
     isPending,
+    hasFilesToUpload,
     dropzone,
     vectorStoreId,
     removeFile,

diff --git a/src/modules/knowledge/create/CreateKnowledgeModal.tsx b/src/modules/knowledge/create/CreateKnowledgeModal.tsx
@@ -39,7 +39,7 @@ import {
   VectorStoreFilesUploadProvider,
 } from '../files/VectorStoreFilesUploadProvider';
 import { useToast } from '@/layout/providers/ToastProvider';
-import { AppProvider, useAppContext } from '@/layout/providers/AppProvider';
+import { useAppContext } from '@/layout/providers/AppProvider';
 
 export interface CreateKnowledgeValues {
   name: string;

diff --git a/src/modules/knowledge/files/VectorStoreFilesUploadProvider.tsx b/src/modules/knowledge/files/VectorStoreFilesUploadProvider.tsx
@@ -56,6 +56,7 @@ export type VectoreStoreFileUpload = {
 const Context = createContext<{
   files: VectoreStoreFileUpload[];
   isPending: boolean;
+  hasFilesToUpload: boolean;
   vectorStoreId: string | null;
   removeFile: (id: string) => void;
   clearFiles: () => void;
@@ -65,6 +66,7 @@ const Context = createContext<{
 }>({
   files: [],
   isPending: false,
+  hasFilesToUpload: false,
   vectorStoreId: null,
   removeFile: noop,
   clearFiles: noop,
@@ -262,7 +264,10 @@ export const VectorStoreFilesUploadProvider = ({
   const value = useMemo(() => {
     return {
       files,
-      isPending: files.some((file) => file.status !== 'complete'),
+      isPending: files.some(
+        (file) => file.status === 'uploading' || file.status === 'embedding',
+      ),
+      hasFilesToUpload: files.some((file) => file.status === 'new'),
       vectorStoreId,
       onFileSubmit: (inputFile: VectoreStoreFileUpload, thread?: Thread) => {
         if (inputFile.isReadable && !vectorStoreIdRef.current)