ai-test-kit 0.0.1

package/dist/ui/index.mjs

@@ -0,0 +1,266 @@
+import { t as tokenize } from "../tokenize-C-Zp26iY.mjs";
+
+//#region src/ui/parts.ts
+/**
+* Builds the {@link UIParts} namespace bound to a message's `DATA` and `TOOLS` types. The runtime is
+* identical for every binding; the type parameters only sharpen `data` and `tool`. Use the top-level
+* {@link UIParts} for the loose default, or `fromUIMessage` to bind.
+*/
+const createUIParts = () => ({
+	text: (text, opts) => ({
+		type: "text",
+		text,
+		...opts
+	}),
+	reasoning: (text, opts) => ({
+		type: "reasoning",
+		text,
+		...opts
+	}),
+	sourceUrl: (args) => ({
+		type: "source-url",
+		...args
+	}),
+	sourceDocument: (args) => ({
+		type: "source-document",
+		...args
+	}),
+	file: (args) => ({
+		type: "file",
+		...args
+	}),
+	stepStart: () => ({ type: "step-start" }),
+	data: (name, data, opts = {}) => ({
+		type: `data-${name}`,
+		data,
+		...opts
+	}),
+	tool: (name, invocation) => ({
+		type: `tool-${name}`,
+		...invocation
+	}),
+	dynamicTool: (args) => ({
+		type: "dynamic-tool",
+		...args
+	})
+});
+/** Builders for `UIMessagePart`s, with loose default types. Use `fromUIMessage` to bind to a message type. */
+const UIParts = createUIParts();
+
+//#endregion
+//#region src/ui/chunks.ts
+/**
+* Builds the {@link UIChunks} namespace bound to a message's `METADATA` and `DATA` types. The runtime
+* is identical for every binding; the type parameters only sharpen `data`, `start`, `finish`, and
+* `messageMetadata`. Use the top-level {@link UIChunks} for the loose default, or `fromUIMessage` to bind.
+*/
+const createUIChunks = () => ({
+	textStart: (args) => ({
+		type: "text-start",
+		...args
+	}),
+	textDelta: (args) => ({
+		type: "text-delta",
+		...args
+	}),
+	textEnd: (args) => ({
+		type: "text-end",
+		...args
+	}),
+	reasoningStart: (args) => ({
+		type: "reasoning-start",
+		...args
+	}),
+	reasoningDelta: (args) => ({
+		type: "reasoning-delta",
+		...args
+	}),
+	reasoningEnd: (args) => ({
+		type: "reasoning-end",
+		...args
+	}),
+	error: (errorText) => ({
+		type: "error",
+		errorText
+	}),
+	toolInputStart: (args) => ({
+		type: "tool-input-start",
+		...args
+	}),
+	toolInputDelta: (args) => ({
+		type: "tool-input-delta",
+		...args
+	}),
+	toolInputAvailable: (args) => ({
+		type: "tool-input-available",
+		...args
+	}),
+	toolInputError: (args) => ({
+		type: "tool-input-error",
+		...args
+	}),
+	toolApprovalRequest: (args) => ({
+		type: "tool-approval-request",
+		...args
+	}),
+	toolOutputAvailable: (args) => ({
+		type: "tool-output-available",
+		...args
+	}),
+	toolOutputError: (args) => ({
+		type: "tool-output-error",
+		...args
+	}),
+	toolOutputDenied: (args) => ({
+		type: "tool-output-denied",
+		...args
+	}),
+	sourceUrl: (args) => ({
+		type: "source-url",
+		...args
+	}),
+	sourceDocument: (args) => ({
+		type: "source-document",
+		...args
+	}),
+	file: (args) => ({
+		type: "file",
+		...args
+	}),
+	data: (name, data, opts = {}) => ({
+		type: `data-${name}`,
+		data,
+		...opts
+	}),
+	startStep: () => ({ type: "start-step" }),
+	finishStep: () => ({ type: "finish-step" }),
+	start: (args) => ({
+		type: "start",
+		...args
+	}),
+	finish: (args) => ({
+		type: "finish",
+		...args
+	}),
+	abort: (args) => ({
+		type: "abort",
+		...args
+	}),
+	messageMetadata: (messageMetadata) => ({
+		type: "message-metadata",
+		messageMetadata
+	}),
+	text: (text, { id = "1", length, separator } = {}) => [
+		{
+			type: "text-start",
+			id
+		},
+		...tokenize(text, {
+			length,
+			separator
+		}).map((delta) => ({
+			type: "text-delta",
+			id,
+			delta
+		})),
+		{
+			type: "text-end",
+			id
+		}
+	],
+	reasoning: (text, { id = "1", length, separator } = {}) => [
+		{
+			type: "reasoning-start",
+			id
+		},
+		...tokenize(text, {
+			length,
+			separator
+		}).map((delta) => ({
+			type: "reasoning-delta",
+			id,
+			delta
+		})),
+		{
+			type: "reasoning-end",
+			id
+		}
+	],
+	toolInput: (args) => [
+		{
+			type: "tool-input-start",
+			toolCallId: args.toolCallId,
+			toolName: args.toolName
+		},
+		...tokenize(JSON.stringify(args.input), { length: args.length }).map((inputTextDelta) => ({
+			type: "tool-input-delta",
+			toolCallId: args.toolCallId,
+			inputTextDelta
+		})),
+		{
+			type: "tool-input-available",
+			toolCallId: args.toolCallId,
+			toolName: args.toolName,
+			input: args.input
+		}
+	]
+});
+/** Builders for `UIMessageChunk`s, with loose default types. Use `fromUIMessage` to bind to a message type. */
+const UIChunks = createUIChunks();
+
+//#endregion
+//#region src/ui/message.ts
+/** Monotonic counter backing the auto-generated message ids. */
+let messageCounter = 0;
+/** Returns the next unique auto-generated message id. */
+const nextMessageId = () => {
+	messageCounter += 1;
+	return `mock-message-${messageCounter}`;
+};
+/**
+* Builds the {@link UIMessages} namespace bound to a message's `METADATA`, `DATA`, and `TOOLS` types.
+* Each role helper accepts a `string` shortcut (becomes a single text part) or a full array of parts.
+* Use the top-level {@link UIMessages} for the loose default, or `fromUIMessage` to bind.
+*/
+const createUIMessages = () => {
+	/** Normalizes a string shortcut into a single text part, or passes an array of parts through. */
+	const toParts = (content) => typeof content === "string" ? [{
+		type: "text",
+		text: content
+	}] : content;
+	/** Assembles a message of the given role from content and options. */
+	const build = (role, content, opts) => ({
+		id: opts.id ?? nextMessageId(),
+		role,
+		parts: toParts(content),
+		...opts.metadata !== void 0 ? { metadata: opts.metadata } : {}
+	});
+	return {
+		user: (content, opts = {}) => build("user", content, opts),
+		assistant: (content, opts = {}) => build("assistant", content, opts),
+		system: (content, opts = {}) => build("system", content, opts)
+	};
+};
+/** Builders for `UIMessage`s, with loose default types. Use `fromUIMessage` to bind to a message type. */
+const UIMessages = createUIMessages();
+
+//#endregion
+//#region src/ui/from-ui-message.ts
+/**
+* Binds the UI builders to a concrete `UIMessage` type, so `data`/`tool`/`messageMetadata` and message
+* metadata infer their names and payloads from that type. Returns typed `UIParts`, `UIChunks`, and
+* `UIMessages` namespaces; for loose, untyped builders use the top-level exports instead.
+*
+* @example
+* const { UIParts, UIChunks, UIMessages } = fromUIMessage<MyUIMessage>();
+* UIChunks.data('weather', { city: 'Tokyo' }); // name and payload typed
+* UIMessages.assistant([UIParts.text('hi')]);
+*/
+const fromUIMessage = () => ({
+	UIParts: createUIParts(),
+	UIChunks: createUIChunks(),
+	UIMessages: createUIMessages()
+});
+
+//#endregion
+export { UIChunks, UIMessages, UIParts, createUIChunks, createUIMessages, createUIParts, fromUIMessage };

package/package.json

@@ -0,0 +1,71 @@
+{
+  "name": "ai-test-kit",
+  "version": "0.0.1",
+  "description": "Test utilities for the AI SDK: mock models, content and stream-part builders, fully type-safe",
+  "keywords": [],
+  "license": "MIT",
+  "author": "Chris Cook",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/zirkelc/ai-test-kit.git"
+  },
+  "files": [
+    "dist"
+  ],
+  "type": "module",
+  "types": "dist/index.d.mts",
+  "exports": {
+    ".": "./dist/index.mjs",
+    "./language": "./dist/language/index.mjs",
+    "./ui": "./dist/ui/index.mjs",
+    "./package.json": "./package.json"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "prepublishOnly": "pnpm build",
+    "build": "tsdown",
+    "test": "vitest",
+    "lint": "oxlint --fix",
+    "lint:ci": "oxlint",
+    "format": "oxfmt --write",
+    "format:ci": "oxfmt --check",
+    "prepare": "husky"
+  },
+  "devDependencies": {
+    "@ai-sdk/provider": "^3.0.10",
+    "@ai-sdk/provider-utils": "^4.0.27",
+    "@arethetypeswrong/cli": "^0.18.2",
+    "@total-typescript/tsconfig": "^1.0.4",
+    "@types/node": "^25.2.3",
+    "ai": "^6.0.197",
+    "husky": "^9.1.7",
+    "lint-staged": "^16.2.7",
+    "oxfmt": "^0.30.0",
+    "oxlint": "^1.45.0",
+    "oxlint-tsgolint": "^0.11.5",
+    "pkg-pr-new": "^0.0.63",
+    "publint": "^0.3.17",
+    "tsdown": "^0.20.3",
+    "tsx": "^4.21.0",
+    "typescript": "^5.9.3",
+    "vitest": "^4.0.18"
+  },
+  "peerDependencies": {
+    "@ai-sdk/provider": "^3.0.0",
+    "@ai-sdk/provider-utils": "^4.0.0",
+    "ai": "5.x || 6.x",
+    "vitest": ">=3"
+  },
+  "lint-staged": {
+    "*.{js,ts,tsx,jsx}": [
+      "pnpm lint",
+      "pnpm format"
+    ],
+    "*.{json,jsonc}": [
+      "pnpm format"
+    ]
+  },
+  "packageManager": "pnpm@10.29.2"
+}