okengine 0.12.0 → 0.14.1

package/site/content/docs/reference/environment-variables.mdx

@@ -33,11 +33,11 @@ OKE reads environment variables at boot for connection detail and secrets — ne
 
 ## KV store
 
-| Variable           | Used for                 | Default when unset |
-| ------------------ | ------------------------ | ------------------ |
-| `REDIS_URL`        | Redis connection         | driver default     |
-| `OKE_STORE_KV_URL` | Explicit KV URL override | —                  |
-| `OKE_KV_DRIVER`    | Force the kv driver id   | config map         |
+| Variable           | Used for                       | Default when unset |
+| ------------------ | ------------------------------ | ------------------ |
+| `REDIS_URL`        | Cache Redis connection         | driver default     |
+| `OKE_STORE_KV_URL` | Explicit cache KV URL override | —                  |
+| `OKE_KV_DRIVER`    | Force the kv driver id         | config map         |
 
 ## Files store
 

package/site/content/docs/reference/fx.mdx

@@ -33,9 +33,10 @@ See [Store](/docs/elements/store) for the query-builder surface.
 
 ## Signals
 
-| Signature                             | Records | Notes                                                                                                                                                                                                     |
-| ------------------------------------- | ------- | --------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
-| `fx.emit(signal, payload?, { key? })` | `emit`  | Commits the signal outbox when the call resolves; optional `key` serializes `once` per key; stamps producer run id as `parentRunId` for trace chains; throws **OKE1042** (orphan) or **OKE1043** (schema) |
+| Signature                             | Records                | Notes                                                                                                                                                                                                     |
+| ------------------------------------- | ---------------------- | --------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| `fx.emit(signal, payload?, { key? })` | `emit`                 | Commits the signal outbox when the call resolves; optional `key` serializes `once` per key; stamps producer run id as `parentRunId` for trace chains; throws **OKE1042** (orphan) or **OKE1043** (schema) |
+| `fx.deadLetters(signal)`              | `read` `signal:<name>` | Dead-lettered messages for that signal. Payload typed from `SignalDecl<T>`. Page with `fx.json.withQuery`. Cross-signal throws **OKE1001**.                                                               |
 
 ## Runs (observability read)
 
@@ -145,12 +146,12 @@ contact a provider. Channel bodies use `{{field}}` catalogs — not ICU (see [Ch
 
 ## AI
 
-| Signature                                             | Records                   | Returns                                                                       |
-| ----------------------------------------------------- | ------------------------- | ----------------------------------------------------------------------------- |
-| `fx.ask(prompt, input?, { via?, tools?, maxSteps? })` | `ask` (+ `call` per tool) | Object validated against the prompt's `out`                                   |
-| `fx.run(agent, input?)`                               | `ask`                     | Agent result                                                                  |
-| `fx.stream(model, { prompt?, data? })`                | `ask`                     | `AsyncIterable<string>` — real driver stream; cancels via ambient `fx.signal` |
-| `fx.search(embed, query, { topK? })`                  | `read`                    | Matches from the index/embed                                                  |
+| Signature                                             | Records                   | Returns                                                                                                  |
+| ----------------------------------------------------- | ------------------------- | -------------------------------------------------------------------------------------------------------- |
+| `fx.ask(prompt, input?, { via?, tools?, maxSteps? })` | `ask` (+ `call` per tool) | Object validated against the prompt's `out`                                                              |
+| `fx.run(agent, input?)`                               | `ask`                     | Agent result                                                                                             |
+| `fx.stream(model, { prompt?, data?, via? })`          | `ask`                     | `AsyncIterable<string>` — real driver stream; cancels via ambient `fx.signal` (HTTP disconnect included) |
+| `fx.search(embed, query, { topK? })`                  | `read`                    | Matches from the index/embed                                                                             |
 
 AI calls are nondeterministic: journaling is forced on and auto-cache disabled around them. `tools` are Flow refs — each model tool call goes through `fx.call` (same capability and Runs path).
 
@@ -169,12 +170,15 @@ AI calls are nondeterministic: journaling is forced on and auto-cache disabled a
 
 ## Clock
 
-| Signature                         | Notes                                                 |
-| --------------------------------- | ----------------------------------------------------- |
-| `fx.clock.now()`                  | Epoch-ms, injectable — the only legal "now"           |
-| `fx.clock.sleep(label, duration)` | Durable sleep in `durable` flows; immediate otherwise |
+| Signature                         | Notes                                                               |
+| --------------------------------- | ------------------------------------------------------------------- |
+| `fx.clock.now()`                  | Epoch-ms, injectable — the only legal "now"                         |
+| `fx.clock.ago(duration)`          | Instant before now (`"30d"` → now − 30 days)                        |
+| `fx.clock.fromNow(duration)`      | Instant after now (`"14d"` → now + 14 days)                         |
+| `fx.clock.duration(duration)`     | Span in ms — offset a stored instant (`createdAt + duration("7d")`) |
+| `fx.clock.sleep(label, duration)` | Durable sleep in `durable` flows; immediate otherwise               |
 
-Durations: `"200ms"` · `"30s"` · `"2m"` · `"1h"` · `"7d"`.
+Durations: `"200ms"` · `"30s"` · `"2m"` · `"1h"` · `"7d"`. A `"d"` is 86_400_000 ms, not a calendar day. Unknown strings parse as `0`.
 
 ## Cache
 
@@ -192,15 +196,16 @@ keys. Use `cache: false` to opt out, or `cache: "30s"` for a TTL.
 
 ## Responses
 
-| Helper                                    | Status | Body                                                        |
-| ----------------------------------------- | ------ | ----------------------------------------------------------- |
-| `fx.json.ok(value, { meta? })`            | 200    | `{ data, meta?, error: null }`                              |
-| `fx.json.create(value)`                   | 201    | `{ data, error: null }`                                     |
-| `fx.json.empty()`                         | 204    | no body                                                     |
-| `fx.json.with(page)` / `with(data, meta)` | 200    | `{ data, meta, error: null }` — already-built pager         |
-| `fx.json.withQuery(rows, input, spec?)`   | 200    | In-memory list page — zero-config `q` / auto-eq / PostgREST |
+| Helper                                    | Status | Body                                                           |
+| ----------------------------------------- | ------ | -------------------------------------------------------------- |
+| `fx.json.ok(value, { meta? })`            | 200    | `{ data, meta?, error: null }`                                 |
+| `fx.json.create(value)`                   | 201    | `{ data, error: null }`                                        |
+| `fx.json.empty()`                         | 204    | no body                                                        |
+| `fx.json.with(page)` / `with(data, meta)` | 200    | `{ data, meta, error: null }` — already-built pager            |
+| `fx.json.withQuery(rows, input, spec?)`   | 200    | In-memory list page — zero-config `q` / auto-eq / PostgREST    |
+| `fx.json.stream(chunks)`                  | 200    | `text/event-stream` — JSON `data:` frames, then `data: [DONE]` |
 
-Returning a plain value instead answers 200 with `{ data: value, error: null }` — the helpers exist for status and `meta` control.
+Returning a plain value instead answers 200 with `{ data: value, error: null }` — the helpers exist for status and `meta` control. Pass `fx.stream(...)` into `fx.json.stream` to reach the HTTP client token-by-token.
 
 ## Logging, i18n, ids
 

package/src/cli/ai-setup/recommend.test.ts

@@ -197,4 +197,29 @@ describe("formatModelRow", () => {
     expect(row).toContain("text · code");
     expect(row).not.toContain("  ·  ≈");
   });
+
+  test("Arabic and CJK labels keep Caps aligned", () => {
+    const header = formatModelTableHeader();
+    const capsWidth = Bun.stringWidth(header.slice(0, header.indexOf("Caps")));
+    const ar = formatModelRow({
+      id: "ar",
+      label: "مرحبا بالعالم",
+      hint: "",
+      role: "chat",
+      ramGb: 8,
+      tier: "fast",
+      modalities: ["text"],
+    });
+    const cjk = formatModelRow({
+      id: "cjk",
+      label: "你好世界模型",
+      hint: "",
+      role: "chat",
+      ramGb: 8,
+      tier: "fast",
+      modalities: ["text"],
+    });
+    expect(Bun.stringWidth(ar.slice(0, ar.indexOf("text")))).toBe(capsWidth);
+    expect(Bun.stringWidth(cjk.slice(0, cjk.indexOf("text")))).toBe(capsWidth);
+  });
 });

package/src/cli/ai-setup/recommend.ts

@@ -195,9 +195,14 @@ const MODEL_COL_RAM = 7;
  * @param width - Fixed width
  */
 function clipPad(value: string, width: number): string {
-  if (value.length === width) return value;
-  if (value.length < width) return value.padEnd(width);
-  return `${value.slice(0, Math.max(0, width - 1))}…`;
+  const bunAnsi = Bun as typeof Bun & {
+    stringWidth(text: string): number;
+    sliceAnsi(text: string, start: number, end: number, omission?: string): string;
+  };
+  const display = bunAnsi.stringWidth(value);
+  if (display === width) return value;
+  if (display < width) return value + " ".repeat(width - display);
+  return bunAnsi.sliceAnsi(value, 0, width, "…");
 }
 
 /**

package/src/cli/load-config.images.test.ts

@@ -132,6 +132,7 @@ describe("resolveImages", () => {
     });
     expect(images["store.sql"]).toBe("postgres:18-alpine");
     expect(images["store.kv"]).toBe("redis:8-alpine");
+    expect(images["store.kv.durable"]).toBeUndefined();
   });
 });
 

package/src/compiler/effects-infer.ts

@@ -12,6 +12,7 @@ import type {
   ResourceRef,
   SecretRef,
   SignalRef,
+  SignalResourceRef,
   TemplateRef,
   FlowRef,
 } from "../manifest/types.ts";
@@ -56,6 +57,8 @@ export interface InferBinding {
     | "flow"
     | "embed"
     | "table"
+    | "mcp-server"
+    | "mcp-tool"
     | "unknown";
   /** Resolved resource / name. */
   readonly ref: string;
@@ -142,7 +145,7 @@ const TABLE_ARG_METHODS = new Set([
  * @param options - Handler AST, bindings, and annotation flag
  */
 export function inferEffects(options: InferEffectsOptions): InferredEffects {
-  const reads = new Set<ResourceRef>();
+  const reads = new Set<ResourceRef | SignalResourceRef>();
   const writes = new Set<ResourceRef>();
   const emits = new Set<SignalRef>();
   const sends = new Set<TemplateRef>();
@@ -173,6 +176,12 @@ export function inferEffects(options: InferEffectsOptions): InferredEffects {
       continue;
     }
 
+    if (chain.rootMethod === "deadLetters" && call === chain.rootCall) {
+      const ref = resolveNamed(call.arguments[0], options.bindings, "signal");
+      if (ref) reads.add(`signal:${ref}`);
+      continue;
+    }
+
     if (chain.rootMethod === "send" && call === chain.rootCall) {
       const ref = resolveNamed(call.arguments[0], options.bindings, "template");
       if (ref) sends.add(ref);
@@ -203,7 +212,7 @@ export function inferEffects(options: InferEffectsOptions): InferredEffects {
     }
 
     if (chain.rootMethod === "call" && call === chain.rootCall) {
-      const ref = resolveNamed(call.arguments[0], options.bindings, "flow");
+      const ref = resolveCallTarget(call.arguments[0], options.bindings);
       if (ref) calls.add(ref);
       continue;
     }
@@ -541,12 +550,58 @@ function toolsFromAskOptions(
   );
   const out: FlowRef[] = [];
   for (const el of els) {
-    const ref = resolveNamed(el, bindings, "flow");
+    const ref = resolveCallTarget(el, bindings);
     if (ref) out.push(ref as FlowRef);
   }
   return out;
 }
 
+/**
+ * Resolve `fx.call` / ask-tools: flow name, `mcp:` ref, or `server.tool("x")`.
+ *
+ * @param node - Argument AST
+ * @param bindings - Known bindings
+ */
+export function resolveCallTarget(
+  node: AstNode | undefined,
+  bindings: ReadonlyMap<string, InferBinding>,
+): string | undefined {
+  const mcp = resolveMcpToolExpr(node, bindings);
+  if (mcp) return mcp;
+  return resolveNamed(node, bindings, "flow");
+}
+
+/**
+ * Resolve `server.tool("name")` or a bound `mcp-tool` identifier.
+ *
+ * @param node - AST
+ * @param bindings - Known bindings
+ */
+export function resolveMcpToolExpr(
+  node: AstNode | undefined,
+  bindings: ReadonlyMap<string, InferBinding>,
+): string | undefined {
+  if (!node) return undefined;
+  if (node.type === "CallExpression") {
+    const callee = (node as CallExpression).callee;
+    if (callee.type === "MemberExpression") {
+      const member = callee as AstNode & { object: AstNode; property: AstNode };
+      const obj = identifierName(member.object);
+      const prop = identifierName(member.property);
+      const tool = stringArg((node as CallExpression).arguments[0]);
+      if (obj && prop === "tool" && tool) {
+        const server = bindings.get(obj);
+        if (server?.kind === "mcp-server") {
+          return `mcp:${server.ref}/${tool}`;
+        }
+      }
+    }
+  }
+  const binding = resolveBinding(node, bindings);
+  if (binding?.kind === "mcp-tool") return binding.ref;
+  return undefined;
+}
+
 /**
  * String literal argument.
  *

package/src/compiler/extract.test.ts

@@ -135,12 +135,15 @@ export const flow_${i} = on(
     }
     sources["src/flows/all.ts"] = parts.join("\n");
 
+    await extractFromSources(sources);
     const start = performance.now();
     const manifest = await extractFromSources(sources);
     const elapsed = performance.now() - start;
 
     expect(Object.keys(manifest.flows ?? {}).length).toBe(200);
-    expect(elapsed).toBeLessThan(2000);
+    // Local ~400ms after warmup. GHA shared runners measured ~2.9s cold.
+    const budgetMs = process.env.CI ? 4_000 : 2_000;
+    expect(elapsed).toBeLessThan(budgetMs);
   });
 });
 
@@ -293,6 +296,7 @@ export const stripeKey = vault.secret("STRIPE_KEY", {
     expect(manifest.stores?.embeddings?.description).toBe("Document embeddings");
     expect(manifest.stores?.sessions?.description).toBe("Session cache");
     expect(manifest.stores?.sessions?.namespaces).toEqual(["sessions"]);
+    expect(manifest.stores?.sessions?.durable).toBeUndefined();
     expect(manifest.signals?.["order-placed"]?.description).toBe("Order placed event");
     expect(manifest.channels?.["booking-confirmed"]?.description).toBe(
       "Booking confirmation email",
@@ -301,6 +305,19 @@ export const stripeKey = vault.secret("STRIPE_KEY", {
     expect(manifest.gates?.member?.description).toBe("Verified members only");
     expect(manifest.vault?.STRIPE_KEY?.description).toBe("Payments gateway key");
   });
+
+  test("extracts store.kv durable onto the Manifest store", async () => {
+    const manifest = await extractFromSources({
+      "src/kv.ts": `
+import { store } from "okengine";
+export const sessions = store.kv("sessions", { description: "Session cache" });
+export const ledger = store.kv("ledger", { durable: true, description: "Idempotency keys" });
+`,
+    });
+    expect(manifest.stores?.sessions?.durable).toBeUndefined();
+    expect(manifest.stores?.ledger?.durable).toBe(true);
+    expect(manifest.stores?.ledger?.namespaces).toEqual(["ledger"]);
+  });
 });
 
 describe("extractManifest — bindNamedTableCrud(...)", () => {
@@ -584,6 +601,25 @@ export const expire = on(
   });
 });
 
+describe("extractManifest — fx.deadLetters", () => {
+  test("fx.deadLetters(signal) infers reads: [signal:<name>]", async () => {
+    const source = `
+import { on, flow, http, signal } from "okengine";
+
+export const notify = signal("notify", { delivery: "once" });
+
+export const failed = on(
+  http.get("/notifications/failed").gate.public,
+  flow("notifications.failed", {
+    do: async (input, fx) => fx.json.withQuery(await fx.deadLetters(notify), input),
+  }),
+);
+`;
+    const manifest = await extractFromSources({ "src/flows/failed.ts": source });
+    expect(manifest.flows?.["notifications.failed"]?.effects?.reads).toEqual(["signal:notify"]);
+  });
+});
+
 describe("extractManifest — channel medium binder aliasing", () => {
   test("mail.template(...) resolves through `const mail = channel.email(...)`", async () => {
     const source = `
@@ -762,3 +798,34 @@ export const create = on(
     expect(manifest.gates?.WRITE).toBeUndefined();
   });
 });
+
+describe("extractManifest — ai.mcpServer", () => {
+  test("stamps mcpServers and infers mcp: calls from .tool()", async () => {
+    const source = `
+import { ai, flow, on, http, vault } from "okengine";
+
+export const token = vault.secret("GITHUB_TOKEN");
+export const github = ai.mcpServer("github", {
+  url: "https://mcp.example/github",
+  auth: { bearer: token },
+  tools: ["create_issue"],
+});
+
+export const triage = on(
+  http.post("/triage"),
+  flow("support.triage", {
+    do: async (input, fx) => {
+      await fx.ask("ticket-triage", input, { tools: [github.tool("create_issue")] });
+    },
+  }),
+);
+`;
+    const manifest = await extractFromSources({ "src/ai.ts": source });
+    expect(manifest.ai?.mcpServers?.github).toEqual({
+      url: "https://mcp.example/github",
+      auth: "GITHUB_TOKEN",
+      tools: ["create_issue"],
+    });
+    expect(manifest.flows?.["support.triage"]?.effects?.calls).toEqual(["mcp:github/create_issue"]);
+  });
+});

package/src/compiler/extract.ts

@@ -11,6 +11,7 @@ import { parseSync } from "oxc-parser";
 import type {
   Ai,
   AiAgent,
+  AiMcpServer,
   AiModel,
   AiPrompt,
   Channel,
@@ -35,6 +36,7 @@ import { sqlTableRef } from "../manifest/sql-resource.ts";
 import {
   identifierName,
   inferEffects,
+  resolveCallTarget,
   stringArg,
   walk,
   type AstNode,
@@ -177,11 +179,12 @@ export async function extractManifest(options: ExtractManifestOptions = {}): Pro
   if (Object.keys(gates).length > 0) manifest.gates = gates;
   if (Object.keys(vault).length > 0) manifest.vault = vault;
   if (Object.keys(channels).length > 0) manifest.channels = channels;
-  if (scope.ai.models || scope.ai.prompts || scope.ai.agents) {
+  if (scope.ai.models || scope.ai.prompts || scope.ai.agents || scope.ai.mcpServers) {
     manifest.ai = {
       ...(scope.ai.models ? { models: sortRecord(scope.ai.models) } : {}),
       ...(scope.ai.prompts ? { prompts: sortRecord(scope.ai.prompts) } : {}),
       ...(scope.ai.agents ? { agents: sortRecord(scope.ai.agents) } : {}),
+      ...(scope.ai.mcpServers ? { mcpServers: sortRecord(scope.ai.mcpServers) } : {}),
     };
   }
   if (Object.keys(journeys).length > 0) manifest.journeys = journeys;
@@ -301,6 +304,13 @@ function finalizeRefs(scope: ProjectScope): void {
       );
     }
   }
+  if (scope.ai.mcpServers) {
+    for (const server of Object.values(scope.ai.mcpServers)) {
+      if (!server.auth) continue;
+      const binding = scope.bindings.get(server.auth);
+      if (binding?.kind === "secret") server.auth = binding.ref;
+    }
+  }
   for (const journey of Object.values(scope.journeys)) {
     if (!journey.flows) continue;
     journey.flows = journey.flows.map(
@@ -650,6 +660,7 @@ function visitDeclarationCall(call: CallExpression, program: AstNode, scope: Pro
           const namespaces = new Set(storeEntry.namespaces ?? []);
           namespaces.add(storeName);
           storeEntry.namespaces = [...namespaces].sort();
+          if (boolProp(storeOpts, "durable")) storeEntry.durable = true;
         } else if (facet === "files") {
           const buckets = new Set(storeEntry.buckets ?? []);
           buckets.add(storeName);
@@ -815,6 +826,7 @@ function visitDeclarationCall(call: CallExpression, program: AstNode, scope: Pro
         const model: AiModel = {
           ...(stringProp(opts, "provider") ? { provider: stringProp(opts, "provider") } : {}),
           ...(stringProp(opts, "tier") ? { tier: stringProp(opts, "tier") } : {}),
+          ...(stringProp(opts, "driverId") ? { driverId: stringProp(opts, "driverId") } : {}),
         };
         scope.ai.models = scope.ai.models ?? {};
         scope.ai.models[modelName] = model;
@@ -832,6 +844,24 @@ function visitDeclarationCall(call: CallExpression, program: AstNode, scope: Pro
       collectAgent(call, scope);
     }
 
+    if (obj === "ai" && prop === "mcpServer") {
+      collectMcpServer(call, program, scope);
+    }
+
+    if (prop === "tool" && obj) {
+      const toolName = stringArg(call.arguments[0]);
+      const server = scope.bindings.get(obj);
+      if (server?.kind === "mcp-server" && toolName) {
+        const bindingName = enclosingConstName(call, program);
+        if (bindingName) {
+          scope.bindings.set(bindingName, {
+            kind: "mcp-tool",
+            ref: `mcp:${server.ref}/${toolName}`,
+          });
+        }
+      }
+    }
+
     if (obj === "ai" && prop === "embed") {
       const embedName = stringArg(call.arguments[0]);
       if (embedName) {
@@ -1025,7 +1055,11 @@ function collectAgent(call: CallExpression, scope: ProjectScope): void {
   if (!agentName || !opts) return;
   const toolsArr = arrayProp(opts, "tools");
   const tools = toolsArr
-    ?.map((el) => identifierName(el) ?? stringArg(el))
+    ?.map((el) => {
+      const resolved = resolveCallTarget(el, scope.bindings);
+      if (resolved) return resolved;
+      return identifierName(el) ?? stringArg(el);
+    })
     .filter((x): x is string => typeof x === "string")
     .map((id) => scope.flowExports.get(id) ?? scope.bindings.get(id)?.ref ?? id);
   const agent: AiAgent = {
@@ -1038,6 +1072,40 @@ function collectAgent(call: CallExpression, scope: ProjectScope): void {
   scope.ai.agents[agentName] = agent;
 }
 
+function collectMcpServer(call: CallExpression, program: AstNode, scope: ProjectScope): void {
+  const serverName = stringArg(call.arguments[0]);
+  const opts = objectArg(call.arguments[1]);
+  if (!serverName || !opts) return;
+  const tools = stringArrayProp(opts, "tools") ?? [];
+  const url = stringProp(opts, "url");
+  const command = stringProp(opts, "command");
+  const args = stringArrayProp(opts, "args");
+  const authObj = objectProp(opts, "auth");
+  const bearerNode = authObj ? objectProp(authObj, "bearer") : undefined;
+  const bearerLit = stringArg(bearerNode);
+  const bearerId = identifierName(bearerNode);
+  const auth =
+    bearerLit ??
+    (bearerId
+      ? scope.bindings.get(bearerId)?.kind === "secret"
+        ? scope.bindings.get(bearerId)?.ref
+        : bearerId
+      : undefined);
+  const server: AiMcpServer = {
+    tools,
+    ...(url ? { url } : {}),
+    ...(command ? { command } : {}),
+    ...(args && args.length > 0 ? { args } : {}),
+    ...(auth ? { auth } : {}),
+  };
+  scope.ai.mcpServers = scope.ai.mcpServers ?? {};
+  scope.ai.mcpServers[serverName] = server;
+  const bindingName = enclosingConstName(call, program);
+  if (bindingName) {
+    scope.bindings.set(bindingName, { kind: "mcp-server", ref: serverName });
+  }
+}
+
 function collectConfig(opts: AstNode | undefined, scope: ProjectScope): void {
   if (!opts) return;
 

package/src/compiler/response.ts

@@ -5,7 +5,7 @@
  */
 
 import type { FlowFailure } from "../kernel/errors.ts";
-import { isJsonResult } from "../kernel/fx.ts";
+import { isJsonResult, isJsonStreamResult, type JsonStreamResult } from "../kernel/fx.ts";
 import { isFlowFailure } from "../kernel/hooks.ts";
 import { VALIDATION_ERROR_CODE } from "../validation/standard-schema.ts";
 
@@ -53,6 +53,9 @@ export function statusForFailure(failure: FlowFailure): number {
  * @param output - Handler output (`undefined` → 204)
  */
 export function encodeSuccess(output: unknown): Response {
+  if (isJsonStreamResult(output)) {
+    return encodeSseStream(output);
+  }
   if (isJsonResult(output)) {
     if (output.status === 204) {
       return new Response(null, { status: 204 });
@@ -110,3 +113,44 @@ export function encodeExecuteResult(result: {
   }
   return encodeSuccess(result.output);
 }
+
+/**
+ * Encode {@link JsonStreamResult} as SSE (`text/event-stream`).
+ *
+ * @param carrier - Stream carrier from `fx.json.stream`
+ */
+function encodeSseStream(carrier: JsonStreamResult): Response {
+  const encoder = new TextEncoder();
+  let finalized = false;
+  const finish = async (): Promise<void> => {
+    if (finalized) return;
+    finalized = true;
+    await carrier.finalize?.();
+  };
+  const body = new ReadableStream<Uint8Array>({
+    async start(controller) {
+      try {
+        for await (const chunk of carrier.chunks) {
+          controller.enqueue(encoder.encode(`data: ${JSON.stringify(chunk)}\n\n`));
+        }
+        controller.enqueue(encoder.encode("data: [DONE]\n\n"));
+        controller.close();
+      } catch (err) {
+        controller.error(err);
+      } finally {
+        await finish();
+      }
+    },
+    cancel() {
+      void finish();
+    },
+  });
+  return new Response(body, {
+    status: 200,
+    headers: {
+      "content-type": "text/event-stream; charset=utf-8",
+      "cache-control": "no-cache",
+      connection: "keep-alive",
+    },
+  });
+}

package/src/console/server/ai.ts

@@ -459,7 +459,7 @@ function buildVersionMetrics(input: {
         costs: [],
         latencies: [],
         evals: [],
-        outcomes: { ok: 0, provider_error: 0, schema_invalid: 0 },
+        outcomes: { ok: 0, provider_error: 0, schema_invalid: 0, budget_exceeded: 0 },
         overBudget: 0,
         budget: meta?.budget ?? null,
       };
@@ -514,7 +514,10 @@ function buildVersionMetrics(input: {
   const out: PromptVersionMetrics[] = [];
   for (const acc of map.values()) {
     const totalOutcomes =
-      acc.outcomes.ok + acc.outcomes.provider_error + acc.outcomes.schema_invalid;
+      acc.outcomes.ok +
+      acc.outcomes.provider_error +
+      acc.outcomes.schema_invalid +
+      acc.outcomes.budget_exceeded;
     out.push({
       prompt: acc.prompt,
       version: acc.version,

package/src/console/server/flows.ts

@@ -834,6 +834,7 @@ const AiListOut = z.object({
         ok: z.number(),
         provider_error: z.number(),
         schema_invalid: z.number(),
+        budget_exceeded: z.number(),
       }),
     }),
   ),

package/src/console/server/serve.ts

@@ -244,7 +244,12 @@ export async function serveConsole(
           new Response("WebSocket upgrade failed", { status: 400 }),
         );
       }
-      return fetchHandler(req);
+      return (async () => {
+        const res = await fetchHandler(req);
+        const ct = (res.headers.get("content-type") ?? "").split(";")[0]?.trim() ?? "";
+        if (/^text\/event-stream$/i.test(ct)) srv.timeout(req, 0);
+        return res;
+      })();
     },
     websocket: {
       open: live.open,
@@ -259,6 +264,15 @@ export async function serveConsole(
     boundHost.includes(":") && !boundHost.startsWith("[") ? `[${boundHost}]` : boundHost;
   const url = new URL(`http://${hostForUrl}:${boundPort}/`);
 
+  const closeIdle = (
+    server as typeof server & { closeIdleConnections(): void }
+  ).closeIdleConnections.bind(server);
+  const onPressure = (): void => {
+    closeIdle();
+  };
+  const proc = process as NodeJS.EventEmitter;
+  proc.on("memoryPressure", onPressure);
+
   return {
     console: handle,
     url,
@@ -266,9 +280,10 @@ export async function serveConsole(
     hostname: boundHost,
     fetch: fetchHandler,
     stop(closeActive = false) {
-      server.stop(closeActive);
+      proc.off("memoryPressure", onPressure);
       void handle.app.stop();
       void persistence?.close();
+      return server.stop(closeActive);
     },
   };
 }

package/src/console/ui-next/dist/assets/cache-glyph-BanhLsEY.js

@@ -0,0 +1 @@
+import{Cr as e,F as t,L as n,P as r,wr as i}from"./shortcut-keys-DO4IsVqv.js";import{A as a,f as o}from"./xyflow-D7n4g6go.js";import{t as s}from"./dagre.esm-ZwcdTuZZ.js";import{D as c,E as l,T as u,b as d,o as f,w as p,y as m}from"./http-method-CJCBYL2j.js";function h(e){let t=e.indexOf(`.`);return t===-1?e:e.slice(0,t)}function g(e){let t=e.indexOf(`.`);return t===-1?e:e.slice(t+1)}function _(e){let t=e?.flows??{},n=new Map;for(let e of Object.keys(t)){let r=t[e]?.effects?.calls??[];for(let t of r){let r=n.get(t)??[];r.push(e),n.set(t,r)}}for(let[e,t]of n)t.sort((e,t)=>e.localeCompare(t)),n.set(e,t);return n}function v(e,t){return _(e).get(t)??[]}function y(e){return e.startsWith(`sql:`)?{id:e,kind:`store`,label:e.slice(4),facet:`sql`}:e.startsWith(`kv:`)?{id:e,kind:`store`,label:e.slice(3),facet:`kv`}:e.startsWith(`files:`)?{id:e,kind:`store`,label:e.slice(6),facet:`files`}:e.startsWith(`index:`)?{id:e,kind:`store`,label:e.slice(6),facet:`index`}:e.startsWith(`signal:`)?{id:e,kind:`signal`,label:e.slice(7)}:e.startsWith(`ai:`)?{id:e,kind:`ai`,label:e.slice(3)}:e.startsWith(`vault:`)?{id:e,kind:`vault`,label:e.slice(6)}:e.startsWith(`channel:`)?{id:e,kind:`channel`,label:e.slice(8)}:e.startsWith(`gate:`)?{id:e,kind:`gate`,label:e.slice(5)}:e.startsWith(`clock:`)?{id:e,kind:`clock`,label:e.slice(6)}:null}function b(e,t,n,r,i=!1){let a=m[r];return{id:e,source:t,target:n,type:`smoothstep`,animated:i,data:{kind:r},style:{stroke:a,strokeWidth:r===`calls`?2.5:2.25},pathOptions:{borderRadius:28},zIndex:d.edge,markerEnd:{type:o.ArrowClosed,width:14,height:14,color:a}}}function x(e){let t=e?.flows??{},n=new Set(Object.keys(t)),r=[],i=new Map;for(let e of[...n].sort()){let t=h(e),n=i.get(t)??[];n.push(e),i.set(t,n)}let a=new Map,o=new Map;for(let e of n){let n=t[e];if(!n)continue;if(n.trigger?.signal){let t=`signal:${n.trigger.signal}`,i=y(t);i&&a.set(t,i),r.push(b(`e:${t}->flow:${e}`,t,`flow:${e}`,`trigger`))}if(n.trigger?.cron||n.trigger?.every){let t=`clock:${e}`,i=n.trigger.cron??n.trigger.every??e;a.set(t,{id:t,kind:`clock`,label:i,facet:n.trigger.cron?`cron`:`every`}),r.push(b(`e:${t}->flow:${e}`,t,`flow:${e}`,`trigger`))}for(let t of n.gates??[]){let n=`gate:${t}`;a.set(n,{id:n,kind:`gate`,label:t}),o.set(n,(o.get(n)??0)+1),r.push(b(`e:flow:${e}-gates->${n}`,`flow:${e}`,n,`gates`))}let i=n.effects;if(i){for(let t of i.reads??[]){let n=y(t);n&&(a.set(t,n),o.set(t,(o.get(t)??0)+1),r.push(b(`e:flow:${e}-reads->${t}`,`flow:${e}`,t,`reads`)))}for(let t of i.writes??[]){let n=y(t);n&&(a.set(t,n),o.set(t,(o.get(t)??0)+1),r.push(b(`e:flow:${e}-writes->${t}`,`flow:${e}`,t,`writes`)))}for(let t of i.emits??[]){let n=`signal:${t}`,i=y(n);i&&(a.set(n,i),o.set(n,(o.get(n)??0)+1),r.push(b(`e:flow:${e}->${n}`,`flow:${e}`,n,`emits`)))}for(let t of i.asks??[]){let n=`ai:${t}`,i=y(n);i&&(a.set(n,i),o.set(n,(o.get(n)??0)+1),r.push(b(`e:flow:${e}->${n}`,`flow:${e}`,n,`asks`)))}for(let t of i.sends??[]){let n=`channel:${t}`,i=y(n);i&&(a.set(n,i),o.set(n,(o.get(n)??0)+1),r.push(b(`e:flow:${e}->${n}`,`flow:${e}`,n,`sends`)))}for(let t of i.secrets??[]){let n=`vault:${t}`,i=y(n);i&&(a.set(n,i),o.set(n,(o.get(n)??0)+1),r.push(b(`e:flow:${e}->${n}`,`flow:${e}`,n,`secrets`)))}for(let t of i.calls??[]){let n=c(t);if(n){let t=l(n.server);a.set(t,{id:t,kind:`ai`,label:n.server}),o.set(t,(o.get(t)??0)+1),r.push(b(`e:flow:${e}->${t}`,`flow:${e}`,t,`calls`,!0));continue}r.push(b(`e:flow:${e}->flow:${t}`,`flow:${e}`,`flow:${t}`,`calls`,!0))}}}return{nodes:S(i,t,a,o),edges:r,flowIds:n}}function S(e,t,n,r){let i=new s.graphlib.Graph({compound:!0});i.setGraph({rankdir:`LR`,align:`UL`,nodesep:16,ranksep:88,edgesep:12,marginx:8,marginy:8}),i.setDefaultEdgeLabel(()=>({}));for(let[t,n]of[...e.entries()].sort()){i.setNode(`unit:${t}`,{});for(let e of n)i.setNode(`flow:${e}`,{...p.flow}),i.setParent(`flow:${e}`,`unit:${t}`)}for(let e of n.values())i.setNode(e.id,{...p[e.kind]});let a=new Set;for(let n of e.values())for(let e of n){let n=t[e];if(!n)continue;let r=(e,t)=>{let n=`${e}->${t}`;a.has(n)||!i.hasNode(e)||!i.hasNode(t)||(a.add(n),i.setEdge(e,t))};n.trigger?.signal&&r(`signal:${n.trigger.signal}`,`flow:${e}`),(n.trigger?.cron||n.trigger?.every)&&r(`clock:${e}`,`flow:${e}`);for(let t of n.gates??[])r(`flow:${e}`,`gate:${t}`);let o=n.effects;if(o){for(let t of o.reads??[])r(`flow:${e}`,t);for(let t of o.writes??[])r(`flow:${e}`,t);for(let t of o.emits??[])r(`flow:${e}`,`signal:${t}`);for(let t of o.asks??[])r(`flow:${e}`,`ai:${t}`);for(let t of o.sends??[])r(`flow:${e}`,`channel:${t}`);for(let t of o.secrets??[])r(`flow:${e}`,`vault:${t}`);for(let t of o.calls??[]){let n=c(t);n?r(`flow:${e}`,l(n.server)):r(`flow:${e}`,`flow:${t}`)}}}s.layout(i);let o=[],{headerH:f,padX:m,padBottom:h}=u;for(let[n,r]of[...e.entries()].sort()){let e=`unit:${n}`,a=i.node(e);if(!a)continue;let s=a.x??0,c=a.y??0,l=r.length*p.flow.height+Math.max(0,r.length-1)*10,u=p.flow.width+m*2,_=f+l+h,v=s-u/2,y=c-_/2;o.push({id:e,type:`unit`,position:{x:v,y},data:{kind:`unit`,label:n,refId:e,badge:String(r.length)},selectable:!1,draggable:!1,zIndex:d.unit,width:u,height:_,style:{width:u,height:_}}),r.forEach((r,i)=>{let a=t[r];a&&o.push({id:`flow:${r}`,type:`flow`,position:{x:m,y:f+i*(p.flow.height+10)},parentId:e,extent:`parent`,zIndex:d.leaf,data:{kind:`flow`,label:g(r),refId:r,unit:n,plane:a.plane??`user`,badge:a.plane??`user`},draggable:!1,width:p.flow.width,height:p.flow.height,style:{width:p.flow.width,height:p.flow.height}})})}let _=[...n.values()].sort((e,t)=>e.kind.localeCompare(t.kind)||e.label.localeCompare(t.label));for(let e of _){let t=i.node(e.id),n=p[e.kind],a=r.get(e.id)??0;o.push({id:e.id,type:e.kind,position:{x:(t?.x??0)-n.width/2,y:(t?.y??0)-n.height/2},data:{kind:e.kind,label:e.label,refId:e.id,...e.facet===void 0?{}:{facet:e.facet},badge:e.facet??(a>0?String(a):e.kind)},draggable:!1,zIndex:d.leaf,width:n.width,height:n.height,style:{width:n.width,height:n.height}})}return o}function C(e,t,n,r={}){let i=t.size>0||n.size>0;return e.map(e=>{if(e.data.kind===`unit`||e.data.kind===`law`)return e;let a=e.data.kind===`flow`?t.has(e.data.refId):n.has(e.id);return{...e,data:{...e.data,highlighted:a,dimmed:i&&!a,active:r.activeNodeId!=null&&e.id===r.activeNodeId,focused:r.focusedNodeId!=null&&e.id===r.focusedNodeId}}})}function w(e,t){let n=new Map(t.map(e=>[e.id,e]));return t.some(e=>e.data.highlighted===!0)?e.map(e=>{let t=n.get(e.source),r=n.get(e.target),i=t?.data.highlighted===!0&&r?.data.highlighted===!0,a=m[e.data?.kind??`reads`];return{...e,animated:i?!0:e.animated,style:{...e.style,stroke:a,opacity:i?1:.32,strokeWidth:i?2.75:1.75},markerEnd:e.markerEnd&&typeof e.markerEnd==`object`?{...e.markerEnd,color:a}:e.markerEnd}}):e.map(e=>({...e,style:{...e.style,opacity:1}}))}var T=100;function E(e,t){return e.filter(e=>t.has(e.flow)).sort((e,t)=>t.startedAt-e.startedAt).slice(0,T)}var D=a();function O({cache:a,dataSlot:o}){let s=f(a);return(0,D.jsxs)(r,{children:[(0,D.jsx)(n,{render:t=>(0,D.jsx)(`span`,{...t,className:e(`flex w-4 shrink-0 items-center justify-center`,s.className),"data-slot":o,"data-cache":a,"aria-label":s.label,children:(0,D.jsx)(i,{icon:s.icon,className:`size-3`,"aria-hidden":!0})})}),(0,D.jsx)(t,{side:`top`,children:s.label})]})}export{w as a,h as c,C as i,E as n,x as o,g as r,v as s,O as t};