feat(backend): ThtRunner.ollamaEnsure (parses tht ollama ensure --json)

2026-06-29 19:31:49 +02:00
parent deb8081bc2
commit a0af089d00
2 changed files with 49 additions and 0 deletions
@@ -114,3 +114,28 @@ test("documents parses the JSON array", async () => {
  const docs = await r.documents("sid");
  expect(docs[0].key).toBe("question");
 });
+
+test("ollamaEnsure builds argv with --json --timeout and the workspace -c", async () => {
+  let calledArgs: string[] = [];
+  let calledWs: string | undefined;
+  const r = new ThtRunner({ thtBin: "tht", harnessDir: "/nope", configPath: "config/tht.yaml" });
+  r.run = async (args, ws) => { calledArgs = args; calledWs = ws; return { code: 0, stdout: '{"ok":true,"server":"up","model":"warmed","model_name":"m"}', stderr: "" }; };
+  const res = await r.ollamaEnsure("psd", 60);
+  expect(calledArgs).toEqual(["ollama", "ensure", "--json", "--timeout", "60"]);
+  expect(calledWs).toBe("psd");
+  expect(res).toEqual({ ok: true, server: "up", model: "warmed", model_name: "m" });
+});
+
+test("ollamaEnsure maps a non-zero exit to ok:false with stage/error from stdout JSON", async () => {
+  const r = new ThtRunner({ thtBin: "tht", harnessDir: "/nope", configPath: "config/tht.yaml" });
+  r.run = async () => ({ code: 1, stdout: '{"ok":false,"stage":"model","error":"missing"}', stderr: "" });
+  expect(await r.ollamaEnsure("psd", 60)).toEqual({ ok: false, stage: "model", error: "missing" });
+});
+
+test("ollamaEnsure falls back to stderr when stdout is not JSON on failure", async () => {
+  const r = new ThtRunner({ thtBin: "tht", harnessDir: "/nope", configPath: "config/tht.yaml" });
+  r.run = async () => ({ code: 1, stdout: "", stderr: "boom" });
+  const res = await r.ollamaEnsure("psd", 60);
+  expect(res.ok).toBe(false);
+  expect(res.error).toContain("boom");
+});