@kuckit/ai-module 2.0.0

package/dist/index.d.ts

@@ -0,0 +1,7 @@
+import * as _kuckit_sdk0 from "@kuckit/sdk";
+
+//#region src/server/module.d.ts
+type AIModuleConfig = Record<string, never>;
+declare const kuckitModule: _kuckit_sdk0.KuckitModuleDefinition<AIModuleConfig>;
+//#endregion
+export { type AIModuleConfig, kuckitModule };

package/dist/index.js

@@ -0,0 +1,115 @@
+import { asFunction, defineKuckitModule } from "@kuckit/sdk";
+import { convertToModelMessages, streamText } from "ai";
+import { Router } from "express";
+import { z } from "zod";
+
+//#region src/server/use-cases/stream-chat.ts
+const makeStreamChat = ({ aiProvider, logger }) => {
+	return async (input) => {
+		logger.debug("Starting AI chat stream", { messageCount: input.messages.length });
+		return streamText({
+			model: aiProvider,
+			messages: convertToModelMessages(input.messages)
+		});
+	};
+};
+
+//#endregion
+//#region src/server/validation/chat.schema.ts
+const messagePartSchema = z.discriminatedUnion("type", [
+	z.object({
+		type: z.literal("text"),
+		text: z.string(),
+		state: z.string().optional()
+	}),
+	z.object({
+		type: z.literal("image"),
+		image: z.string()
+	}),
+	z.object({ type: z.literal("step-start") })
+]);
+const uiMessageSchema = z.object({
+	id: z.string(),
+	role: z.enum([
+		"user",
+		"assistant",
+		"system"
+	]),
+	content: z.string().optional(),
+	parts: z.array(messagePartSchema).optional()
+});
+const chatRequestSchema = z.object({ messages: z.array(uiMessageSchema).min(1, "At least one message is required") });
+
+//#endregion
+//#region src/server/router/ai.router.ts
+const createAiRouter = () => {
+	const router = Router();
+	router.post("/chat", async (req, res) => {
+		const scope = req.scope;
+		if (!scope) {
+			res.status(500).json({ error: "Internal server error" });
+			return;
+		}
+		const { session, requestLogger: logger, streamChat } = scope.cradle;
+		if (!session?.user?.id) {
+			logger.warn("Unauthenticated AI chat request");
+			res.status(401).json({ error: "Authentication required" });
+			return;
+		}
+		const parseResult = chatRequestSchema.safeParse(req.body);
+		if (!parseResult.success) {
+			logger.warn("Invalid AI chat request", { errors: parseResult.error.flatten() });
+			res.status(400).json({
+				error: "Invalid request",
+				details: parseResult.error.flatten()
+			});
+			return;
+		}
+		try {
+			const { messages } = parseResult.data;
+			(await streamChat({ messages })).pipeUIMessageStreamToResponse(res);
+		} catch (error) {
+			if (!res.headersSent) {
+				logger.error("AI chat failed", { error });
+				res.status(500).json({ error: "AI chat failed" });
+			} else logger.error("AI chat stream failed after headers sent", { error });
+		}
+	});
+	return router;
+};
+
+//#endregion
+//#region src/server/module.ts
+const kuckitModule = defineKuckitModule({
+	id: "kuckit.ai",
+	displayName: "AI Chat",
+	description: "Streaming AI chat powered by Vercel AI SDK",
+	version: "0.1.0",
+	capabilities: ["nav.item", "api.public"],
+	async register(ctx) {
+		const { container } = ctx;
+		container.register({ streamChat: asFunction(({ aiProvider, logger }) => makeStreamChat({
+			aiProvider,
+			logger
+		})).scoped() });
+	},
+	registerApi(ctx) {
+		const aiRouter = createAiRouter();
+		ctx.addApiRegistration({
+			type: "rest-router",
+			name: "ai",
+			router: aiRouter
+		});
+	},
+	async onBootstrap(ctx) {
+		const { container } = ctx;
+		container.resolve("logger").info("AI Chat module initialized");
+	},
+	async onShutdown(ctx) {
+		const { container } = ctx;
+		container.resolve("logger").info("AI Chat module shutting down");
+	}
+});
+
+//#endregion
+export { kuckitModule };

package/package.json

@@ -0,0 +1,48 @@
+{
+	"name": "@kuckit/ai-module",
+	"version": "2.0.0",
+	"description": "AI chat streaming module for Kuckit applications",
+	"type": "module",
+	"main": "dist/server/module.js",
+	"types": "dist/server/module.d.ts",
+	"files": [
+		"dist"
+	],
+	"exports": {
+		".": {
+			"types": "./dist/server/module.d.ts",
+			"default": "./dist/server/module.js"
+		},
+		"./client": {
+			"types": "./dist/client/index.d.ts",
+			"default": "./dist/client/index.js"
+		}
+	},
+	"kuckit": {
+		"id": "kuckit.ai",
+		"server": ".",
+		"client": "./client"
+	},
+	"scripts": {
+		"build": "tsdown",
+		"prepublishOnly": "npm run build && node ../../scripts/resolve-workspace-protocols.cjs && node ../../scripts/check-no-workspace-protocol.cjs"
+	},
+	"peerDependencies": {
+		"@kuckit/sdk": "^2.0.0",
+		"@kuckit/sdk-react": "^2.0.0",
+		"react": "^18 || ^19",
+		"typescript": "^5",
+		"express": "^4 || ^5"
+	},
+	"dependencies": {
+		"@ai-sdk/react": "^2.0.39",
+		"ai": "^5.0.49",
+		"zod": "^4.1.11",
+		"lucide-react": "^0.511.0"
+	},
+	"devDependencies": {
+		"@types/react": "^19.0.0",
+		"@types/express": "catalog:",
+		"tsdown": "catalog:"
+	}
+}