@yuer678/create-kb 0.3.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 anyuer678
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/index.js

@@ -0,0 +1,196 @@
+#!/usr/bin/env node
+
+// src/index.ts
+import { Command } from "commander";
+
+// src/prompts.ts
+import prompts from "prompts";
+async function promptMissing(options) {
+  if (options.projectName && options.template && options.packageManager) {
+    return {
+      projectName: options.projectName,
+      template: options.template,
+      packageManager: options.packageManager,
+      git: options.git ?? true
+    };
+  }
+  const answers = await prompts(
+    [
+      {
+        type: "text",
+        name: "projectName",
+        message: "\u9879\u76EE\u540D\u79F0",
+        initial: "kb-app",
+        validate: validateProjectName,
+        skip: () => Boolean(options.projectName)
+      },
+      {
+        type: "select",
+        name: "template",
+        message: "\u9009\u62E9\u6A21\u677F",
+        choices: [
+          { title: "base - \u6700\u5C0F\u53EF\u7528", value: "base" },
+          { title: "starter - \u5B8C\u6574\u793A\u4F8B", value: "starter" },
+          { title: "api - Express \u540E\u7AEF\u670D\u52A1", value: "api" },
+          { title: "fullstack - \u524D\u7AEF+API+Docker", value: "fullstack" },
+          { title: "electron - \u684C\u9762\u5E94\u7528", value: "electron" },
+          { title: "react - React 19 \u524D\u7AEF", value: "react" },
+          { title: "ai - LLM \u5DE5\u4F5C\u53F0", value: "ai" }
+        ],
+        skip: () => Boolean(options.template)
+      },
+      {
+        type: "select",
+        name: "packageManager",
+        message: "\u5305\u7BA1\u7406\u5668",
+        choices: [
+          { title: "pnpm", value: "pnpm" },
+          { title: "npm", value: "npm" },
+          { title: "yarn", value: "yarn" }
+        ],
+        skip: () => Boolean(options.packageManager)
+      }
+    ],
+    { onCancel: () => process.exit(1) }
+  );
+  return {
+    projectName: options.projectName ?? answers.projectName,
+    template: options.template ?? answers.template,
+    packageManager: options.packageManager ?? answers.packageManager,
+    git: options.git ?? true
+  };
+}
+function validateProjectName(name) {
+  if (!/^[a-z0-9][a-z0-9-]*$/.test(name)) {
+    return "\u9879\u76EE\u540D\u53EA\u80FD\u5305\u542B\u5C0F\u5199\u5B57\u6BCD\u3001\u6570\u5B57\u548C\u8FDE\u5B57\u7B26\uFF08-\uFF09\uFF0C\u4E14\u4E0D\u80FD\u4EE5\u8FDE\u5B57\u7B26\u5F00\u5934";
+  }
+  return true;
+}
+function toPackageName(name) {
+  return name;
+}
+
+// src/create.ts
+import { cpSync, existsSync, mkdirSync, readdirSync, readFileSync, writeFileSync } from "fs";
+import { join, dirname } from "path";
+import { fileURLToPath } from "url";
+import { execSync } from "child_process";
+import { createRequire } from "module";
+var __dirname = dirname(fileURLToPath(import.meta.url));
+function templatesRoot() {
+  return join(__dirname, "../templates");
+}
+function kbVersion() {
+  const require2 = createRequire(import.meta.url);
+  try {
+    const pkg = require2("kb-ui-vue/package.json");
+    return pkg.version;
+  } catch {
+    console.warn('\u26A0 \u672A\u8BFB\u53D6\u5230 kb-ui-vue \u7248\u672C\uFF0C\u6A21\u677F\u5C06\u4F7F\u7528 "latest" \u4F5C\u4E3A\u4F9D\u8D56\u7248\u672C');
+    return "latest";
+  }
+}
+var PLACEHOLDER_EXTENSIONS = /* @__PURE__ */ new Set([
+  ".json",
+  ".ts",
+  ".vue",
+  ".html",
+  ".md",
+  ".js",
+  ".mjs",
+  ".css",
+  ".gitignore",
+  ".yml",
+  ".yaml"
+]);
+function shouldReplace(fileName) {
+  if (fileName === ".gitignore") return true;
+  const ext = fileName.slice(fileName.lastIndexOf("."));
+  return PLACEHOLDER_EXTENSIONS.has(ext);
+}
+function replacePlaceholders(filePath, options, version) {
+  const content = readFileSync(filePath, "utf-8");
+  const replaced = content.replaceAll("{{projectName}}", options.projectName).replaceAll("{{packageName}}", toPackageName(options.projectName)).replaceAll("{{kbVersion}}", version);
+  if (replaced !== content) {
+    writeFileSync(filePath, replaced, "utf-8");
+  }
+}
+function walkAndReplace(dir, options, version) {
+  for (const entry of readdirSync(dir, { withFileTypes: true })) {
+    const full = join(dir, entry.name);
+    if (entry.isDirectory()) {
+      walkAndReplace(full, options, version);
+    } else if (shouldReplace(entry.name)) {
+      replacePlaceholders(full, options, version);
+    }
+  }
+}
+async function createProject(options) {
+  const targetDir = join(process.cwd(), options.projectName);
+  const templateDir = join(templatesRoot(), options.template);
+  if (!existsSync(templateDir)) {
+    console.error(`\u2716 \u6A21\u677F\u4E0D\u5B58\u5728\uFF1A${options.template}`);
+    process.exit(1);
+  }
+  if (existsSync(targetDir)) {
+    const entries = readdirSync(targetDir);
+    if (entries.length > 0) {
+      console.error(`\u2716 \u76EE\u6807\u76EE\u5F55 ${targetDir} \u5DF2\u5B58\u5728\u4E14\u975E\u7A7A\uFF0C\u8BF7\u9009\u62E9\u5176\u4ED6\u9879\u76EE\u540D\u6216\u5220\u9664\u8BE5\u76EE\u5F55`);
+      process.exit(1);
+    }
+  } else {
+    mkdirSync(targetDir, { recursive: true });
+  }
+  console.log(`
+\u521B\u5EFA\u9879\u76EE ${options.projectName}\uFF08\u6A21\u677F\uFF1A${options.template}\uFF09...`);
+  cpSync(templateDir, targetDir, { recursive: true });
+  const version = kbVersion();
+  walkAndReplace(targetDir, options, version);
+  if (options.git) {
+    try {
+      execSync("git init", { cwd: targetDir, stdio: "ignore" });
+    } catch {
+      console.warn("\u26A0 git init \u5931\u8D25\uFF08\u672A\u5B89\u88C5 git\uFF1F\uFF09\uFF0C\u5DF2\u8DF3\u8FC7\uFF0C\u4E0D\u5F71\u54CD\u9879\u76EE\u521B\u5EFA");
+    }
+  }
+  console.log(`
+\u2714 \u9879\u76EE\u521B\u5EFA\u5B8C\u6210\uFF1A${targetDir}
+`);
+  console.log("\u4E0B\u4E00\u6B65\uFF1A");
+  console.log(`  cd ${options.projectName}`);
+  console.log(`  ${options.packageManager} install`);
+  console.log(`  ${options.packageManager} run dev
+`);
+}
+
+// src/index.ts
+var program = new Command();
+program.name("create-kb").description("\u521B\u5EFA\u4E00\u4E2A\u9884\u88C5 @kb/ui \u7684 Vue 3 + Vite + TypeScript \u9879\u76EE").version("0.1.0").argument("[project-name]", "\u9879\u76EE\u540D\u79F0").option("--template <template>", "\u6A21\u677F\uFF1Abase\uFF08\u6700\u5C0F\uFF09| starter\uFF08\u5B8C\u6574\u793A\u4F8B\uFF09", "base").option("--no-git", "\u4E0D\u521D\u59CB\u5316 git \u4ED3\u5E93").option("-p, --package-manager <pm>", "\u5305\u7BA1\u7406\u5668\uFF1Apnpm | npm | yarn", "pnpm").action(async (projectName) => {
+  const cliOptions = program.opts();
+  if (projectName !== void 0) {
+    const result = validateProjectName(projectName);
+    if (result !== true) {
+      console.error(`\u2716 ${result}`);
+      process.exit(1);
+    }
+  }
+  const validTemplates = ["base", "starter", "api", "fullstack", "electron", "react", "ai"];
+  const template = validTemplates.includes(cliOptions.template) ? cliOptions.template : "base";
+  if (cliOptions.template !== template) {
+    console.warn(`\u26A0 \u672A\u77E5\u6A21\u677F ${cliOptions.template}\uFF0C\u4F7F\u7528\u9ED8\u8BA4\u6A21\u677F base`);
+  }
+  const pm = cliOptions.packageManager;
+  const pmValid = pm === "pnpm" || pm === "npm" || pm === "yarn";
+  if (!pmValid) {
+    console.error("\u2716 \u5305\u7BA1\u7406\u5668\u5FC5\u987B\u662F pnpm\u3001npm \u6216 yarn");
+    process.exit(1);
+  }
+  const options = await promptMissing({
+    projectName,
+    template,
+    packageManager: pm,
+    git: cliOptions.git
+  });
+  await createProject(options);
+});
+program.parse();

package/package.json

@@ -0,0 +1,48 @@
+{
+  "name": "@yuer678/create-kb",
+  "version": "0.3.0",
+  "license": "MIT",
+  "author": "anyuer678",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/anyuer678/kb-ui.git",
+    "directory": "packages/create-kb"
+  },
+  "homepage": "https://anyuer678.github.io/kb-ui/",
+  "bugs": {
+    "url": "https://github.com/anyuer678/kb-ui/issues"
+  },
+  "type": "module",
+  "bin": {
+    "create-kb": "./dist/index.js"
+  },
+  "files": [
+    "dist",
+    "templates"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "commander": "^15.0.0",
+    "prompts": "^2.4.2",
+    "kb-ui-vue": "0.4.0"
+  },
+  "devDependencies": {
+    "@types/node": "^26.6.1",
+    "@types/prompts": "^2.4.9",
+    "tsup": "^8.0.0",
+    "typescript": "^5.9.0"
+  },
+  "overrides": {
+    "js-yaml": "^4.1.1",
+    "fast-uri": "^3.1.0",
+    "vite": "^6.3.5"
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "typecheck": "tsc --noEmit",
+    "lint": "eslint src"
+  }
+}

package/templates/ai/.env.example

@@ -0,0 +1,7 @@
+# LLM 配置（复制为 .env 并填写）
+OPENAI_API_KEY=sk-your-key
+# 兼容 OpenAI 协议的任意服务（DeepSeek/Moonshot/Ollama/OpenAI）
+OPENAI_BASE_URL=https://api.openai.com/v1
+# 默认模型
+DEFAULT_MODEL=gpt-4o-mini
+PORT=3000

package/templates/ai/README.md

@@ -0,0 +1,39 @@
+# {{projectName}} · AI 工作台模板
+
+OpenAI 兼容协议的 LLM 接入骨架：非流式 + SSE 流式对话、多模型切换、温度调节、内置聊天前端。
+
+## 快速开始
+
+```bash
+cp .env.example .env    # 填入 OPENAI_API_KEY（可选换 baseURL）
+pnpm install
+pnpm dev                # http://localhost:3000
+```
+
+支持任意 OpenAI 兼容服务：OpenAI / DeepSeek / Moonshot / Ollama（本地）——改 `.env` 的 `OPENAI_BASE_URL` 即可。
+
+## 接口
+
+| 方法 | 路径 | 说明 |
+|------|------|------|
+| GET | / | 内置聊天页面 |
+| POST | /chat | 非流式对话 |
+| POST | /chat/stream | SSE 流式对话 |
+
+## 结构
+
+```
+src/
+├── app.ts          # Express 应用（可测试）
+├── config.ts       # 环境配置 + 启动校验
+├── lib/llm.ts      # OpenAI 客户端单例 + 模型列表
+└── routes/chat.ts  # 对话路由（zod 校验 + 流式）
+public/             # 聊天前端（原生 JS）
+test/               # 校验层单测
+```
+
+## 扩展
+
+- 加角色：messages 支持 system 指令，前端或服务端预设
+- 加 Skill：在 chat 路由前加命令解析（如 /sql），命中则拼入 system
+- 会话持久化：把 history 存入数据库/文件

package/templates/ai/package.json

@@ -0,0 +1,42 @@
+{
+  "name": "{{packageName}}",
+  "private": true,
+  "version": "0.0.0",
+  "type": "module",
+  "engines": {
+    "node": ">=20"
+  },
+  "scripts": {
+    "dev": "tsx watch src/index.ts",
+    "start": "tsx src/index.ts",
+    "test": "vitest run",
+    "typecheck": "tsc --noEmit"
+  },
+  "dependencies": {
+    "express": "^5.1.0",
+    "openai": "^6.0.0",
+    "zod": "^4.0.0"
+  },
+  "devDependencies": {
+    "@types/express": "^5.0.0",
+    "@types/node": "^24.0.0",
+    "supertest": "^7.1.0",
+    "@types/supertest": "^6.0.0",
+    "tsx": "^4.20.0",
+    "typescript": "^5.9.0",
+    "vitest": "^4.1.0"
+  },
+  "overrides": {
+    "js-yaml": "^4.3.2",
+    "fast-uri": "^3.1.0",
+    "vite": "^6.3.5",
+    "electron": "^39.8.10"
+  },
+  "pnpm": {
+    "overrides": {
+      "js-yaml": "^4.1.1",
+      "fast-uri": "^3.1.0",
+      "vite": "^6.3.5"
+    }
+  }
+}

package/templates/ai/public/index.html

@@ -0,0 +1,117 @@
+<!doctype html>
+<html lang="zh-CN">
+<head>
+<meta charset="UTF-8" />
+<title>{{projectName}} · AI 工作台</title>
+<style>
+* { box-sizing: border-box; }
+body { margin: 0; font-family: system-ui, sans-serif; background: #f6f7f9; height: 100vh; display: flex; }
+.sidebar { width: 200px; background: #1e293b; color: #fff; padding: 16px; display: flex; flex-direction: column; gap: 8px; }
+.sidebar h2 { font-size: 15px; margin: 0 0 8px; }
+.sidebar label { font-size: 12px; color: #94a3b8; }
+.sidebar select { width: 100%; padding: 6px; border-radius: 6px; border: none; background: #334155; color: #fff; }
+.chat { flex: 1; display: flex; flex-direction: column; }
+.messages { flex: 1; overflow-y: auto; padding: 16px; display: flex; flex-direction: column; gap: 12px; }
+.msg { max-width: 70%; padding: 10px 14px; border-radius: 12px; font-size: 14px; line-height: 1.6; white-space: pre-wrap; }
+.msg.user { align-self: flex-end; background: #3b82f6; color: #fff; }
+.msg.assistant { align-self: flex-start; background: #fff; border: 1px solid #e2e8f0; }
+.inputbar { display: flex; gap: 8px; padding: 12px 16px; background: #fff; border-top: 1px solid #e2e8f0; }
+.inputbar textarea { flex: 1; padding: 10px; border: 1px solid #cbd5e1; border-radius: 8px; resize: none; font-family: inherit; }
+.inputbar button { padding: 10px 20px; border: none; border-radius: 8px; background: #3b82f6; color: #fff; cursor: pointer; }
+.inputbar button:disabled { opacity: 0.5; }
+</style>
+</head>
+<body>
+<aside class="sidebar">
+  <h2>{{projectName}}</h2>
+  <label>模型</label>
+  <select id="model">
+    <option value="gpt-4o-mini">GPT-4o mini</option>
+    <option value="gpt-4o">GPT-4o</option>
+    <option value="deepseek-chat">DeepSeek Chat</option>
+    <option value="deepseek-reasoner">DeepSeek Reasoner</option>
+    <option value="llama3.1">本地 Ollama</option>
+  </select>
+  <label>温度</label>
+  <input id="temp" type="range" min="0" max="2" step="0.1" value="0.7" />
+</aside>
+<main class="chat">
+  <div class="messages" id="messages">
+    <div class="msg assistant">你好！我是 AI 工作台演示，输入问题开始对话（流式输出）。</div>
+  </div>
+  <div class="inputbar">
+    <textarea id="input" rows="1" placeholder="输入消息…（Enter 发送，Shift+Enter 换行）"></textarea>
+    <button id="send">发送</button>
+  </div>
+</main>
+<script>
+const messagesEl = document.getElementById('messages')
+const inputEl = document.getElementById('input')
+const sendEl = document.getElementById('send')
+const modelEl = document.getElementById('model')
+const tempEl = document.getElementById('temp')
+
+const history = []
+function addMessage(role, content) {
+  const el = document.createElement('div')
+  el.className = `msg ${role}`
+  el.textContent = content
+  messagesEl.appendChild(el)
+  messagesEl.scrollTop = messagesEl.scrollHeight
+  return el
+}
+
+async function send() {
+  const text = inputEl.value.trim()
+  if (!text) return
+  inputEl.value = ''
+  history.push({ role: 'user', content: text })
+  addMessage('user', text)
+  const assistant = addMessage('assistant', '')
+  sendEl.disabled = true
+
+  try {
+    const res = await fetch('/chat/stream', {
+      method: 'POST',
+      headers: { 'content-type': 'application/json' },
+      body: JSON.stringify({ messages: history, model: modelEl.value, temperature: Number(tempEl.value) }),
+    })
+    const reader = res.body.getReader()
+    const decoder = new TextDecoder()
+    let buffer = ''
+    let full = ''
+    while (true) {
+      const { done, value } = await reader.read()
+      if (done) break
+      buffer += decoder.decode(value, { stream: true })
+      const lines = buffer.split('\n\n')
+      buffer = lines.pop() || ''
+      for (const line of lines) {
+        if (!line.startsWith('data: ')) continue
+        const payload = line.slice(6)
+        if (payload === '[DONE]') continue
+        const data = JSON.parse(payload)
+        if (data.error) { assistant.textContent = `⚠ ${data.error}`; continue }
+        full += data.delta
+        assistant.textContent = full
+        messagesEl.scrollTop = messagesEl.scrollHeight
+      }
+    }
+    history.push({ role: 'assistant', content: full })
+  } catch {
+    assistant.textContent = '⚠ 请求失败，请确认后端已启动且 .env 配置正确'
+  } finally {
+    sendEl.disabled = false
+  }
+}
+
+sendEl.addEventListener('click', send)
+inputEl.addEventListener('keydown', (e) => {
+  if (e.key === 'Enter' && !e.shiftKey) {
+    e.preventDefault()
+    send()
+  }
+})
+</script>
+</body>
+</html>

package/templates/ai/src/app.ts

@@ -0,0 +1,22 @@
+import express from 'express'
+import { join, dirname } from 'node:path'
+import { fileURLToPath } from 'node:url'
+import { chatRouter } from './routes/chat'
+import { checkConfig } from './config'
+
+const __dirname = dirname(fileURLToPath(import.meta.url))
+
+export function createApp() {
+  const app = express()
+
+  app.use(express.json())
+
+  app.get('/', (_req, res) => {
+    res.sendFile(join(__dirname, '../public/index.html'))
+  })
+  app.use('/chat', chatRouter)
+
+  return app
+}
+
+checkConfig()

package/templates/ai/src/config.ts

@@ -0,0 +1,13 @@
+/** 环境配置（启动时校验，缺失给出明确提示） */
+export const config = {
+  apiKey: process.env.OPENAI_API_KEY ?? '',
+  baseURL: process.env.OPENAI_BASE_URL ?? 'https://api.openai.com/v1',
+  defaultModel: process.env.DEFAULT_MODEL ?? 'gpt-4o-mini',
+  port: Number(process.env.PORT ?? 3000),
+}
+
+export function checkConfig(): void {
+  if (!config.apiKey) {
+    console.warn('⚠ 未设置 OPENAI_API_KEY（复制 .env.example 为 .env 并填写）')
+  }
+}

package/templates/ai/src/index.ts

@@ -0,0 +1,6 @@
+import { createApp } from './app'
+import { config } from './config'
+
+createApp().listen(config.port, () => {
+  console.log(`AI 工作台已启动: http://localhost:${config.port}`)
+})

package/templates/ai/src/lib/llm.ts

@@ -0,0 +1,24 @@
+import OpenAI from 'openai'
+import { config } from '../config'
+
+let client: OpenAI | null = null
+
+/** 单例 OpenAI 客户端（兼容协议：DeepSeek/Moonshot/Ollama 等） */
+export function getClient(): OpenAI {
+  if (!client) {
+    client = new OpenAI({
+      apiKey: config.apiKey || 'sk-placeholder',
+      baseURL: config.baseURL,
+    })
+  }
+  return client
+}
+
+/** 可选模型列表（按服务能力增减） */
+export const modelOptions = [
+  { label: 'GPT-4o mini', value: 'gpt-4o-mini' },
+  { label: 'GPT-4o', value: 'gpt-4o' },
+  { label: 'DeepSeek Chat', value: 'deepseek-chat' },
+  { label: 'DeepSeek Reasoner', value: 'deepseek-reasoner' },
+  { label: '本地 Ollama', value: 'llama3.1' },
+]

package/templates/ai/src/routes/chat.ts

@@ -0,0 +1,72 @@
+import { Router } from 'express'
+import { z } from 'zod'
+import { getClient } from '../lib/llm'
+import { config } from '../config'
+
+export const chatRouter = Router()
+
+const messageSchema = z.object({
+  role: z.enum(['system', 'user', 'assistant']),
+  content: z.string(),
+})
+
+export const chatSchema = z.object({
+  messages: z.array(messageSchema).min(1),
+  model: z.string().optional(),
+  temperature: z.number().min(0).max(2).optional(),
+})
+
+/** 非流式对话：POST /chat */
+chatRouter.post('/', async (req, res) => {
+  const body = chatSchema.safeParse(req.body)
+  if (!body.success) {
+    res.status(400).json({ code: 400, message: '参数校验失败', errors: body.error.flatten() })
+    return
+  }
+  const { messages, model, temperature } = body.data
+  try {
+    const completion = await getClient().chat.completions.create({
+      model: model ?? config.defaultModel,
+      messages,
+      temperature: temperature ?? 0.7,
+    })
+    res.json({ content: completion.choices[0]?.message?.content ?? '', model: completion.model })
+  } catch (error) {
+    console.error('[chat error]', error)
+    res.status(502).json({ code: 502, message: '调用模型服务失败，请检查 .env 配置' })
+  }
+})
+
+/** 流式对话：POST /chat/stream（SSE） */
+chatRouter.post('/stream', async (req, res) => {
+  const body = chatSchema.safeParse(req.body)
+  if (!body.success) {
+    res.status(400).json({ code: 400, message: '参数校验失败' })
+    return
+  }
+  const { messages, model, temperature } = body.data
+
+  res.setHeader('Content-Type', 'text/event-stream; charset=utf-8')
+  res.setHeader('Cache-Control', 'no-cache')
+  res.setHeader('Connection', 'keep-alive')
+  res.flushHeaders()
+
+  try {
+    const stream = await getClient().chat.completions.create({
+      model: model ?? config.defaultModel,
+      messages,
+      temperature: temperature ?? 0.7,
+      stream: true,
+    })
+    for await (const chunk of stream) {
+      const delta = chunk.choices[0]?.delta?.content
+      if (delta) res.write(`data: ${JSON.stringify({ delta })}\n\n`)
+    }
+    res.write('data: [DONE]\n\n')
+  } catch (error) {
+    console.error('[stream error]', error)
+    res.write(`data: ${JSON.stringify({ error: '调用模型服务失败' })}\n\n`)
+  } finally {
+    res.end()
+  }
+})

package/templates/ai/test/chat.spec.ts

@@ -0,0 +1,27 @@
+import { describe, it, expect } from 'vitest'
+import { chatSchema } from '../src/routes/chat'
+
+// 校验层单测（不触发真实模型调用）
+describe('chatSchema', () => {
+  it('合法消息通过', () => {
+    const result = chatSchema.safeParse({
+      messages: [{ role: 'user', content: '你好' }],
+    })
+    expect(result.success).toBe(true)
+  })
+
+  it('空消息拒绝', () => {
+    const result = chatSchema.safeParse({ messages: [] })
+    expect(result.success).toBe(false)
+  })
+
+  it('非法角色拒绝', () => {
+    const result = chatSchema.safeParse({ messages: [{ role: 'robot', content: 'hi' }] })
+    expect(result.success).toBe(false)
+  })
+
+  it('temperature 越界拒绝', () => {
+    const result = chatSchema.safeParse({ messages: [{ role: 'user', content: 'x' }], temperature: 5 })
+    expect(result.success).toBe(false)
+  })
+})

package/templates/ai/tsconfig.json

@@ -0,0 +1,13 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "ESNext",
+    "moduleResolution": "bundler",
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "types": ["node"],
+    "noEmit": true
+  },
+  "include": ["src", "test"]
+}

package/templates/ai/vitest.config.ts

@@ -0,0 +1,8 @@
+import { defineConfig } from 'vitest/config'
+
+export default defineConfig({
+  test: {
+    environment: 'node',
+    include: ['test/**/*.spec.ts'],
+  },
+})