dsh-prompt-refine 0.1.0

package/lib/index.mjs

@@ -0,0 +1,272 @@
+import { readFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import { createAssistantMessage, createUserMessage } from "@deepseek-ai/dsh-llm";
+//#region src/shared/prompt.ts
+/** 优化器 system prompt —— 关键是要 LLM 返回可被 JSON.parse 的结构 */
+const SYSTEM_PROMPT = `你是一个提示词优化助手。用户给你:
+1. 一段最近的对话历史(可能为空)
+2. 用户即将发送的最新草稿
+
+你的任务:分析草稿的不足,给出恰好 3 条改进建议。每条建议 = 一个「问题标签」+「问题描述」+「具体补丁」。
+
+要求:
+- 只输出合法 JSON,不要任何额外文字、不要 markdown 代码块标记
+- patch 必须是可以直接追加到原草稿末尾或合适位置的一句话,长度 5-50 字
+- 优先指出:范围不清、缺少输出格式、缺少约束条件、模糊指代
+- 不要重写整个草稿,只补漏
+- 如果草稿已经足够好,issue 里如实写「无需修改」,patch 留空
+
+输出 schema(严格遵守):
+{"suggestions": [{"tag": "scope|format|constraint|clarity|other", "issue": "一句话问题", "patch": "一句话补丁"}, ...共 3 条]}
+`;
+/** 用户消息模板:把 history + draft 装进 user message */
+function buildUserMessage(history, draft) {
+	const lines = [];
+	if (history.length > 0) {
+		lines.push("【最近对话历史】");
+		for (const m of history) lines.push(`${m.role === "user" ? "用户" : "助手"}: ${m.content}`);
+		lines.push("");
+	}
+	lines.push("【用户当前草稿】");
+	lines.push(draft);
+	lines.push("");
+	lines.push("请按 system 要求输出 JSON。");
+	return lines.join("\n");
+}
+/**
+* 从 LLM 输出里提取 JSON。
+* 策略:
+*   1. 去掉可能的 markdown ```json``` 包裹
+*   2. 找到第一个 { 到最后一个 } 的子串,parse
+*   3. parse 失败抛错(调用方会回退)
+*/
+function parseSuggestions(raw) {
+	let s = raw.trim();
+	s = s.replace(/^```(?:json)?\s*/i, "").replace(/\s*```$/, "");
+	const first = s.indexOf("{");
+	const last = s.lastIndexOf("}");
+	if (first < 0 || last <= first) throw new Error("LLM 输出不含 JSON 对象");
+	const json = s.slice(first, last + 1);
+	const arr = JSON.parse(json).suggestions;
+	if (!Array.isArray(arr) || arr.length === 0) throw new Error("JSON 里没有 suggestions 数组");
+	return arr.slice(0, 3).map((raw, i) => {
+		const x = raw && typeof raw === "object" ? raw : {};
+		return {
+			tag: typeof x.tag === "string" ? x.tag : "other",
+			issue: typeof x.issue === "string" ? x.issue : `建议 #${i + 1}`,
+			patch: typeof x.patch === "string" ? x.patch : ""
+		};
+	});
+}
+//#endregion
+//#region src/host.ts
+const ROUTE_PREFIX = "/dsh-prompt-refine";
+/** 读请求体 */
+function readBody(req) {
+	return new Promise((resolve, reject) => {
+		const chunks = [];
+		req.on("data", (c) => chunks.push(c));
+		req.on("end", () => resolve(Buffer.concat(chunks).toString("utf8")));
+		req.on("error", reject);
+	});
+}
+/** 写 JSON 响应 */
+function sendJson(res, status, body) {
+	res.writeHead(status, { "Content-Type": "application/json; charset=utf-8" });
+	res.end(JSON.stringify(body));
+}
+/** 拿到 llm 服务(懒注入:即使它晚于 webServer 就绪也能等到) */
+function resolveLlm(ctx) {
+	return new Promise((resolve, reject) => {
+		try {
+			ctx.inject(["llm"], (llmCtx) => {
+				const llm = llmCtx.llm;
+				if (!llm) {
+					reject(/* @__PURE__ */ new Error("ctx.llm 不存在"));
+					return;
+				}
+				resolve(llm);
+			});
+		} catch (err) {
+			reject(err);
+		}
+	});
+}
+/**
+* 读用户 settings.yaml 里的 agent-default-model,拿到默认 provider/model。
+* 用逐行扫描而不是 YAML 库:结构固定,避免引入依赖。
+*/
+async function resolveDefaultModel() {
+	const fallback = {
+		provider: "deepseek-official",
+		model: "deepseek-v4-flash"
+	};
+	try {
+		const home = process.env.DSH_HOME ?? join(homedir(), ".dsh");
+		const lines = (await readFile(join(home, "settings.yaml"), "utf8")).split(/\r?\n/);
+		let i = lines.findIndex((l) => /^agent-default-model\s*:/.test(l));
+		if (i < 0) return fallback;
+		let provider = "";
+		let model = "";
+		for (i += 1; i < lines.length; i++) {
+			const line = lines[i];
+			if (/^\S/.test(line)) break;
+			const pm = line.match(/^\s+provider\s*:\s*(\S+)/);
+			const mm = line.match(/^\s+model\s*:\s*(\S+)/);
+			if (pm) provider = pm[1];
+			if (mm) model = mm[1];
+		}
+		if (provider && model) return {
+			provider,
+			model
+		};
+		return fallback;
+	} catch {
+		return fallback;
+	}
+}
+/**
+* 调一次 LLM,拼出完整文本。
+* Message 用官方工厂函数构造。
+*/
+async function callLlm(ctx, req) {
+	const llm = await resolveLlm(ctx);
+	const historyMessages = req.history.map((m) => m.role === "user" ? createUserMessage({
+		content: [{
+			type: "text",
+			text: m.content
+		}],
+		source: { kind: "user" }
+	}) : createAssistantMessage({
+		content: [{
+			type: "text",
+			text: m.content
+		}],
+		source: {
+			provider: req.provider,
+			model: req.model
+		}
+	}));
+	const draftMessage = createUserMessage({
+		content: [{
+			type: "text",
+			text: buildUserMessage([], req.draft)
+		}],
+		source: { kind: "user" }
+	});
+	const stream = llm.stream({
+		provider: req.provider,
+		model: req.model,
+		system: SYSTEM_PROMPT,
+		messages: [...historyMessages, draftMessage]
+	});
+	let out = "";
+	let lastError = "";
+	for await (const chunk of stream) if (chunk.type === "text-delta" && typeof chunk.text === "string") out += chunk.text;
+	else if (chunk.type === "finish") {
+		const reason = chunk.reason;
+		if (reason?.kind === "error") lastError = reason.failure?.message ?? "未知模型错误";
+	}
+	if (!out && lastError) throw new Error(`模型调用失败: ${lastError}`);
+	return out;
+}
+/** 兜底建议 —— LLM 挂掉或解析失败时给用户一个能用的结果 */
+function fallbackSuggestions() {
+	return [
+		{
+			tag: "clarity",
+			issue: "请补充更多背景信息(语言、目标、约束)",
+			patch: "请补充:编程语言 / 目标读者 / 期望长度"
+		},
+		{
+			tag: "format",
+			issue: "请指定输出格式(代码 / 表格 / 列表 / 段落)",
+			patch: "请用 Markdown 格式返回"
+		},
+		{
+			tag: "constraint",
+			issue: "请说明边界条件或偏好",
+			patch: "请不要使用外部依赖"
+		}
+	];
+}
+/** 注册 HTTP 路由 */
+function registerHttpRoutes(ctx) {
+	const webServer = ctx.webServer;
+	if (!webServer) {
+		console.log("[dsh-prompt-refine] webServer 服务不可用,HTTP 端点未注册");
+		return;
+	}
+	webServer.register({
+		kind: "prefix",
+		path: ROUTE_PREFIX,
+		handler: async (req, res) => {
+			const url = (req.url ?? "/").split("?")[0];
+			const method = req.method ?? "GET";
+			if (url !== "/dsh-prompt-refine/suggest" || method !== "POST") {
+				sendJson(res, 404, {
+					ok: false,
+					error: `not found: ${method} ${url}`
+				});
+				return;
+			}
+			try {
+				const body = await readBody(req);
+				const parsed = JSON.parse(body);
+				const draft = typeof parsed.draft === "string" ? parsed.draft.trim() : "";
+				const history = Array.isArray(parsed.history) ? parsed.history.filter((m) => !!m && typeof m.content === "string" && (m.role === "user" || m.role === "assistant")) : [];
+				if (!draft) {
+					sendJson(res, 400, {
+						ok: false,
+						error: "draft 不能为空"
+					});
+					return;
+				}
+				const resolved = typeof parsed.provider === "string" && typeof parsed.model === "string" && parsed.provider && parsed.model ? {
+					provider: parsed.provider,
+					model: parsed.model
+				} : await resolveDefaultModel();
+				console.log(`[dsh-prompt-refine] 建议请求: draft=${draft.length}字, history=${history.length}条, model=${resolved.provider}/${resolved.model}`);
+				try {
+					sendJson(res, 200, {
+						ok: true,
+						suggestions: parseSuggestions(await callLlm(ctx, {
+							draft,
+							history,
+							...resolved
+						})),
+						usedModel: `${resolved.provider}/${resolved.model}`
+					});
+				} catch (err) {
+					const msg = err instanceof Error ? err.message : String(err);
+					console.log(`[dsh-prompt-refine] LLM 调用或解析失败: ${msg}`);
+					sendJson(res, 200, {
+						ok: true,
+						suggestions: fallbackSuggestions(),
+						error: "模型调用失败,已用兜底建议: " + msg,
+						usedModel: `${resolved.provider}/${resolved.model}`
+					});
+				}
+			} catch (err) {
+				sendJson(res, 500, {
+					ok: false,
+					error: "服务器错误: " + (err instanceof Error ? err.message : String(err))
+				});
+			}
+		}
+	});
+	console.log("[dsh-prompt-refine] HTTP 端点已注册: POST " + ROUTE_PREFIX + "/suggest");
+}
+//#endregion
+//#region src/dsh-prompt-refine.ts
+const name = "dsh-prompt-refine";
+const inject = ["webServer"];
+function apply(ctx) {
+	registerHttpRoutes(ctx);
+	ctx.inject(["llm"], (llmCtx) => {
+		console.log("[dsh-prompt-refine] llm 服务已就绪,可以处理优化请求");
+	});
+}
+//#endregion
+export { apply, inject, name };

package/package.json

@@ -0,0 +1,57 @@
+{
+  "name": "dsh-prompt-refine",
+  "version": "0.1.0",
+  "type": "module",
+  "main": "lib/index.mjs",
+  "description": "DSH 提示词优化插件:在发送前拦截用户提示词,结合上下文给出 3 条改进建议",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/sojo-negai/dsh-prompt-refine.git"
+  },
+  "keywords": [
+    "dsh",
+    "dsh-plugin",
+    "deepseek-harness",
+    "prompt",
+    "prompt-engineering"
+  ],
+  "exports": {
+    ".": {
+      "default": "./lib/index.mjs"
+    },
+    "./client": {
+      "default": "./lib/client.iife.js"
+    },
+    "./src/*": "./src/*",
+    "./cordis.patch.yml": "./cordis.patch.yml",
+    "./package.json": "./package.json"
+  },
+  "files": [
+    "lib",
+    "src",
+    "scripts",
+    "cordis.patch.yml",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsdown && tsdown --config ./tsdown.client.config.ts",
+    "prepack": "npm run build"
+  },
+  "dsh": {
+    "bundle": {
+      "patch": "./cordis.patch.yml"
+    },
+    "client": {
+      "platform": "web"
+    }
+  },
+  "peerDependencies": {
+    "@deepseek-ai/dsh-llm": ">=0.1.0-0 <0.1.1-0 || >=0.1.1-0 <0.1.2-0 || >=0.1.2-0 <0.2.0-0"
+  },
+  "devDependencies": {
+    "tsdown": "^0.23.0",
+    "typescript": "^5.9.2"
+  }
+}

package/scripts/install.mjs

@@ -0,0 +1,97 @@
+// scripts/install.mjs —— 把 dsh-prompt-refine 装进指定 DSH profile
+//
+// 用法:
+//   node scripts/install.mjs --profile web         # 装到单个 profile
+//   node scripts/install.mjs --all                 # 装到所有 profile
+//   node scripts/install.mjs --profile web --remove  # 从 web 卸载
+
+import { readFileSync, writeFileSync, existsSync } from 'node:fs'
+import { dirname, join } from 'node:path'
+import { fileURLToPath } from 'node:url'
+
+const __dirname = dirname(fileURLToPath(import.meta.url))
+const PLUGIN_PATH = dirname(__dirname)
+
+const PROFILE_DIR = process.env.DSH_HOME ?? join(process.env.USERPROFILE ?? process.env.HOME, '.dsh')
+const ALL_PROFILES = ['desktop', 'web', 'open-design']
+
+function parseArgs(argv) {
+  const args = { profile: null, all: false, remove: false, help: false }
+  for (let i = 2; i < argv.length; i++) {
+    const a = argv[i]
+    if (a === '--profile') args.profile = argv[++i]
+    else if (a === '--all') args.all = true
+    else if (a === '--remove') args.remove = true
+    else if (a === '--help' || a === '-h') args.help = true
+  }
+  return args
+}
+
+function usage() {
+  console.log(`用法:
+  node scripts/install.mjs --profile <name>            安装到指定 profile
+  node scripts/install.mjs --all                        安装到所有 profile
+  node scripts/install.mjs --profile <name> --remove   从指定 profile 卸载
+
+可用 profile: ${ALL_PROFILES.join(', ')}`)
+}
+
+function listProfiles() {
+  return ALL_PROFILES.filter((n) => existsSync(join(PROFILE_DIR, 'profiles', n, 'package.json')))
+}
+
+function applyProfile(profileName, remove) {
+  const pkgPath = join(PROFILE_DIR, 'profiles', profileName, 'package.json')
+  const pkg = JSON.parse(readFileSync(pkgPath, 'utf8'))
+  const deps = pkg.dependencies ?? {}
+  const dshProfile = pkg.dsh?.profile ?? {}
+  const bundles = dshProfile.bundles ?? []
+
+  if (remove) {
+    let changed = false
+    if (deps['dsh-prompt-refine']) {
+      delete deps['dsh-prompt-refine']
+      changed = true
+    }
+    const idx = bundles.indexOf('dsh-prompt-refine')
+    if (idx >= 0) {
+      bundles.splice(idx, 1)
+      changed = true
+    }
+    if (changed) {
+      pkg.dependencies = deps
+      pkg.dsh = pkg.dsh ?? {}
+      pkg.dsh.profile = dshProfile
+      pkg.dsh.profile.bundles = bundles
+      writeFileSync(pkgPath, JSON.stringify(pkg, null, 2) + '\n', 'utf8')
+      console.log(`  ✓ 已从 ${profileName} 卸载`)
+    } else {
+      console.log(`  - ${profileName} 本来就没装`)
+    }
+  } else {
+    deps['dsh-prompt-refine'] = `file:${PLUGIN_PATH}`
+    if (!bundles.includes('dsh-prompt-refine')) bundles.push('dsh-prompt-refine')
+    pkg.dependencies = deps
+    pkg.dsh = pkg.dsh ?? {}
+    pkg.dsh.profile = dshProfile
+    pkg.dsh.profile.bundles = bundles
+    writeFileSync(pkgPath, JSON.stringify(pkg, null, 2) + '\n', 'utf8')
+    console.log(`  ✓ 已装到 ${profileName}`)
+  }
+}
+
+const args = parseArgs(process.argv)
+if (args.help || (!args.profile && !args.all)) {
+  usage()
+  process.exit(args.help ? 0 : 1)
+}
+
+const targets = args.all ? listProfiles() : [args.profile]
+for (const name of targets) {
+  if (!existsSync(join(PROFILE_DIR, 'profiles', name))) {
+    console.warn(`! ${name} 不存在,跳过`)
+    continue
+  }
+  applyProfile(name, args.remove)
+}
+console.log('完成。')