create-interview-cockpit 0.3.0 → 0.4.0

package/template/server/src/index.ts

@@ -447,6 +447,37 @@ app.delete(
   },
 );
 
+// ─── AI Settings ────────────────────────────────────────────
+
+app.get("/api/settings", async (_req, res) => {
+  const settings = await storage.getAiSettings();
+  res.json(settings);
+});
+
+app.patch("/api/settings", async (req, res) => {
+  const current = await storage.getAiSettings();
+  const updated: storage.AiSettings = {
+    systemPrompt:
+      typeof req.body.systemPrompt === "string"
+        ? req.body.systemPrompt
+        : current.systemPrompt,
+    responseProfiles:
+      req.body.responseProfiles != null
+        ? req.body.responseProfiles
+        : current.responseProfiles,
+    vizGuide:
+      typeof req.body.vizGuide === "string"
+        ? req.body.vizGuide
+        : current.vizGuide,
+    promptGroups:
+      req.body.promptGroups != null
+        ? req.body.promptGroups
+        : current.promptGroups,
+  };
+  await storage.saveAiSettings(updated);
+  res.json(updated);
+});
+
 // ─── Chat ────────────────────────────────────────────────
 
 app.post("/api/chat", async (req, res) => {
@@ -462,38 +493,12 @@ app.post("/api/chat", async (req, res) => {
     responseLength,
   } = req.body;
 
-  const responseProfiles: Record<
-    string,
-    { maxOutputTokens: number; maxSteps: number }
-  > = {
-    concise: {
-      maxOutputTokens: 1000,
-      maxSteps: 3,
-    },
-    moderate: {
-      maxOutputTokens: 1000,
-      maxSteps: 5,
-    },
-    normal: {
-      maxOutputTokens: 3000,
-      maxSteps: 5,
-    },
-  };
-  const selectedResponseProfile =
-    responseProfiles[responseLength] || responseProfiles.normal;
-
-  let system = `You are a senior engineering interview coach.
+  const aiSettings = await storage.getAiSettings();
+  const { responseProfiles, vizGuide } = aiSettings;
+  const selectedResponseProfile = responseProfiles[responseLength] ??
+    responseProfiles["normal"] ?? { maxOutputTokens: 3000, maxSteps: 5 };
 
-Highest priority: follow the user's explicit response preferences and current conversation context. If they conflict with your default teaching behavior, the user's preference wins.
-Explain clearly, accurately, and practically.
-Only include Mermaid diagrams, code blocks, or tables when the user explicitly asks for them or when they materially improve the answer.
-If you show code, use a fenced code block with the correct language.
-
-Mermaid syntax rules (follow strictly):
-- Wrap node labels in quotes when they contain special characters: A["Microservice A (Producer)"]
-- Edge labels use |text| syntax: A -->|sends message| B
-- Never put parentheses or brackets inside [] without quoting the label
-- Use simple node IDs (letters/numbers) and put descriptive text in the label`;
+  let system = aiSettings.systemPrompt;
 
   if (topicTitle || questionTitle) {
     system += `\n\n--- Current Context ---`;
@@ -625,8 +630,14 @@ Only use coderef:// for [code] files. Do not use it for uploaded documents.`;
       }),
       system,
       messages: modelMessages,
-      tools:
-        fileRegistry.size > 0
+      tools: {
+        getVizGuide: tool({
+          description:
+            "Get the full viz diagram spec reference. Call this before writing a ```viz block so you have the correct schema, rules, and examples.",
+          inputSchema: z.object({}),
+          execute: async () => ({ guide: vizGuide }),
+        }),
+        ...(fileRegistry.size > 0
           ? {
               readFile: tool({
                 description:
@@ -650,7 +661,8 @@ Only use coderef:// for [code] files. Do not use it for uploaded documents.`;
                 },
               }),
             }
-          : undefined,
+          : {}),
+      },
       stopWhen: stepCountIs(selectedResponseProfile.maxSteps),
     });
 
@@ -666,6 +678,7 @@ Only use coderef:// for [code] files. Do not use it for uploaded documents.`;
             id: message.id || randomUUID(),
             role: message.role,
             content: getTextFromUIMessage(message),
+            ...(Array.isArray(message.parts) ? { parts: message.parts } : {}),
           }));
 
           await storage.updateQuestionMessages(questionId, normalized);
@@ -899,6 +912,43 @@ Their question: "${prompt}"`,
   }
 });
 
+// ─── Fix Viz ────────────────────────────────────────────
+
+app.post("/api/fix-viz", async (req, res) => {
+  const { spec, error: renderError } = req.body;
+  if (typeof spec !== "string" || !spec.trim()) {
+    return res.status(400).json({ error: "spec is required" });
+  }
+
+  try {
+    const isGoogle = ["google", "gemini"].includes(
+      (process.env.AI_PROVIDER || "openai").toLowerCase(),
+    );
+
+    const { text } = await generateText({
+      model: getModel(),
+      maxOutputTokens: 1200,
+      ...(isGoogle && {
+        providerOptions: {
+          google: { thinkingConfig: { thinkingBudget: 0 } },
+        },
+      }),
+      prompt: `The following viz diagram spec failed to render.${renderError ? `\n\nRender error:\n${renderError}` : ""}\n\nFix the spec so it renders correctly. Common issues to check:\n- x/y coordinates must be numbers, not strings\n- node ids must be kebab-case with no spaces\n- chain arrays in autoSignals/signals must reference valid node ids\n- YAML flow sequences ([a, b]) must be correctly indented and closed\n- Do NOT mix autoSignals and steps in the same spec\n- YAML string quoting: if a label or value contains special characters (hyphens like --, colons, brackets, slashes, or leading/trailing spaces), wrap the ENTIRE value in double quotes. Do NOT use single quotes — single-quoted YAML strings must be self-contained; trailing words after the closing quote cause parse errors. Example of wrong: label: 'update-index --skip-worktree' changes file metadata  Example of correct: label: "update-index --skip-worktree changes file metadata"\n- Never split a label across multiple YAML lines unless using a literal block scalar (| or >)\n\nReturn ONLY the corrected YAML (or JSON) spec with no explanation and no markdown fences — just the raw spec.\n\nFailed spec:\n${spec}`,
+    });
+
+    // Strip any fences the model might have added
+    const fixed = text
+      .replace(/^```(?:yaml|json|viz)?\s*/i, "")
+      .replace(/```\s*$/, "")
+      .trim();
+
+    res.json({ spec: fixed });
+  } catch (err: any) {
+    console.error("fix-viz error:", err?.message || err);
+    res.status(500).json({ error: err?.message || "Failed to fix viz" });
+  }
+});
+
 // ─── Fix Diagram ────────────────────────────────────────
 
 app.post("/api/fix-diagram", async (req, res) => {

package/template/server/src/storage.ts

@@ -57,6 +57,7 @@ export interface Message {
   id: string;
   role: string;
   content: string;
+  parts?: any[];
   createdAt?: string;
 }
 
@@ -601,3 +602,98 @@ export async function updateQuestionMessages(
   q.messages = messages;
   await saveQuestion(q);
 }
+
+// ── AI Settings ───────────────────────────────────────────────
+
+const AI_SETTINGS_FILE = path.join(DATA_DIR, "ai-settings.json");
+
+export interface ResponseProfile {
+  maxOutputTokens: number;
+  maxSteps: number;
+}
+
+export interface PromptGroup {
+  label: string;
+  description?: string;
+  default: string;
+  options: Record<string, string>;
+}
+
+export interface AiSettings {
+  /** Base system prompt sent to the AI on every chat turn. */
+  systemPrompt: string;
+  /** Per-length-preference token and step limits. */
+  responseProfiles: Record<string, ResponseProfile>;
+  /** Full viz diagram spec reference — returned by the getVizGuide tool. */
+  vizGuide: string;
+  /** All user-selectable prompt groups. Add new entries here to extend the UI. */
+  promptGroups: Record<string, PromptGroup>;
+}
+
+const DEFAULT_AI_SETTINGS: AiSettings = {
+  systemPrompt:
+    "You are a senior engineering interview coach.\n\nExplain clearly, accurately, and practically.",
+  responseProfiles: {
+    concise: { maxOutputTokens: 1000, maxSteps: 3 },
+    moderate: { maxOutputTokens: 1000, maxSteps: 5 },
+    normal: { maxOutputTokens: 3000, maxSteps: 5 },
+  },
+  vizGuide: "No viz guide configured.",
+  promptGroups: {
+    length: {
+      label: "Response Length",
+      description:
+        "Appended to the user message when the selected length changes.",
+      default: "normal",
+      options: {
+        concise: "Keep the response concise.",
+        moderate: "Keep the response moderately detailed.",
+        normal:
+          "Use a fuller answer with enough context to explain the idea clearly.",
+      },
+    },
+    style: {
+      label: "Response Style",
+      description:
+        "Appended to the user message when the selected style changes.",
+      default: "prose",
+      options: {
+        prose: "Use natural prose with short paragraphs.",
+        bullets: "Use bullet points and short lists as the main format.",
+        structured:
+          "Use structured sections with headings and numbered steps when helpful.",
+      },
+    },
+    audience: {
+      label: "Response Audience",
+      description:
+        "Appended to the user message when the selected audience changes.",
+      default: "normal",
+      options: {
+        normal: "",
+        beginner:
+          "When using technical terms or abbreviations, immediately expand their meaning in square brackets.",
+      },
+    },
+  },
+};
+
+export async function getAiSettings(): Promise<AiSettings> {
+  try {
+    const raw = await fs.readFile(AI_SETTINGS_FILE, "utf-8");
+    return {
+      ...DEFAULT_AI_SETTINGS,
+      ...(JSON.parse(raw) as Partial<AiSettings>),
+    };
+  } catch {
+    return DEFAULT_AI_SETTINGS;
+  }
+}
+
+export async function saveAiSettings(settings: AiSettings): Promise<void> {
+  await fs.writeFile(
+    AI_SETTINGS_FILE,
+    JSON.stringify(settings, null, 2),
+    "utf-8",
+  );
+}