jason/google-mcp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

implement single user mode

Browse Source
This commit is contained in:
Taylor Wilsdon
2025-05-23 11:22:23 -04:00
parent 0a353a5bbc
commit 558d61fc9e
5 changed files with 122 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
gdrive/drive_tools.py
View File

@@ -73,7 +73,9 @@ async def search_drive_files(
try:
service = build('drive', 'v3', credentials=credentials)
user_email_from_creds = credentials.id_token.get('email') if credentials.id_token else 'Unknown (Drive)'
user_email_from_creds = 'Unknown (Drive)'
if credentials.id_token and isinstance(credentials.id_token, dict):
user_email_from_creds = credentials.id_token.get('email', 'Unknown (Drive)')
# Check if the query looks like a structured Drive query or free text
# Basic check for operators or common keywords used in structured queries
@@ -243,7 +245,9 @@ async def list_drive_items(
try:
service = build('drive', 'v3', credentials=credentials)
user_email_from_creds = credentials.id_token.get('email') if credentials.id_token else 'Unknown (Drive)'
user_email_from_creds = 'Unknown (Drive)'
if credentials.id_token and isinstance(credentials.id_token, dict):
user_email_from_creds = credentials.id_token.get('email', 'Unknown (Drive)')
results = await asyncio.to_thread(
service.files().list(
@@ -317,7 +321,9 @@ async def create_drive_file(
try:
service = build('drive', 'v3', credentials=credentials)
user_email_from_creds = credentials.id_token.get('email') if credentials.id_token else 'Unknown (Drive)'
user_email_from_creds = 'Unknown (Drive)'
if credentials.id_token and isinstance(credentials.id_token, dict):
user_email_from_creds = credentials.id_token.get('email', 'Unknown (Drive)')
file_metadata = {
'name': file_name,