feat(agent): return diagnostics from resource loaders

packages/agent/test/harness/prompt-templates.test.ts

@@ -2,7 +2,11 @@ import { symlink } from "node:fs/promises";
 import { join } from "node:path";
 import { describe, expect, it } from "vitest";
 import { NodeExecutionEnv } from "../../src/harness/execution-env.js";
-import { expandPromptTemplate, loadPromptTemplates } from "../../src/harness/prompt-templates.js";
+import {
+	expandPromptTemplate,
+	loadPromptTemplates,
+	loadSourcedPromptTemplates,
+} from "../../src/harness/prompt-templates.js";
 import { createTempDir } from "./session-test-utils.js";
 
 describe("loadPromptTemplates", () => {
@@ -15,21 +19,61 @@ describe("loadPromptTemplates", () => {
 		await env.writeFile("a/nested/ignored.md", "Ignored");
 		await env.writeFile("b/two.md", "First line description\nBody");
 
-		const templates = await loadPromptTemplates(env, ["a", "b"]);
+		const { promptTemplates, diagnostics } = await loadPromptTemplates(env, ["a", "b"]);
 
-		expect(templates).toEqual([
+		expect(diagnostics).toEqual([]);
+		expect(promptTemplates).toEqual([
 			{ name: "one", description: "One template", content: "Hello $1" },
 			{ name: "two", description: "First line description", content: "First line description\nBody" },
 		]);
 	});
 
+	it("preserves source info for sourced prompt templates", async () => {
+		const root = createTempDir();
+		const env = new NodeExecutionEnv({ cwd: root });
+		await env.createDir("prompts", { recursive: true });
+		await env.writeFile("prompts/example.md", "---\ndescription: Example\n---\nExample body");
+
+		const { promptTemplates, diagnostics } = await loadSourcedPromptTemplates(env, [
+			{ path: "prompts", source: { type: "project" as const } },
+		]);
+
+		expect(diagnostics).toEqual([]);
+		expect(promptTemplates).toEqual([
+			{
+				promptTemplate: { name: "example", description: "Example", content: "Example body" },
+				source: { type: "project" },
+			},
+		]);
+	});
+
+	it("attaches source info to diagnostics", async () => {
+		const root = createTempDir();
+		const env = new NodeExecutionEnv({ cwd: root });
+		await env.writeFile("broken.md", "---\ndescription: [unterminated\n---\nBody");
+
+		const { promptTemplates, diagnostics } = await loadSourcedPromptTemplates(env, [
+			{ path: "broken.md", source: { type: "user" as const } },
+		]);
+
+		expect(promptTemplates).toEqual([]);
+		expect(diagnostics).toHaveLength(1);
+		expect(diagnostics[0]).toMatchObject({
+			type: "warning",
+			path: join(root, "broken.md"),
+			source: { type: "user" },
+		});
+	});
+
 	it("loads explicit markdown files and symlinked files", async () => {
 		const root = createTempDir();
 		const env = new NodeExecutionEnv({ cwd: root });
 		await env.writeFile("target.md", "---\ndescription: Target\n---\nTarget body");
 		await symlink(join(root, "target.md"), join(root, "link.md"));
 
-		expect(await loadPromptTemplates(env, ["target.md", "link.md"])).toEqual([
+		const { promptTemplates } = await loadPromptTemplates(env, ["target.md", "link.md"]);
+
+		expect(promptTemplates).toEqual([
 			{ name: "target", description: "Target", content: "Target body" },
 			{ name: "link", description: "Target", content: "Target body" },
 		]);