fix(FN-2349): migrate agent-generation diagnostics to shared helper

- Replace raw console diagnostics in agent-generation with structured ai-session diagnostics events
- Add a test-only cleanup hook and regression tests for cleanup telemetry and AI generation failure logging
- Expand diagnostics guardrail coverage to include agent-generation and generalize guardrail naming to AI-session modules
This commit is contained in:
Fusion
2026-04-23 14:32:49 -07:00
committed by gsxdsm
parent eefc6fb1d5
commit dbb3d9911f
3 changed files with 118 additions and 15 deletions

View File

@@ -8,9 +8,16 @@ import {
getRateLimitResetTime,
parseGenerationResponse,
__resetAgentGenerationState,
__runAgentGenerationCleanupForTests,
RateLimitError,
SessionNotFoundError,
} from "./agent-generation.js";
import {
setDiagnosticsSink,
resetDiagnosticsSink,
type DiagnosticsContext,
type DiagnosticsLevel,
} from "./ai-session-diagnostics.js";
// Counter for unique IPs per test
let ipCounter = 0;
@@ -18,6 +25,13 @@ function getUniqueIp(): string {
return `127.0.0.${++ipCounter}`;
}
interface CapturedDiagnostic {
level: DiagnosticsLevel;
scope: string;
message: string;
context: DiagnosticsContext;
}
describe("agent-generation module", () => {
beforeEach(() => {
vi.useFakeTimers();
@@ -25,7 +39,9 @@ describe("agent-generation module", () => {
});
afterEach(() => {
resetDiagnosticsSink();
vi.useRealTimers();
vi.restoreAllMocks();
});
describe("startAgentGeneration", () => {
@@ -101,6 +117,45 @@ describe("agent-generation module", () => {
generateAgentSpec("non-existent-session-id", "/tmp")
).rejects.toThrow(SessionNotFoundError);
});
it("logs structured error diagnostics with sessionId and rethrows when AI generation fails", async () => {
vi.resetModules();
const generationFailure = new Error("engine failed to generate spec");
vi.doMock("@fusion/engine", () => ({
createFnAgent: vi.fn(async () => {
throw generationFailure;
}),
}));
const diagnostics: CapturedDiagnostic[] = [];
const diagnosticsModule = await import("./ai-session-diagnostics.js");
diagnosticsModule.setDiagnosticsSink((level, scope, message, context) => {
diagnostics.push({ level, scope, message, context });
});
const agentGenerationModule = await import("./agent-generation.js");
const session = await agentGenerationModule.startAgentGeneration(getUniqueIp(), "Role requiring AI spec");
await expect(agentGenerationModule.generateAgentSpec(session.id, "/tmp")).rejects.toThrow(generationFailure);
expect(diagnostics).toHaveLength(1);
expect(diagnostics[0]).toMatchObject({
level: "error",
scope: "agent-generation",
message: "AI generation failed for session",
context: expect.objectContaining({
sessionId: session.id,
operation: "generate-agent-spec",
error: expect.objectContaining({
message: "engine failed to generate spec",
}),
}),
});
diagnosticsModule.resetDiagnosticsSink();
});
});
describe("getAgentGenerationSession", () => {
@@ -334,6 +389,36 @@ describe("agent-generation module", () => {
expect(retrieved).toBeDefined();
expect(retrieved!.roleDescription).toBe("Security auditor role");
});
it("emits one structured cleanup diagnostic when expired sessions and stale rate limits are removed", async () => {
const diagnostics: CapturedDiagnostic[] = [];
setDiagnosticsSink((level, scope, message, context) => {
diagnostics.push({ level, scope, message, context });
});
const staleIp = getUniqueIp();
const session = await startAgentGeneration(staleIp, "Temporary role");
checkRateLimit("192.168.0.99");
vi.setSystemTime(Date.now() + 61 * 60 * 1000);
__runAgentGenerationCleanupForTests();
expect(getAgentGenerationSession(session.id)).toBeUndefined();
expect(getRateLimitResetTime(staleIp)).toBeNull();
expect(getRateLimitResetTime("192.168.0.99")).toBeNull();
expect(diagnostics).toHaveLength(1);
expect(diagnostics[0]).toMatchObject({
level: "info",
scope: "agent-generation",
message: "Cleanup completed",
context: expect.objectContaining({
cleanedSessions: 1,
cleanedRateLimits: 2,
operation: "cleanup-expired",
}),
});
});
});
describe("prompt override support", () => {