Merge branch 'develop' into refactor/argilla-server/better-oauth2-int…

…egration
argilla-io · Nov 18, 2024 · 8700b81 · 8700b81
2 parents a1dbc01 + f9ab910
commit 8700b81
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,
   };
 };