Draft LinkedIn posts inspired by high-performing ones you collect

from datetime import date
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, http, ai

AUTHOR, SHEET_ID = "urn:li:person:abc123", "your-spreadsheet-id"
VOICE = "Sales coach for B2B founders. Direct, practical, short sentences, real examples."

@on_trigger
def run(event):
    values = build("sheets", "v4", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).spreadsheets().values()
    swipe = [r[0][:800] for r in values.get(spreadsheetId=SHEET_ID, range="Swipe!A2:A").execute().get("values", []) if r][-15:]
    out = ai.generate(f"Voice: {VOICE}\nHigh-performing posts:\n" + "\n---\n".join(swipe) +
                      "\nWrite 3 original drafts on my own topics using what made these work. Mark the best.", max_output_tokens=1500,
                      json_schema={"type": "object", "properties": {"drafts": {"type": "array", "items": {"type": "string"}}, "best": {"type": "integer"}}}).json
    best = out["drafts"][out["best"]]
    http.post("https://api.linkedin.com/rest/posts",
              headers={"Authorization": f"Bearer {credentials.connection('linkedin', 'work')}", "LinkedIn-Version": "202409"},
              json={"author": AUTHOR, "commentary": best, "visibility": "PUBLIC", "lifecycleState": "PUBLISHED",
                    "distribution": {"feedDistribution": "MAIN_FEED"}})
    rows = [[date.today().isoformat(), d, "posted" if d == best else ""] for d in out["drafts"]]
    values.append(spreadsheetId=SHEET_ID, range="Drafts!A:C", valueInputOption="USER_ENTERED", body={"values": rows}).execute()
    return {"drafts": len(rows)}