Get realistic close dates for open deals

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from datetime import date
from clickruntime import on_trigger, credentials, http, ai
SHEET_ID = "your-spreadsheet-id"
INSTANCE = "https://yourcompany.my.salesforce.com"
API = f"{INSTANCE}/services/data/v61.0"
SCHEMA = {"type": "object", "properties": {"dates": {"type": "array", "items": {"type": "string"}}}, "required": ["dates"]}

@on_trigger
def run(event):
    headers = {"Authorization": f"Bearer {credentials.connection('salesforce', 'work')}"}
    opps = http.get(f"{API}/query", headers=headers, params={"q": "SELECT Id, Name, StageName, Amount, CloseDate, "
                    "CreatedDate, PushCount, LastActivityDate FROM Opportunity WHERE IsClosed = false LIMIT 60"}).json()["records"]
    if not opps:
        return {"updated": 0}
    lines = "\n".join(f"{i + 1}. {o['Name']} | {o['StageName']} | {o['Amount']} | created {o['CreatedDate'][:10]} | "
                      f"close {o['CloseDate']} | pushed {o['PushCount']} | last activity {o['LastActivityDate']}" for i, o in enumerate(opps))
    dates = ai.generate(f"Today is {date.today()}. Estimate a realistic close date (YYYY-MM-DD) for each numbered deal, in order.\n"
                        + lines, json_schema=SCHEMA, max_output_tokens=900).json["dates"]
    log = []
    for o, new in zip(opps, dates):
        http.patch(f"{API}/sobjects/Opportunity/{o['Id']}", headers=headers,
                   json={"Predicted_Close_Date__c": new}).raise_for_status()
        log.append([str(date.today()), o["Name"], o["StageName"], o["CloseDate"], new])
    sheets = build("sheets", "v4", credentials=Credentials(credentials.connection("google", "work")),
                   cache_discovery=False).spreadsheets()
    sheets.values().append(spreadsheetId=SHEET_ID, range="Close dates!A1", valueInputOption="RAW",
                           body={"values": log}).execute()
    return {"updated": len(log)}