Amazon Q Business: Build a Google Drive Upload Plugin

import json
import boto3
from google.oauth2 import service_account
from googleapiclient.discovery import build
from googleapiclient.http import MediaIoBaseUpload
import io

def get_secret():
    client = boto3.client('secretsmanager')
    response = client.get_secret_value(SecretId='google/drive/service-account')
    return json.loads(response['SecretString'])

def lambda_handler(event, context):
    try:
        # Extract file info from Amazon Q request body
        body = json.loads(event['body'])
        file_name = body.get('fileName', 'upload.txt')
        file_content = body.get('content', '')
        folder_id = body.get('folderId')

        creds_json = get_secret()
        scopes = ['https://www.googleapis.com/auth/drive']
        creds = service_account.Credentials.from_service_account_info(creds_json, scopes=scopes)
        
        service = build('drive', 'v3', credentials=creds)
        
        file_metadata = {'name': file_name, 'parents': [folder_id]}
        media = MediaIoBaseUpload(io.BytesIO(file_content.encode('utf-8')), mimetype='text/plain')
        
        file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
        
        return {
            'statusCode': 200,
            'body': json.dumps({'message': 'File uploaded', 'fileId': file.get('id')})
        }
    except Exception as e:
        return {'statusCode': 500, 'body': json.dumps({'error': str(e)})}

openapi: 3.0.0
info:
  title: Google Drive Uploader
  version: 1.0.0
paths:
  /upload:
    post:
      summary: Upload a text file to Google Drive
      description: Uploads content to a specific Google Drive folder.
      operationId: uploadFile
      requestBody:
        required: true
        content:
          application/json:
            schema:
              type: object
              properties:
                fileName:
                  type: string
                  description: The name of the file to save.
                content:
                  type: string
                  description: The text content to write inside the file.
                folderId:
                  type: string
                  description: The Google Drive Folder ID to upload into.
      responses:
        '200':
          description: Successful upload
          content:
            application/json:
              schema:
                type: object
                properties:
                  fileId:
                    type: string