Turn PDFs in Drive into two versions of a multiple-choice test

import io
from pypdf import PdfReader
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, state, ai, report_fetched

FOLDER_ID = "your-test-material-folder-id"

@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    drive, docs = build("drive", "v3", credentials=creds, cache_discovery=False), build("docs", "v1", credentials=creds, cache_discovery=False)
    seen = set(state.get("seen") or [])
    files = drive.files().list(q=f"'{FOLDER_ID}' in parents and mimeType='application/pdf' and trashed=false", fields="files(id,name)").execute()["files"]
    new = [f for f in files if f["id"] not in seen]
    report_fetched(len(new))
    if not new:
        return {"summary": "Nothing new"}
    for f in new:
        text = "".join(p.extract_text() or "" for p in PdfReader(io.BytesIO(drive.files().get_media(fileId=f["id"]).execute())).pages)[:20000]
        for version in ("A", "B"):
            test = ai.generate(f"Write test version {version}: 10 multiple-choice questions (4 options each) on this material, "
                               f"different from any other version, then an answer key.\n{text}", max_output_tokens=2000).text
            doc = docs.documents().create(body={"title": f"{f['name']} - Test {version}"}).execute()
            docs.documents().batchUpdate(documentId=doc["documentId"], body={"requests": [{"insertText": {"location": {"index": 1}, "text": test}}]}).execute()
            drive.files().update(fileId=doc["documentId"], addParents=FOLDER_ID, removeParents="root").execute()
        seen.add(f["id"])
    state.set("seen", list(seen)[-500:])
    return {"pdfs": len(new)}