mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-09 03:07:46 +08:00
dayuan.jiang
d32eb25523
- The three XML examples (swimlanes, two edges, waypoints) that the web system prompt, the display_diagram description and the MCP drawing guide each had a copy of now live in packages/mcp-server/src/xml-examples.ts - The shape library group list and reader come from the MCP shape-library.ts; getShapeLibrary takes the folder, and the web route passes docs/shape-libraries under the working directory as before. Only the 30 known library names are read. Error texts differ slightly - next.config.ts traces docs/shape-libraries/*.md for /api/chat, since the read now happens in another module - Every prompt is byte-for-byte unchanged: the web system prompt for five model ids with and without minimal style, and the MCP drawing guide, compared before and after; the library list and swimlane example match the old tool description text exactly
74 lines
2.3 KiB
TypeScript
74 lines
2.3 KiB
TypeScript
/**
|
|
* Shape and icon library docs (docs/shape-libraries/*.md), the same files
|
|
* the web app's get_shape_library tool reads (app/api/chat/route.ts).
|
|
*/
|
|
|
|
import { readFile } from "node:fs/promises"
|
|
import { dirname, join, resolve } from "node:path"
|
|
import { fileURLToPath } from "node:url"
|
|
|
|
export const SHAPE_LIBRARY_GROUPS: Record<string, string[]> = {
|
|
Cloud: [
|
|
"aws4",
|
|
"azure2",
|
|
"gcp2",
|
|
"alibaba_cloud",
|
|
"openstack",
|
|
"salesforce",
|
|
],
|
|
Networking: ["cisco19", "network", "kubernetes", "vvd", "rack"],
|
|
Business: ["bpmn", "lean_mapping"],
|
|
General: ["flowchart", "basic", "arrows2", "infographic", "sitemap"],
|
|
"UI/Mockups": ["android", "material_design"],
|
|
Enterprise: ["citrix", "sap", "mscae", "atlassian"],
|
|
Engineering: ["fluidpower", "electrical", "pid", "cabinets", "floorplan"],
|
|
Icons: ["webicons"],
|
|
}
|
|
|
|
/** The libraries by group, one "- Group: a, b" line each, for tool descriptions */
|
|
export const SHAPE_LIBRARY_LIST = Object.entries(SHAPE_LIBRARY_GROUPS)
|
|
.map(([group, names]) => `- ${group}: ${names.join(", ")}`)
|
|
.join("\n")
|
|
|
|
const LIBRARIES = new Set(Object.values(SHAPE_LIBRARY_GROUPS).flat())
|
|
|
|
/**
|
|
* The build copies the docs to dist/shape-libraries; running from src (tsx)
|
|
* reads them from the repository instead. The web app passes its own folder.
|
|
*/
|
|
function libraryDirs(dir?: string): string[] {
|
|
if (dir) return [dir]
|
|
const here = dirname(fileURLToPath(import.meta.url))
|
|
return [
|
|
join(here, "shape-libraries"),
|
|
resolve(here, "../../../docs/shape-libraries"),
|
|
]
|
|
}
|
|
|
|
export async function getShapeLibrary(
|
|
name: string,
|
|
dir?: string,
|
|
): Promise<{ ok: true; text: string } | { ok: false; error: string }> {
|
|
const library = name.trim().toLowerCase()
|
|
if (!LIBRARIES.has(library)) {
|
|
return {
|
|
ok: false,
|
|
error: `Library "${name}" not found. Available: ${Array.from(LIBRARIES).join(", ")}`,
|
|
}
|
|
}
|
|
for (const folder of libraryDirs(dir)) {
|
|
try {
|
|
return {
|
|
ok: true,
|
|
text: await readFile(join(folder, `${library}.md`), "utf-8"),
|
|
}
|
|
} catch {
|
|
// Try the next location
|
|
}
|
|
}
|
|
return {
|
|
ok: false,
|
|
error: `Library "${library}" is missing from this installation.`,
|
|
}
|
|
}
|