unwrap kb tags before API call, added more query invalidation for chunks

Author: waleedlatif1

apps/sim/app/workspace/[workspaceId]/knowledge/components/create-base-modal/create-base-modal.tsx

@@ -246,7 +246,7 @@ export function CreateBaseModal({ open, onOpenChange }: CreateBaseModalProps) {
   }
 
   const isSubmitting =
-    createKnowledgeBaseMutation.isPending || deleteKnowledgeBaseMutation.isPending
+    createKnowledgeBaseMutation.isPending || deleteKnowledgeBaseMutation.isPending || isUploading
 
   const onSubmit = async (data: FormValues) => {
     setSubmitStatus(null)

apps/sim/hooks/queries/knowledge.ts

@@ -379,10 +379,13 @@ export function useUpdateChunk() {
 
   return useMutation({
     mutationFn: updateChunk,
-    onSuccess: (_, { knowledgeBaseId }) => {
+    onSuccess: (_, { knowledgeBaseId, documentId }) => {
       queryClient.invalidateQueries({
         queryKey: knowledgeKeys.detail(knowledgeBaseId),
       })
+      queryClient.invalidateQueries({
+        queryKey: knowledgeKeys.document(knowledgeBaseId, documentId),
+      })
     },
   })
 }
@@ -419,10 +422,13 @@ export function useDeleteChunk() {
 
   return useMutation({
     mutationFn: deleteChunk,
-    onSuccess: (_, { knowledgeBaseId }) => {
+    onSuccess: (_, { knowledgeBaseId, documentId }) => {
       queryClient.invalidateQueries({
         queryKey: knowledgeKeys.detail(knowledgeBaseId),
       })
+      queryClient.invalidateQueries({
+        queryKey: knowledgeKeys.document(knowledgeBaseId, documentId),
+      })
     },
   })
 }
@@ -464,10 +470,13 @@ export function useCreateChunk() {
 
   return useMutation({
     mutationFn: createChunk,
-    onSuccess: (_, { knowledgeBaseId }) => {
+    onSuccess: (_, { knowledgeBaseId, documentId }) => {
       queryClient.invalidateQueries({
         queryKey: knowledgeKeys.detail(knowledgeBaseId),
       })
+      queryClient.invalidateQueries({
+        queryKey: knowledgeKeys.document(knowledgeBaseId, documentId),
+      })
     },
   })
 }
@@ -784,10 +793,13 @@ export function useBulkChunkOperation() {
 
   return useMutation({
     mutationFn: bulkChunkOperation,
-    onSuccess: (_, { knowledgeBaseId }) => {
+    onSuccess: (_, { knowledgeBaseId, documentId }) => {
       queryClient.invalidateQueries({
         queryKey: knowledgeKeys.detail(knowledgeBaseId),
       })
+      queryClient.invalidateQueries({
+        queryKey: knowledgeKeys.document(knowledgeBaseId, documentId),
+      })
     },
   })
 }
@@ -806,7 +818,7 @@ export async function updateDocumentTags({
   const response = await fetch(`/api/knowledge/${knowledgeBaseId}/documents/${documentId}`, {
     method: 'PUT',
     headers: { 'Content-Type': 'application/json' },
-    body: JSON.stringify({ tags }),
+    body: JSON.stringify(tags),
   })
 
   if (!response.ok) {