llm-mock-server 1.0.0 → 1.0.2

package/dist/rule-engine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rule-engine.js","sourceRoot":"","sources":["../src/rule-engine.ts"],"names":[],"mappings":"AAEA,SAAS,SAAS,CAAC,EAAU;IAC3B,OAAO,CAAC,EAAE,CAAC,MAAM,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AAC9F,CAAC;AAED,SAAS,cAAc,CAAC,OAAwB;IAC9C,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACpC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC;IACD,MAAM,EAAE,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC;IAC9B,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,cAAc,CAAC,KAAY;IAClC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACxC,CAAC;IACD,IAAI,KAAK,YAAY,MAAM,EAAE,CAAC;QAC5B,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACxC,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;QAChC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,GAAG,GAAG,KAAK,CAAC;IAClB,MAAM,WAAW,GAAG,GAAG,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACxF,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAClF,MAAM,UAAU,GAAG,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACrF,OAAO,CAAC,GAAG,EAAE,EAAE;QACb,IAAI,WAAW,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC;YAAE,OAAO,KAAK,CAAC;QAC/D,IAAI,SAAS,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACrD,IAAI,UAAU,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,aAAa,CAAC;YAAE,OAAO,KAAK,CAAC;QAC/D,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QACxE,IAAI,GAAG,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC;YAAE,OAAO,KAAK,CAAC;QACtF,IAAI,GAAG,CAAC,UAAU,KAAK,SAAS,IAAI,GAAG,CAAC,cAAc,KAAK,GAAG,CAAC,UAAU;YAAE,OAAO,KAAK,CAAC;QACxF,IAAI,GAAG,CAAC,SAAS,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC;YAAE,OAAO,KAAK,CAAC;QACvD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAAY;IACjC,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,IAAI,KAAK,GAAG,CAAC;IACnD,IAAI,KAAK,YAAY,MAAM;QAAE,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;IACrD,IAAI,OAAO,KAAK,KAAK,UAAU;QAAE,OAAO,aAAa,CAAC;IACtD,MAAM,GAAG,GAAgB,KAAK,CAAC;IAC/B,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;SAC9B,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,IAAI,OAAO,CAAC,KAAK,UAAU,CAAC;SAC7D,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACxC,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;AACjC,CAAC;AAED,SAAS,UAAU,CAAC,KAAY,EAAE,OAAiB,EAAE,OAAqB,EAAE,WAAoB;IAC9F,OAAO;QACL,WAAW,EAAE,WAAW,IAAI,aAAa,CAAC,KAAK,CAAC;QAChD,KAAK,EAAE,cAAc,CAAC,KAAK,CAAC;QAC5B,OAAO;QACP,OAAO;QACP,SAAS,EAAE,QAAQ;KACpB,CAAC;AACJ,CAAC;AAOD,MAAM,UAAU,sBAAsB,CACpC,KAA8B,EAC9B,IAA+B;IAE/B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;IACjF,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC;IACtC,OAAO;QACL,QAAQ,EAAE,GAAG,EAAE;YACb,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC,IAAI,IAAI,CAAC;YACpC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;YAClC,OAAO,IAAI,CAAC,KAAK,CAAC;QACpB,CAAC;QACD,UAAU,EAAE,KAAK,CAAC,MAAM;KACzB,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,UAAU;IACJ,KAAK,GAAW,EAAE,CAAC;IAEpC,GAAG,CAAC,KAAY,EAAE,OAAiB,EAAE,UAAwB,EAAE;QAC7D,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QACjD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,WAAW,CAAC,IAAU;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACrC,IAAI,GAAG,GAAG,CAAC,EAAE,CAAC;YACZ,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YAC1B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,UAAU,CAAC,OAAsC,EAAE,OAAiB,EAAE,WAAW,GAAG,WAAW;QAC7F,MAAM,IAAI,GAAG,UAAU,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,CAAC,CAAC;QAC3D,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,GAAgB;QACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;YAE5B,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC;gBAAE,SAAS;YAClC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC;gBAAE,SAAS;YAE/B,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC,EAAE,CAAC;gBACxB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC1B,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC;IACpF,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;IAC3B,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,SAAS,EAAE,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;IACzF,CAAC;IAED,KAAK;QACH,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;IACxB,CAAC;CACF"}

package/dist/sse-writer.d.ts

@@ -0,0 +1,5 @@
+import type { FastifyReply } from "fastify";
+import type { SSEChunk } from "./formats/types.js";
+import type { ReplyOptions } from "./types.js";
+export declare function writeSSE(reply: FastifyReply, chunks: readonly SSEChunk[], options?: ReplyOptions): Promise<void>;
+//# sourceMappingURL=sse-writer.d.ts.map

package/dist/sse-writer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sse-writer.d.ts","sourceRoot":"","sources":["../src/sse-writer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAC5C,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAa/C,wBAAsB,QAAQ,CAC5B,KAAK,EAAE,YAAY,EACnB,MAAM,EAAE,SAAS,QAAQ,EAAE,EAC3B,OAAO,GAAE,YAAiB,GACzB,OAAO,CAAC,IAAI,CAAC,CAef"}

package/dist/sse-writer.js

@@ -0,0 +1,23 @@
+const HTTP_OK = 200;
+function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function formatSSEChunk(chunk) {
+    const eventLine = chunk.event ? `event: ${chunk.event}\n` : "";
+    return `${eventLine}data: ${chunk.data}\n\n`;
+}
+export async function writeSSE(reply, chunks, options = {}) {
+    const latency = options.latency ?? 0;
+    reply.raw.writeHead(HTTP_OK, {
+        "Content-Type": "text/event-stream",
+        "Cache-Control": "no-cache",
+        Connection: "keep-alive",
+    });
+    for (const chunk of chunks) {
+        reply.raw.write(formatSSEChunk(chunk));
+        if (latency > 0)
+            await sleep(latency);
+    }
+    reply.raw.end();
+}
+//# sourceMappingURL=sse-writer.js.map

package/dist/sse-writer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sse-writer.js","sourceRoot":"","sources":["../src/sse-writer.ts"],"names":[],"mappings":"AAIA,MAAM,OAAO,GAAG,GAAG,CAAC;AAEpB,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED,SAAS,cAAc,CAAC,KAAe;IACrC,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/D,OAAO,GAAG,SAAS,SAAS,KAAK,CAAC,IAAI,MAAM,CAAC;AAC/C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,KAAmB,EACnB,MAA2B,EAC3B,UAAwB,EAAE;IAE1B,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,CAAC,CAAC;IAErC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,EAAE;QAC3B,cAAc,EAAE,mBAAmB;QACnC,eAAe,EAAE,UAAU;QAC3B,UAAU,EAAE,YAAY;KACzB,CAAC,CAAC;IAEH,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;QACvC,IAAI,OAAO,GAAG,CAAC;YAAE,MAAM,KAAK,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;AAClB,CAAC"}

package/{src/types/index.ts → dist/types/index.d.ts}

@@ -2,3 +2,4 @@ export type { FormatName, MockRequest, Message, ToolDef } from "./request.js";
 export type { Reply, ReplyObject, ErrorReply, ToolCall, Resolver, ReplyOptions, SequenceEntry } from "./reply.js";
 export type { Match, MatchObject, PendingRule, RuleHandle, RuleSummary, Handler, Rule } from "./rule.js";
 export type { RecordedRequest } from "../history.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/types/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAC9E,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAClH,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACzG,YAAY,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC"}

package/dist/types/index.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=index.js.map

package/dist/types/index.js.map

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

package/dist/types/reply.d.ts

@@ -0,0 +1,45 @@
+import type { MockRequest } from "./request.js";
+/** A reply is either a plain string (turns into `{ text: "..." }`) or a full reply object. */
+export type Reply = string | ReplyObject;
+/** A structured reply. Text, reasoning, tool calls, usage, and errors are all optional. */
+export interface ReplyObject {
+    readonly text?: string | undefined;
+    /** Extended thinking or chain-of-thought. Works with Anthropic and Responses formats. */
+    readonly reasoning?: string | undefined;
+    readonly tools?: readonly ToolCall[] | undefined;
+    /** Falls back to `{ input: 10, output: 5 }` if omitted. */
+    readonly usage?: {
+        readonly input: number;
+        readonly output: number;
+    } | undefined;
+    /** When set, the server responds with this HTTP error instead of a normal reply. */
+    readonly error?: ErrorReply | undefined;
+}
+/** An HTTP error response. The server returns this status code with a format-appropriate body. */
+export interface ErrorReply {
+    readonly status: number;
+    readonly message: string;
+    /** Each format has its own default if omitted. */
+    readonly type?: string | undefined;
+}
+export interface ToolCall {
+    /** Auto-generated if omitted. */
+    readonly id?: string | undefined;
+    readonly name: string;
+    readonly args: Readonly<Record<string, unknown>>;
+}
+/** A reply value or a function that produces one. Async functions are supported. */
+export type Resolver = Reply | ((req: MockRequest) => Reply | Promise<Reply>);
+/** Per-rule streaming options. Merged with server-level defaults, with per-rule values winning. */
+export interface ReplyOptions {
+    /** Milliseconds between SSE chunks. */
+    readonly latency?: number | undefined;
+    /** Characters per SSE chunk for more realistic streaming. */
+    readonly chunkSize?: number | undefined;
+}
+/** A single entry in a reply sequence. Can be a plain reply or a reply with per-step options. */
+export type SequenceEntry = Reply | {
+    readonly reply: Reply;
+    readonly options?: ReplyOptions;
+};
+//# sourceMappingURL=reply.d.ts.map

package/dist/types/reply.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reply.d.ts","sourceRoot":"","sources":["../../src/types/reply.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAEhD,8FAA8F;AAC9F,MAAM,MAAM,KAAK,GAAG,MAAM,GAAG,WAAW,CAAC;AAEzC,2FAA2F;AAC3F,MAAM,WAAW,WAAW;IAC1B,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACnC,yFAAyF;IACzF,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxC,QAAQ,CAAC,KAAK,CAAC,EAAE,SAAS,QAAQ,EAAE,GAAG,SAAS,CAAC;IACjD,2DAA2D;IAC3D,QAAQ,CAAC,KAAK,CAAC,EAAE;QAAE,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,SAAS,CAAC;IACjF,oFAAoF;IACpF,QAAQ,CAAC,KAAK,CAAC,EAAE,UAAU,GAAG,SAAS,CAAC;CACzC;AAED,kGAAkG;AAClG,MAAM,WAAW,UAAU;IACzB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,kDAAkD;IAClD,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACpC;AAED,MAAM,WAAW,QAAQ;IACvB,iCAAiC;IACjC,QAAQ,CAAC,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;CAClD;AAED,oFAAoF;AACpF,MAAM,MAAM,QAAQ,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,EAAE,WAAW,KAAK,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;AAE9E,mGAAmG;AACnG,MAAM,WAAW,YAAY;IAC3B,uCAAuC;IACvC,QAAQ,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACtC,6DAA6D;IAC7D,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACzC;AAED,iGAAiG;AACjG,MAAM,MAAM,aAAa,GAAG,KAAK,GAAG;IAAE,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;IAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,YAAY,CAAA;CAAE,CAAC"}

package/dist/types/reply.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=reply.js.map

package/dist/types/reply.js.map

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

package/dist/types/request.d.ts

@@ -0,0 +1,39 @@
+/** The LLM API wire format that was detected for a request. */
+export type FormatName = "openai" | "anthropic" | "responses";
+/** A normalised view of an incoming request, regardless of the original wire format. */
+export interface MockRequest {
+    readonly format: FormatName;
+    readonly model: string;
+    readonly streaming: boolean;
+    /** Full conversation, normalised from whatever format came in. */
+    readonly messages: readonly Message[];
+    /** The last user message's text. This is what most matchers check. */
+    readonly lastMessage: string;
+    /** Empty string if there wasn't one. */
+    readonly systemMessage: string;
+    readonly tools?: readonly ToolDef[] | undefined;
+    /** Pulled out from `tools` for quick lookups. */
+    readonly toolNames: readonly string[];
+    /** Set when the last message was a tool result. */
+    readonly lastToolCallId: string | undefined;
+    /** The raw request body, for anything we don't extract. */
+    readonly raw: unknown;
+    readonly headers: Readonly<Record<string, string | undefined>>;
+    /** e.g. `/v1/chat/completions` */
+    readonly path: string;
+}
+/** A single conversation message, normalised across all supported formats. */
+export interface Message {
+    readonly role: "system" | "user" | "assistant" | "tool";
+    readonly content: string;
+    /** Links the result back to its tool call. Only set on `"tool"` messages. */
+    readonly toolCallId?: string | undefined;
+}
+/** A tool definition from the request's `tools` array, normalised across formats. */
+export interface ToolDef {
+    readonly name: string;
+    readonly description?: string | undefined;
+    /** JSON Schema, passed through as-is. */
+    readonly parameters?: unknown;
+}
+//# sourceMappingURL=request.d.ts.map

package/dist/types/request.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"request.d.ts","sourceRoot":"","sources":["../../src/types/request.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,MAAM,MAAM,UAAU,GAAG,QAAQ,GAAG,WAAW,GAAG,WAAW,CAAC;AAE9D,wFAAwF;AACxF,MAAM,WAAW,WAAW;IAC1B,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAC;IAC5B,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC;IAC5B,kEAAkE;IAClE,QAAQ,CAAC,QAAQ,EAAE,SAAS,OAAO,EAAE,CAAC;IACtC,sEAAsE;IACtE,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,wCAAwC;IACxC,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,KAAK,CAAC,EAAE,SAAS,OAAO,EAAE,GAAG,SAAS,CAAC;IAChD,iDAAiD;IACjD,QAAQ,CAAC,SAAS,EAAE,SAAS,MAAM,EAAE,CAAC;IACtC,mDAAmD;IACnD,QAAQ,CAAC,cAAc,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5C,2DAA2D;IAC3D,QAAQ,CAAC,GAAG,EAAE,OAAO,CAAC;IACtB,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC;IAC/D,kCAAkC;IAClC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;CACvB;AAED,8EAA8E;AAC9E,MAAM,WAAW,OAAO;IACtB,QAAQ,CAAC,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;IACxD,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,6EAA6E;IAC7E,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1C;AAED,qFAAqF;AACrF,MAAM,WAAW,OAAO;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1C,yCAAyC;IACzC,QAAQ,CAAC,UAAU,CAAC,EAAE,OAAO,CAAC;CAC/B"}

package/dist/types/request.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=request.js.map

package/dist/types/request.js.map

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

package/dist/types/rule.d.ts

@@ -0,0 +1,57 @@
+import type { MockRequest, FormatName } from "./request.js";
+import type { Resolver, ReplyOptions, Reply, SequenceEntry } from "./reply.js";
+/**
+ * Determines whether a rule matches an incoming request.
+ *
+ * A `string` does a case-insensitive substring match on the last user message.
+ * A `RegExp` gets tested against the last user message.
+ * A `MatchObject` checks multiple fields at once with AND logic.
+ * A function receives the normalised request and returns a boolean.
+ */
+export type Match = string | RegExp | MatchObject | ((req: MockRequest) => boolean);
+/** A structured matcher. Every field you set must match for the rule to fire. */
+export interface MatchObject {
+    readonly message?: string | RegExp;
+    readonly model?: string | RegExp;
+    readonly system?: string | RegExp;
+    readonly format?: FormatName;
+    /** Match when the request includes a tool definition with this name. */
+    readonly toolName?: string;
+    /** Match when the last tool-result message has this `tool_call_id`. */
+    readonly toolCallId?: string;
+    /** Extra check that runs after all other fields pass. */
+    readonly predicate?: (req: MockRequest) => boolean;
+}
+/** Returned by `when()`. Call `.reply()` or `.replySequence()` on it to complete the rule. */
+export interface PendingRule {
+    reply(response: Resolver, options?: ReplyOptions): RuleHandle;
+    /** Each match advances through the array. The last entry repeats once exhausted. */
+    replySequence(entries: readonly SequenceEntry[]): RuleHandle;
+}
+/** A handle to a registered rule. All methods return `this` for chaining. */
+export interface RuleHandle {
+    times(n: number): RuleHandle;
+    first(): RuleHandle;
+}
+/**
+ * The shape of a handler file's default export.
+ * You can export a single handler or an array of them.
+ */
+export interface Handler {
+    match: (req: MockRequest) => boolean;
+    respond: (req: MockRequest) => Reply | Promise<Reply>;
+}
+/** A summary of a registered rule, for inspection. */
+export interface RuleSummary {
+    readonly description: string;
+    /** `Infinity` means unlimited. */
+    readonly remaining: number;
+}
+export interface Rule {
+    readonly description: string;
+    readonly match: (req: MockRequest) => boolean;
+    resolve: Resolver;
+    options: ReplyOptions;
+    remaining: number;
+}
+//# sourceMappingURL=rule.d.ts.map

package/dist/types/rule.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rule.d.ts","sourceRoot":"","sources":["../../src/types/rule.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC5D,OAAO,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAE/E;;;;;;;GAOG;AACH,MAAM,MAAM,KAAK,GACb,MAAM,GACN,MAAM,GACN,WAAW,GACX,CAAC,CAAC,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC,CAAC;AAEpC,iFAAiF;AACjF,MAAM,WAAW,WAAW;IAC1B,QAAQ,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IACnC,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IACjC,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAClC,QAAQ,CAAC,MAAM,CAAC,EAAE,UAAU,CAAC;IAC7B,wEAAwE;IACxE,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAC3B,uEAAuE;IACvE,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAC7B,yDAAyD;IACzD,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC;CACpD;AAED,8FAA8F;AAC9F,MAAM,WAAW,WAAW;IAC1B,KAAK,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,UAAU,CAAC;IAC9D,oFAAoF;IACpF,aAAa,CAAC,OAAO,EAAE,SAAS,aAAa,EAAE,GAAG,UAAU,CAAC;CAC9D;AAED,6EAA6E;AAC7E,MAAM,WAAW,UAAU;IACzB,KAAK,CAAC,CAAC,EAAE,MAAM,GAAG,UAAU,CAAC;IAC7B,KAAK,IAAI,UAAU,CAAC;CACrB;AAED;;;GAGG;AACH,MAAM,WAAW,OAAO;IACtB,KAAK,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC;IACrC,OAAO,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;CACvD;AAED,sDAAsD;AACtD,MAAM,WAAW,WAAW;IAC1B,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,kCAAkC;IAClC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;CAC5B;AAED,MAAM,WAAW,IAAI;IACnB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC;IAC9C,OAAO,EAAE,QAAQ,CAAC;IAClB,OAAO,EAAE,YAAY,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CACnB"}

package/dist/types/rule.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=rule.js.map

package/dist/types/rule.js.map

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

package/dist/types.d.ts

@@ -0,0 +1,4 @@
+export type { FormatName, MockRequest, Message, ToolDef } from "./types/request.js";
+export type { Reply, ReplyObject, ErrorReply, ToolCall, Resolver, ReplyOptions, SequenceEntry } from "./types/reply.js";
+export type { Match, MatchObject, PendingRule, RuleHandle, RuleSummary, Handler, Rule } from "./types/rule.js";
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACpF,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACxH,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/types.js

@@ -0,0 +1,2 @@
+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":""}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "llm-mock-server",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "A standalone mock LLM server for deterministic testing: OpenAI, Anthropic, and Responses API formats",
   "type": "module",
   "engines": {
@@ -41,6 +41,7 @@
     "commander": "14.0.3",
     "fastify": "5.8.2",
     "json5": "2.2.3",
+    "llm-schemas": "1.0.1",
     "picocolors": "1.1.1",
     "zod": "4.3.6"
   },

package/src/cli-validators.ts

@@ -28,6 +28,9 @@ export function parseLogLevel(value: string): LogLevel {
 }
 
 export async function parseHost(value: string): Promise<string> {
+  if (!value) {
+    throw new Error(`Invalid host "${value}". Must be a resolvable hostname or IP address.`);
+  }
   if (value === "localhost" || isIP(value) !== 0) {
     return value;
   }

package/src/cli.ts

@@ -1,12 +1,16 @@
 #!/usr/bin/env node
 
 import { watch } from "node:fs";
+import { createRequire } from "node:module";
 import { Command } from "commander";
 import pc from "picocolors";
 import { MockServer } from "./mock-server.js";
 import { Logger } from "./logger.js";
 import { parsePort, parseHost, parseLogLevel, parseChunkSize, parseLatency } from "./cli-validators.js";
 
+const require = createRequire(import.meta.url);
+const { version } = require("../package.json") as { version: string };
+
 const WATCH_DEBOUNCE_MS = 100;
 
 interface StartOptions {
@@ -54,7 +58,7 @@ async function start(options: StartOptions): Promise<void> {
 
   if (!quiet) {
     console.log();
-    console.log(`  ${pc.bold(pc.cyan("llm-mock-server"))} ${pc.dim("v1.0.0")}`);
+    console.log(`  ${pc.bold(pc.cyan("llm-mock-server"))} ${pc.dim(`v${version}`)}`);
     console.log();
     console.log(`  ${pc.dim("Port")}       ${pc.bold(String(port))}`);
     console.log(`  ${pc.dim("Rules")}      ${pc.bold(String(server.ruleCount))} loaded`);
@@ -106,7 +110,7 @@ async function start(options: StartOptions): Promise<void> {
 const program = new Command()
   .name("llm-mock-server")
   .description("Mock LLM server for deterministic testing")
-  .version("1.0.0");
+  .version(version);
 
 program
   .command("start", { isDefault: true })

package/src/formats/anthropic/index.ts

@@ -1,5 +1,5 @@
 import type { Format } from "../types.js";
-import { isStreaming } from "../parse-helpers.js";
+import { isStreaming } from "../request-helpers.js";
 import { parseRequest } from "./parse.js";
 import { serialize, serializeComplete, serializeError } from "./serialize.js";
 

package/src/formats/anthropic/parse.ts

@@ -1,5 +1,5 @@
 import type { MockRequest, Message, ToolDef } from "../../types.js";
-import { buildMockRequest, type RequestMeta } from "../parse-helpers.js";
+import { buildMockRequest, type RequestMeta } from "../request-helpers.js";
 import { AnthropicRequestSchema, type AnthropicRequest } from "./schema.js";
 
 function extractSystem(system: AnthropicRequest["system"]): Message[] {
@@ -12,11 +12,11 @@ function extractSystem(system: AnthropicRequest["system"]): Message[] {
 function extractContent(content: AnthropicRequest["messages"][number]["content"]): { content: string; toolCallId?: string | undefined } {
   if (typeof content === "string") return { content };
   const text = content
-    .filter((b) => b.type === "text")
+    .filter((b): b is { type: "text"; text: string } => b.type === "text")
     .map((b) => b.text)
     .join("\n");
-  const toolResult = content.find((b) => b.type === "tool_result");
-  const toolCallId = toolResult?.type === "tool_result" ? toolResult.tool_use_id : undefined;
+  const toolResult = content.find((b): b is { type: "tool_result"; tool_use_id: string } => b.type === "tool_result");
+  const toolCallId = toolResult?.tool_use_id;
   return { content: text, toolCallId };
 }
 

package/src/formats/anthropic/schema.ts

@@ -1,73 +1,6 @@
 import { z } from "zod";
 
-const TextBlockSchema = z.object({ type: z.literal("text"), text: z.string() });
-
-const ToolUseBlockSchema = z.object({
-  type: z.literal("tool_use"),
-  id: z.string(),
-  name: z.string(),
-  input: z.record(z.string(), z.unknown()),
-});
-
-const ToolResultBlockSchema = z.object({
-  type: z.literal("tool_result"),
-  tool_use_id: z.string(),
-  content: z.union([z.string(), z.array(TextBlockSchema)]).optional(),
-});
-
-const KnownContentBlockSchema = z.discriminatedUnion("type", [
-  TextBlockSchema,
-  ToolUseBlockSchema,
-  ToolResultBlockSchema,
-]);
-
-const LooseContentBlockSchema = z.union([
-  KnownContentBlockSchema,
-  z.looseObject({ type: z.string() }),
-]);
-
-const KNOWN_BLOCK_TYPES = new Set(["text", "tool_use", "tool_result"]);
-
-type KnownBlock = z.infer<typeof KnownContentBlockSchema>;
-
-const MessageSchema = z.object({
-  role: z.enum(["user", "assistant"]),
-  content: z.union([
-    z.string(),
-    z.array(LooseContentBlockSchema).transform((blocks) =>
-      blocks.filter((b): b is KnownBlock => KNOWN_BLOCK_TYPES.has(b.type)),
-    ),
-  ]),
-});
-
-const ToolDefinitionSchema = z.object({
-  name: z.string(),
-  description: z.string().optional(),
-  input_schema: z.record(z.string(), z.unknown()),
-});
-
-export const AnthropicRequestSchema = z.looseObject({
-  model: z.string().min(1),
-  max_tokens: z.number().int().positive(),
-  system: z.union([z.string(), z.array(TextBlockSchema)]).optional(),
-  messages: z.array(MessageSchema).min(1),
-  tools: z.array(ToolDefinitionSchema).optional(),
-  stream: z.boolean().optional(),
-  temperature: z.number().optional(),
-  top_p: z.number().optional(),
-  top_k: z.number().optional(),
-  stop_sequences: z.array(z.string()).optional(),
-  metadata: z.record(z.string(), z.unknown()).optional(),
-  cache_control: z.unknown().optional(),
-  container: z.string().optional(),
-  inference_geo: z.string().optional(),
-  output_config: z.unknown().optional(),
-  service_tier: z.string().optional(),
-  thinking: z.unknown().optional(),
-  tool_choice: z.unknown().optional(),
-});
-
-export type AnthropicRequest = z.infer<typeof AnthropicRequestSchema>;
+export { AnthropicRequestSchema, type AnthropicRequest } from "llm-schemas/anthropic";
 
 const ResponseContentBlockSchema = z.object({
   type: z.string(),

package/src/formats/anthropic/serialize.ts

@@ -1,6 +1,10 @@
 import type { ReplyObject, ReplyOptions } from "../../types.js";
 import type { SSEChunk } from "../types.js";
-import { splitText, genId, toolId, shouldEmitText, finishReason, DEFAULT_USAGE } from "../parse-helpers.js";
+import { splitText, genId, toolId, shouldEmitText, finishReason, DEFAULT_USAGE } from "../serialize-helpers.js";
+
+function buildUsage(usage: { input: number; output: number }) {
+  return { input_tokens: usage.input, output_tokens: usage.output };
+}
 
 function contentBlock(index: number, startBlock: unknown, deltas: SSEChunk[]): SSEChunk[] {
   return [
@@ -52,7 +56,7 @@ export function serialize(reply: ReplyObject, model: string, options: ReplyOptio
   return [
     { event: "message_start", data: JSON.stringify({
       type: "message_start",
-      message: { id, type: "message", role: "assistant", model, content: [], stop_reason: null, usage: { input_tokens: usage.input, output_tokens: 0 } },
+      message: { id, type: "message", role: "assistant", model, content: [], stop_reason: null, usage: { ...buildUsage(usage), output_tokens: 0 } },
     })},
     ...reasoningChunks,
     ...textChunks,
@@ -66,7 +70,7 @@ export function serialize(reply: ReplyObject, model: string, options: ReplyOptio
   ];
 }
 
-export function serializeComplete(reply: ReplyObject, model: string): unknown {
+export function serializeComplete(reply: ReplyObject, model: string): Record<string, unknown> {
   const id = genId("msg");
   const usage = reply.usage ?? DEFAULT_USAGE;
 
@@ -82,10 +86,10 @@ export function serializeComplete(reply: ReplyObject, model: string): unknown {
     id, type: "message", role: "assistant", model, content,
     stop_reason: finishReason(reply, "tool_use", "end_turn"),
     stop_sequence: null,
-    usage: { input_tokens: usage.input, output_tokens: usage.output },
+    usage: buildUsage(usage),
   };
 }
 
-export function serializeError(error: { status: number; message: string; type?: string }): unknown {
+export function serializeError(error: { status: number; message: string; type?: string }): Record<string, unknown> {
   return { type: "error", error: { type: error.type ?? "api_error", message: error.message } };
 }

package/src/formats/openai/index.ts

@@ -1,5 +1,5 @@
 import type { Format } from "../types.js";
-import { isStreaming } from "../parse-helpers.js";
+import { isStreaming } from "../request-helpers.js";
 import { parseRequest } from "./parse.js";
 import { serialize, serializeComplete, serializeError } from "./serialize.js";
 

package/src/formats/openai/parse.ts

@@ -1,5 +1,5 @@
 import type { MockRequest, Message, ToolDef } from "../../types.js";
-import { buildMockRequest, type RequestMeta } from "../parse-helpers.js";
+import { buildMockRequest, type RequestMeta } from "../request-helpers.js";
 import { OpenAIRequestSchema, type OpenAIRequest } from "./schema.js";
 
 function extractContent(content: OpenAIRequest["messages"][number]["content"]): string {

package/src/formats/openai/schema.ts

@@ -1,74 +1,6 @@
 import { z } from "zod";
 
-const ContentPartSchema = z.looseObject({
-  type: z.string(),
-  text: z.string().optional(),
-});
-
-const MessageSchema = z.object({
-  role: z.enum(["system", "developer", "user", "assistant", "tool"]).optional(),
-  content: z.union([z.string(), z.array(ContentPartSchema), z.null()]).optional(),
-  name: z.string().optional(),
-  tool_calls: z.array(z.object({
-    index: z.number().optional(),
-    id: z.string().optional(),
-    type: z.string().optional(),
-    function: z.object({
-      name: z.string(),
-      arguments: z.string(),
-    }),
-  })).optional(),
-  tool_call_id: z.string().optional(),
-});
-
-const ToolSchema = z.object({
-  type: z.string(),
-  function: z.object({
-    name: z.string(),
-    description: z.string().optional(),
-    parameters: z.record(z.string(), z.unknown()).optional(),
-  }),
-});
-
-export const OpenAIRequestSchema = z.looseObject({
-  model: z.string().min(1),
-  messages: z.array(MessageSchema).min(1),
-  temperature: z.number().optional(),
-  top_p: z.number().optional(),
-  n: z.number().optional(),
-  stop: z.union([z.string(), z.array(z.string())]).optional(),
-  max_tokens: z.number().optional(),
-  presence_penalty: z.number().optional(),
-  frequency_penalty: z.number().optional(),
-  tools: z.array(ToolSchema).optional(),
-  stream: z.boolean().optional(),
-  tool_choice: z.unknown().optional(),
-  user: z.string().optional(),
-  audio: z.unknown().optional(),
-  function_call: z.unknown().optional(),
-  functions: z.array(z.unknown()).optional(),
-  logit_bias: z.record(z.string(), z.number()).optional(),
-  logprobs: z.boolean().optional(),
-  max_completion_tokens: z.number().optional(),
-  metadata: z.record(z.string(), z.string()).optional(),
-  modalities: z.array(z.string()).optional(),
-  parallel_tool_calls: z.boolean().optional(),
-  prediction: z.unknown().optional(),
-  prompt_cache_key: z.string().optional(),
-  prompt_cache_retention: z.string().optional(),
-  reasoning_effort: z.string().optional(),
-  response_format: z.unknown().optional(),
-  safety_identifier: z.string().optional(),
-  seed: z.number().optional(),
-  service_tier: z.string().optional(),
-  store: z.boolean().optional(),
-  stream_options: z.unknown().optional(),
-  top_logprobs: z.number().optional(),
-  verbosity: z.string().optional(),
-  web_search_options: z.unknown().optional(),
-});
-
-export type OpenAIRequest = z.infer<typeof OpenAIRequestSchema>;
+export { OpenAIRequestSchema, type OpenAIRequest } from "llm-schemas/openai/chat-completions";
 
 const ToolCallResponseSchema = z.object({
   id: z.string(),

package/src/formats/openai/serialize.ts

@@ -1,6 +1,16 @@
 import type { ReplyObject, ReplyOptions } from "../../types.js";
 import type { SSEChunk } from "../types.js";
-import { splitText, genId, toolId, finishReason, MS_PER_SECOND, DEFAULT_USAGE } from "../parse-helpers.js";
+import { splitText, genId, toolId, finishReason, MS_PER_SECOND, DEFAULT_USAGE } from "../serialize-helpers.js";
+
+function buildUsage(usage: { input: number; output: number }) {
+  return {
+    prompt_tokens: usage.input,
+    completion_tokens: usage.output,
+    total_tokens: usage.input + usage.output,
+    prompt_tokens_details: { cached_tokens: 0, audio_tokens: 0 },
+    completion_tokens_details: { reasoning_tokens: 0, audio_tokens: 0, accepted_prediction_tokens: 0, rejected_prediction_tokens: 0 },
+  };
+}
 
 function chunkEnvelope(
   id: string, created: number, model: string,
@@ -38,13 +48,7 @@ export function serialize(reply: ReplyObject, model: string, options: ReplyOptio
     }),
   );
 
-  const usageChunk = {
-    prompt_tokens: usage.input,
-    completion_tokens: usage.output,
-    total_tokens: usage.input + usage.output,
-    prompt_tokens_details: { cached_tokens: 0, audio_tokens: 0 },
-    completion_tokens_details: { reasoning_tokens: 0, audio_tokens: 0, accepted_prediction_tokens: 0, rejected_prediction_tokens: 0 },
-  };
+  const usageChunk = buildUsage(usage);
 
   return [
     chunkEnvelope(id, created, model, { role: "assistant" }),
@@ -56,7 +60,7 @@ export function serialize(reply: ReplyObject, model: string, options: ReplyOptio
   ];
 }
 
-export function serializeComplete(reply: ReplyObject, model: string): unknown {
+export function serializeComplete(reply: ReplyObject, model: string): Record<string, unknown> {
   const id = genId("chatcmpl");
   const created = Math.floor(Date.now() / MS_PER_SECOND);
   const usage = reply.usage ?? DEFAULT_USAGE;
@@ -77,16 +81,10 @@ export function serializeComplete(reply: ReplyObject, model: string): unknown {
     system_fingerprint: null,
     service_tier: "default",
     choices: [{ index: 0, message, logprobs: null, finish_reason: finishReason(reply, "tool_calls", "stop") }],
-    usage: {
-      prompt_tokens: usage.input,
-      completion_tokens: usage.output,
-      total_tokens: usage.input + usage.output,
-      prompt_tokens_details: { cached_tokens: 0, audio_tokens: 0 },
-      completion_tokens_details: { reasoning_tokens: 0, audio_tokens: 0, accepted_prediction_tokens: 0, rejected_prediction_tokens: 0 },
-    },
+    usage: buildUsage(usage),
   };
 }
 
-export function serializeError(error: { status: number; message: string; type?: string }): unknown {
+export function serializeError(error: { status: number; message: string; type?: string }): Record<string, unknown> {
   return { error: { message: error.message, type: error.type ?? "server_error", code: null } };
 }