@rivet-dev/flue 0.0.0-main.79e97f1

package/dist/workflow-coordinator.js

@@ -0,0 +1,154 @@
+import { admitDetachedWorkflow, failRecoveredRun, handleRunRouteRequest, handleStreamHead, handleStreamRead, handleWorkflowRequest, } from '@flue/runtime/adapter-kit';
+import { createRivetRunStore } from './run-store-composite.js';
+import { createAsyncRunStore, ensureAsyncSqlSchema } from './stores/index.js';
+export function createRivetWorkflowRuntime(options) {
+    const coordinators = new WeakMap();
+    const getCoordinator = (actor) => {
+        const coordinator = coordinators.get(actor);
+        if (!coordinator) {
+            throw new Error('[flue] Generated Rivet workflow coordinator was not initialized.');
+        }
+        return coordinator;
+    };
+    return {
+        async prepare({ db, workflowName, actor }) {
+            await ensureAsyncSqlSchema(db);
+            const primary = createAsyncRunStore(db);
+            const mirror = options.createRegistryRunStore?.(actor);
+            return {
+                workflowName,
+                runStore: createRivetRunStore(primary, mirror),
+            };
+        },
+        attach(actor, prepared) {
+            const coordinator = new RivetWorkflowCoordinator(actor, prepared, options, options.createEventStreamStore(actor));
+            coordinators.set(actor, coordinator);
+            return coordinator;
+        },
+        onWake(actor, inherited = () => undefined) {
+            return getCoordinator(actor).onWake(inherited);
+        },
+        admitWorkflow(actor, input) {
+            return getCoordinator(actor).admitWorkflow(input);
+        },
+        onRequest(actor, request) {
+            return getCoordinator(actor).onRequest(request);
+        },
+    };
+}
+export class RivetWorkflowCoordinator {
+    actor;
+    prepared;
+    options;
+    eventStreamStore;
+    constructor(actor, prepared, options, eventStreamStore) {
+        this.actor = actor;
+        this.prepared = prepared;
+        this.options = options;
+        this.eventStreamStore = eventStreamStore;
+    }
+    async onWake(inherited) {
+        await inherited();
+        const run = await this.prepared.runStore.getRun(this.runId);
+        if (run?.status !== 'active')
+            return;
+        await failRecoveredRun({
+            workflowName: this.workflowName,
+            runId: this.runId,
+            error: new Error('Flue workflow execution was interrupted. Start a new workflow run explicitly if retry is appropriate.'),
+            runStore: this.prepared.runStore,
+            eventStreamStore: this.eventStreamStore,
+            createContext: ({ runId, request, initialEventIndex }) => this.createContext(request, runId, initialEventIndex),
+        });
+    }
+    async onRequest(request) {
+        const runRoute = parseRunRoute(request);
+        if (runRoute) {
+            if (runRoute.action === 'ds-stream') {
+                const streamPath = runStreamPath(runRoute.runId);
+                if (request.method === 'HEAD')
+                    return handleStreamHead(this.eventStreamStore, streamPath);
+                return handleStreamRead({ store: this.eventStreamStore, path: streamPath, request });
+            }
+            return handleRunRouteRequest({
+                workflowName: this.workflowName,
+                runId: runRoute.runId,
+                runStore: this.prepared.runStore,
+            });
+        }
+        if (!parseWorkflowStart(request, this.workflowName))
+            return null;
+        const workflow = this.options.workflows.find((record) => record.name === this.workflowName)?.definition;
+        if (!workflow)
+            return null;
+        return handleWorkflowRequest({
+            request,
+            workflowName: this.workflowName,
+            runId: this.runId,
+            workflow,
+            runStore: this.prepared.runStore,
+            eventStreamStore: this.eventStreamStore,
+            createContext: ({ runId, request, initialEventIndex }) => this.createContext(request, runId, initialEventIndex),
+            startWorkflowAdmission: (_runId, run) => {
+                const completion = this.actor.keepAwake(Promise.resolve().then(run));
+                return { admitted: Promise.resolve(), completion };
+            },
+        });
+    }
+    async admitWorkflow(options) {
+        const workflow = this.options.workflows.find((record) => record.name === this.workflowName)?.definition;
+        if (!workflow) {
+            throw new Error(`[flue] Workflow target "${this.workflowName}" is unavailable.`);
+        }
+        return admitDetachedWorkflow({
+            workflowName: this.workflowName,
+            runId: options.runId,
+            workflow,
+            input: options.input,
+            runStore: this.prepared.runStore,
+            eventStreamStore: this.eventStreamStore,
+            createContext: ({ runId, request, initialEventIndex }) => this.createContext(request, runId, initialEventIndex),
+            startWorkflowAdmission: (_runId, run) => {
+                const completion = this.actor.keepAwake(Promise.resolve().then(run));
+                return { admitted: Promise.resolve(), completion };
+            },
+        });
+    }
+    get workflowName() {
+        return this.prepared.workflowName;
+    }
+    get runId() {
+        return this.options.resolveRunId?.(this.actor) ?? this.actor.key[0] ?? this.actor.actorId;
+    }
+    createContext(request, runId, initialEventIndex) {
+        return this.options.createContext({
+            actor: this.actor,
+            request,
+            runId,
+            initialEventIndex,
+        });
+    }
+}
+function parseWorkflowStart(request, workflowName) {
+    if (request.method !== 'POST')
+        return false;
+    const url = new URL(request.url);
+    const segments = url.pathname.split('/').filter(Boolean);
+    if (segments.length !== 2 || segments[0] !== 'workflows')
+        return false;
+    return decodeURIComponent(segments[1] ?? '') === workflowName;
+}
+function parseRunRoute(request) {
+    const url = new URL(request.url);
+    const segments = url.pathname.split('/').filter(Boolean);
+    if (segments.length !== 2 || segments[0] !== 'runs')
+        return null;
+    const runId = decodeURIComponent(segments[1] ?? '');
+    if (!runId)
+        return null;
+    return { action: request.method === 'GET' && url.searchParams.has('meta') ? 'meta' : 'ds-stream', runId };
+}
+function runStreamPath(runId) {
+    return `runs/${runId}`;
+}
+//# sourceMappingURL=workflow-coordinator.js.map

package/dist/workflow-coordinator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-coordinator.js","sourceRoot":"","sources":["../src/workflow-coordinator.ts"],"names":[],"mappings":"AAMA,OAAO,EACN,qBAAqB,EACrB,gBAAgB,EAChB,qBAAqB,EACrB,gBAAgB,EAChB,gBAAgB,EAChB,qBAAqB,GACrB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAC/D,OAAO,EAAE,mBAAmB,EAAE,oBAAoB,EAAmB,MAAM,mBAAmB,CAAC;AAsD/F,MAAM,UAAU,0BAA0B,CAAC,OAAoC;IAC9E,MAAM,YAAY,GAAG,IAAI,OAAO,EAAuD,CAAC;IAExF,MAAM,cAAc,GAAG,CAAC,KAAgC,EAA4B,EAAE;QACrF,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC5C,IAAI,CAAC,WAAW,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,kEAAkE,CAAC,CAAC;QACrF,CAAC;QACD,OAAO,WAAW,CAAC;IACpB,CAAC,CAAC;IAEF,OAAO;QACN,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE;YACxC,MAAM,oBAAoB,CAAC,EAAE,CAAC,CAAC;YAC/B,MAAM,OAAO,GAAG,mBAAmB,CAAC,EAAE,CAAC,CAAC;YACxC,MAAM,MAAM,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC,KAAK,CAAC,CAAC;YACvD,OAAO;gBACN,YAAY;gBACZ,QAAQ,EAAE,mBAAmB,CAAC,OAAO,EAAE,MAAM,CAAC;aAC9C,CAAC;QACH,CAAC;QACD,MAAM,CAAC,KAAK,EAAE,QAAQ;YACrB,MAAM,WAAW,GAAG,IAAI,wBAAwB,CAC/C,KAAK,EACL,QAAQ,EACR,OAAO,EACP,OAAO,CAAC,sBAAsB,CAAC,KAAK,CAAC,CACrC,CAAC;YACF,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YACrC,OAAO,WAAW,CAAC;QACpB,CAAC;QACD,MAAM,CAAC,KAAK,EAAE,SAAS,GAAG,GAAG,EAAE,CAAC,SAAS;YACxC,OAAO,cAAc,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAChD,CAAC;QACD,aAAa,CAAC,KAAK,EAAE,KAAK;YACzB,OAAO,cAAc,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC;QACD,SAAS,CAAC,KAAK,EAAE,OAAO;YACvB,OAAO,cAAc,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACjD,CAAC;KACD,CAAC;AACH,CAAC;AAED,MAAM,OAAO,wBAAwB;IACnB,KAAK,CAA4B;IACjC,QAAQ,CAAmC;IAC3C,OAAO,CAA8B;IACrC,gBAAgB,CAAmB;IAEpD,YACC,KAAgC,EAChC,QAA0C,EAC1C,OAAoC,EACpC,gBAAkC;QAElC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAA2C;QACvD,MAAM,SAAS,EAAE,CAAC;QAClB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,IAAI,GAAG,EAAE,MAAM,KAAK,QAAQ;YAAE,OAAO;QACrC,MAAM,gBAAgB,CAAC;YACtB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,KAAK,EAAE,IAAI,KAAK,CACf,uGAAuG,CACvG;YACD,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;YAChC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,aAAa,EAAE,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,iBAAiB,EAAE,EAAE,EAAE,CACxD,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,KAAK,EAAE,iBAAiB,CAAC;SACtD,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,OAAgB;QAC/B,MAAM,QAAQ,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QACxC,IAAI,QAAQ,EAAE,CAAC;YACd,IAAI,QAAQ,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBACrC,MAAM,UAAU,GAAG,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACjD,IAAI,OAAO,CAAC,MAAM,KAAK,MAAM;oBAAE,OAAO,gBAAgB,CAAC,IAAI,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;gBAC1F,OAAO,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC;YACtF,CAAC;YACD,OAAO,qBAAqB,CAAC;gBAC5B,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;aAChC,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC;YAAE,OAAO,IAAI,CAAC;QACjE,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,YAAY,CAAC,EAAE,UAAU,CAAC;QACxG,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC;QAC3B,OAAO,qBAAqB,CAAC;YAC5B,OAAO;YACP,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ;YACR,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;YAChC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,aAAa,EAAE,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,iBAAiB,EAAE,EAAE,EAAE,CACxD,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,KAAK,EAAE,iBAAiB,CAAC;YACtD,sBAAsB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;gBACvC,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;gBACrE,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YACpD,CAAC;SACD,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAGnB;QACA,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,CAC3C,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,YAAY,CAC7C,EAAE,UAAU,CAAC;QACd,IAAI,CAAC,QAAQ,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,2BAA2B,IAAI,CAAC,YAAY,mBAAmB,CAAC,CAAC;QAClF,CAAC;QACD,OAAO,qBAAqB,CAAC;YAC5B,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ;YACR,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;YAChC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,aAAa,EAAE,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,iBAAiB,EAAE,EAAE,EAAE,CACxD,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,KAAK,EAAE,iBAAiB,CAAC;YACtD,sBAAsB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;gBACvC,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;gBACrE,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YACpD,CAAC;SACD,CAAC,CAAC;IACJ,CAAC;IAED,IAAY,YAAY;QACvB,OAAO,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;IACnC,CAAC;IAED,IAAY,KAAK;QAChB,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;IAC3F,CAAC;IAEO,aAAa,CACpB,OAA4B,EAC5B,KAAa,EACb,iBAA0B;QAE1B,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;YACjC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,OAAO;YACP,KAAK;YACL,iBAAiB;SACjB,CAAC,CAAC;IACJ,CAAC;CAED;AAED,SAAS,kBAAkB,CAAC,OAAgB,EAAE,YAAoB;IACjE,IAAI,OAAO,CAAC,MAAM,KAAK,MAAM;QAAE,OAAO,KAAK,CAAC;IAC5C,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACzD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,KAAK,WAAW;QAAE,OAAO,KAAK,CAAC;IACvE,OAAO,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,YAAY,CAAC;AAC/D,CAAC;AAED,SAAS,aAAa,CAAC,OAAgB;IACtC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACzD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,KAAK,MAAM;QAAE,OAAO,IAAI,CAAC;IACjE,MAAM,KAAK,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IACpD,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,KAAK,KAAK,IAAI,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC;AAC3G,CAAC;AAED,SAAS,aAAa,CAAC,KAAa;IACnC,OAAO,QAAQ,KAAK,EAAE,CAAC;AACxB,CAAC"}

package/package.json

@@ -0,0 +1,62 @@
+{
+	"name": "@rivet-dev/flue",
+	"version": "0.0.0-main.79e97f1",
+	"description": "Run Flue agents and workflows as durable Rivet Actors.",
+	"license": "Apache-2.0",
+	"publishConfig": {
+		"access": "public"
+	},
+	"type": "module",
+	"keywords": [
+		"rivetkit",
+		"rivet",
+		"flue",
+		"agents",
+		"workflows",
+		"actor",
+		"durable",
+		"ai"
+	],
+	"files": [
+		"dist",
+		"README.md"
+	],
+	"exports": {
+		".": {
+			"types": "./dist/index.d.ts",
+			"default": "./dist/index.js"
+		},
+		"./runtime": {
+			"types": "./dist/runtime.d.ts",
+			"default": "./dist/runtime.js"
+		},
+		"./internal": {
+			"types": "./dist/internal.d.ts",
+			"default": "./dist/internal.js"
+		}
+	},
+	"main": "./dist/index.js",
+	"types": "./dist/index.d.ts",
+	"scripts": {
+		"build": "rm -rf dist && tsc -p tsconfig.build.json",
+		"check-types": "tsc --noEmit",
+		"check:types": "tsc --noEmit",
+		"test": "pnpm run test:node && pnpm run test:contracts",
+		"test:node": "bash -c 'for file in test/*.test.ts; do case \"$file\" in *contract*) continue ;; esac; tsx --test \"$file\" || exit $?; done'",
+		"test:contracts": "vitest run test/rivet-store-contracts.test.ts"
+	},
+	"dependencies": {
+		"@flue/cli": "npm:@rivet-dev/labs-flue-cli@1.0.0-beta.9-rivet.2",
+		"@flue/runtime": "npm:@rivet-dev/labs-flue-runtime@1.0.0-beta.9-rivet.2",
+		"rivetkit": "0.0.0-main.79e97f1"
+	},
+	"devDependencies": {
+		"@rivetkit/engine-cli": "0.0.0-main.79e97f1",
+		"tsx": "^4.22.4",
+		"typescript": "6.0.3",
+		"vitest": "^4.0.15"
+	},
+	"engines": {
+		"node": ">=22.19.0"
+	}
+}