fix: add sync user id aget_files (#3261)

QuivrHQ · Sep 25, 2024 · 0c90254 · 0c90254
1 parent ebeb41e
commit 0c90254
Showing 1 changed file with 28 additions and 6 deletions.
diff --git a/backend/api/quivr_api/modules/sync/utils/sync.py b/backend/api/quivr_api/modules/sync/utils/sync.py
@@ -305,7 +305,11 @@ def get_files(
             raise Exception("Failed to retrieve files")
 
     async def aget_files(
-        self, credentials: Dict, folder_id: str | None = None, recursive: bool = False
+        self,
+        credentials: Dict,
+        folder_id: str | None = None,
+        recursive: bool = False,
+        sync_user_id: int | None = None,
     ) -> List[SyncFile]:
         return self.get_files(credentials, folder_id, recursive)
 
@@ -476,7 +480,11 @@ def fetch_files(endpoint, headers, max_retries=1):
         return files
 
     async def aget_files(
-        self, credentials: Dict, folder_id: str | None = None, recursive: bool = False
+        self,
+        credentials: Dict,
+        folder_id: str | None = None,
+        recursive: bool = False,
+        sync_user_id: int | None = None,
     ) -> List[SyncFile]:
         return self.get_files(credentials, folder_id, recursive)
 
@@ -670,7 +678,11 @@ def fetch_files(metadata):
             raise Exception("Failed to retrieve files")
 
     async def aget_files(
-        self, credentials: Dict, folder_id: str | None = None, recursive: bool = False
+        self,
+        credentials: Dict,
+        folder_id: str | None = None,
+        recursive: bool = False,
+        sync_user_id: int | None = None,
     ) -> List[SyncFile]:
         return self.get_files(credentials, folder_id, recursive)
 
@@ -788,10 +800,11 @@ def check_and_refresh_access_token(self, credentials: Dict) -> Dict:
     async def aget_files(
         self,
         credentials: Dict,
-        sync_user_id: int,
         folder_id: str | None = None,
         recursive: bool = False,
+        sync_user_id: int | None = None,
     ) -> List[SyncFile]:
+        assert sync_user_id, "should not be optional for notion"
         pages = []
 
         if not self.notion:
@@ -818,7 +831,12 @@ async def aget_files(
             pages.append(page_info)
 
             if recursive:
-                sub_pages = await self.aget_files(credentials=credentials, sync_user_id=sync_user_id, folder_id=str(page.id), recursive=recursive)
+                sub_pages = await self.aget_files(
+                    credentials=credentials,
+                    sync_user_id=sync_user_id,
+                    folder_id=str(page.id),
+                    recursive=recursive,
+                )
                 pages.extend(sub_pages)
         return pages
 
@@ -1035,7 +1053,11 @@ def get_files(
             return self.list_github_repos(credentials, recursive=recursive)
 
     async def aget_files(
-        self, credentials: Dict, folder_id: str | None = None, recursive: bool = False
+        self,
+        credentials: Dict,
+        folder_id: str | None = None,
+        recursive: bool = False,
+        sync_user_id: int | None = None,
     ) -> List[SyncFile]:
         return self.get_files(credentials, folder_id, recursive)