@prometheus-ai/agent 0.5.4 → 0.5.8

package/src/cli/gallery-fixtures/agentic.ts

@@ -0,0 +1,407 @@
+// Gallery fixtures for the agentic orchestration tools (task, irc, goal, job).
+import type { Usage } from "@prometheus-ai/ai";
+import type { TaskToolDetails } from "../../task/types";
+import type { IrcDetails } from "../../tools/irc";
+import type { GalleryFixture } from "./types";
+
+/** Message/activity timestamps are offsets from load time so gallery ages stay plausible. */
+const FIXTURE_NOW = Date.now();
+
+/** Plausible cumulative usage for a fixture subagent run. */
+const fixtureUsage = (tokens: { input: number; output: number }, costTotal: number): Usage => ({
+	input: tokens.input,
+	output: tokens.output,
+	cacheRead: 0,
+	cacheWrite: 0,
+	totalTokens: tokens.input + tokens.output,
+	cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: costTotal },
+});
+
+export const agenticFixtures: Record<string, GalleryFixture> = {
+	task: {
+		label: "Task",
+		customRendered: true,
+		// Streaming: agent chosen, assignment still landing.
+		streamingArgs: {
+			agent: "task",
+			id: "AuthLoader",
+			description: "Load auth middleware",
+			assignment: "Read packages/server/src/auth/*.ts and summarize the session-cookie",
+		},
+		args: {
+			agent: "task",
+			id: "AuthLoader",
+			description: "Load auth middleware",
+			assignment:
+				"Read packages/server/src/auth/session.ts and middleware.ts, then document the session-cookie validation flow and any TODOs.",
+		},
+		result: {
+			content: [
+				{
+					type: "text",
+					text: "Agent AuthLoader completed.",
+				},
+			],
+			details: {
+				projectAgentsDir: null,
+				totalDurationMs: 48_200,
+				usage: fixtureUsage({ input: 52_600, output: 8_800 }, 0.12),
+				progress: [
+					{
+						index: 0,
+						id: "AuthLoader",
+						agent: "task",
+						agentSource: "bundled",
+						status: "completed",
+						task: "Read packages/server/src/auth/session.ts and middleware.ts",
+						description: "Load auth middleware",
+						lastIntent: "Documenting session-cookie flow",
+						recentTools: [
+							{ tool: "read", args: "packages/server/src/auth/session.ts", endMs: 1_749_200_040_000 },
+							{ tool: "read", args: "packages/server/src/auth/middleware.ts", endMs: 1_749_200_052_000 },
+						],
+						recentOutput: ["Session validation runs in middleware.ts:42 via verifySessionCookie()."],
+						toolCount: 9,
+						requests: 6,
+						tokens: 61_400,
+						contextTokens: 23_100,
+						contextWindow: 200_000,
+						cost: 0.12,
+						durationMs: 41_900,
+						resolvedModel: "anthropic/claude-sonnet",
+					},
+				],
+				results: [
+					{
+						index: 0,
+						id: "AuthLoader",
+						agent: "task",
+						agentSource: "bundled",
+						description: "Load auth middleware",
+						task: "Read packages/server/src/auth/session.ts and middleware.ts",
+						assignment:
+							"Read packages/server/src/auth/session.ts and middleware.ts, then document the session-cookie validation flow and any TODOs.",
+						exitCode: 0,
+						output: [
+							"Session validation runs in middleware.ts:42 via verifySessionCookie().",
+							"Cookies are HMAC-signed (SHA-256) and checked against the session store.",
+							"TODO at session.ts:88 — sliding-expiration refresh is stubbed.",
+						].join("\n"),
+						stderr: "",
+						truncated: false,
+						durationMs: 41_900,
+						tokens: 61_400,
+						requests: 6,
+						contextTokens: 23_100,
+						contextWindow: 200_000,
+						resolvedModel: "anthropic/claude-sonnet",
+						usage: fixtureUsage({ input: 52_600, output: 8_800 }, 0.12),
+						outputMeta: { lineCount: 3, charCount: 214 },
+					},
+				],
+			} satisfies TaskToolDetails,
+		},
+		errorResult: {
+			isError: true,
+			content: [
+				{
+					type: "text",
+					text: "Agent RateLimiter failed.",
+				},
+			],
+			details: {
+				projectAgentsDir: null,
+				totalDurationMs: 9_800,
+				usage: fixtureUsage({ input: 10_900, output: 1_400 }, 0.1),
+				results: [
+					{
+						index: 0,
+						id: "RateLimiter",
+						agent: "task",
+						agentSource: "bundled",
+						description: "Audit rate limiter",
+						task: "Inspect packages/server/src/auth/rate-limit.ts",
+						assignment:
+							"Inspect packages/server/src/auth/rate-limit.ts. Confirm the 429 path sets Retry-After and report gaps.",
+						exitCode: 1,
+						output: "",
+						stderr: "ENOENT: packages/server/src/auth/rate-limit.ts",
+						truncated: false,
+						durationMs: 9_800,
+						tokens: 12_300,
+						requests: 3,
+						contextTokens: 6_400,
+						contextWindow: 200_000,
+						resolvedModel: "anthropic/claude-sonnet",
+						usage: fixtureUsage({ input: 10_900, output: 1_400 }, 0.1),
+						error: "Subagent exited 1: target file packages/server/src/auth/rate-limit.ts does not exist.",
+						outputMeta: { lineCount: 0, charCount: 0 },
+					},
+				],
+			} satisfies TaskToolDetails,
+		},
+	},
+
+	irc: {
+		label: "IRC",
+		// Streaming: recipient known; the message body still arriving.
+		streamingArgs: { op: "send", to: "AuthLoader", message: "Are you still touching" },
+		args: {
+			op: "send",
+			to: "AuthLoader",
+			message: "Are you still touching src/server/auth.ts? I need to add a 401 path.",
+			await: true,
+		},
+		result: {
+			content: [
+				{
+					type: "text",
+					text: [
+						"Delivered to 1 peer(s):",
+						"- AuthLoader: revived",
+						"",
+						"Reply from AuthLoader:",
+						"Done with auth.ts — go ahead, just rebase past my session-store rename.",
+					].join("\n"),
+				},
+			],
+			details: {
+				op: "send",
+				from: "Main",
+				to: "AuthLoader",
+				receipts: [{ to: "AuthLoader", outcome: "revived" }],
+				waited: {
+					id: "7181122334455667789",
+					from: "AuthLoader",
+					to: "Main",
+					body: "Done with auth.ts — go ahead, just rebase past my session-store rename.",
+					ts: FIXTURE_NOW - 5_000,
+					replyTo: "7181122334455667788",
+				},
+			} satisfies IrcDetails,
+		},
+		errorResult: {
+			isError: true,
+			content: [
+				{
+					type: "text",
+					text: 'No recipients received the message.\n- RateLimiter: failed — unknown agent "RateLimiter"',
+				},
+			],
+			details: {
+				op: "send",
+				from: "Main",
+				to: "RateLimiter",
+				receipts: [{ to: "RateLimiter", outcome: "failed", error: 'unknown agent "RateLimiter"' }],
+			} satisfies IrcDetails,
+		},
+	},
+
+	irc_wait: {
+		label: "IRC (wait)",
+		customRendered: true,
+		renderer: "irc",
+		streamingArgs: { op: "wait", from: "AuthLoader" },
+		args: { op: "wait", from: "AuthLoader", timeoutMs: 60_000 },
+		result: {
+			content: [
+				{
+					type: "text",
+					text: "[7181122334455667790] AuthLoader: session-store rename is merged; auth.ts is yours.",
+				},
+			],
+			details: {
+				op: "wait",
+				from: "Main",
+				waited: {
+					id: "7181122334455667790",
+					from: "AuthLoader",
+					to: "Main",
+					body: "session-store rename is merged; auth.ts is yours.",
+					ts: FIXTURE_NOW - 30_000,
+				},
+			} satisfies IrcDetails,
+		},
+	},
+
+	irc_inbox: {
+		label: "IRC (inbox)",
+		customRendered: true,
+		renderer: "irc",
+		streamingArgs: { op: "inbox" },
+		args: { op: "inbox", peek: true },
+		result: {
+			content: [
+				{
+					type: "text",
+					text: [
+						"2 unread message(s):",
+						"- [7181122334455667791] AuthLoader: hub table reads unreadCount — ping me when the bus lands.",
+						"- [7181122334455667792] RateLimiter (reply to 7181122334455667791): bus is in; receipts carry outcome.",
+					].join("\n"),
+				},
+			],
+			details: {
+				op: "inbox",
+				from: "Main",
+				inbox: [
+					{
+						id: "7181122334455667791",
+						from: "AuthLoader",
+						to: "Main",
+						body: "hub table reads unreadCount — ping me when the bus lands.",
+						ts: FIXTURE_NOW - 4 * 60_000,
+					},
+					{
+						id: "7181122334455667792",
+						from: "RateLimiter",
+						to: "Main",
+						body: "bus is in; receipts carry outcome.",
+						ts: FIXTURE_NOW - 60_000,
+						replyTo: "7181122334455667791",
+					},
+				],
+			} satisfies IrcDetails,
+		},
+	},
+
+	irc_list: {
+		label: "IRC (list)",
+		customRendered: true,
+		renderer: "irc",
+		streamingArgs: { op: "list" },
+		args: { op: "list" },
+		result: {
+			content: [
+				{
+					type: "text",
+					text: [
+						"2 peer(s):",
+						"- AuthLoader [task · sub · idle] — parent Main, active 2m ago",
+						"- RateLimiter [task · sub · parked] — unread 2, parent Main, active 12m ago",
+						"",
+						"Parked agents are revived automatically when you message them.",
+					].join("\n"),
+				},
+			],
+			details: {
+				op: "list",
+				from: "Main",
+				peers: [
+					{
+						id: "AuthLoader",
+						displayName: "task",
+						kind: "sub",
+						status: "idle",
+						parentId: "Main",
+						unread: 0,
+						lastActivity: FIXTURE_NOW - 2 * 60_000,
+					},
+					{
+						id: "RateLimiter",
+						displayName: "task",
+						kind: "sub",
+						status: "parked",
+						parentId: "Main",
+						unread: 2,
+						lastActivity: FIXTURE_NOW - 12 * 60_000,
+					},
+				],
+			} satisfies IrcDetails,
+		},
+	},
+
+	goal: {
+		label: "Goal",
+		// Streaming: op is "create"; objective text still being typed.
+		streamingArgs: { op: "create", objective: "Ship the auth hardening" },
+		args: {
+			op: "create",
+			objective: "Ship the auth hardening pass: per-account rate limits and sliding session expiry.",
+			token_budget: 500_000,
+		},
+		result: {
+			content: [
+				{
+					type: "text",
+					text: "Goal set. Working toward: Ship the auth hardening pass.",
+				},
+			],
+			details: {
+				op: "create",
+				remainingTokens: 451_800,
+				completionBudgetReport: null,
+				goal: {
+					id: "goal_8f2a",
+					objective: "Ship the auth hardening pass: per-account rate limits and sliding session expiry.",
+					status: "active",
+					tokenBudget: 500_000,
+					tokensUsed: 48_200,
+					timeUsedSeconds: 312,
+					createdAt: 1_749_200_000_000,
+					updatedAt: 1_749_200_312_000,
+				},
+			},
+		},
+		errorResult: {
+			isError: true,
+			content: [{ type: "text", text: "Goal tool failed: objective is required when op=create." }],
+			details: { op: "create" },
+		},
+	},
+
+	job: {
+		label: "Job",
+		// Streaming: polling a single job id; the second id is still arriving.
+		streamingArgs: { poll: ["job_a1"] },
+		args: { poll: ["job_a1", "job_b2", "job_c3"] },
+		result: {
+			content: [{ type: "text", text: "3 jobs settled." }],
+			details: {
+				jobs: [
+					{
+						id: "job_a1",
+						type: "bash",
+						status: "completed",
+						label: "bun test packages/server/test/auth.test.ts",
+						durationMs: 18_400,
+						resultText: "42 pass, 0 fail (18.4s)",
+					},
+					{
+						id: "job_b2",
+						type: "task",
+						status: "completed",
+						label: "Migrate rate limiter to a sliding window",
+						durationMs: 96_700,
+						resultText: "Rewrote rate-limit.ts to a token-bucket; added per-account keys.",
+					},
+					{
+						id: "job_c3",
+						type: "bash",
+						status: "failed",
+						label: "bunx biome check packages/server/src/auth",
+						durationMs: 4_100,
+						errorText: "biome: 2 errors in tokens.ts — noUnusedVariables, useConst",
+					},
+				],
+			},
+		},
+		errorResult: {
+			isError: true,
+			content: [{ type: "text", text: "Job cancelled by user." }],
+			details: {
+				jobs: [
+					{
+						id: "job_d4",
+						type: "task",
+						status: "cancelled",
+						label: "Refactor the session store to Redis",
+						durationMs: 52_300,
+						errorText: "Aborted: superseded by goal re-scope.",
+					},
+				],
+				cancelled: [{ id: "job_d4", status: "cancelled" }],
+			},
+		},
+	},
+};

package/src/cli/gallery-fixtures/codeintel.ts

@@ -0,0 +1,187 @@
+/** Gallery fixtures for the code-intelligence tools (lsp, debug). */
+import type { GalleryFixture } from "./types";
+
+export const codeintelFixtures: Record<string, GalleryFixture> = {
+	lsp: {
+		label: "LSP",
+		streamingArgs: {
+			action: "references",
+			file: "src/server/auth.ts",
+		},
+		args: {
+			action: "references",
+			file: "src/server/auth.ts",
+			line: 42,
+			symbol: "validateToken",
+		},
+		result: {
+			content: [
+				{
+					type: "text",
+					text: [
+						"Found 6 reference(s):",
+						"  src/server/auth.ts:42:14",
+						"    41: ",
+						"    42: export function validateToken(token: string): Claims {",
+						"    43:   const claims = verifyJwt(token);",
+						"  src/server/auth.ts:118:21",
+						'    117:   if (!header) throw new HttpError(401, "missing token");',
+						"    118:   const claims = validateToken(stripBearer(header));",
+						"    119:   return claims.sub;",
+						"  src/server/middleware/session.ts:57:18",
+						"    56:   const token = req.cookies.session;",
+						"    57:   const claims = validateToken(token);",
+						"    58:   req.userId = claims.sub;",
+						"  src/server/router.ts:153:20",
+						"    152: router.use(async (req, res, next) => {",
+						"    153:   req.claims = await validateToken(req.token);",
+						"    154:   next();",
+						"  test/auth.test.ts:24:9",
+						'    23: it("rejects expired tokens", () => {',
+						"    24:   expect(() => validateToken(expired)).toThrow(/expired/);",
+						"    25: });",
+						"  test/auth.test.ts:41:9",
+						'    40: it("accepts valid tokens", () => {',
+						"    41:   const claims = validateToken(signed);",
+						'    42:   expect(claims.sub).toBe("u_123");',
+					].join("\n"),
+				},
+			],
+			details: {
+				serverName: "typescript-language-server",
+				action: "references",
+				success: true,
+				request: {
+					action: "references",
+					file: "src/server/auth.ts",
+					line: 42,
+					symbol: "validateToken",
+				},
+			},
+		},
+		errorResult: {
+			content: [
+				{
+					type: "text",
+					text: "No language server found for this file",
+				},
+			],
+			isError: true,
+			details: {
+				serverName: "typescript-language-server",
+				action: "references",
+				success: false,
+				request: {
+					action: "references",
+					file: "src/server/auth.ts",
+					line: 42,
+					symbol: "validateToken",
+				},
+			},
+		},
+	},
+
+	debug: {
+		label: "Debug",
+		streamingArgs: {
+			action: "stack_trace",
+		},
+		args: {
+			action: "stack_trace",
+			levels: 20,
+		},
+		result: {
+			content: [
+				{
+					type: "text",
+					text: [
+						"Stack trace:",
+						"- #1000 validate_token @ app/server.py:42:14",
+						"- #1001 authenticate @ app/server.py:88:9",
+						"- #1002 handle_request @ app/router.py:153:20",
+						"- #1003 dispatch @ app/router.py:97:5",
+						"- #1004 <module> @ app/server.py:212:1",
+					].join("\n"),
+				},
+			],
+			details: {
+				action: "stack_trace",
+				success: true,
+				snapshot: {
+					id: "dbg-1",
+					adapter: "debugpy",
+					cwd: "/Users/dev/project",
+					program: "./app/server.py",
+					status: "stopped",
+					launchedAt: "2026-06-06T14:21:08.412Z",
+					lastUsedAt: "2026-06-06T14:22:55.901Z",
+					threadId: 1,
+					frameId: 1000,
+					stopReason: "breakpoint",
+					stopDescription: "breakpoint 2",
+					frameName: "validate_token",
+					instructionPointerReference: "0x00000001000034a8",
+					source: { name: "server.py", path: "app/server.py" },
+					line: 42,
+					column: 14,
+					breakpointFiles: 1,
+					breakpointCount: 2,
+					functionBreakpointCount: 0,
+					outputBytes: 248,
+					outputTruncated: false,
+					needsConfigurationDone: false,
+				},
+				stackFrames: [
+					{
+						id: 1000,
+						name: "validate_token",
+						source: { name: "server.py", path: "app/server.py" },
+						line: 42,
+						column: 14,
+					},
+					{
+						id: 1001,
+						name: "authenticate",
+						source: { name: "server.py", path: "app/server.py" },
+						line: 88,
+						column: 9,
+					},
+					{
+						id: 1002,
+						name: "handle_request",
+						source: { name: "router.py", path: "app/router.py" },
+						line: 153,
+						column: 20,
+					},
+					{
+						id: 1003,
+						name: "dispatch",
+						source: { name: "router.py", path: "app/router.py" },
+						line: 97,
+						column: 5,
+					},
+					{
+						id: 1004,
+						name: "<module>",
+						source: { name: "server.py", path: "app/server.py" },
+						line: 212,
+						column: 1,
+					},
+				],
+			},
+		},
+		errorResult: {
+			content: [
+				{
+					type: "text",
+					text: "No active debug session. Launch or attach first.",
+				},
+			],
+			isError: true,
+			details: {
+				action: "stack_trace",
+				success: false,
+			},
+		},
+	},
+};