feat: upload to blob based on the user selected namespace

instill-ai · Nov 25, 2024 · c416bc7 · c416bc7
1 parent 5b3260e
commit c416bc7
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 10 deletions.
diff --git a/packages/toolkit/src/view/pipeline/view-pipeline/PipelinePlayground.tsx b/packages/toolkit/src/view/pipeline/view-pipeline/PipelinePlayground.tsx
@@ -187,6 +187,19 @@ export const PipelinePlayground = ({
 
     setIsPipelineRunning(true);
 
+    const targetNamespace = userNamespaces.data.find(
+      (namespace) => namespace.id === navigationNamespaceAnchor,
+    );
+
+    if (!targetNamespace) {
+      toastInstillError({
+        title: "Something went wrong, please refresh the page and try again",
+        toast,
+        error: new Error("Failed to find the target namespace"),
+      });
+      return;
+    }
+
     // Backend need to have the encoded JSON input. So we need to double check
     // the metadata whether this field is a semi-structured object and parse it
     const semiStructuredObjectKeys: string[] = [];
@@ -276,7 +289,7 @@ export const PipelinePlayground = ({
           continue;
         }
         const downloadURL = await uploadAndGetDownloadNamespaceObjectURL({
-          namespaceId: routeInfo.data.namespaceId,
+          namespaceId: targetNamespace.id,
           accessToken,
           object: targetValue,
         });
@@ -318,10 +331,6 @@ export const PipelinePlayground = ({
           );
         }
 
-        const targetNamespace = userNamespaces.data.find(
-          (namespace) => namespace.id === navigationNamespaceAnchor,
-        );
-
         const data = await triggerPipeline.mutateAsync({
           namespaceId: routeInfo.data.namespaceId,
           pipelineId: routeInfo.data.resourceId,

diff --git a/packages/toolkit/src/view/recipe-editor/input/Input.tsx b/packages/toolkit/src/view/recipe-editor/input/Input.tsx
@@ -213,6 +213,12 @@ export const Input = ({
         }
       }
 
+      // We use the current route namespace as the requester namespace
+      // This only happens when the user is on the recipe editor page
+      const tartgetNamespace = userNamespaces.data.find(
+        (ns) => ns.id === routeInfo.data.namespaceId,
+      );
+
       // The data comes from the form is either File or URL for these file related fields
       // like image, video, audio, file
       for (const key of uploadedToArtifactKeys) {
@@ -265,11 +271,6 @@ export const Input = ({
         ),
       );
 
-      // We use the current route namespace as the requester namespace
-      const tartgetNamespace = userNamespaces.data.find(
-        (ns) => ns.id === routeInfo.data.namespaceId,
-      );
-
       const downloadedFromArtifactKeys: string[] = [];
 
       if (outputSchema && outputSchema.properties) {