diff --git a/llama-index-integrations/readers/llama-index-readers-google/llama_index/readers/google/drive/base.py b/llama-index-integrations/readers/llama-index-readers-google/llama_index/readers/google/drive/base.py index d07a2de0b2341527aa12015415541b36d8b58503..bbe652eeed5efb7eeb9d032dc9a0654f89befc89 100644 --- a/llama-index-integrations/readers/llama-index-readers-google/llama_index/readers/google/drive/base.py +++ b/llama-index-integrations/readers/llama-index-readers-google/llama_index/readers/google/drive/base.py @@ -128,11 +128,11 @@ class GoogleDriveReader(BasePydanticReader): # First, we need the Google API credentials for the app creds = None - if Path(self.authorized_user_info).exists(): + if self.authorized_user_info is not None: creds = Credentials.from_authorized_user_info( self.authorized_user_info, SCOPES ) - elif Path(self.service_account_key).exists(): + elif self.service_account_key is not None: return service_account.Credentials.from_service_account_info( self.service_account_key, scopes=SCOPES ) @@ -396,6 +396,7 @@ class GoogleDriveReader(BasePydanticReader): def _load_from_file_ids( self, + drive_id: Optional[str], file_ids: List[str], mime_types: Optional[List[str]], query_string: Optional[str], @@ -414,6 +415,7 @@ class GoogleDriveReader(BasePydanticReader): for file_id in file_ids: fileids_meta.extend( self._get_fileids_meta( + drive_id=drive_id, file_id=file_id, mime_types=mime_types, query_string=query_string, @@ -427,7 +429,7 @@ class GoogleDriveReader(BasePydanticReader): def _load_from_folder( self, - drive_id: str, + drive_id: Optional[str], folder_id: str, mime_types: Optional[List[str]], query_string: Optional[str],