@aztec/prover-client 0.0.0-test.0

package/dest/proving_broker/proof_store/gcs_proof_store.d.ts

@@ -0,0 +1,14 @@
+import { type ProofUri, type ProvingJobId, type ProvingJobInputs, type ProvingJobInputsMap, type ProvingJobResult, type ProvingJobResultsMap } from '@aztec/stdlib/interfaces/server';
+import { ProvingRequestType } from '@aztec/stdlib/proofs';
+import type { ProofStore } from './proof_store.js';
+export declare class GoogleCloudStorageProofStore implements ProofStore {
+    private readonly bucketName;
+    private readonly path;
+    private readonly storage;
+    constructor(bucketName: string, path: string);
+    saveProofInput<T extends ProvingRequestType>(id: ProvingJobId, type: T, inputs: ProvingJobInputsMap[T]): Promise<ProofUri>;
+    saveProofOutput<T extends ProvingRequestType>(_id: ProvingJobId, _type: T, _result: ProvingJobResultsMap[T]): Promise<ProofUri>;
+    getProofInput(uri: ProofUri): Promise<ProvingJobInputs>;
+    getProofOutput(_uri: ProofUri): Promise<ProvingJobResult>;
+}
+//# sourceMappingURL=gcs_proof_store.d.ts.map

package/dest/proving_broker/proof_store/gcs_proof_store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gcs_proof_store.d.ts","sourceRoot":"","sources":["../../../src/proving_broker/proof_store/gcs_proof_store.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,QAAQ,EACb,KAAK,YAAY,EACjB,KAAK,gBAAgB,EACrB,KAAK,mBAAmB,EACxB,KAAK,gBAAgB,EACrB,KAAK,oBAAoB,EAE1B,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAK1D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAInD,qBAAa,4BAA6B,YAAW,UAAU;IAGjD,OAAO,CAAC,QAAQ,CAAC,UAAU;IAAU,OAAO,CAAC,QAAQ,CAAC,IAAI;IAFtE,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAU;gBAEL,UAAU,EAAE,MAAM,EAAmB,IAAI,EAAE,MAAM;IAIjE,cAAc,CAAC,CAAC,SAAS,kBAAkB,EACtD,EAAE,EAAE,YAAY,EAChB,IAAI,EAAE,CAAC,EACP,MAAM,EAAE,mBAAmB,CAAC,CAAC,CAAC,GAC7B,OAAO,CAAC,QAAQ,CAAC;IAOpB,eAAe,CAAC,CAAC,SAAS,kBAAkB,EAC1C,GAAG,EAAE,YAAY,EACjB,KAAK,EAAE,CAAC,EACR,OAAO,EAAE,oBAAoB,CAAC,CAAC,CAAC,GAC/B,OAAO,CAAC,QAAQ,CAAC;IAIP,aAAa,CAAC,GAAG,EAAE,QAAQ,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAwBpE,cAAc,CAAC,IAAI,EAAE,QAAQ,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAG1D"}

package/dest/proving_broker/proof_store/gcs_proof_store.js

@@ -0,0 +1,51 @@
+import { getProvingJobInputClassFor } from '@aztec/stdlib/interfaces/server';
+import { ProvingRequestType } from '@aztec/stdlib/proofs';
+import { Storage } from '@google-cloud/storage';
+import { join } from 'path';
+const INPUTS_PATH = 'inputs';
+export class GoogleCloudStorageProofStore {
+    bucketName;
+    path;
+    storage;
+    constructor(bucketName, path){
+        this.bucketName = bucketName;
+        this.path = path;
+        this.storage = new Storage();
+    }
+    async saveProofInput(id, type, inputs) {
+        const path = join(this.path, INPUTS_PATH, ProvingRequestType[type], id);
+        const file = this.storage.bucket(this.bucketName).file(path);
+        await file.save(inputs.toBuffer());
+        return file.cloudStorageURI.toString();
+    }
+    saveProofOutput(_id, _type, _result) {
+        throw new Error('Not implemented');
+    }
+    async getProofInput(uri) {
+        try {
+            const url = new URL(uri);
+            const bucket = this.storage.bucket(url.host);
+            const path = url.pathname.replace(/^\/+/, '');
+            const file = bucket.file(path);
+            if (!await file.exists()) {
+                throw new Error(`File at ${uri} does not exist`);
+            }
+            const typeString = path.split('/').at(-2);
+            const type = typeString ? ProvingRequestType[typeString] : undefined;
+            if (type === undefined) {
+                throw new Error(`Unrecognized proof type ${type} in path ${path}`);
+            }
+            const contents = await file.download();
+            const inputs = getProvingJobInputClassFor(type).fromBuffer(contents[0]);
+            return {
+                inputs,
+                type
+            };
+        } catch (err) {
+            throw new Error(`Error getting proof input at ${uri}: ${err}`);
+        }
+    }
+    getProofOutput(_uri) {
+        throw new Error('Not implemented');
+    }
+}

package/dest/proving_broker/proof_store/index.d.ts

@@ -0,0 +1,4 @@
+export * from './proof_store.js';
+export * from './inline_proof_store.js';
+export * from './factory.js';
+//# sourceMappingURL=index.d.ts.map

package/dest/proving_broker/proof_store/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/proving_broker/proof_store/index.ts"],"names":[],"mappings":"AAAA,cAAc,kBAAkB,CAAC;AACjC,cAAc,yBAAyB,CAAC;AACxC,cAAc,cAAc,CAAC"}

package/dest/proving_broker/proof_store/index.js

@@ -0,0 +1,3 @@
+export * from './proof_store.js';
+export * from './inline_proof_store.js';
+export * from './factory.js';

package/dest/proving_broker/proof_store/inline_proof_store.d.ts

@@ -0,0 +1,15 @@
+import { type ProofUri, type ProvingJobId, ProvingJobInputs, type ProvingJobInputsMap, ProvingJobResult, type ProvingJobResultsMap } from '@aztec/stdlib/interfaces/server';
+import type { ProvingRequestType } from '@aztec/stdlib/proofs';
+import type { ProofStore } from './proof_store.js';
+/**
+ * An implementation of a proof input/output database that stores data inline in the URI.
+ */
+export declare class InlineProofStore implements ProofStore {
+    saveProofInput<T extends ProvingRequestType>(_id: ProvingJobId, type: T, inputs: ProvingJobInputsMap[T]): Promise<ProofUri>;
+    saveProofOutput<T extends ProvingRequestType>(_id: ProvingJobId, type: T, result: ProvingJobResultsMap[T]): Promise<ProofUri>;
+    getProofInput(uri: ProofUri): Promise<ProvingJobInputs>;
+    getProofOutput(uri: ProofUri): Promise<ProvingJobResult>;
+    private encode;
+    private decode;
+}
+//# sourceMappingURL=inline_proof_store.d.ts.map

package/dest/proving_broker/proof_store/inline_proof_store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inline_proof_store.d.ts","sourceRoot":"","sources":["../../../src/proving_broker/proof_store/inline_proof_store.ts"],"names":[],"mappings":"AACA,OAAO,EACL,KAAK,QAAQ,EACb,KAAK,YAAY,EACjB,gBAAgB,EAChB,KAAK,mBAAmB,EACxB,gBAAgB,EAChB,KAAK,oBAAoB,EAC1B,MAAM,iCAAiC,CAAC;AACzC,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAG/D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAOnD;;GAEG;AACH,qBAAa,gBAAiB,YAAW,UAAU;IACjD,cAAc,CAAC,CAAC,SAAS,kBAAkB,EACzC,GAAG,EAAE,YAAY,EACjB,IAAI,EAAE,CAAC,EACP,MAAM,EAAE,mBAAmB,CAAC,CAAC,CAAC,GAC7B,OAAO,CAAC,QAAQ,CAAC;IAKpB,eAAe,CAAC,CAAC,SAAS,kBAAkB,EAC1C,GAAG,EAAE,YAAY,EACjB,IAAI,EAAE,CAAC,EACP,MAAM,EAAE,oBAAoB,CAAC,CAAC,CAAC,GAC9B,OAAO,CAAC,QAAQ,CAAC;IAKpB,aAAa,CAAC,GAAG,EAAE,QAAQ,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAIvD,cAAc,CAAC,GAAG,EAAE,QAAQ,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAIxD,OAAO,CAAC,MAAM;IAKd,OAAO,CAAC,MAAM;CAQf"}

package/dest/proving_broker/proof_store/inline_proof_store.js

@@ -0,0 +1,41 @@
+import { jsonParseWithSchema, jsonStringify } from '@aztec/foundation/json-rpc';
+import { ProvingJobInputs, ProvingJobResult } from '@aztec/stdlib/interfaces/server';
+// use an ASCII encoded data uri https://datatracker.ietf.org/doc/html/rfc2397#section-2
+// we do this to avoid double encoding to base64 (since the inputs already serialize to a base64 string)
+const PREFIX = 'data:application/json;charset=utf-8';
+const SEPARATOR = ',';
+/**
+ * An implementation of a proof input/output database that stores data inline in the URI.
+ */ export class InlineProofStore {
+    saveProofInput(_id, type, inputs) {
+        const jobInputs = {
+            type,
+            inputs
+        };
+        return Promise.resolve(this.encode(jobInputs));
+    }
+    saveProofOutput(_id, type, result) {
+        const jobResult = {
+            type,
+            result
+        };
+        return Promise.resolve(this.encode(jobResult));
+    }
+    getProofInput(uri) {
+        return Promise.resolve(this.decode(uri, ProvingJobInputs));
+    }
+    getProofOutput(uri) {
+        return Promise.resolve(this.decode(uri, ProvingJobResult));
+    }
+    encode(obj) {
+        const encoded = encodeURIComponent(jsonStringify(obj));
+        return PREFIX + SEPARATOR + encoded;
+    }
+    decode(uri, schema) {
+        const [prefix, data] = uri.split(SEPARATOR);
+        if (prefix !== PREFIX) {
+            throw new Error('Invalid proof input URI: ' + prefix);
+        }
+        return jsonParseWithSchema(decodeURIComponent(data), schema);
+    }
+}

package/dest/proving_broker/proof_store/proof_store.d.ts

@@ -0,0 +1,36 @@
+import type { ProofUri, ProvingJobId, ProvingJobInputs, ProvingJobInputsMap, ProvingJobResult, ProvingJobResultsMap } from '@aztec/stdlib/interfaces/server';
+import type { ProvingRequestType } from '@aztec/stdlib/proofs';
+/**
+ * A database for storing proof inputs and outputs.
+ */
+export interface ProofStore {
+    /**
+     * Save a proof input to the database.
+     * @param jobId - The ID of the job the proof input is associated with.
+     * @param type - The type of the proving request.
+     * @param inputs - The proof input to save.
+     * @returns The URI of the saved proof input.
+     */
+    saveProofInput<T extends ProvingRequestType>(jobId: ProvingJobId, type: T, inputs: ProvingJobInputsMap[T]): Promise<ProofUri>;
+    /**
+     * Save a proof output to the database.
+     * @param jobId - The ID of the job the proof input is associated with.
+     * @param type - The type of the proving request.
+     * @param result - The proof output to save.
+     * @returns The URI of the saved proof output.
+     */
+    saveProofOutput<T extends ProvingRequestType>(id: ProvingJobId, type: T, result: ProvingJobResultsMap[T]): Promise<ProofUri>;
+    /**
+     * Retrieve a proof input from the database.
+     * @param uri - The URI of the proof input to retrieve.
+     * @returns The proof input.
+     */
+    getProofInput(uri: ProofUri): Promise<ProvingJobInputs>;
+    /**
+     * Retrieve a proof output from the database.
+     * @param uri - The URI of the proof output to retrieve.
+     * @returns The proof output.
+     */
+    getProofOutput(uri: ProofUri): Promise<ProvingJobResult>;
+}
+//# sourceMappingURL=proof_store.d.ts.map

package/dest/proving_broker/proof_store/proof_store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proof_store.d.ts","sourceRoot":"","sources":["../../../src/proving_broker/proof_store/proof_store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,QAAQ,EACR,YAAY,EACZ,gBAAgB,EAChB,mBAAmB,EACnB,gBAAgB,EAChB,oBAAoB,EACrB,MAAM,iCAAiC,CAAC;AACzC,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE/D;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB;;;;;;OAMG;IACH,cAAc,CAAC,CAAC,SAAS,kBAAkB,EACzC,KAAK,EAAE,YAAY,EACnB,IAAI,EAAE,CAAC,EACP,MAAM,EAAE,mBAAmB,CAAC,CAAC,CAAC,GAC7B,OAAO,CAAC,QAAQ,CAAC,CAAC;IAErB;;;;;;OAMG;IACH,eAAe,CAAC,CAAC,SAAS,kBAAkB,EAC1C,EAAE,EAAE,YAAY,EAChB,IAAI,EAAE,CAAC,EACP,MAAM,EAAE,oBAAoB,CAAC,CAAC,CAAC,GAC9B,OAAO,CAAC,QAAQ,CAAC,CAAC;IAErB;;;;OAIG;IACH,aAAa,CAAC,GAAG,EAAE,QAAQ,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAExD;;;;OAIG;IACH,cAAc,CAAC,GAAG,EAAE,QAAQ,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;CAC1D"}

package/dest/proving_broker/proof_store/proof_store.js

@@ -0,0 +1,3 @@
+/**
+ * A database for storing proof inputs and outputs.
+ */ export { };

package/dest/proving_broker/proving_agent.d.ts

@@ -0,0 +1,46 @@
+import type { ProvingJobConsumer, ProvingJobId, ProvingJobResultsMap, ServerCircuitProver } from '@aztec/stdlib/interfaces/server';
+import { ProvingRequestType } from '@aztec/stdlib/proofs';
+import { type TelemetryClient, type Traceable, type Tracer } from '@aztec/telemetry-client';
+import type { ProofStore } from './proof_store/index.js';
+/**
+ * A helper class that encapsulates a circuit prover and connects it to a job source.
+ */
+export declare class ProvingAgent implements Traceable {
+    /** The source of proving jobs */
+    private broker;
+    /** Database holding proof inputs and outputs */
+    private proofStore;
+    /** The prover implementation to defer jobs to */
+    private circuitProver;
+    /** Optional list of allowed proof types to build */
+    private proofAllowList;
+    /** How long to wait between jobs */
+    private pollIntervalMs;
+    private log;
+    private currentJobController?;
+    private runningPromise;
+    private instrumentation;
+    private idleTimer;
+    readonly tracer: Tracer;
+    constructor(
+    /** The source of proving jobs */
+    broker: ProvingJobConsumer, 
+    /** Database holding proof inputs and outputs */
+    proofStore: ProofStore, 
+    /** The prover implementation to defer jobs to */
+    circuitProver: ServerCircuitProver, 
+    /** Optional list of allowed proof types to build */
+    proofAllowList?: Array<ProvingRequestType>, 
+    /** How long to wait between jobs */
+    pollIntervalMs?: number, 
+    /** A telemetry client through which to emit metrics */
+    client?: TelemetryClient, log?: import("@aztec/foundation/log").Logger);
+    setCircuitProver(circuitProver: ServerCircuitProver): void;
+    isRunning(): boolean;
+    start(): void;
+    stop(): Promise<void>;
+    private work;
+    private startJob;
+    handleJobResult: <T extends ProvingRequestType>(jobId: ProvingJobId, type: T, err: Error | undefined, result: ProvingJobResultsMap[T] | undefined) => Promise<void>;
+}
+//# sourceMappingURL=proving_agent.d.ts.map

package/dest/proving_broker/proving_agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proving_agent.d.ts","sourceRoot":"","sources":["../../src/proving_broker/proving_agent.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAEV,kBAAkB,EAClB,YAAY,EAEZ,oBAAoB,EACpB,mBAAmB,EACpB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EACL,KAAK,eAAe,EACpB,KAAK,SAAS,EACd,KAAK,MAAM,EAGZ,MAAM,yBAAyB,CAAC;AAEjC,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAIzD;;GAEG;AACH,qBAAa,YAAa,YAAW,SAAS;IAS1C,iCAAiC;IACjC,OAAO,CAAC,MAAM;IACd,gDAAgD;IAChD,OAAO,CAAC,UAAU;IAClB,iDAAiD;IACjD,OAAO,CAAC,aAAa;IACrB,oDAAoD;IACpD,OAAO,CAAC,cAAc;IACtB,oCAAoC;IACpC,OAAO,CAAC,cAAc;IAGtB,OAAO,CAAC,GAAG;IApBb,OAAO,CAAC,oBAAoB,CAAC,CAAuB;IACpD,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,eAAe,CAA8B;IACrD,OAAO,CAAC,SAAS,CAAoB;IAErC,SAAgB,MAAM,EAAE,MAAM,CAAC;;IAG7B,iCAAiC;IACzB,MAAM,EAAE,kBAAkB;IAClC,gDAAgD;IACxC,UAAU,EAAE,UAAU;IAC9B,iDAAiD;IACzC,aAAa,EAAE,mBAAmB;IAC1C,oDAAoD;IAC5C,cAAc,GAAE,KAAK,CAAC,kBAAkB,CAAM;IACtD,oCAAoC;IAC5B,cAAc,SAAO;IAC7B,uDAAuD;IACvD,MAAM,GAAE,eAAsC,EACtC,GAAG,yCAA8C;IAOpD,gBAAgB,CAAC,aAAa,EAAE,mBAAmB,GAAG,IAAI;IAI1D,SAAS,IAAI,OAAO;IAIpB,KAAK,IAAI,IAAI;IAKP,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;YAMpB,IAAI;YA6BJ,QAAQ;IAmDtB,eAAe,wCACN,YAAY,QACb,CAAC,OACF,KAAK,GAAG,SAAS,UACd,oBAAoB,CAAC,CAAC,CAAC,GAAG,SAAS,mBAmB3C;CACH"}

package/dest/proving_broker/proving_agent.js

@@ -0,0 +1,134 @@
+function _ts_decorate(decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for(var i = decorators.length - 1; i >= 0; i--)if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+import { createLogger } from '@aztec/foundation/log';
+import { RunningPromise } from '@aztec/foundation/running-promise';
+import { truncate } from '@aztec/foundation/string';
+import { Timer } from '@aztec/foundation/timer';
+import { ProvingError } from '@aztec/stdlib/errors';
+import { ProvingRequestType } from '@aztec/stdlib/proofs';
+import { getTelemetryClient, trackSpan } from '@aztec/telemetry-client';
+import { ProvingAgentInstrumentation } from './proving_agent_instrumentation.js';
+import { ProvingJobController, ProvingJobControllerStatus } from './proving_job_controller.js';
+/**
+ * A helper class that encapsulates a circuit prover and connects it to a job source.
+ */ export class ProvingAgent {
+    broker;
+    proofStore;
+    circuitProver;
+    proofAllowList;
+    pollIntervalMs;
+    log;
+    currentJobController;
+    runningPromise;
+    instrumentation;
+    idleTimer;
+    tracer;
+    constructor(/** The source of proving jobs */ broker, /** Database holding proof inputs and outputs */ proofStore, /** The prover implementation to defer jobs to */ circuitProver, /** Optional list of allowed proof types to build */ proofAllowList = [], /** How long to wait between jobs */ pollIntervalMs = 1000, /** A telemetry client through which to emit metrics */ client = getTelemetryClient(), log = createLogger('prover-client:proving-agent')){
+        this.broker = broker;
+        this.proofStore = proofStore;
+        this.circuitProver = circuitProver;
+        this.proofAllowList = proofAllowList;
+        this.pollIntervalMs = pollIntervalMs;
+        this.log = log;
+        this.handleJobResult = async (jobId, type, err, result)=>{
+            let maybeJob;
+            if (err) {
+                const retry = err.name === ProvingError.NAME ? err.retry : false;
+                this.log.error(`Job id=${jobId} type=${ProvingRequestType[type]} failed err=${err.message} retry=${retry}`, err);
+                maybeJob = await this.broker.reportProvingJobError(jobId, err.message, retry, {
+                    allowList: this.proofAllowList
+                });
+            } else if (result) {
+                const outputUri = await this.proofStore.saveProofOutput(jobId, type, result);
+                this.log.info(`Job id=${jobId} type=${ProvingRequestType[type]} completed outputUri=${truncate(outputUri)}`);
+                maybeJob = await this.broker.reportProvingJobSuccess(jobId, outputUri, {
+                    allowList: this.proofAllowList
+                });
+            }
+            if (maybeJob) {
+                const { job, time } = maybeJob;
+                await this.startJob(job, time);
+            } else {
+                this.idleTimer = new Timer();
+            }
+        };
+        this.tracer = client.getTracer('ProvingAgent');
+        this.instrumentation = new ProvingAgentInstrumentation(client);
+        this.runningPromise = new RunningPromise(this.work.bind(this), this.log, this.pollIntervalMs);
+    }
+    setCircuitProver(circuitProver) {
+        this.circuitProver = circuitProver;
+    }
+    isRunning() {
+        return this.runningPromise?.isRunning() ?? false;
+    }
+    start() {
+        this.idleTimer = new Timer();
+        this.runningPromise.start();
+    }
+    async stop() {
+        this.currentJobController?.abort();
+        await this.runningPromise.stop();
+    }
+    async work() {
+        // every tick we need to
+        // (1) either do a heartbeat, telling the broker that we're working
+        // (2) get a new job
+        // If during (1) the broker returns a new job that means we can cancel the current job and start the new one
+        let maybeJob;
+        if (this.currentJobController?.getStatus() === ProvingJobControllerStatus.PROVING) {
+            maybeJob = await this.broker.reportProvingJobProgress(this.currentJobController.getJobId(), this.currentJobController.getStartedAt(), {
+                allowList: this.proofAllowList
+            });
+        } else {
+            maybeJob = await this.broker.getProvingJob({
+                allowList: this.proofAllowList
+            });
+        }
+        if (!maybeJob) {
+            return;
+        }
+        if (this.idleTimer) {
+            this.instrumentation.recordIdleTime(this.idleTimer);
+        }
+        this.idleTimer = undefined;
+        const { job, time } = maybeJob;
+        await this.startJob(job, time);
+    }
+    async startJob(job, startedAt) {
+        let abortedProofJobId;
+        let abortedProofName;
+        if (this.currentJobController?.getStatus() === ProvingJobControllerStatus.PROVING) {
+            abortedProofJobId = this.currentJobController.getJobId();
+            abortedProofName = this.currentJobController.getProofTypeName();
+            this.currentJobController?.abort();
+        }
+        let inputs;
+        try {
+            inputs = await this.proofStore.getProofInput(job.inputsUri);
+        } catch (err) {
+            const maybeJob = await this.broker.reportProvingJobError(job.id, 'Failed to load proof inputs', true, {
+                allowList: this.proofAllowList
+            });
+            if (maybeJob) {
+                return this.startJob(maybeJob.job, maybeJob.time);
+            }
+            return;
+        }
+        this.currentJobController = new ProvingJobController(job.id, inputs, job.epochNumber, startedAt, this.circuitProver, this.handleJobResult);
+        if (abortedProofJobId) {
+            this.log.info(`Aborting job id=${abortedProofJobId} type=${abortedProofName} to start new job id=${this.currentJobController.getJobId()} type=${this.currentJobController.getProofTypeName()} inputsUri=${truncate(job.inputsUri)}`);
+        } else {
+            this.log.info(`Starting job id=${this.currentJobController.getJobId()} type=${this.currentJobController.getProofTypeName()} inputsUri=${truncate(job.inputsUri)}`);
+        }
+        this.currentJobController.start();
+    }
+    handleJobResult;
+}
+_ts_decorate([
+    trackSpan('ProvingAgent.safeWork')
+], ProvingAgent.prototype, "work", null);

package/dest/proving_broker/proving_agent_instrumentation.d.ts

@@ -0,0 +1,8 @@
+import type { Timer } from '@aztec/foundation/timer';
+import { type TelemetryClient } from '@aztec/telemetry-client';
+export declare class ProvingAgentInstrumentation {
+    private idleTime;
+    constructor(client: TelemetryClient, name?: string);
+    recordIdleTime(msOrTimer: Timer | number): void;
+}
+//# sourceMappingURL=proving_agent_instrumentation.d.ts.map

package/dest/proving_broker/proving_agent_instrumentation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proving_agent_instrumentation.d.ts","sourceRoot":"","sources":["../../src/proving_broker/proving_agent_instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AACrD,OAAO,EAA2B,KAAK,eAAe,EAAa,MAAM,yBAAyB,CAAC;AAEnG,qBAAa,2BAA2B;IACtC,OAAO,CAAC,QAAQ,CAAY;gBAEhB,MAAM,EAAE,eAAe,EAAE,IAAI,SAAiB;IAU1D,cAAc,CAAC,SAAS,EAAE,KAAK,GAAG,MAAM;CAIzC"}

package/dest/proving_broker/proving_agent_instrumentation.js

@@ -0,0 +1,16 @@
+import { Metrics, ValueType } from '@aztec/telemetry-client';
+export class ProvingAgentInstrumentation {
+    idleTime;
+    constructor(client, name = 'ProvingAgent'){
+        const meter = client.getMeter(name);
+        this.idleTime = meter.createHistogram(Metrics.PROVING_AGENT_IDLE, {
+            description: 'Records how long an agent was idle',
+            unit: 's',
+            valueType: ValueType.DOUBLE
+        });
+    }
+    recordIdleTime(msOrTimer) {
+        const duration = typeof msOrTimer === 'number' ? msOrTimer : msOrTimer.ms();
+        this.idleTime.record(duration / 1000);
+    }
+}

package/dest/proving_broker/proving_broker.d.ts

@@ -0,0 +1,64 @@
+import type { GetProvingJobResponse, ProofUri, ProvingJob, ProvingJobConsumer, ProvingJobFilter, ProvingJobId, ProvingJobProducer, ProvingJobStatus } from '@aztec/stdlib/interfaces/server';
+import { type TelemetryClient, type Traceable, type Tracer } from '@aztec/telemetry-client';
+import { type ProverBrokerConfig } from './config.js';
+import type { ProvingBrokerDatabase } from './proving_broker_database.js';
+/**
+ * A broker that manages proof requests and distributes them to workers based on their priority.
+ * It takes a backend that is responsible for storing and retrieving proof requests and results.
+ */
+export declare class ProvingBroker implements ProvingJobProducer, ProvingJobConsumer, Traceable {
+    #private;
+    private database;
+    private logger;
+    private queues;
+    private jobsCache;
+    private resultsCache;
+    private enqueuedAt;
+    private inProgress;
+    private retries;
+    private promises;
+    private cleanupPromise;
+    private msTimeSource;
+    private jobTimeoutMs;
+    private maxRetries;
+    private instrumentation;
+    readonly tracer: Tracer;
+    private completedJobNotifications;
+    /**
+     * The broker keeps track of the highest epoch its seen.
+     * This information is used for garbage collection: once it reaches the next epoch, it can start pruning the database of old state.
+     * It is important that this value is initialised to zero. This ensures that we don't delete any old jobs until the current
+     * process instance receives a job request informing it of the actual current highest epoch
+     * Example:
+     * proving epoch 11 - the broker will wipe all jobs for epochs 9 and lower
+     * finished proving epoch 11 and got first job for epoch 12 -> the broker will wipe all settled jobs for epochs 10 and lower
+     * reorged back to end of epoch 10 -> epoch 11 is skipped and epoch 12 starts -> the broker will wipe all settled jobs for epochs 10 and lower
+     */
+    private epochHeight;
+    private maxEpochsToKeepResultsFor;
+    private started;
+    constructor(database: ProvingBrokerDatabase, { proverBrokerJobTimeoutMs, proverBrokerPollIntervalMs, proverBrokerJobMaxRetries, proverBrokerMaxEpochsToKeepResultsFor, }?: Required<Pick<ProverBrokerConfig, 'proverBrokerJobTimeoutMs' | 'proverBrokerPollIntervalMs' | 'proverBrokerJobMaxRetries' | 'proverBrokerMaxEpochsToKeepResultsFor'>>, client?: TelemetryClient, logger?: import("@aztec/foundation/log").Logger);
+    private measureQueueDepth;
+    private countActiveJobs;
+    start(): Promise<void>;
+    stop(): Promise<void>;
+    enqueueProvingJob(job: ProvingJob): Promise<ProvingJobStatus>;
+    cancelProvingJob(id: ProvingJobId): Promise<void>;
+    getProvingJobStatus(id: ProvingJobId): Promise<ProvingJobStatus>;
+    getCompletedJobs(ids: ProvingJobId[]): Promise<ProvingJobId[]>;
+    getProvingJob(filter?: ProvingJobFilter): Promise<GetProvingJobResponse | undefined>;
+    reportProvingJobSuccess(id: ProvingJobId, value: ProofUri, filter?: ProvingJobFilter): Promise<GetProvingJobResponse | undefined>;
+    reportProvingJobError(id: ProvingJobId, err: string, retry?: boolean, filter?: ProvingJobFilter): Promise<GetProvingJobResponse | undefined>;
+    reportProvingJobProgress(id: ProvingJobId, startedAt: number, filter?: ProvingJobFilter): Promise<{
+        job: ProvingJob;
+        time: number;
+    } | undefined>;
+    private cleanUpProvingJobState;
+    private cleanupPass;
+    private cleanupStaleJobs;
+    private reEnqueueExpiredJobs;
+    private enqueueJobInternal;
+    private isJobStale;
+    private oldestEpochToKeep;
+}
+//# sourceMappingURL=proving_broker.d.ts.map

package/dest/proving_broker/proving_broker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proving_broker.d.ts","sourceRoot":"","sources":["../../src/proving_broker/proving_broker.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,qBAAqB,EACrB,QAAQ,EACR,UAAU,EACV,kBAAkB,EAClB,gBAAgB,EAChB,YAAY,EACZ,kBAAkB,EAElB,gBAAgB,EACjB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,EACL,KAAK,eAAe,EACpB,KAAK,SAAS,EACd,KAAK,MAAM,EAGZ,MAAM,yBAAyB,CAAC;AAIjC,OAAO,EAAE,KAAK,kBAAkB,EAA6B,MAAM,aAAa,CAAC;AACjF,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAW1E;;;GAGG;AACH,qBAAa,aAAc,YAAW,kBAAkB,EAAE,kBAAkB,EAAE,SAAS;;IAkEnF,OAAO,CAAC,QAAQ;IAgBhB,OAAO,CAAC,MAAM;IAjFhB,OAAO,CAAC,MAAM,CAgBZ;IAIF,OAAO,CAAC,SAAS,CAAuC;IAExD,OAAO,CAAC,YAAY,CAAoD;IAGxE,OAAO,CAAC,UAAU,CAAkC;IAMpD,OAAO,CAAC,UAAU,CAA+C;IAGjE,OAAO,CAAC,OAAO,CAAmC;IAGlD,OAAO,CAAC,QAAQ,CAA0E;IAE1F,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,YAAY,CAAoB;IACxC,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,UAAU,CAAS;IAE3B,OAAO,CAAC,eAAe,CAA+B;IACtD,SAAgB,MAAM,EAAE,MAAM,CAAC;IAE/B,OAAO,CAAC,yBAAyB,CAAsB;IAEvD;;;;;;;;;OASG;IACH,OAAO,CAAC,WAAW,CAAK;IACxB,OAAO,CAAC,yBAAyB,CAAK;IAEtC,OAAO,CAAC,OAAO,CAAS;gBAGd,QAAQ,EAAE,qBAAqB,EACvC,EACE,wBAAwB,EACxB,0BAA0B,EAC1B,yBAAyB,EACzB,qCAAqC,GACtC,GAAE,QAAQ,CACT,IAAI,CACF,kBAAkB,EAChB,0BAA0B,GAC1B,4BAA4B,GAC5B,2BAA2B,GAC3B,uCAAuC,CAC1C,CAC0B,EAC7B,MAAM,GAAE,eAAsC,EACtC,MAAM,yCAA+C;IAU/D,OAAO,CAAC,iBAAiB,CAEvB;IAEF,OAAO,CAAC,eAAe,CAUrB;IAEW,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA+BtB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ3B,iBAAiB,CAAC,GAAG,EAAE,UAAU,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAI7D,gBAAgB,CAAC,EAAE,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAIjD,mBAAmB,CAAC,EAAE,EAAE,YAAY,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAIhE,gBAAgB,CAAC,GAAG,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAI9D,aAAa,CAAC,MAAM,CAAC,EAAE,gBAAgB,GAAG,OAAO,CAAC,qBAAqB,GAAG,SAAS,CAAC;IAIpF,uBAAuB,CAC5B,EAAE,EAAE,YAAY,EAChB,KAAK,EAAE,QAAQ,EACf,MAAM,CAAC,EAAE,gBAAgB,GACxB,OAAO,CAAC,qBAAqB,GAAG,SAAS,CAAC;IAItC,qBAAqB,CAC1B,EAAE,EAAE,YAAY,EAChB,GAAG,EAAE,MAAM,EACX,KAAK,UAAQ,EACb,MAAM,CAAC,EAAE,gBAAgB,GACxB,OAAO,CAAC,qBAAqB,GAAG,SAAS,CAAC;IAItC,wBAAwB,CAC7B,EAAE,EAAE,YAAY,EAChB,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,gBAAgB,GACxB,OAAO,CAAC;QAAE,GAAG,EAAE,UAAU,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,GAAG,SAAS,CAAC;IAoDzD,OAAO,CAAC,sBAAsB;YA6QhB,WAAW;IAUzB,OAAO,CAAC,gBAAgB;IAgBxB,OAAO,CAAC,oBAAoB;IAqB5B,OAAO,CAAC,kBAAkB;IAY1B,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,iBAAiB;CAG1B"}