@rubrion/agent-kit-server 0.0.0-stage → 0.1.1

package/dist/gateway-entry.d.ts

@@ -0,0 +1,3 @@
+export { createAgentGateway, type GatewayOptions, type Preflight, type Blocked } from './gateway.js';
+export { HEADER, readCaller, writeCaller, writeModel } from './headers.js';
+export type { Caller, ModelChoice, Provider } from './types.js';

package/dist/gateway-entry.js

@@ -0,0 +1,5 @@
+// @rubrion/agent-kit-server/gateway: only the gateway, with no Mastra in the
+// bundle. Import this from the app's API Worker.
+export { createAgentGateway } from './gateway.js';
+export { HEADER, readCaller, writeCaller, writeModel } from './headers.js';
+//# sourceMappingURL=gateway-entry.js.map

package/dist/gateway-entry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gateway-entry.js","sourceRoot":"","sources":["../src/gateway-entry.ts"],"names":[],"mappings":"AAAA,6EAA6E;AAC7E,iDAAiD;AACjD,OAAO,EAAE,kBAAkB,EAAqD,MAAM,cAAc,CAAC;AACrG,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC"}

package/dist/gateway.d.ts

@@ -0,0 +1,25 @@
+import { Hono, type Context } from 'hono';
+import type { Caller, ModelChoice } from './types.js';
+export type Blocked = {
+    status: number;
+    reason: string;
+    message: string;
+};
+/** ok with an optional model (e.g. the tenant's own key), or why not. */
+export type Preflight = {
+    ok: true;
+    model?: ModelChoice;
+} | ({
+    ok: false;
+} & Blocked);
+export interface GatewayOptions<C extends Context = Context> {
+    /** The agent worker's service binding. */
+    agent: (c: C) => Fetcher | undefined;
+    /** The person from the app's session, or null (401). */
+    identify: (c: C) => Caller | null | Promise<Caller | null>;
+    /** Runs before /status and /stream: missing key, budget reached… */
+    preflight?: (c: C, caller: Caller) => Preflight | Promise<Preflight>;
+    /** Per-agent gate (e.g. the thinking agent is only on bigger plans). */
+    canUse?: (c: C, caller: Caller, agentId: string) => true | Blocked | Promise<true | Blocked>;
+}
+export declare function createAgentGateway<C extends Context = Context>(o: GatewayOptions<C>): Hono<import("hono/types").BlankEnv, import("hono/types").BlankSchema, "/">;

package/dist/gateway.js

@@ -0,0 +1,102 @@
+// The gateway: the app's public side of the assistant. It turns the app's own
+// session into a Caller, checks whether the assistant may answer (key set, budget
+// left, plan includes the agent…), and forwards to the agent worker through its
+// service binding. Answers stream back untouched.
+//
+// Mount it under the app's API, e.g. app.route('/api/agent', createAgentGateway(...)):
+//   GET    /agents        agents this caller may use
+//   GET    /status        { ready, reason, message, provider, model }: the UI checks on open
+//   POST   /stream        a question (SSE)
+//   GET    /threads       conversations
+//   GET    /threads/:id   one conversation (?agent=)
+//   DELETE /threads/:id
+import { Hono } from 'hono';
+import { writeCaller, writeModel } from './headers.js';
+const OFFLINE = 'The assistant is offline.';
+export function createAgentGateway(o) {
+    const app = new Hono();
+    async function forward(c, caller, path, init = {}) {
+        const binding = o.agent(c);
+        if (!binding)
+            return c.json({ error: OFFLINE }, 503);
+        const headers = new Headers(init.headers);
+        writeCaller(headers, caller);
+        if (init.model)
+            writeModel(headers, init.model);
+        try {
+            const r = await binding.fetch(new Request(`https://agent${path}`, { method: init.method ?? 'GET', body: init.body, headers, signal: c.req.raw.signal }));
+            return new Response(r.body, { status: r.status, headers: r.headers });
+        }
+        catch (e) {
+            console.error('agent worker unreachable', e);
+            return c.json({ error: OFFLINE }, 503);
+        }
+    }
+    const who = async (c) => {
+        const caller = await o.identify(c);
+        return caller ?? null;
+    };
+    app.get('/agents', async (c) => {
+        const ctx = c;
+        const caller = await who(ctx);
+        if (!caller)
+            return c.json({ error: 'Not signed in.' }, 401);
+        const r = await forward(ctx, caller, '/agents');
+        if (!r.ok || !o.canUse)
+            return r;
+        const list = (await r.json());
+        const allowed = await Promise.all(list.map(async (a) => (await o.canUse(ctx, caller, a.id)) === true));
+        return c.json(list.filter((_, i) => allowed[i]));
+    });
+    app.get('/status', async (c) => {
+        const ctx = c;
+        const caller = await who(ctx);
+        if (!caller)
+            return c.json({ error: 'Not signed in.' }, 401);
+        const p = (await o.preflight?.(ctx, caller)) ?? { ok: true };
+        return c.json(p.ok
+            ? { ready: true, reason: null, message: null, provider: p.model?.provider ?? 'workers-ai', model: p.model?.model ?? null }
+            : { ready: false, reason: p.reason, message: p.message });
+    });
+    app.post('/stream', async (c) => {
+        const ctx = c;
+        const caller = await who(ctx);
+        if (!caller)
+            return c.json({ error: 'Not signed in.' }, 401);
+        const p = (await o.preflight?.(ctx, caller)) ?? { ok: true };
+        if (!p.ok)
+            return c.json({ error: p.message, reason: p.reason }, p.status);
+        const body = await c.req.text();
+        let agentId = '';
+        try {
+            agentId = String(JSON.parse(body).agent ?? '');
+        }
+        catch {
+            return c.json({ error: 'Invalid body.' }, 400);
+        }
+        if (agentId && o.canUse) {
+            const ok = await o.canUse(ctx, caller, agentId);
+            if (ok !== true)
+                return c.json({ error: ok.message, reason: ok.reason }, ok.status);
+        }
+        return forward(ctx, caller, '/stream', { method: 'POST', body, headers: { 'content-type': 'application/json' }, model: p.model });
+    });
+    app.get('/threads', async (c) => {
+        const caller = await who(c);
+        return caller ? forward(c, caller, '/threads') : c.json([], 401);
+    });
+    app.get('/threads/:id', async (c) => {
+        const caller = await who(c);
+        if (!caller)
+            return c.json([], 401);
+        return forward(c, caller, `/threads/${encodeURIComponent(c.req.param('id'))}?agent=${encodeURIComponent(c.req.query('agent') ?? '')}`);
+    });
+    app.delete('/threads/:id', async (c) => {
+        const caller = await who(c);
+        if (!caller)
+            return c.json({ ok: false }, 401);
+        return forward(c, caller, `/threads/${encodeURIComponent(c.req.param('id'))}`, { method: 'DELETE' });
+    });
+    return app;
+}
+//# sourceMappingURL=gateway.js.map

package/dist/gateway.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gateway.js","sourceRoot":"","sources":["../src/gateway.ts"],"names":[],"mappings":"AAAA,8EAA8E;AAC9E,kFAAkF;AAClF,gFAAgF;AAChF,kDAAkD;AAClD,EAAE;AACF,uFAAuF;AACvF,qDAAqD;AACrD,6FAA6F;AAC7F,2CAA2C;AAC3C,wCAAwC;AACxC,qDAAqD;AACrD,wBAAwB;AACxB,OAAO,EAAE,IAAI,EAAgB,MAAM,MAAM,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAkBvD,MAAM,OAAO,GAAG,2BAA2B,CAAC;AAE5C,MAAM,UAAU,kBAAkB,CAA8B,CAAoB;IACnF,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IAEvB,KAAK,UAAU,OAAO,CAAC,CAAI,EAAE,MAAc,EAAE,IAAY,EAAE,OAA8C,EAAE;QAC1G,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,IAAI,CAAC,OAAO;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,GAAG,CAAC,CAAC;QACrD,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1C,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7B,IAAI,IAAI,CAAC,KAAK;YAAE,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAChD,IAAI,CAAC;YACJ,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,gBAAgB,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACzJ,OAAO,IAAI,QAAQ,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QACvE,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACZ,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,CAAC,CAAC,CAAC;YAC7C,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC;IACF,CAAC;IAED,MAAM,GAAG,GAAG,KAAK,EAAE,CAAI,EAAE,EAAE;QAC1B,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QACnC,OAAO,MAAM,IAAI,IAAI,CAAC;IACvB,CAAC,CAAC;IAEF,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC9B,MAAM,GAAG,GAAG,CAAiB,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,GAAG,CAAC,CAAC;QAC7D,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;QAChD,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC;QACjC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAqB,CAAC;QAClD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,MAAO,CAAC,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;QACxG,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC9B,MAAM,GAAG,GAAG,CAAiB,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,GAAG,CAAC,CAAC;QAC7D,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,EAAE,IAAa,EAAE,CAAC;QACtE,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE;YACjB,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,KAAK,EAAE,QAAQ,IAAI,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,IAAI,IAAI,EAAE;YAC1H,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC/B,MAAM,GAAG,GAAG,CAAiB,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,GAAG,CAAC,CAAC;QAC7D,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,EAAE,IAAa,EAAE,CAAC;QACtE,IAAI,CAAC,CAAC,CAAC,EAAE;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAa,CAAC,CAAC;QAClF,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;QAChC,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,CAAC;YAAC,OAAO,GAAG,MAAM,CAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAyB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC;YAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,EAAE,GAAG,CAAC,CAAC;QAAC,CAAC;QAC1I,IAAI,OAAO,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACzB,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAChD,IAAI,EAAE,KAAK,IAAI;gBAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC,MAAa,CAAC,CAAC;QAC5F,CAAC;QACD,OAAO,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;IACnI,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC/B,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,CAAiB,CAAC,CAAC;QAC5C,OAAO,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAiB,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QACnC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,CAAiB,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QACpC,OAAO,OAAO,CAAC,CAAiB,EAAE,MAAM,EAAE,YAAY,kBAAkB,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,UAAU,kBAAkB,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;IACxJ,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QACtC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,CAAiB,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,GAAG,CAAC,CAAC;QAC/C,OAAO,OAAO,CAAC,CAAiB,EAAE,MAAM,EAAE,YAAY,kBAAkB,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IACtH,CAAC,CAAC,CAAC;IAEH,OAAO,GAAG,CAAC;AACZ,CAAC"}

package/dist/headers.d.ts

@@ -0,0 +1,13 @@
+import type { Caller, ModelChoice } from './types.js';
+export declare const HEADER: {
+    readonly tenant: "X-Agent-Tenant";
+    readonly person: "X-Agent-Person";
+    readonly role: "X-Agent-Role";
+    /** URI-encoded: names carry accents. */
+    readonly name: "X-Agent-Name";
+    /** JSON ModelChoice; carries a decrypted key only inside this request. */
+    readonly model: "X-Agent-Model";
+};
+export declare function writeCaller(headers: Headers, caller: Caller): void;
+export declare function readCaller(headers: Headers): Caller | null;
+export declare const writeModel: (headers: Headers, choice: ModelChoice) => void;

package/dist/headers.js

@@ -0,0 +1,31 @@
+export const HEADER = {
+    tenant: 'X-Agent-Tenant',
+    person: 'X-Agent-Person',
+    role: 'X-Agent-Role',
+    /** URI-encoded: names carry accents. */
+    name: 'X-Agent-Name',
+    /** JSON ModelChoice; carries a decrypted key only inside this request. */
+    model: 'X-Agent-Model',
+};
+/** Ids are kept to a safe alphabet: they become part of thread ids. */
+const ID = /^[A-Za-z0-9_.@-]{1,64}$/;
+export function writeCaller(headers, caller) {
+    headers.set(HEADER.tenant, caller.tenant);
+    headers.set(HEADER.person, caller.person);
+    headers.set(HEADER.role, caller.role);
+    headers.set(HEADER.name, encodeURIComponent(caller.name).slice(0, 300));
+}
+export function readCaller(headers) {
+    const tenant = headers.get(HEADER.tenant) ?? '';
+    const person = headers.get(HEADER.person) ?? '';
+    if (!ID.test(tenant) || !ID.test(person))
+        return null;
+    let name = headers.get(HEADER.name) ?? '';
+    try {
+        name = decodeURIComponent(name);
+    }
+    catch { /* sent raw */ }
+    return { tenant, person, role: headers.get(HEADER.role) ?? '', name };
+}
+export const writeModel = (headers, choice) => headers.set(HEADER.model, JSON.stringify(choice));
+//# sourceMappingURL=headers.js.map

package/dist/headers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"headers.js","sourceRoot":"","sources":["../src/headers.ts"],"names":[],"mappings":"AAIA,MAAM,CAAC,MAAM,MAAM,GAAG;IACrB,MAAM,EAAE,gBAAgB;IACxB,MAAM,EAAE,gBAAgB;IACxB,IAAI,EAAE,cAAc;IACpB,wCAAwC;IACxC,IAAI,EAAE,cAAc;IACpB,0EAA0E;IAC1E,KAAK,EAAE,eAAe;CACb,CAAC;AAEX,uEAAuE;AACvE,MAAM,EAAE,GAAG,yBAAyB,CAAC;AAErC,MAAM,UAAU,WAAW,CAAC,OAAgB,EAAE,MAAc;IAC3D,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC1C,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC1C,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IACtC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,kBAAkB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;AACzE,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,OAAgB;IAC1C,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IAChD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IAChD,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;QAAE,OAAO,IAAI,CAAC;IACtD,IAAI,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;IAC1C,IAAI,CAAC;QAAC,IAAI,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC;IAAC,CAAC;IAAC,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC;IACjE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC;AACvE,CAAC;AAED,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,OAAgB,EAAE,MAAmB,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,8 @@
+export { createAgentWorker, type AgentWorkerOptions } from './worker.js';
+export { toTurns } from './turns.js';
+export { createAgentGateway, type GatewayOptions, type Preflight, type Blocked } from './gateway.js';
+export { HEADER, readCaller, writeCaller, writeModel } from './headers.js';
+export { DEFAULT_MODEL, PRICES, isWorkersAI, priceOf, readModelChoice, resolveModel, type Providers } from './models.js';
+export { defaultErrorMessage, slim, usageOf } from './stream.js';
+export type { AgentDefinition, Caller, ModelChoice, Price, Provider, StreamEvent, Usage } from './types.js';
+export { createTool, z } from '@rubrion/mastra-cloudflare';

package/dist/index.js

@@ -0,0 +1,12 @@
+// @rubrion/agent-kit-server: the server half of the agent kit.
+//   createAgentWorker   the agent worker (Mastra + D1 memory + slim SSE), behind a service binding
+//   createAgentGateway  the app-side routes: identity, preflight, forwarding
+export { createAgentWorker } from './worker.js';
+export { toTurns } from './turns.js';
+export { createAgentGateway } from './gateway.js';
+export { HEADER, readCaller, writeCaller, writeModel } from './headers.js';
+export { DEFAULT_MODEL, PRICES, isWorkersAI, priceOf, readModelChoice, resolveModel } from './models.js';
+export { defaultErrorMessage, slim, usageOf } from './stream.js';
+// So an agent project needs only this package to declare tools.
+export { createTool, z } from '@rubrion/mastra-cloudflare';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,mGAAmG;AACnG,6EAA6E;AAC7E,OAAO,EAAE,iBAAiB,EAA2B,MAAM,aAAa,CAAC;AACzE,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AACrC,OAAO,EAAE,kBAAkB,EAAqD,MAAM,cAAc,CAAC;AACrG,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,eAAe,EAAE,YAAY,EAAkB,MAAM,aAAa,CAAC;AACzH,OAAO,EAAE,mBAAmB,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAEjE,gEAAgE;AAChE,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,MAAM,4BAA4B,CAAC"}

package/dist/models.d.ts

@@ -0,0 +1,32 @@
+import { type AiBinding, type Model } from '@rubrion/mastra-cloudflare';
+import type { ModelChoice, Price } from './types.js';
+export declare const DEFAULT_MODEL = "@cf/google/gemma-4-26b-a4b-it";
+/** Dollars per million tokens, used to report the cost of each answer. Extend through options.prices. */
+export declare const PRICES: Record<string, Price>;
+/** The choice in the header; anything malformed becomes the default Workers AI model. */
+export declare function readModelChoice(header: string | null | undefined, fallback?: ModelChoice): ModelChoice;
+export declare const priceOf: (choice: ModelChoice, prices?: Record<string, Price>) => Price;
+/** Only Workers AI takes temperature and the thinking switch; the others decide on their own. */
+export declare const isWorkersAI: (choice: ModelChoice) => boolean;
+/**
+ * Bring-your-own-key providers, opt-in so their SDKs only enter the bundle of
+ * apps that offer them:
+ *
+ *     providers: { anthropic: () => import('@ai-sdk/anthropic'), openai: () => import('@ai-sdk/openai') }
+ */
+export interface Providers {
+    anthropic?: () => Promise<{
+        createAnthropic: (o: {
+            apiKey?: string;
+        }) => (model: string) => unknown;
+    }>;
+    openai?: () => Promise<{
+        createOpenAI: (o: {
+            apiKey?: string;
+        }) => {
+            chat: (model: string) => unknown;
+        };
+    }>;
+}
+/** The AI SDK model for a choice. A provider not enabled falls back to Workers AI. */
+export declare function resolveModel(choice: ModelChoice, ai: AiBinding, thinking: boolean, providers?: Providers): Promise<Model>;

package/dist/models.js

@@ -0,0 +1,49 @@
+// Which model answers each request. Workers AI is the default (no key, billed to
+// the account running the worker); a tenant may bring its own Anthropic or OpenAI
+// key, which the gateway sends in X-Agent-Model and only lives in that request.
+import { workersAI } from '@rubrion/mastra-cloudflare';
+export const DEFAULT_MODEL = '@cf/google/gemma-4-26b-a4b-it';
+/** Dollars per million tokens, used to report the cost of each answer. Extend through options.prices. */
+export const PRICES = {
+    [DEFAULT_MODEL]: { input: 0.1, output: 0.3 },
+    'claude-haiku-5-5': { input: 0.1, output: 0.5 },
+    'claude-sonnet-5-5': { input: 2, output: 10 },
+    'claude-opus-5-5': { input: 4, output: 20 },
+    'gpt-4o-mini': { input: 0.15, output: 0.6 },
+    'gpt-4o': { input: 2.5, output: 10 },
+};
+/** Used when a model is not in the table. */
+const PROVIDER_PRICE = {
+    'workers-ai': PRICES[DEFAULT_MODEL],
+    anthropic: PRICES['claude-sonnet-5-5'],
+    openai: PRICES['gpt-4o'],
+};
+/** The choice in the header; anything malformed becomes the default Workers AI model. */
+export function readModelChoice(header, fallback = { provider: 'workers-ai', model: DEFAULT_MODEL }) {
+    try {
+        const c = JSON.parse(header ?? '');
+        if ((c.provider === 'anthropic' || c.provider === 'openai') && typeof c.model === 'string' && c.model && typeof c.key === 'string' && c.key) {
+            return { provider: c.provider, model: c.model, key: c.key };
+        }
+        if (c.provider === 'workers-ai' && typeof c.model === 'string' && c.model.startsWith('@cf/'))
+            return { provider: 'workers-ai', model: c.model };
+    }
+    catch { /* no header */ }
+    return fallback;
+}
+export const priceOf = (choice, prices = PRICES) => prices[choice.model] ?? PROVIDER_PRICE[choice.provider];
+/** Only Workers AI takes temperature and the thinking switch; the others decide on their own. */
+export const isWorkersAI = (choice) => choice.provider === 'workers-ai';
+/** The AI SDK model for a choice. A provider not enabled falls back to Workers AI. */
+export async function resolveModel(choice, ai, thinking, providers = {}) {
+    if (choice.provider === 'anthropic' && providers.anthropic) {
+        const { createAnthropic } = await providers.anthropic();
+        return createAnthropic({ apiKey: choice.key })(choice.model);
+    }
+    if (choice.provider === 'openai' && providers.openai) {
+        const { createOpenAI } = await providers.openai();
+        return createOpenAI({ apiKey: choice.key }).chat(choice.model);
+    }
+    return workersAI(ai, isWorkersAI(choice) ? choice.model : DEFAULT_MODEL, { thinking });
+}
+//# sourceMappingURL=models.js.map

package/dist/models.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../src/models.ts"],"names":[],"mappings":"AAAA,iFAAiF;AACjF,kFAAkF;AAClF,gFAAgF;AAChF,OAAO,EAAE,SAAS,EAA8B,MAAM,4BAA4B,CAAC;AAGnF,MAAM,CAAC,MAAM,aAAa,GAAG,+BAA+B,CAAC;AAE7D,yGAAyG;AACzG,MAAM,CAAC,MAAM,MAAM,GAA0B;IAC5C,CAAC,aAAa,CAAC,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;IAC5C,kBAAkB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;IAC/C,mBAAmB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;IAC7C,iBAAiB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;IAC3C,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE;IAC3C,QAAQ,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE;CACpC,CAAC;AAEF,6CAA6C;AAC7C,MAAM,cAAc,GAA4B;IAC/C,YAAY,EAAE,MAAM,CAAC,aAAa,CAAC;IACnC,SAAS,EAAE,MAAM,CAAC,mBAAmB,CAAC;IACtC,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC;CACxB,CAAC;AAEF,yFAAyF;AACzF,MAAM,UAAU,eAAe,CAAC,MAAiC,EAAE,WAAwB,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,aAAa,EAAE;IAC1I,IAAI,CAAC;QACJ,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,EAAE,CAAyB,CAAC;QAC3D,IAAI,CAAC,CAAC,CAAC,QAAQ,KAAK,WAAW,IAAI,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,IAAI,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ,IAAI,CAAC,CAAC,KAAK,IAAI,OAAO,CAAC,CAAC,GAAG,KAAK,QAAQ,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC;YAC7I,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC;QAC7D,CAAC;QACD,IAAI,CAAC,CAAC,QAAQ,KAAK,YAAY,IAAI,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ,IAAI,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC;YAAE,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;IACjJ,CAAC;IAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC;IAC3B,OAAO,QAAQ,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,MAAM,OAAO,GAAG,CAAC,MAAmB,EAAE,SAAgC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;AAEhJ,iGAAiG;AACjG,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,MAAmB,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,KAAK,YAAY,CAAC;AAarF,sFAAsF;AACtF,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,MAAmB,EAAE,EAAa,EAAE,QAAiB,EAAE,YAAuB,EAAE;IAClH,IAAI,MAAM,CAAC,QAAQ,KAAK,WAAW,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;QAC5D,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,SAAS,CAAC,SAAS,EAAE,CAAC;QACxD,OAAO,eAAe,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAU,CAAC;IACvE,CAAC;IACD,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;QACtD,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,SAAS,CAAC,MAAM,EAAE,CAAC;QAClD,OAAO,YAAY,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAU,CAAC;IACzE,CAAC;IACD,OAAO,SAAS,CAAC,EAAE,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;AACxF,CAAC"}

package/dist/stream.d.ts

@@ -0,0 +1,11 @@
+import type { StreamEvent } from './types.js';
+export declare function slim(chunk: unknown, errorMessage?: (raw: string) => string): StreamEvent | null;
+/** Token usage reported by a step-finish or finish chunk. */
+export declare function usageOf(chunk: unknown): {
+    input: number;
+    output: number;
+};
+/** Provider errors the person can act on; the rest pass through. */
+export declare function defaultErrorMessage(m: string): string;
+export declare const sse: (o: unknown) => string;
+export declare const SSE_DONE = "data: [DONE]\n\n";

package/dist/stream.js

@@ -0,0 +1,49 @@
+export function slim(chunk, errorMessage = (m) => m) {
+    if (!chunk || typeof chunk !== 'object')
+        return null;
+    const { type, payload = {} } = chunk;
+    switch (type) {
+        case 'text-delta':
+        case 'reasoning-delta':
+            return { type, text: String(payload.text ?? '') };
+        case 'reasoning-start':
+        case 'reasoning-end':
+        case 'text-start':
+        case 'text-end':
+        case 'step-start':
+        case 'step-finish':
+        case 'finish':
+            return { type };
+        case 'tool-call-input-streaming-start':
+            return { type: 'tool-start', id: String(payload.toolCallId), name: String(payload.toolName) };
+        case 'tool-call':
+            return { type: 'tool-call', id: String(payload.toolCallId), name: String(payload.toolName), args: payload.args };
+        case 'tool-result':
+            return { type: 'tool-result', id: String(payload.toolCallId), name: String(payload.toolName), result: payload.result, error: Boolean(payload.isError) };
+        case 'tool-error':
+            return { type: 'tool-result', id: String(payload.toolCallId), name: String(payload.toolName), result: null, error: true };
+        case 'error':
+            return { type: 'error', message: errorMessage(String(payload.error?.message ?? payload.error ?? 'error')) };
+        default:
+            return null;
+    }
+}
+/** Token usage reported by a step-finish or finish chunk. */
+export function usageOf(chunk) {
+    const p = (chunk.payload ?? {});
+    const u = p.output?.usage ?? p.totalUsage ?? p.usage ?? {};
+    return { input: Number(u.inputTokens ?? u.promptTokens ?? 0), output: Number(u.outputTokens ?? u.completionTokens ?? 0) };
+}
+/** Provider errors the person can act on; the rest pass through. */
+export function defaultErrorMessage(m) {
+    if (/\b401\b|unauthori[sz]ed|invalid.{0,20}(api.?key|x-api-key)|authentication|incorrect api key/i.test(m)) {
+        return 'The AI provider rejected the access key. Check the key in the AI settings.';
+    }
+    if (/\b429\b|rate.?limit|quota|credit balance/i.test(m)) {
+        return 'The AI provider refused because of a rate limit or account credit. Check the provider account.';
+    }
+    return m;
+}
+export const sse = (o) => `data: ${JSON.stringify(o)}\n\n`;
+export const SSE_DONE = 'data: [DONE]\n\n';
+//# sourceMappingURL=stream.js.map

package/dist/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../src/stream.ts"],"names":[],"mappings":"AAIA,MAAM,UAAU,IAAI,CAAC,KAAc,EAAE,eAAwC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACpF,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IACrD,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,EAAE,EAAE,GAAG,KAA4D,CAAC;IAC5F,QAAQ,IAAI,EAAE,CAAC;QACd,KAAK,YAAY,CAAC;QAClB,KAAK,iBAAiB;YACrB,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,CAAC;QACnD,KAAK,iBAAiB,CAAC;QACvB,KAAK,eAAe,CAAC;QACrB,KAAK,YAAY,CAAC;QAClB,KAAK,UAAU,CAAC;QAChB,KAAK,YAAY,CAAC;QAClB,KAAK,aAAa,CAAC;QACnB,KAAK,QAAQ;YACZ,OAAO,EAAE,IAAI,EAAE,CAAC;QACjB,KAAK,iCAAiC;YACrC,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC/F,KAAK,WAAW;YACf,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC;QAClH,KAAK,aAAa;YACjB,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACzJ,KAAK,YAAY;YAChB,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAC3H,KAAK,OAAO;YACX,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,CAAC,MAAM,CAAE,OAAO,CAAC,KAAe,EAAE,OAAO,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,CAAC,EAAE,CAAC;QACxH;YACC,OAAO,IAAI,CAAC;IACd,CAAC;AACF,CAAC;AAED,6DAA6D;AAC7D,MAAM,UAAU,OAAO,CAAC,KAAc;IACrC,MAAM,CAAC,GAAG,CAAE,KAA+C,CAAC,OAAO,IAAI,EAAE,CAAwB,CAAC;IAClG,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;IAC3D,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,gBAAgB,IAAI,CAAC,CAAC,EAAE,CAAC;AAC3H,CAAC;AAED,oEAAoE;AACpE,MAAM,UAAU,mBAAmB,CAAC,CAAS;IAC5C,IAAI,8FAA8F,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5G,OAAO,4EAA4E,CAAC;IACrF,CAAC;IACD,IAAI,2CAA2C,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;QACzD,OAAO,gGAAgG,CAAC;IACzG,CAAC;IACD,OAAO,CAAC,CAAC;AACV,CAAC;AAED,MAAM,CAAC,MAAM,GAAG,GAAG,CAAC,CAAU,EAAE,EAAE,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC;AACpE,MAAM,CAAC,MAAM,QAAQ,GAAG,kBAAkB,CAAC"}

package/dist/turns.d.ts

@@ -0,0 +1,9 @@
+export type StoredMessage = {
+    role: string;
+    content: {
+        parts?: Record<string, any>[];
+        content?: string;
+    };
+};
+/** Stored Mastra messages → turns. One answer may be stored as several messages (a tool step, then the text). */
+export declare function toTurns(messages: StoredMessage[], agent: string): Record<string, any>[];

package/dist/turns.js

@@ -0,0 +1,28 @@
+/** Stored Mastra messages → turns. One answer may be stored as several messages (a tool step, then the text). */
+export function toTurns(messages, agent) {
+    const turns = [];
+    for (const m of messages) {
+        const parts = m.content.parts ?? [];
+        if (m.role === 'user') {
+            const text = parts.filter((p) => p.type === 'text').map((p) => p.text).join('\n') || m.content.content || '';
+            if (text)
+                turns.push({ id: crypto.randomUUID(), role: 'user', text });
+        }
+        else if (m.role === 'assistant') {
+            const steps = parts.flatMap((p) => p.type === 'text' && p.text
+                ? [{ kind: 'text', text: p.text }]
+                : p.type === 'reasoning' && (p.reasoning || p.text)
+                    ? [{ kind: 'think', text: p.reasoning ?? p.text, start: 0, end: 0 }]
+                    : p.type === 'tool-invocation'
+                        ? [{ kind: 'tool', id: p.toolInvocation.toolCallId, name: p.toolInvocation.toolName, args: p.toolInvocation.args, result: p.toolInvocation.result, state: 'done', start: 0, end: 0 }]
+                        : []);
+            const last = turns[turns.length - 1];
+            if (last?.role === 'assistant')
+                last.steps.push(...steps);
+            else if (steps.length)
+                turns.push({ id: crypto.randomUUID(), role: 'assistant', agent, steps, done: true });
+        }
+    }
+    return turns;
+}
+//# sourceMappingURL=turns.js.map

package/dist/turns.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"turns.js","sourceRoot":"","sources":["../src/turns.ts"],"names":[],"mappings":"AAGA,iHAAiH;AACjH,MAAM,UAAU,OAAO,CAAC,QAAyB,EAAE,KAAa;IAC/D,MAAM,KAAK,GAA0B,EAAE,CAAC;IACxC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;QACpC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACvB,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;YAC7G,IAAI,IAAI;gBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACvE,CAAC;aAAM,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAA6B,EAAE,CAC5D,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI;gBAC1B,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBAClC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,IAAI,CAAC;oBAClD,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;oBACpE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB;wBAC7B,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;wBACrL,CAAC,CAAC,EAAE,CACP,CAAC;YACF,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACrC,IAAI,IAAI,EAAE,IAAI,KAAK,WAAW;gBAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;iBACrD,IAAI,KAAK,CAAC,MAAM;gBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7G,CAAC;IACF,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,82 @@
+/**
+ * Who is asking. The gateway resolves it from the app's own session and sends it
+ * to the agent worker in headers; the agent worker trusts it because it is only
+ * reachable through a service binding.
+ */
+export interface Caller {
+    /** The account, office, workspace… everything this person's data lives under. */
+    tenant: string;
+    /** The person inside the tenant. Conversations are private to tenant + person. */
+    person: string;
+    /** Free-form role, available to tools and context (e.g. to hide financial data). */
+    role: string;
+    /** Display name, given to the agent so it can address the person. */
+    name: string;
+}
+export type Provider = 'workers-ai' | 'anthropic' | 'openai';
+/** The model that answers a request. `key` exists only for bring-your-own-key providers. */
+export interface ModelChoice {
+    provider: Provider;
+    model: string;
+    key?: string;
+}
+/** Dollars per million tokens. */
+export interface Price {
+    input: number;
+    output: number;
+}
+/**
+ * One agent the assistant offers. Each becomes a Mastra Agent, built once per
+ * isolate; the model is resolved per request (the tenant may bring its own key).
+ */
+export interface AgentDefinition {
+    id: string;
+    name: string;
+    /** One line for the agent picker. */
+    tagline: string;
+    /** Reasoning on (Workers AI): slower and pricier, and the UI shows the thinking. */
+    thinking?: boolean;
+    /** Mastra tools (createTool). Prefer small read-only tools that return facts. */
+    tools: Record<string, unknown>;
+    /** Output cap on Workers AI. External models get more room, since their thinking counts as output. */
+    maxOutputTokens?: number;
+    instructions: string;
+}
+/** What one answer cost, reported when the stream ends. */
+export interface Usage {
+    caller: Caller;
+    agent: string;
+    thread: string;
+    provider: Provider;
+    model: string;
+    inputTokens: number;
+    outputTokens: number;
+    costUsd: number;
+    status: 'ok' | 'error';
+    error?: string;
+}
+/** A stream event as the browser receives it (`data: {json}` lines, then `data: [DONE]`). */
+export type StreamEvent = {
+    type: 'text-delta' | 'reasoning-delta';
+    text: string;
+} | {
+    type: 'reasoning-start' | 'reasoning-end' | 'text-start' | 'text-end' | 'step-start' | 'step-finish' | 'finish';
+} | {
+    type: 'tool-start';
+    id: string;
+    name: string;
+} | {
+    type: 'tool-call';
+    id: string;
+    name: string;
+    args: unknown;
+} | {
+    type: 'tool-result';
+    id: string;
+    name: string;
+    result: unknown;
+    error: boolean;
+} | {
+    type: 'error';
+    message: string;
+};

package/dist/types.js

@@ -0,0 +1,4 @@
+// Shared shapes of the server side: who is asking, which model answers, and how
+// an agent is declared.
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,gFAAgF;AAChF,wBAAwB"}

package/dist/worker.d.ts

@@ -0,0 +1,54 @@
+import { Hono } from 'hono';
+import { type AiBinding } from '@rubrion/mastra-cloudflare';
+import { type Providers } from './models.js';
+import type { AgentDefinition, Caller, ModelChoice, Price, Usage } from './types.js';
+export interface AgentWorkerOptions<E> {
+    agents: AgentDefinition[];
+    /** The Workers AI binding. */
+    ai: (env: E) => AiBinding;
+    /** The D1 database for Mastra Memory (tables are created on first use). */
+    memory: (env: E) => D1Database;
+    /** Messages recalled into each prompt. Default 8. */
+    lastMessages?: number;
+    /** Conversations kept per person; the oldest is deleted past this. Default 50. */
+    maxThreads?: number;
+    /** Tool-call rounds per answer. Default 8. */
+    maxSteps?: number;
+    /** Longest question accepted, in characters. Default 4000. */
+    maxQuestion?: number;
+    /** The model when the gateway sends none. Default: Workers AI Gemma. */
+    defaultModel?: ModelChoice;
+    prices?: Record<string, Price>;
+    /** Bring-your-own-key providers the gateway may select (see Providers). Off by default. */
+    providers?: Providers;
+    /**
+     * Text appended to the instructions on every question: a snapshot of what
+     * exists (projects, team…), so the agent knows the ground before calling tools.
+     */
+    context?: (args: {
+        caller: Caller;
+        env: E;
+        agent: AgentDefinition;
+    }) => string | Promise<string>;
+    /**
+     * Wraps each request, e.g. to put the tenant in an AsyncLocalStorage that the
+     * tools read so every query is scoped. Must return what `next` returns.
+     */
+    scope?: (args: {
+        caller: Caller;
+        env: E;
+    }, next: () => Promise<Response>) => Promise<Response>;
+    /** Reported when an answer ends: log it, bill it, enforce a monthly cap. */
+    onUsage?: (usage: Usage, env: E) => void | Promise<void>;
+    /** Turns provider errors into something the person can act on. */
+    errorMessage?: (raw: string) => string;
+}
+export declare function createAgentWorker<E extends object = Record<string, unknown>>(o: AgentWorkerOptions<E>): {
+    fetch: (request: Request, env?: {} | E | undefined, executionCtx?: import("hono").ExecutionContext) => Response | Promise<Response>;
+    app: Hono<{
+        Bindings: E;
+        Variables: {
+            caller: Caller;
+        };
+    }, import("hono/types").BlankSchema, "/">;
+};

package/dist/worker.js

@@ -0,0 +1,186 @@
+// The agent worker: Mastra agents behind a few plain routes, reached only through
+// a service binding from the gateway (no public route).
+//
+//   GET    /agents        the agents (id, name, tagline, thinking)
+//   POST   /stream        one question: Mastra's stream as slim SSE
+//   GET    /threads       the person's recent conversations (all agents)
+//   GET    /threads/:id   one conversation, shaped as the UI draws it (?agent=)
+//   DELETE /threads/:id   delete a conversation
+//
+// Own routes rather than Mastra's REST server: importing that route table at load
+// costs most of a Worker's startup budget.
+import { Hono } from 'hono';
+import { Agent, Mastra, RequestContext, d1Memory, workersAIOptions } from '@rubrion/mastra-cloudflare';
+import { readCaller, HEADER } from './headers.js';
+import { DEFAULT_MODEL, isWorkersAI, priceOf, readModelChoice, resolveModel, PRICES } from './models.js';
+import { defaultErrorMessage, slim, sse, SSE_DONE, usageOf } from './stream.js';
+import { toTurns } from './turns.js';
+const CLIENT_ID = /^[a-z0-9]{6,32}$/i;
+const TITLE_CHARS = 60;
+export function createAgentWorker(o) {
+    const lastMessages = o.lastMessages ?? 8;
+    const maxThreads = o.maxThreads ?? 50;
+    const maxSteps = o.maxSteps ?? 8;
+    const maxQuestion = o.maxQuestion ?? 4000;
+    const prices = { ...PRICES, ...o.prices };
+    const errorMessage = o.errorMessage ?? defaultErrorMessage;
+    const fallback = o.defaultModel ?? { provider: 'workers-ai', model: DEFAULT_MODEL };
+    if (!o.agents.length)
+        throw new Error('createAgentWorker: at least one agent');
+    // Bindings are stable per isolate; memory and agents are built on the first request.
+    let current;
+    const getMemory = d1Memory(() => o.memory(current), { id: 'agent-kit-memory', lastMessages });
+    const built = new Map();
+    let mastra;
+    const definition = (id) => o.agents.find((a) => a.id === id) ?? o.agents[0];
+    function agentOf(def) {
+        let agent = built.get(def.id);
+        if (!agent) {
+            agent = new Agent({
+                id: def.id,
+                name: def.name,
+                instructions: def.instructions,
+                // the model is resolved per request: the tenant may bring its own key
+                model: (({ requestContext }) => resolveModel(requestContext?.get('model') ?? fallback, o.ai(current), Boolean(def.thinking), o.providers)),
+                tools: def.tools,
+                memory: () => getMemory(),
+            });
+            built.set(def.id, agent);
+        }
+        return agent;
+    }
+    /** Conversations belong to tenant + person; nobody opens another's. */
+    const ownerOf = (c) => `t${c.tenant}-p${c.person}`;
+    const clientId = (thread) => (typeof thread === 'string' && CLIENT_ID.test(thread) ? thread : 'default');
+    const threadId = (owner, thread, agent) => `${owner}-${clientId(thread)}-${agent}`;
+    async function threadsOf(owner) {
+        const r = await getMemory()
+            .listThreads({ filter: { resourceId: owner }, orderBy: { field: 'updatedAt', direction: 'DESC' }, perPage: false })
+            .catch(() => ({ threads: [] }));
+        return r.threads;
+    }
+    /** Creates the thread on its first message, titled by it, and keeps only the newest maxThreads. */
+    async function openThread(owner, thread, agent, message) {
+        const memory = getMemory();
+        const id = threadId(owner, thread, agent);
+        if (await memory.getThreadById({ threadId: id }).catch(() => null))
+            return id;
+        const now = new Date();
+        const title = message.replace(/\s+/g, ' ').trim().slice(0, TITLE_CHARS) || '…';
+        await memory.saveThread({ thread: { id, resourceId: owner, title, metadata: { agent, client: clientId(thread) }, createdAt: now, updatedAt: now } });
+        const old = (await threadsOf(owner)).slice(maxThreads);
+        await Promise.all(old.map((t) => memory.deleteThread(t.id).catch(() => undefined)));
+        return id;
+    }
+    const app = new Hono();
+    app.use(async (c, next) => {
+        current = c.env;
+        const caller = readCaller(c.req.raw.headers);
+        if (!caller)
+            return c.json({ error: 'Unknown caller.' }, 401);
+        c.set('caller', caller);
+        if (!o.scope)
+            return next();
+        return o.scope({ caller, env: c.env }, async () => { await next(); return c.res; });
+    });
+    app.get('/agents', (c) => c.json(o.agents.map(({ id, name, tagline, thinking }) => ({ id, name, tagline, thinking: Boolean(thinking) }))));
+    app.post('/stream', async (c) => {
+        const caller = c.get('caller');
+        const body = (await c.req.json().catch(() => null));
+        const message = typeof body?.message === 'string' ? body.message.trim().slice(0, maxQuestion) : '';
+        if (!message)
+            return c.json({ error: 'Empty question.' }, 400);
+        mastra ??= new Mastra({ agents: Object.fromEntries(o.agents.map((a) => [a.id, agentOf(a)])) });
+        const def = definition(body?.agent);
+        const agent = agentOf(def);
+        const asked = readModelChoice(c.req.header(HEADER.model), fallback);
+        const choice = asked.provider === 'workers-ai' || o.providers?.[asked.provider] ? asked : fallback;
+        const price = priceOf(choice, prices);
+        const owner = ownerOf(caller);
+        const thread = await openThread(owner, body?.thread, def.id, message);
+        const extra = await Promise.resolve(o.context?.({ caller, env: c.env, agent: def })).catch(() => '');
+        const who = caller.name ? `The person asking: ${caller.name}.` : '';
+        const instructions = [def.instructions, who, extra].filter(Boolean).join('\n\n');
+        const result = await agent.stream([{ role: 'user', content: message }], {
+            requestContext: new RequestContext([['caller', caller], ['model', choice]]),
+            // Mastra Memory recalls the thread's last messages; the UI sends only the new one.
+            memory: { thread, resource: owner },
+            maxSteps,
+            // temperature 0 on Workers AI: the same question over the same data gives the same answer
+            modelSettings: isWorkersAI(choice) ? { maxOutputTokens: def.maxOutputTokens ?? 1200, temperature: 0 } : { maxOutputTokens: 8000 },
+            providerOptions: (isWorkersAI(choice) ? workersAIOptions(Boolean(def.thinking)) : undefined),
+            instructions,
+            abortSignal: c.req.raw.signal,
+        });
+        const { readable, writable } = new TransformStream();
+        const writer = writable.getWriter();
+        const enc = new TextEncoder();
+        const send = (s) => writer.write(enc.encode(s));
+        // No await: the response goes out while this keeps writing to it.
+        c.executionCtx.waitUntil((async () => {
+            const reader = result.fullStream.getReader();
+            let input = 0;
+            let output = 0;
+            let error = '';
+            try {
+                for (;;) {
+                    const { done, value } = await reader.read();
+                    if (done)
+                        break;
+                    const ev = slim(value, errorMessage);
+                    if (!ev)
+                        continue;
+                    if (ev.type === 'step-finish') {
+                        const u = usageOf(value);
+                        input += u.input;
+                        output += u.output;
+                    }
+                    else if (ev.type === 'finish' && !input && !output) {
+                        const u = usageOf(value);
+                        input = u.input;
+                        output = u.output;
+                    }
+                    if (ev.type === 'error')
+                        error = ev.message;
+                    await send(sse(ev));
+                }
+            }
+            catch (e) {
+                error = e.message;
+                await send(sse({ type: 'error', message: errorMessage(error) })).catch(() => undefined);
+            }
+            finally {
+                const usage = {
+                    caller, agent: def.id, thread, provider: choice.provider, model: choice.model,
+                    inputTokens: input, outputTokens: output, costUsd: (input * price.input + output * price.output) / 1_000_000,
+                    status: error ? 'error' : 'ok', ...(error ? { error: error.slice(0, 500) } : {}),
+                };
+                await Promise.resolve(o.onUsage?.(usage, c.env)).catch((e) => console.error('onUsage failed', e));
+                await send(SSE_DONE).catch(() => undefined);
+                await writer.close().catch(() => undefined);
+            }
+        })());
+        return new Response(readable, { headers: { 'content-type': 'text/event-stream', 'cache-control': 'no-cache' } });
+    });
+    app.get('/threads', async (c) => {
+        const threads = (await threadsOf(ownerOf(c.get('caller')))).filter((t) => t.metadata?.agent && t.metadata?.client);
+        return c.json(threads.slice(0, maxThreads).map((t) => ({ id: t.metadata.client, agent: t.metadata.agent, title: t.title ?? '', updatedAt: t.updatedAt })));
+    });
+    /** A stored conversation as the UI draws it: the person's text; the agent's steps and text. */
+    app.get('/threads/:id', async (c) => {
+        const owner = ownerOf(c.get('caller'));
+        const agent = definition(c.req.query('agent')).id;
+        const { messages } = await getMemory()
+            .recall({ threadId: threadId(owner, c.req.param('id'), agent), resourceId: owner, perPage: false })
+            .catch(() => ({ messages: [] }));
+        return c.json(toTurns(messages, agent));
+    });
+    app.delete('/threads/:id', async (c) => {
+        const owner = ownerOf(c.get('caller'));
+        const memory = getMemory();
+        await Promise.all(o.agents.map((a) => memory.deleteThread(threadId(owner, c.req.param('id'), a.id)).catch(() => undefined)));
+        return c.json({ ok: true });
+    });
+    return { fetch: app.fetch, app };
+}
+//# sourceMappingURL=worker.js.map

package/dist/worker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.js","sourceRoot":"","sources":["../src/worker.ts"],"names":[],"mappings":"AAAA,kFAAkF;AAClF,wDAAwD;AACxD,EAAE;AACF,mEAAmE;AACnE,oEAAoE;AACpE,yEAAyE;AACzE,gFAAgF;AAChF,gDAAgD;AAChD,EAAE;AACF,kFAAkF;AAClF,2CAA2C;AAC3C,OAAO,EAAE,IAAI,EAAgB,MAAM,MAAM,CAAC;AAC1C,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,gBAAgB,EAAkB,MAAM,4BAA4B,CAAC;AACvH,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,OAAO,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,EAAkB,MAAM,aAAa,CAAC;AACzH,OAAO,EAAE,mBAAmB,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAChF,OAAO,EAAE,OAAO,EAAsB,MAAM,YAAY,CAAC;AAsCzD,MAAM,SAAS,GAAG,mBAAmB,CAAC;AACtC,MAAM,WAAW,GAAG,EAAE,CAAC;AAIvB,MAAM,UAAU,iBAAiB,CAA6C,CAAwB;IACrG,MAAM,YAAY,GAAG,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,CAAC,CAAC,UAAU,IAAI,EAAE,CAAC;IACtC,MAAM,QAAQ,GAAG,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC;IACjC,MAAM,WAAW,GAAG,CAAC,CAAC,WAAW,IAAI,IAAI,CAAC;IAC1C,MAAM,MAAM,GAAG,EAAE,GAAG,MAAM,EAAE,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;IAC1C,MAAM,YAAY,GAAG,CAAC,CAAC,YAAY,IAAI,mBAAmB,CAAC;IAC3D,MAAM,QAAQ,GAAG,CAAC,CAAC,YAAY,IAAI,EAAE,QAAQ,EAAE,YAAqB,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;IAC7F,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;IAE/E,qFAAqF;IACrF,IAAI,OAAsB,CAAC;IAC3B,MAAM,SAAS,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,OAAQ,CAAC,EAAE,EAAE,EAAE,EAAE,kBAAkB,EAAE,YAAY,EAAE,CAAC,CAAC;IAC/F,MAAM,KAAK,GAAG,IAAI,GAAG,EAAiB,CAAC;IACvC,IAAI,MAA0B,CAAC;IAE/B,MAAM,UAAU,GAAG,CAAC,EAAW,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACrF,SAAS,OAAO,CAAC,GAAoB;QACpC,IAAI,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC9B,IAAI,CAAC,KAAK,EAAE,CAAC;YACZ,KAAK,GAAG,IAAI,KAAK,CAAC;gBACjB,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,sEAAsE;gBACtE,KAAK,EAAE,CAAC,CAAC,EAAE,cAAc,EAAuC,EAAE,EAAE,CACnE,YAAY,CAAE,cAAc,EAAE,GAAG,CAAC,OAAO,CAA6B,IAAI,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,OAAQ,CAAC,EAAE,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAU;gBAClJ,KAAK,EAAE,GAAG,CAAC,KAAc;gBACzB,MAAM,EAAE,GAAG,EAAE,CAAC,SAAS,EAAE;aACzB,CAAC,CAAC;YACH,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAC1B,CAAC;QACD,OAAO,KAAK,CAAC;IACd,CAAC;IAED,uEAAuE;IACvE,MAAM,OAAO,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC;IAC3D,MAAM,QAAQ,GAAG,CAAC,MAAe,EAAE,EAAE,CAAC,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAClH,MAAM,QAAQ,GAAG,CAAC,KAAa,EAAE,MAAe,EAAE,KAAa,EAAE,EAAE,CAAC,GAAG,KAAK,IAAI,QAAQ,CAAC,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;IAE5G,KAAK,UAAU,SAAS,CAAC,KAAa;QACrC,MAAM,CAAC,GAAG,MAAM,SAAS,EAAE;aACzB,WAAW,CAAC,EAAE,MAAM,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;aAClH,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,EAAoB,EAAE,CAAC,CAAC,CAAC;QACnD,OAAO,CAAC,CAAC,OAAyB,CAAC;IACpC,CAAC;IAED,mGAAmG;IACnG,KAAK,UAAU,UAAU,CAAC,KAAa,EAAE,MAAe,EAAE,KAAa,EAAE,OAAe;QACvF,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,EAAE,GAAG,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QAC1C,IAAI,MAAM,MAAM,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;YAAE,OAAO,EAAE,CAAC;QAC9E,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,IAAI,GAAG,CAAC;QAC/E,MAAM,MAAM,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QACrJ,MAAM,GAAG,GAAG,CAAC,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACvD,MAAM,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QACpF,OAAO,EAAE,CAAC;IACX,CAAC;IAGD,MAAM,GAAG,GAAG,IAAI,IAAI,EAAO,CAAC;IAE5B,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE;QACzB,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC;QAChB,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,iBAAiB,EAAE,EAAE,GAAG,CAAC,CAAC;QAC9D,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACxB,IAAI,CAAC,CAAC,CAAC,KAAK;YAAE,OAAO,IAAI,EAAE,CAAC;QAC5B,OAAO,CAAC,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,EAAE,EAAE,KAAK,IAAI,EAAE,GAAG,MAAM,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACrF,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAE3I,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC/B,MAAM,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAiE,CAAC;QACpH,MAAM,OAAO,GAAG,OAAO,IAAI,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACnG,IAAI,CAAC,OAAO;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,iBAAiB,EAAE,EAAE,GAAG,CAAC,CAAC;QAE/D,MAAM,KAAK,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC/F,MAAM,GAAG,GAAG,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACpC,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;QAC3B,MAAM,KAAK,GAAG,eAAe,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,QAAQ,CAAC,CAAC;QACpE,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,KAAK,YAAY,IAAI,CAAC,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC;QACnG,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACtC,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAEtE,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QACrG,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,sBAAsB,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QACpE,MAAM,YAAY,GAAG,CAAC,GAAG,CAAC,YAAY,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEjF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAU,EAAE;YAChF,cAAc,EAAE,IAAI,cAAc,CAAC,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;YAC3E,mFAAmF;YACnF,MAAM,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE;YACnC,QAAQ;YACR,0FAA0F;YAC1F,aAAa,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,GAAG,CAAC,eAAe,IAAI,IAAI,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE;YACjI,eAAe,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAU;YACrG,YAAY;YACZ,WAAW,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM;SAC7B,CAAC,CAAC;QAEH,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,IAAI,eAAe,EAA0B,CAAC;QAC7E,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC;QACpC,MAAM,GAAG,GAAG,IAAI,WAAW,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAExD,kEAAkE;QAClE,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,KAAK,IAAI,EAAE;YACpC,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC;YAC7C,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,IAAI,MAAM,GAAG,CAAC,CAAC;YACf,IAAI,KAAK,GAAG,EAAE,CAAC;YACf,IAAI,CAAC;gBACJ,SAAS,CAAC;oBACT,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;oBAC5C,IAAI,IAAI;wBAAE,MAAM;oBAChB,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;oBACrC,IAAI,CAAC,EAAE;wBAAE,SAAS;oBAClB,IAAI,EAAE,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;wBAC/B,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;wBACzB,KAAK,IAAI,CAAC,CAAC,KAAK,CAAC;wBACjB,MAAM,IAAI,CAAC,CAAC,MAAM,CAAC;oBACpB,CAAC;yBAAM,IAAI,EAAE,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI,CAAC,MAAM,EAAE,CAAC;wBACtD,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;wBACzB,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;wBAChB,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;oBACnB,CAAC;oBACD,IAAI,EAAE,CAAC,IAAI,KAAK,OAAO;wBAAE,KAAK,GAAG,EAAE,CAAC,OAAO,CAAC;oBAC5C,MAAM,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;gBACrB,CAAC;YACF,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACZ,KAAK,GAAI,CAAW,CAAC,OAAO,CAAC;gBAC7B,MAAM,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACzF,CAAC;oBAAS,CAAC;gBACV,MAAM,KAAK,GAAU;oBACpB,MAAM,EAAE,KAAK,EAAE,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK;oBAC7E,WAAW,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,GAAG,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,GAAG,SAAS;oBAC5G,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAChF,CAAC;gBACF,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,CAAC;gBAClG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBAC5C,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAC7C,CAAC;QACF,CAAC,CAAC,EAAE,CAAC,CAAC;QAEN,OAAO,IAAI,QAAQ,CAAC,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,mBAAmB,EAAE,eAAe,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;IAClH,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC/B,MAAM,OAAO,GAAG,CAAC,MAAM,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,IAAI,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACnH,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,QAAS,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,QAAS,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;IAC9J,CAAC,CAAC,CAAC;IAEH,+FAA+F;IAC/F,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE,KAAK,EAAE,CAAe,EAAE,EAAE;QACjD,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QACvC,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;QAClD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,SAAS,EAAE;aACpC,MAAM,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;aAClG,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAe,EAAE,CAAC,CAAC,CAAC;QAC/C,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,QAA2B,EAAE,KAAK,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QACtC,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAC7H,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC;AAClC,CAAC"}

package/package.json

@@ -1,6 +1,51 @@
 {
   "name": "@rubrion/agent-kit-server",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "version": "0.1.1",
+  "description": "Agent worker (Mastra on Cloudflare Workers, D1 memory, slim SSE) and the gateway that fronts it.",
+  "license": "MIT",
+  "author": "Rubrion",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/rubrion/agent-kit.git",
+    "directory": "packages/server"
+  },
+  "homepage": "https://github.com/rubrion/agent-kit#readme",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    },
+    "./gateway": {
+      "types": "./dist/gateway-entry.d.ts",
+      "import": "./dist/gateway-entry.js"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "sideEffects": false,
+  "scripts": {
+    "build": "rm -rf dist && tsc -p tsconfig.json",
+    "test": "vitest run",
+    "prepublishOnly": "npm run build"
+  },
+  "dependencies": {
+    "@rubrion/mastra-cloudflare": "^0.3.1",
+    "hono": "^4.13.14"
+  },
+  "devDependencies": {
+    "@cloudflare/workers-types": "^5.20260908.1",
+    "typescript": "^5.9.3",
+    "vitest": "^5.0.5"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "engines": {
+    "node": ">=20"
+  }
+}

package/README.md

@@ -1,3 +0,0 @@
-# Temporary Holding Version
-
-This version is a temporary placeholder for this package. An operational version to replace this has been submitted for review and is awaiting a staged release.