@aglyn/plugins-ai 1.0.0-beta.227 → 1.0.0-beta.229

package/src/lib/jobs/ai-jobs.js

@@ -43,6 +43,8 @@ import { assistExchangeExpiry, assistUsageDay, assistUsageMonth, recordAssistCos
 import { aiJobRefundCredits, refundJobCredits } from "../usage/assist-job-refund.js";
 import { freeAssistAccount } from "../usage/assist-free-taste.js";
 import { aiJobAutoConfirms } from "./ai-job-auto-confirm.js";
+import { AI_SITE_GUIDED_BUILD_FAILED_COPY } from "../model/ai-job-failure-copy.js";
+import { aiSiteStartInputsOf } from "../model/ai-site-start.js";
 import { addAdminAudit } from "@aglyn/tenant-data-admin/server/admin-audit-write";
 /**
  * AI generation jobs — the Firestore state machine (AGL-2904).
@@ -425,14 +427,38 @@ export async function getAiJob(firestore, orgId, jobId) {
     else if ((_options_statuses = options.statuses) == null ? void 0 : _options_statuses.length) query = query.where('status', 'in', [
         ...options.statuses
     ]);
-    const snapshot = await query.orderBy('createdAt', 'desc').limit(limit).get();
-    return snapshot.docs.map((doc)=>jobFrom(doc)).filter((job)=>job !== null);
+    else if (options.hostId) query = query.where('hostId', '==', options.hostId);
+    const read = async (from, count)=>(await from.orderBy('createdAt', 'desc').limit(count).get()).docs.map((doc)=>jobFrom(doc)).filter((job)=>job !== null);
+    try {
+        return await read(query, limit);
+    } catch (error) {
+        var _options_statuses1;
+        // A site's list while its (hostId, createdAt) index is still building
+        // (AGL-3596): Firestore refuses the query with FAILED_PRECONDITION until
+        // the index is ready, which answered the site's AI jobs page with a 500.
+        // The org's own newest jobs — the index every org list already uses —
+        // filtered to the site in memory stand in, bounded to
+        // `AI_JOBS_HOST_FALLBACK_SCAN` jobs, so a site whose jobs are older than
+        // that lists fewer until the index is ready rather than none.
+        if (!options.hostId || options.status || ((_options_statuses1 = options.statuses) == null ? void 0 : _options_statuses1.length) || !isFailedPrecondition(error)) throw error;
+        console.warn('ai jobs: host index not ready; listing from the org', {
+            orgId,
+            hostId: options.hostId
+        });
+        const recent = await read(jobsCollection(firestore, orgId), AI_JOBS_HOST_FALLBACK_SCAN);
+        return recent.filter((job)=>job.hostId === options.hostId).slice(0, limit);
+    }
+}
+/** The most of an org's newest jobs a site's list scans while its index builds (AGL-3596). */ export const AI_JOBS_HOST_FALLBACK_SCAN = 300;
+/** Firestore's "this query needs an index that is not ready" (gRPC 9). */ function isFailedPrecondition(error) {
+    const code = error == null ? void 0 : error.code;
+    return code === 9 || code === 'failed-precondition' || code === 'FAILED_PRECONDITION';
 }
 /** The wire form. `running` is the lease, read against `now`. */ export function aiJobSummary(job, now = new Date()) {
     var _job_hostId, _job_steps, _job_outputs, _job_creditsReserved, _job_creditsSpent, _job_refundedCredits, _toIso, _toIso1, _job_error, _job_plan_labels, _job_plan_confirmedBy, _job_review, _job_applied_versions, _job_applied_staged;
     var _job_lease, _job_inputs;
     const leaseUntil = toMillis((_job_lease = job.lease) == null ? void 0 : _job_lease.until);
-    return {
+    return _extends({
         id: job.$id,
         orgId: job.orgId,
         hostId: (_job_hostId = job.hostId) != null ? _job_hostId : null,
@@ -473,7 +499,11 @@ export async function getAiJob(firestore, orgId, jobId) {
             versions: (_job_applied_versions = job.applied.versions) != null ? _job_applied_versions : {},
             staged: (_job_applied_staged = job.applied.staged) != null ? _job_applied_staged : []
         } : null
-    };
+    }, job.sitePublish ? {
+        sitePublish: job.sitePublish
+    } : {}, job.kind === 'site' ? {
+        siteInputs: aiSiteStartInputsOf(job.inputs)
+    } : {});
 }
 function leaseIsLive(job, owner, nowMs) {
     const lease = job.lease;
@@ -688,6 +718,8 @@ function nextStepIndex(job) {
             lease: null
         } : {}, input.plan ? {
             plan: input.plan
+        } : {}, input.sitePublish ? {
+            sitePublish: input.sitePublish
         } : {}, parksForReview && input.review ? {
             review: input.review,
             error: input.review.reason === 'plan' ? null : input.review.message
@@ -702,10 +734,11 @@ function nextStepIndex(job) {
             parked: parksForReview && job.status !== 'needs_review'
         };
     });
-    if (recorded.parked) await announceAiJobTransition(recorded.job, 'needs-review');
+    if (recorded.parked && !input.quiet) await announceAiJobTransition(recorded.job, 'needs-review');
     return {
         job: recorded.job,
-        remaining: recorded.remaining
+        remaining: recorded.remaining,
+        parked: recorded.parked
     };
 }
 async function transition(firestore, orgId, jobId, apply) {
@@ -863,6 +896,21 @@ async function transition(firestore, orgId, jobId, apply) {
         return job;
     }
 }
+/**
+ * A failure on our side, given back BEFORE the job is failed (AGL-3596): the
+ * failed state is the last one a follower of the job reads and the one its
+ * notice is written from, so it already carries `refundedCredits`. A job that
+ * ended meanwhile — a person's cancel, which pays for what ran — is left as it
+ * ended and given nothing back.
+ */ async function failOurFailure(firestore, orgId, jobId, message, detail, refund, now) {
+    const current = await getAiJob(firestore, orgId, jobId);
+    if (current && !isAiJobTerminal(current.status)) {
+        await refundOurFailure(firestore, orgId, current, _extends({}, refund, {
+            now
+        }));
+    }
+    return failAiJob(firestore, orgId, jobId, message, detail, now);
+}
 /**
  * Resume a job that stopped for a person (AGL-2935): confirm its plan, or
  * try again the step whose answer broke a building rule. One transaction,
@@ -1000,7 +1048,7 @@ async function transition(firestore, orgId, jobId, apply) {
  * round.
  */ export async function runAiJobStep(firestore, orgId, jobId, options) {
     var _options_now, _step_attempts, _options_org, _data, _ref, _ref1, _outcome_effort, _step_passes, _options_actor;
-    var _reservation_allotment, _reservation_allotment1;
+    var _reservation_allotment, _reservation_allotment1, _outcome_review;
     const now = (_options_now = options.now) != null ? _options_now : new Date();
     // A reservation the caller took before the job existed is a message
     // counted against the org; a step that fails before the provider is
@@ -1041,32 +1089,28 @@ async function transition(firestore, orgId, jobId, apply) {
     const runner = aiJobRunnerForStep(job.kind, step.name);
     if (!runner) {
         await releaseHeld();
-        const failed = await failAiJob(firestore, orgId, jobId, AI_JOB_NOT_AVAILABLE_COPY, {
-            stepIndex,
-            error: `no runner registered for kind ${job.kind}`
-        }, now);
         return {
             outcome: 'failed',
-            job: await refundOurFailure(firestore, orgId, failed, {
+            job: await failOurFailure(firestore, orgId, jobId, AI_JOB_NOT_AVAILABLE_COPY, {
+                stepIndex,
+                error: `no runner registered for kind ${job.kind}`
+            }, {
                 reason: 'unavailable',
-                stepCredits: 0,
-                now
-            })
+                stepCredits: 0
+            }, now)
         };
     }
     if (((_step_attempts = step.attempts) != null ? _step_attempts : 0) > AI_JOB_STEP_MAX_ATTEMPTS) {
         await releaseHeld();
-        const failed = await failAiJob(firestore, orgId, jobId, AI_UPSTREAM_FAILURE_COPY, {
-            stepIndex,
-            error: `step ${step.name} exhausted ${AI_JOB_STEP_MAX_ATTEMPTS} attempts`
-        }, now);
         return {
             outcome: 'failed',
-            job: await refundOurFailure(firestore, orgId, failed, {
+            job: await failOurFailure(firestore, orgId, jobId, AI_UPSTREAM_FAILURE_COPY, {
+                stepIndex,
+                error: `step ${step.name} exhausted ${AI_JOB_STEP_MAX_ATTEMPTS} attempts`
+            }, {
                 reason: 'timeout',
-                stepCredits: 0,
-                now
-            })
+                stepCredits: 0
+            }, now)
         };
     }
     const org = (_options_org = options.org) != null ? _options_org : (_data = (await firestore.collection('orgs').doc(orgId).get()).data()) != null ? _data : {};
@@ -1192,18 +1236,16 @@ async function transition(firestore, orgId, jobId, apply) {
                 job: requeued
             };
         }
-        const failed = await failAiJob(firestore, orgId, jobId, AI_UPSTREAM_FAILURE_COPY, {
-            stepIndex,
-            error
-        }, now);
         return {
             outcome: 'failed',
-            job: await refundOurFailure(firestore, orgId, failed, {
+            job: await failOurFailure(firestore, orgId, jobId, AI_UPSTREAM_FAILURE_COPY, {
+                stepIndex,
+                error
+            }, {
                 reason: 'provider',
                 stepCredits: 0,
-                free: (_reservation_free = reservation.free) != null ? _reservation_free : null,
-                now
-            })
+                free: (_reservation_free = reservation.free) != null ? _reservation_free : null
+            }, now)
         };
     }
     const latencyMs = Math.max(0, Date.now() - runStarted);
@@ -1223,18 +1265,16 @@ async function transition(firestore, orgId, jobId, apply) {
             creditsSpent: 0,
             error: outcome.failure
         }, now);
-        const failed = await failAiJob(firestore, orgId, jobId, outcome.failure, {
-            stepIndex,
-            error: `step failure before the provider: ${outcome.failure}`
-        }, now);
         return {
             outcome: 'failed',
-            job: await refundOurFailure(firestore, orgId, failed, {
+            job: await failOurFailure(firestore, orgId, jobId, outcome.failure, {
+                stepIndex,
+                error: `step failure before the provider: ${outcome.failure}`
+            }, {
                 reason: 'step-failure',
                 stepCredits: 0,
-                free: (_reservation_free1 = reservation.free) != null ? _reservation_free1 : null,
-                now
-            })
+                free: (_reservation_free1 = reservation.free) != null ? _reservation_free1 : null
+            }, now)
         };
     }
     const tokens = {
@@ -1298,19 +1338,45 @@ async function transition(firestore, orgId, jobId, apply) {
             error: message,
             tokens
         }, now);
-        const failed = await failAiJob(firestore, orgId, jobId, message, {
+        const detail = {
             stepIndex,
             error: outcome.refused ? 'stop_reason refusal' : `step failure: ${message}`
-        }, now);
+        };
         // A model declining the brief is not our failure; a step's own is (AGL-3594).
         return {
             outcome: 'failed',
-            job: outcome.refused ? failed : await refundOurFailure(firestore, orgId, failed, {
+            job: outcome.refused ? await failAiJob(firestore, orgId, jobId, message, detail, now) : await failOurFailure(firestore, orgId, jobId, message, detail, {
                 reason: 'step-failure',
                 stepCredits: credits,
-                free: (_reservation_free3 = reservation.free) != null ? _reservation_free3 : null,
-                now
-            })
+                free: (_reservation_free3 = reservation.free) != null ? _reservation_free3 : null
+            }, now)
+        };
+    }
+    // A guided site start whose step broke a building rule after its re-ask
+    // fails, given back whole (AGL-3596): the person never saw the plan, so a
+    // review asking them to look at it again is a review of nothing. The rule
+    // and its findings go to the log for staff; the person reads one sentence,
+    // and Try again starts a fresh job.
+    if (((_outcome_review = outcome.review) == null ? void 0 : _outcome_review.reason) === 'doctrine' && aiJobAutoConfirms(job) && step.name !== AI_JOB_PLAN_STEP) {
+        var _outcome_review_findings, _outcome_review_message, _reservation_free4;
+        await recordStep(firestore, orgId, jobId, options.owner, stepIndex, _extends({
+            status: 'failed',
+            creditsSpent: credits,
+            error: AI_SITE_GUIDED_BUILD_FAILED_COPY,
+            outputs: outcome.outputs
+        }, spentNothing ? {} : {
+            tokens
+        }), now);
+        return {
+            outcome: 'failed',
+            job: await failOurFailure(firestore, orgId, jobId, AI_SITE_GUIDED_BUILD_FAILED_COPY, {
+                stepIndex,
+                error: `doctrine refusal after the re-ask: ${JSON.stringify((_outcome_review_findings = outcome.review.findings) != null ? _outcome_review_findings : [])} ${(_outcome_review_message = outcome.review.message) != null ? _outcome_review_message : ''}`
+            }, {
+                reason: 'doctrine-refused',
+                stepCredits: credits,
+                free: (_reservation_free4 = reservation.free) != null ? _reservation_free4 : null
+            }, now)
         };
     }
     // A step that stopped for a person parks the job in the same write that
@@ -1347,6 +1413,10 @@ async function transition(firestore, orgId, jobId, apply) {
         plan: outcome.plan
     } : {}, review ? {
         review
+    } : {}, outcome.sitePublish ? {
+        sitePublish: outcome.sitePublish
+    } : {}, review ? {
+        quiet: true
     } : {}), now);
     const actor = (_options_actor = options.actor) != null ? _options_actor : {
         uid: job.createdBy
@@ -1403,7 +1473,7 @@ async function transition(firestore, orgId, jobId, apply) {
         };
     }
     if (review && recorded.job.status === 'needs_review') {
-        var _reservation_free4;
+        var _reservation_free5;
         // Nobody parked the job but its own step, so the row names no actor, as
         // the meter's park does.
         await logAiJobNeedsInput(orgId, {
@@ -1413,15 +1483,22 @@ async function transition(firestore, orgId, jobId, apply) {
             kind: job.kind,
             reason: review.reason
         });
-        // A plan the plan rules still refused is our failure (AGL-3594).
+        // A plan the plan rules still refused, and a step that still broke a
+        // building rule after its re-ask, are our failures (AGL-3594, AGL-3596):
+        // given back before the person is told, so the notice can say so. Trying
+        // again spends anew, and a second refusal gives back again, within the
+        // day's bound.
+        const ours = outcome.uncredited ? 'plan-refused' : review.reason === 'doctrine' ? 'doctrine-refused' : null;
+        const parked = ours ? await refundOurFailure(firestore, orgId, recorded.job, {
+            reason: ours,
+            stepCredits: credits,
+            free: (_reservation_free5 = reservation.free) != null ? _reservation_free5 : null,
+            now
+        }) : recorded.job;
+        if (recorded.parked) await announceAiJobTransition(parked, 'needs-review');
         return {
             outcome: 'needs_review',
-            job: outcome.uncredited ? await refundOurFailure(firestore, orgId, recorded.job, {
-                reason: 'plan-refused',
-                stepCredits: credits,
-                free: (_reservation_free4 = reservation.free) != null ? _reservation_free4 : null,
-                now
-            }) : recorded.job
+            job: parked
         };
     }
     if (recorded.remaining > 0 || isAiJobTerminal(recorded.job.status)) {