@argentic/chest-sdk 0.2.0 → 0.4.0

package/dist/src/testing.js

@@ -1,11 +1,12 @@
-import { createHash, createHmac, randomBytes } from "node:crypto";
+import { createHash, randomBytes } from "node:crypto";
 import { createServer } from "node:http";
 import { groupIdPattern, memberIdPattern } from "./member.js";
 import { forget } from "./members.js";
-const encode = (value) => Buffer.from(JSON.stringify(value)).toString("base64url");
+import { fakeOrigins } from "./api.js";
+import { eventChannel, scheduleChannel, sign, signClaims } from "./signed.js";
 // signAssertion is the Chest-Member value the Chest's front would send for
-// that member: HS256 under the key of the token (CHEST_TOKEN by default), for
-// the tool (CHEST_TOOL by default), valid 60 seconds from now.
+// that member: HS256 under the key of the token, for the tool, valid 60
+// seconds from when it is issued.
 export function signAssertion(member, options = {}) {
     const token = options.token ?? process.env["CHEST_TOKEN"];
     const tool = options.tool ?? process.env["CHEST_TOOL"];
@@ -14,27 +15,18 @@ export function signAssertion(member, options = {}) {
     if (!memberIdPattern.test(member.id) || !member.groups.every(g => groupIdPattern.test(g)))
         throw new Error("signAssertion needs identifiers of the Chest's shape (mbr_…, grp_…)");
     const iat = Math.floor((options.now ?? new Date()).getTime() / 1000);
-    const body = encode({ alg: "HS256", typ: "JWT" }) + "." + encode({
+    // Under the label of the assertion's shape, as the Chest signs it and
+    // member() reads it.
+    return signClaims("Chest-Member v2", {
         iss: `https://${tool}-chest.chest.test`, aud: tool, iat, exp: iat + 60, sub: member.id,
         given_name: member.firstName, family_name: member.lastName, name: member.name, picture: member.photo ?? "", role: member.role ?? "",
-        admin: member.isAdmin, builder: member.isBuilder, groups: member.groups, ...(member.email === undefined ? {} : { email: member.email }),
-    });
-    // The key as the Chest derives it, and member() reads it: HMAC-SHA256 of
-    // the label of the assertion's shape under the text of the token.
-    const key = createHmac("sha256", Buffer.from(token, "utf8")).update("Chest-Member v2").digest();
-    return body + "." + createHmac("sha256", key).update(body).digest("base64url");
-}
-// signEvent is the Chest-Event value the Chest would send with that body:
-// HS256 under the key the token derives for events, for the tool, naming the
-// event and the digest of the body, valid 60 seconds.
-function signEvent(id, body, options) {
-    const iat = Math.floor(Date.now() / 1000);
-    const signed = encode({ alg: "HS256", typ: "JWT" }) + "." + encode({ aud: options.tool, iat, exp: iat + 60, jti: id, digest: createHash("sha256").update(body).digest("base64url") });
-    const key = createHmac("sha256", Buffer.from(options.token, "utf8")).update("Chest-Event v1").digest();
-    return signed + "." + createHmac("sha256", key).update(signed).digest("base64url");
+        admin: member.isAdmin, builder: member.isBuilder, groups: member.groups, time_zone: member.timeZone, ...(member.email === undefined ? {} : { email: member.email }),
+        language: member.language,
+    }, token);
 }
-// withMember is the request carrying that member's assertion: a new Web
-// Request, or the same Node request with its header set.
+// withMember is the request carrying that member's assertion, signed with
+// the options of signAssertion: a new Web Request, or the same Node request
+// with its header set.
 export function withMember(request, member, options = {}) {
     const assertion = signAssertion(member, options);
     if (request instanceof Request) {
@@ -49,6 +41,19 @@ export function withMember(request, member, options = {}) {
 const maxLimit = 500, defaultLimit = 100, maxLookup = 200, callsPerMinute = 600;
 const maxObject = 32 << 20, maxObjects = 10000, maxTotal = 1 << 30;
 const namePattern = /^[A-Za-z0-9][A-Za-z0-9._-]{0,99}(\/[A-Za-z0-9][A-Za-z0-9._-]{0,99}){0,7}$/u;
+// Its links and uploads: their paths on the team host, their lives in
+// seconds, the images it makes thumbnails of, the first bytes an upload of
+// a type must start with, the endings of the names it chooses (chest/toolfiles).
+const linkPath = "/_chest/files/", uploadPath = "/_chest/files/upload/", linkLife = 900, uploadLife = 900;
+const mediaPattern = /^[a-z0-9][a-z0-9!#$&^_.+-]{0,62}\/(\*|[a-z0-9][a-z0-9!#$&^_.+-]{0,62})$/u;
+const thumbnailed = ["image/jpeg", "image/png", "image/gif", "image/webp"];
+const starts = (...heads) => (data) => heads.some(head => data.subarray(0, head.length).equals(Buffer.from(head, "latin1")));
+const signatures = {
+    "image/jpeg": starts("\xff\xd8\xff"), "image/png": starts("\x89PNG\r\n\x1a\n"), "image/gif": starts("GIF87a", "GIF89a"),
+    "image/webp": data => data.subarray(0, 4).toString("latin1") === "RIFF" && data.subarray(8, 12).toString("latin1") === "WEBP", "application/pdf": starts("%PDF-"),
+};
+const extensions = { "image/jpeg": ".jpg", "image/png": ".png", "image/gif": ".gif", "image/webp": ".webp", "image/avif": ".avif", "image/heic": ".heic", "application/pdf": ".pdf", "text/plain": ".txt", "text/csv": ".csv", "application/json": ".json", "application/zip": ".zip", "video/mp4": ".mp4", "audio/mpeg": ".mp3" };
+const newToken = () => randomBytes(18).toString("base64url") + "." + randomBytes(12).toString("base64url");
 // Its notifications: recipients and badges a call, text, quotas.
 const maxRecipients = 500, maxTitle = 80, maxText = 280, maxPath = 512, maxCount = 9999;
 const recipientsPerHour = 1000, itemsPerDay = 100, badgesPerMinute = 600;
@@ -57,6 +62,22 @@ const reordering = /[\u061c\u200e\u200f\u202a-\u202e\u2066-\u2069]/gu;
 const cleanTitle = (s) => s.replace(/[\t\r\n]/gu, " ").replace(/\p{Cc}/gu, "").replace(reordering, "").trim();
 const cleanText = (s) => s.replace(/\r\n?/gu, "\n").replace(/\t/gu, " ").replace(/[^\P{Cc}\n]/gu, "").replace(reordering, "").trim();
 const isPath = (p) => typeof p === "string" && p.length <= maxPath && /^\/chest([/?#][\x21-\x5b\x5d-\x7e]*)?$/u.test(p) && !p.includes("//") && !p.split(/[?#]/u)[0].split("/").some(x => /^(\.|%2e){1,2}$/iu.test(x));
+// Its AI: aliases in order, bounds of a request, requests a minute.
+const aliasOrder = ["default", "fast", "smart", "embedding"];
+const chatKeys = ["model", "messages", "max_tokens", "stream", "temperature", "top_p", "stop", "tools", "tool_choice", "response_format", "parallel_tool_calls", "seed", "reasoning_effort", "member"];
+const maxAiBody = 10 << 20, maxAiOutput = 128000, maxInputs = 256, aiPerMinute = 60, fakeDimensions = 8;
+// A fake count of tokens: one per 4 characters.
+const tokensOf = (text) => Math.ceil(text.length / 4);
+// A vector of a text, the same for the same text: unit length.
+function vectorOf(text, dimensions) {
+    const values = [];
+    for (let block = 0; values.length < dimensions; block++) {
+        for (const b of createHash("sha256").update(block + ":" + text).digest())
+            values.push(b / 127.5 - 1);
+    }
+    const kept = values.slice(0, dimensions), norm = Math.hypot(...kept) || 1;
+    return kept.map(x => x / norm);
+}
 const fold = (s) => s.normalize("NFD").replace(/\p{Mn}/gu, "").toLowerCase();
 function send(response, status, value, headers = {}) {
     if (value === undefined)
@@ -77,10 +98,11 @@ async function body(request, limit) {
 }
 // fakeChest starts a Chest's API on 127.0.0.1 and points the environment at
 // it: CHEST_API, CHEST_TOKEN (a new one) and CHEST_TOOL ("tool" unless the
-// environment names one). What member() and the modules of the SDK read is
-// then this Chest's.
+// environment names one), and what the Chest is (CHEST_ORGANIZATION,
+// CHEST_TIME_ZONE, CHEST_LANGUAGE, as options.chest says). What member(),
+// chest and the modules of the SDK read is then this Chest's.
 export async function fakeChest(options = {}) {
-    const capabilities = new Set(options.capabilities ?? ["members", "files", "notifications"]);
+    const capabilities = new Set(options.capabilities ?? ["members", "files", "notifications", "ai"]);
     const email = capabilities.has("members.email");
     const tool = process.env["CHEST_TOOL"] || "tool";
     const token = randomBytes(32).toString("base64url");
@@ -91,12 +113,12 @@ export async function fakeChest(options = {}) {
     const receives = options.receives ?? ["member.*"];
     // The erasures the tool was told of, by emit: those it may acknowledge.
     const erasures = new Set();
-    const chest = { api: "", token, tool, members: [...(options.members ?? [])], groups: [...(options.groups ?? [])], files, notifications: [], badges: new Map(), acknowledged: [], emit: async () => 0, close: async () => { } };
+    const chest = { api: "", token, tool, members: [...(options.members ?? [])], groups: [...(options.groups ?? [])], files, notifications: [], badges: new Map(), acknowledged: [], ai: [], emit: async () => 0, run: async () => 0, close: async () => { } };
     const former = [...(options.former ?? [])];
     let window = 0, calls = 0;
-    const shown = (m) => ({ id: m.id, first_name: m.firstName, last_name: m.lastName, name: m.name, photo: m.photo, role: m.role, admin: m.isAdmin, builder: m.isBuilder, groups: m.groups, ...(email && m.email !== undefined ? { email: m.email } : {}) });
+    const shown = (m) => ({ id: m.id, first_name: m.firstName, last_name: m.lastName, name: m.name, photo: m.photo, role: m.role, admin: m.isAdmin, builder: m.isBuilder, groups: m.groups, language: m.language, time_zone: m.timeZone, ...(email && m.email !== undefined ? { email: m.email } : {}) });
     const key = (m) => fold(m.name) + "\u0000" + m.id;
-    const described = (name, f) => ({ name, type: f.type, size: f.data.byteLength, updated: f.updated });
+    const described = (name, f) => ({ name, type: f.type, size: f.data.byteLength, sha256: createHash("sha256").update(f.data).digest("hex"), updated: f.updated });
     async function members(request, response, url) {
         if (!capabilities.has("members"))
             return send(response, 403, { error: "capability_not_granted" });
@@ -139,8 +161,10 @@ export async function fakeChest(options = {}) {
                 const m = chest.members.find(x => x.id === id), f = former.find(x => x.id === id);
                 if (m)
                     answer.members.push(shown(m));
-                else if (f?.erased)
+                else if (f?.status === "erased")
                     answer.former.push({ id, status: "erased" });
+                else if (f?.status === "no_access")
+                    answer.former.push({ id, name: f.name ?? "Test member", status: "no_access" });
                 else if (f)
                     answer.former.push({ id, ...(f.name ? { name: f.name } : {}), status: "former" });
                 else
@@ -167,22 +191,7 @@ export async function fakeChest(options = {}) {
             const names = [...files.keys()].filter(n => n.startsWith(prefix) && n > after).sort();
             return send(response, 200, { files: names.slice(0, 1000).map(n => described(n, files.get(n))), next: names.length > 1000 ? names[999] : null });
         }
-        if (request.method === "POST" && url.pathname === "/files/url") {
-            const raw = await body(request, 4096);
-            let name;
-            try {
-                name = JSON.parse(raw?.toString() ?? "").name;
-            }
-            catch {
-                name = undefined;
-            }
-            if (typeof name !== "string")
-                return send(response, 400, { error: "invalid_body" });
-            if (!files.has(name))
-                return send(response, 404, { error: "not_found" });
-            return send(response, 200, { url: `https://${tool}-chest.chest.test/_chest/files/${Buffer.from(name).toString("base64url")}.fake`, expires_in: 900 });
-        }
-        if (request.method === "POST" && (url.pathname === "/files/move" || url.pathname === "/files/upload-url")) {
+        if (request.method === "POST" && (url.pathname === "/files/url" || url.pathname === "/files/move" || url.pathname === "/files/upload-url")) {
             const raw = await body(request, 4096);
             let command = {};
             try {
@@ -191,12 +200,33 @@ export async function fakeChest(options = {}) {
             catch {
                 command = {};
             }
+            if (url.pathname === "/files/url") {
+                const { name, thumbnail, download } = command;
+                if (typeof name !== "string" || !(thumbnail === undefined || thumbnail === 256 || thumbnail === 1024) || !(download === undefined || typeof download === "boolean"))
+                    return send(response, 400, { error: "invalid_body" });
+                const object = files.get(name);
+                if (!object)
+                    return send(response, 404, { error: "not_found" });
+                if (thumbnail !== undefined && !thumbnailed.includes(object.type))
+                    return send(response, 400, { error: "no_thumbnail" });
+                const token = newToken();
+                links.set(token, { name, object, until: Date.now() + linkLife * 1000, download: download === true });
+                return send(response, 200, { url: `${chest.api}${linkPath}${token}`, expires_in: linkLife });
+            }
             if (url.pathname === "/files/upload-url") {
-                // The upload itself goes from a member's browser to the team host,
-                // which a fake Chest does not play: it only authorises it.
-                if (typeof command["name"] !== "string")
+                const { name, max_size: maxSize, types, expires_in: expiresIn } = command;
+                const folder = typeof name === "string" && name.endsWith("/");
+                if (typeof name !== "string" || !namePattern.test(folder ? name.slice(0, -1) : name))
+                    return send(response, 400, { error: "invalid_name" });
+                if (!(types === undefined || (Array.isArray(types) && types.length <= 8 && types.every(t => typeof t === "string" && mediaPattern.test(t)))))
+                    return send(response, 400, { error: "invalid_type" });
+                if (!(maxSize === undefined || (typeof maxSize === "number" && Number.isSafeInteger(maxSize) && maxSize > 0)) || !(expiresIn === undefined || (typeof expiresIn === "number" && Number.isInteger(expiresIn) && expiresIn > 0 && expiresIn <= uploadLife)))
                     return send(response, 400, { error: "invalid_body" });
-                return send(response, 200, { url: `https://${tool}-chest.chest.test/_chest/files/upload/${Buffer.from(command["name"]).toString("base64url")}.fake`, method: "PUT", expires_in: typeof command["expires_in"] === "number" ? command["expires_in"] : 900 });
+                if ((maxSize ?? 0) > maxObject)
+                    return send(response, 413, { error: "too_large" });
+                const token = newToken(), life = expiresIn ?? uploadLife;
+                uploads.set(token, { name, maxSize: maxSize ?? maxObject, types: types ?? [], until: Date.now() + life * 1000 });
+                return send(response, 200, { url: `${chest.api}${uploadPath}${token}`, method: "PUT", expires_in: life });
             }
             const from = command["from"], to = command["to"];
             if (typeof from !== "string" || typeof to !== "string" || !namePattern.test(from) || !namePattern.test(to))
@@ -235,6 +265,46 @@ export async function fakeChest(options = {}) {
         }
         send(response, 404, { error: "not_found" });
     }
+    // The team host's part of the files, on the fake's own origin: a link it
+    // signed opens the content it was signed for, for its life, as long as
+    // the file is that content — a thumbnail is the image itself, a fake does
+    // not reduce it —; an upload token takes one file, once, within its life,
+    // of the types it names and up to its size, whose first bytes are those of
+    // its type, named by the Chest in a folder. Nobody's session is checked:
+    // the Chest's own front does that.
+    const links = new Map();
+    const uploads = new Map();
+    async function front(request, response, url) {
+        if (request.method === "GET" && url.pathname.startsWith(linkPath) && !url.pathname.startsWith(uploadPath)) {
+            const link = links.get(url.pathname.slice(linkPath.length));
+            if (!link || link.until < Date.now() || files.get(link.name) !== link.object)
+                return send(response, 404, { error: "not_found" });
+            const headers = { "Content-Type": link.object.type, "Content-Length": String(link.object.data.byteLength), "X-Content-Type-Options": "nosniff" };
+            if (link.download)
+                headers["Content-Disposition"] = `attachment; filename="${link.name.split("/").pop()}"`;
+            return void response.writeHead(200, headers).end(link.object.data);
+        }
+        if (request.method !== "PUT" || !url.pathname.startsWith(uploadPath))
+            return send(response, 404, { error: "not_found" });
+        const token = url.pathname.slice(uploadPath.length), grant = uploads.get(token);
+        uploads.delete(token);
+        if (!grant || grant.until < Date.now())
+            return send(response, 403, { error: "invalid_token" });
+        const type = (request.headers["content-type"] ?? "").split(";")[0].trim().toLowerCase();
+        if (!mediaPattern.test(type) || type.endsWith("/*") || (grant.types.length > 0 && !grant.types.some(t => t === type || (t.endsWith("/*") && type.startsWith(t.slice(0, -1))))))
+            return send(response, 415, { error: "type_refused" });
+        const data = await body(request, grant.maxSize);
+        if (data === null)
+            return send(response, 413, { error: "too_large" });
+        if (signatures[type] && !signatures[type](data))
+            return send(response, 400, { error: "type_mismatch" });
+        const name = grant.name.endsWith("/") ? grant.name + randomBytes(10).toString("hex") + (extensions[type] ?? "") : grant.name;
+        const total = [...files.entries()].reduce((sum, [n, f]) => n === name ? sum : sum + f.data.byteLength, 0);
+        if (total + data.length > maxTotal || (!files.has(name) && files.size >= maxObjects))
+            return send(response, 429, { error: "quota_exceeded" });
+        files.set(name, { data: new Uint8Array(data), type, updated: new Date().toISOString() });
+        send(response, 201, { name, type, size: data.length });
+    }
     const hour = { start: 0, count: 0 }, minute = { start: 0, count: 0 }, days = new Map();
     const live = (w, span, now) => w !== undefined && w.count > 0 && now - w.start < span;
     const wait = (w, span, now) => ({ "Retry-After": String(Math.max(1, Math.ceil((w.start + span - now) / 1000))) });
@@ -359,6 +429,119 @@ export async function fakeChest(options = {}) {
         }
         send(response, 200, { delivered: kept, skipped: ids.filter(id => !access(id)) });
     }
+    // Its AI: the declared aliases, the spending this month, the requests this
+    // minute.
+    const mapped = (options.ai?.models ?? aliasOrder.map((alias) => ({ alias, model: "fake-" + alias }))).map(m => ({ alias: m.alias, model: m.model, provider: m.provider ?? "openrouter", input: m.input ?? 1, output: m.output ?? 2 }));
+    const cap = options.ai?.cap ?? 5;
+    const aiMinute = { start: 0, count: 0 };
+    let spent = 0;
+    const month = () => new Date().toISOString().slice(0, 7);
+    const resets = () => {
+        const now = new Date();
+        return new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth() + 1, 1)).toISOString().replace(".000Z", "Z");
+    };
+    const spend = (m, input, output) => {
+        const cost = Math.round((input * m.input + output * m.output) / 1e6 * 1e6) / 1e6;
+        spent = Math.round((spent + cost) * 1e6) / 1e6;
+        return cost;
+    };
+    // echo is the text of the last user message: its text, or its text parts.
+    const echo = (request) => {
+        const last = [...request["messages"]].reverse().find(m => m?.role === "user")?.content;
+        return typeof last === "string" ? last : Array.isArray(last) ? last.map(p => p?.text).filter(t => typeof t === "string").join("") : "";
+    };
+    async function aiRoute(request, response, url) {
+        const raw = request.method === "POST" ? await body(request, maxAiBody) : null;
+        let command = null;
+        try {
+            const value = JSON.parse(raw?.toString() ?? "");
+            if (value !== null && typeof value === "object" && !Array.isArray(value))
+                command = value;
+        }
+        catch {
+            command = null;
+        }
+        chest.ai.push({ path: url.pathname, body: command });
+        if (!capabilities.has("ai"))
+            return send(response, 403, { error: "capability_not_granted" });
+        if (request.method === "GET" && url.pathname === "/ai/models")
+            return send(response, 200, { models: [...mapped].sort((a, b) => aliasOrder.indexOf(a.alias) - aliasOrder.indexOf(b.alias)) });
+        if (request.method === "GET" && url.pathname === "/ai/usage")
+            return send(response, 200, { month: month(), spent, cap, resets: resets() });
+        if (request.method !== "POST" || (url.pathname !== "/ai/chat" && url.pathname !== "/ai/embeddings"))
+            return send(response, 404, { error: "not_found" });
+        if (raw === null)
+            return send(response, 413, { error: "too_large" });
+        const chat = url.pathname === "/ai/chat";
+        if (!command || !Object.keys(command).every(k => chat ? chatKeys.includes(k) : ["model", "input", "dimensions", "member"].includes(k)))
+            return send(response, 400, { error: "invalid_body" });
+        const model = mapped.find(m => m.alias === command["model"]);
+        if (!model)
+            return send(response, 403, { error: "model_not_allowed" });
+        const member = command["member"];
+        if (member !== undefined && (typeof member !== "string" || !memberIdPattern.test(member)))
+            return send(response, 400, { error: "invalid_body" });
+        const now = Date.now();
+        // refused says the refusal of a valid request, if any: the rate, the
+        // provider, the cap.
+        const refused = () => {
+            const busy = live(aiMinute, 60_000, now) && aiMinute.count >= aiPerMinute;
+            if (busy)
+                send(response, 429, { error: "rate_limited" }, wait(aiMinute, 60_000, now));
+            else {
+                count(aiMinute, 60_000, now, 1);
+                if (options.ai?.unavailable)
+                    send(response, options.ai.unavailable === "provider_key_invalid" ? 502 : 503, { error: options.ai.unavailable });
+                else if (spent >= cap)
+                    send(response, 402, { error: "cap_reached", scope: "tool", resets: resets() });
+            }
+            return response.headersSent;
+        };
+        if (!chat) {
+            const input = command["input"], texts = typeof input === "string" ? [input] : input, dimensions = command["dimensions"] ?? fakeDimensions;
+            if (!Array.isArray(texts) || texts.length < 1 || texts.length > maxInputs || !texts.every(t => typeof t === "string") || typeof dimensions !== "number" || !Number.isInteger(dimensions) || dimensions < 1 || dimensions > 4096)
+                return send(response, 400, { error: "invalid_body" });
+            if (refused())
+                return;
+            const used = texts.reduce((sum, t) => sum + tokensOf(t), 0);
+            const cost = spend(model, used, 0);
+            return send(response, 200, { object: "list", data: texts.map((t, index) => ({ object: "embedding", index, embedding: vectorOf(t, dimensions) })), model: model.model, usage: { prompt_tokens: used, total_tokens: used, cost } });
+        }
+        const messages = command["messages"], limit = command["max_tokens"] ?? 4096;
+        if (!Array.isArray(messages) || messages.length < 1 || !messages.every(m => m !== null && typeof m === "object" && typeof m.role === "string") || typeof limit !== "number" || !Number.isInteger(limit) || limit < 1 || limit > maxAiOutput || !(command["stream"] === undefined || typeof command["stream"] === "boolean"))
+            return send(response, 400, { error: "invalid_body" });
+        if (refused())
+            return;
+        const given = options.ai?.reply ? options.ai.reply(command) : echo(command);
+        const text = typeof given === "string" ? given : given.text ?? "";
+        const calls = (typeof given === "string" ? [] : given.toolCalls ?? []).map((c, i) => ({ id: c.id ?? `call_${i + 1}`, type: "function", function: { name: c.name, arguments: c.arguments } }));
+        const input = tokensOf(JSON.stringify(messages)), output = tokensOf(text + calls.map(c => c.function.name + c.function.arguments).join(""));
+        const usage = { prompt_tokens: input, completion_tokens: output, total_tokens: input + output, prompt_tokens_details: { cached_tokens: 0 }, cost: spend(model, input, output) };
+        const finish = calls.length ? "tool_calls" : "stop";
+        const head = { id: "chatcmpl-fake", created: Math.floor(now / 1000), model: model.model };
+        if (command["stream"] !== true) {
+            return send(response, 200, { ...head, object: "chat.completion", choices: [{ index: 0, message: { role: "assistant", content: calls.length && !text ? null : text, ...(calls.length ? { tool_calls: calls } : {}) }, finish_reason: finish }], usage });
+        }
+        // Streamed: the role, the text word by word, each call's name then its
+        // arguments in two halves, the finish reason, the usage, [DONE].
+        const chunk = (choices, extra = {}) => `data: ${JSON.stringify({ ...head, object: "chat.completion.chunk", choices, ...extra })}\n\n`;
+        const delta = (d, finishReason = null) => chunk([{ index: 0, delta: d, finish_reason: finishReason }]);
+        const parts = [delta({ role: "assistant", content: "" })];
+        for (const word of text.match(/\S+\s*|\s+/gu) ?? [])
+            parts.push(delta({ content: word }));
+        calls.forEach((c, index) => {
+            const half = Math.ceil(c.function.arguments.length / 2);
+            parts.push(delta({ tool_calls: [{ index, id: c.id, type: "function", function: { name: c.function.name, arguments: "" } }] }));
+            for (const piece of [c.function.arguments.slice(0, half), c.function.arguments.slice(half)])
+                if (piece)
+                    parts.push(delta({ tool_calls: [{ index, function: { arguments: piece } }] }));
+        });
+        parts.push(delta({}, finish), chunk([], { usage }), "data: [DONE]\n\n");
+        response.writeHead(200, { "Content-Type": "text/event-stream", "Cache-Control": "no-cache" });
+        for (const part of parts)
+            response.write(part);
+        response.end();
+    }
     // The acknowledgment of an erasure the tool was told of (emit).
     async function erasuresRoute(request, response, url) {
         if (!receives.includes("member.*"))
@@ -377,10 +560,12 @@ export async function fakeChest(options = {}) {
     }
     const server = createServer((request, response) => {
         const url = new URL(request.url ?? "/", "http://127.0.0.1");
-        const route = url.pathname.startsWith("/erasures/") ? erasuresRoute
-            : url.pathname === "/members" || url.pathname.startsWith("/members/") || url.pathname === "/groups" ? members
-                : url.pathname === "/files" || url.pathname.startsWith("/files/") ? filesRoute
-                    : url.pathname === "/badges" || url.pathname.startsWith("/badges/") || url.pathname.startsWith("/notifications") ? notificationsRoute : null;
+        const route = url.pathname.startsWith("/_chest/files/") ? front
+            : url.pathname.startsWith("/erasures/") ? erasuresRoute
+                : url.pathname.startsWith("/ai/") ? aiRoute
+                    : url.pathname === "/members" || url.pathname.startsWith("/members/") || url.pathname === "/groups" ? members
+                        : url.pathname === "/files" || url.pathname.startsWith("/files/") ? filesRoute
+                            : url.pathname === "/badges" || url.pathname.startsWith("/badges/") || url.pathname.startsWith("/notifications") ? notificationsRoute : null;
         if (!route)
             return send(response, 404, { error: "not_found" });
         route(request, response, url).catch(() => { if (!response.headersSent)
@@ -389,21 +574,41 @@ export async function fakeChest(options = {}) {
             response.destroy(); });
     });
     await new Promise(resolve => server.listen(0, "127.0.0.1", resolve));
-    const saved = Object.fromEntries(["CHEST_API", "CHEST_TOKEN", "CHEST_TOOL"].map(name => [name, process.env[name]]));
+    const saved = Object.fromEntries(["CHEST_API", "CHEST_TOKEN", "CHEST_TOOL", "CHEST_ORGANIZATION", "CHEST_TIME_ZONE", "CHEST_LANGUAGE", "CHEST_CURRENCY", "CHEST_TEAM_URL", "CHEST_PUBLIC_URL"].map(name => [name, process.env[name]]));
     chest.api = "http://127.0.0.1:" + server.address().port;
-    Object.assign(process.env, { CHEST_API: chest.api, CHEST_TOKEN: token, CHEST_TOOL: tool });
+    // Its links are taken by the files module of this process, for as long as
+    // it runs (api.ts, fakeOrigins).
+    fakeOrigins.add(chest.api);
+    const publicUrl = options.chest?.publicUrl === undefined ? `https://${tool}.chest.test` : options.chest.publicUrl;
+    Object.assign(process.env, { CHEST_API: chest.api, CHEST_TOKEN: token, CHEST_TOOL: tool, CHEST_ORGANIZATION: options.chest?.organization ?? "Test organization", CHEST_TIME_ZONE: options.chest?.timeZone ?? "UTC", CHEST_LANGUAGE: options.chest?.language ?? "en", CHEST_CURRENCY: options.chest?.currency ?? "EUR", CHEST_TEAM_URL: options.chest?.teamUrl ?? `https://${tool}-chest.chest.test` });
+    if (publicUrl === null)
+        delete process.env["CHEST_PUBLIC_URL"];
+    else
+        process.env["CHEST_PUBLIC_URL"] = publicUrl;
     forget();
+    // post delivers what the Chest posts to the tool on a channel — at its
+    // address, or to a handler — and says the status it answered.
+    const post = async (path, channel, id, body, to) => {
+        const request = new Request((typeof to === "string" ? to.replace(/\/$/u, "") : "http://tool.test") + path, { method: "POST", headers: { "Content-Type": "application/json", [channel.header]: sign(channel, id, body, { token, tool }) }, body });
+        const answer = typeof to === "string" ? await fetch(request, { redirect: "manual" }) : await to(request);
+        await answer.body?.cancel();
+        return answer.status;
+    };
+    const newId = (prefix) => prefix + Array.from(randomBytes(26), b => "abcdefghijklmnopqrstuvwxyz234567"[b & 31]).join("");
     chest.emit = async (event, to) => {
-        const id = event.id ?? "evt_" + Array.from(randomBytes(26), b => "abcdefghijklmnopqrstuvwxyz234567"[b & 31]).join("");
+        const id = event.id ?? newId("evt_");
         const body = JSON.stringify({ id, type: event.type, occurredAt: event.occurredAt ?? new Date().toISOString(), data: event.data });
         if (event.type === "member.erased")
             erasures.add(event.data.erasure);
-        const request = new Request(typeof to === "string" ? to.replace(/\/$/u, "") + "/chest-events" : "http://tool.test/chest-events", { method: "POST", headers: { "Content-Type": "application/json", "Chest-Event": signEvent(id, body, { token, tool }) }, body });
-        const answer = typeof to === "string" ? await fetch(request, { redirect: "manual" }) : await to(request);
-        await answer.body?.cancel();
-        return answer.status;
+        return post("/chest-events", eventChannel, id, body, to);
+    };
+    chest.run = async (name, to, run = {}) => {
+        const id = run.id ?? newId("run_");
+        const body = JSON.stringify({ id, name, scheduledAt: run.scheduledAt ?? new Date().toISOString(), attempt: run.attempt ?? 1 });
+        return post("/chest-schedules", scheduleChannel, id, body, to);
     };
     chest.close = async () => {
+        fakeOrigins.delete(chest.api);
         server.closeAllConnections();
         await new Promise(resolve => server.close(() => resolve()));
         for (const [name, value] of Object.entries(saved)) {