From 12828b265943c5b7b3caabc7409bb06c1d0fad97 Mon Sep 17 00:00:00 2001
From: Matthieu Mayran <mayran@google.com>
Date: Tue, 9 Dec 2025 18:20:31 -0800
Subject: [PATCH] feat: Support custom drive to prevent using prefix in the
 file name

---
 src/scheduler/NotebookScheduler.tsx                | 4 +++-
 src/scheduler/composer/CreateNotebookScheduler.tsx | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/scheduler/NotebookScheduler.tsx b/src/scheduler/NotebookScheduler.tsx
index c3b321e9..8fdd3130 100644
--- a/src/scheduler/NotebookScheduler.tsx
+++ b/src/scheduler/NotebookScheduler.tsx
@@ -96,7 +96,9 @@ const NotebookSchedulerComponent = ({
       const currentTime = new Date().getTime();
       const formattedCurrentTime = formatTimestamp(currentTime);
       setJobNameSelected(`job_${formattedCurrentTime}`);
-      setInputFileSelected(context.path);
+      setInputFileSelected(
+        app.serviceManager.contents.localPath(context.localPath)
+      );
     }
   }, [notebookSelector]);
 
diff --git a/src/scheduler/composer/CreateNotebookScheduler.tsx b/src/scheduler/composer/CreateNotebookScheduler.tsx
index 9a23d550..3877b93e 100644
--- a/src/scheduler/composer/CreateNotebookScheduler.tsx
+++ b/src/scheduler/composer/CreateNotebookScheduler.tsx
@@ -541,7 +541,9 @@ const CreateNotebookScheduler = ({
    */
   useEffect(() => {
     if (context !== '') {
-      setInputFileSelected(context.path);
+      setInputFileSelected(
+        app.serviceManager.contents.localPath(context.localPath)
+      );
     }
     setJobNameSelected('');
     if (!editMode) {