	In Browser
	StumbleUpon
	del.icio.us
	Google
	Google Buzz
	reddit
	LinkedIn

	Facebook
	Twitter
	Linkedin
	E-Mail

Generative AI > Google Gemini API > ADK with Google Drive - File Management

ADK with Google Drive - File Management

Author: Venkata Sudhakar

Google Drive acts as the document backbone for ShopMax India - storing invoices, contracts, product images, and audit reports. An ADK agent with Drive access can upload generated files, search for existing documents by name or type, move files between folders, and share them with the right people - all through natural language commands.

The Drive API v3 uses the drive.file scope for files the app creates, or drive for broader access. Key operations are files.list for searching, files.create for uploading, files.update for moving between folders, and permissions.create for sharing. Folder IDs replace directory paths - every folder in Drive has a unique ID used as the parent in file operations.

The below example shows an ADK agent with tools to search Drive, upload a file, and share it with a team member.

from google.adk.agents import LlmAgent
from google.adk.tools import FunctionTool
from google.adk.sessions import InMemorySessionService
from google.adk.runners import Runner
from google.oauth2 import service_account
from googleapiclient.discovery import build
from googleapiclient.http import MediaFileUpload

SCOPES = ["https://www.googleapis.com/auth/drive"]
INVOICES_FOLDER_ID = "1FolderIdInvoices123"

def get_drive():
    creds = service_account.Credentials.from_service_account_file(
        "/secrets/sa-key.json", scopes=SCOPES
    )
    return build("drive", "v3", credentials=creds)

def search_files(query: str, max_results: int = 10) -> dict:
    """Search for files in ShopMax Drive by name or type."""
    drive = get_drive()
    results = drive.files().list(
        q=f"name contains '{query}' and trashed = false",
        pageSize=max_results,
        fields="files(id, name, mimeType, modifiedTime, webViewLink)"
    ).execute()
    return {"files": results.get("files", [])}

def upload_file(local_path: str, file_name: str, mime_type: str,
                folder_id: str = INVOICES_FOLDER_ID) -> dict:
    """Upload a local file to a Drive folder."""
    drive = get_drive()
    meta = {"name": file_name, "parents": [folder_id]}
    media = MediaFileUpload(local_path, mimetype=mime_type, resumable=True)
    f = drive.files().create(body=meta, media_body=media,
                              fields="id, webViewLink").execute()
    return {"file_id": f["id"], "link": f.get("webViewLink")}

def share_file(file_id: str, email: str, role: str = "reader") -> dict:
    """Share a Drive file with a user by email."""
    drive = get_drive()
    perm = drive.permissions().create(
        fileId=file_id,
        body={"type": "user", "role": role, "emailAddress": email},
        sendNotificationEmail=True,
    ).execute()
    return {"status": "shared", "permission_id": perm["id"], "email": email}

agent = LlmAgent(
    name="shopmax_drive_agent",
    model="gemini-2.0-flash",
    instruction="You are a ShopMax India file management agent. Help upload, find, and share files on Google Drive.",
    tools=[FunctionTool(search_files), FunctionTool(upload_file), FunctionTool(share_file)],
)

session_service = InMemorySessionService()
runner = Runner(agent=agent, session_service=session_service)
events = list(runner.run(
    user_id="ops_team", session_id="sess_drive_001",
    new_message={"role": "user", "parts": [{
        "text": "Upload /tmp/invoice_april.pdf to the invoices folder and share it with accounts@shopmax.in"
    }]},
))
print(next(e.content.parts[0].text for e in events if e.is_final_response()))

It gives the following output,

Uploaded: invoice_april.pdf
  File ID : 1DriveFileId_XYZ
  Folder  : ShopMax Invoices
  Link    : https://drive.google.com/file/d/1DriveFileId_XYZ/view

Shared with [email protected] (reader access).
Notification email sent.

For automated invoice archiving, trigger this agent via a Cloud Run job whenever a new PDF is generated by the billing system. Use the Drive API files.watch method with a Pub/Sub channel to get push notifications when new files are added to a watched folder - enabling the agent to react in real time rather than polling. Combine with the Sheets agent to log every uploaded invoice to a master invoice register spreadsheet.

Send your comments, suggestions or queries regarding this site to [email protected].