[BUGFIX] Show Import data if user is admin or owner (#5688)

FIx #5686
argilla-io · Nov 18, 2024 · f9ab910 · f9ab910
1 parent 31b3c6a
commit f9ab910
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/argilla-frontend/pages/dataset/_id/annotation-mode/index.vue b/argilla-frontend/pages/dataset/_id/annotation-mode/index.vue
@@ -12,6 +12,7 @@
         >
           <template slot="dialog-cta" v-if="dataset && dataset.createdFromUI">
             <ImportData
+              v-if="isAdminOrOwner"
               :snippet="dataset.createCodeSnippetFromHub(getUser())"
             />
           </template>

diff --git a/argilla-frontend/pages/dataset/_id/annotation-mode/useAnnotationModeViewModel.ts b/argilla-frontend/pages/dataset/_id/annotation-mode/useAnnotationModeViewModel.ts
@@ -9,10 +9,11 @@ import { useDatasetViewModel } from "../useDatasetViewModel";
 import { GetDatasetByIdUseCase } from "@/v1/domain/usecases/get-dataset-by-id-use-case";
 import { useDataset } from "@/v1/infrastructure/storage/DatasetStorage";
 import { RecordCriteria } from "~/v1/domain/entities/record/RecordCriteria";
-import { useRoutes, useUser } from "~/v1/infrastructure/services";
+import { useRoutes, useUser, useRole } from "~/v1/infrastructure/services";
 import { RecordStatus } from "~/v1/domain/entities/record/RecordAnswer";
 
 export const useAnnotationModeViewModel = () => {
+  const { isAdminOrOwner } = useRole();
   const router = useRouter();
   const routes = useRoutes();
   const { getUser } = useUser();
@@ -113,5 +114,6 @@ export const useAnnotationModeViewModel = () => {
     breadcrumbs,
     updateQueryParams,
     getUser,
+    isAdminOrOwner,
   };
 };