UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-03 20:41:29 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: Improve Google Drive picker initialization and error handling

Browse Source
This commit is contained in:
Taylor Wilsdon (aider) 2024-12-15 16:36:45 -05:00
parent eed7cfd2a2
commit 85508106a8
2 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/lib/components/chat/MessageInput.svelte
View File

@ -498,11 +498,15 @@
}}
uploadGoogleDriveHandler={async () => {
try {
if (!import.meta.env.VITE_GOOGLE_API_KEY || !import.meta.env.VITE_GOOGLE_CLIENT_ID) {
throw new Error('Google Drive API credentials not configured');
}
const fileData = await createPicker();
if (fileData) {
dispatch('upload', { type: 'google-drive', data: fileData });
}
} catch (error) {
console.error('Google Drive Error:', error);
toast.error('Error accessing Google Drive: ' + error.message);
}
}}

13
src/lib/utils/google-drive-picker.ts
View File

@ -5,6 +5,7 @@ const SCOPE = ['https://www.googleapis.com/auth/drive.readonly'];
let pickerApiLoaded = false;
let oauthToken: string | null = null;
let initialized = false;
export const loadGoogleDriveApi = () => {
return new Promise((resolve, reject) => {
@ -62,13 +63,17 @@ export const getAuthToken = async () => {
return oauthToken;
};
const initialize = async () => {
if (!initialized) {
await Promise.all([loadGoogleDriveApi(), loadGoogleAuthApi()]);
initialized = true;
}
};
export const createPicker = () => {
return new Promise(async (resolve, reject) => {
try {
if (!pickerApiLoaded) {
await loadGoogleDriveApi();
}
await initialize();
const token = await getAuthToken();
if (!token) {
throw new Error('Unable to get OAuth token');