root/fusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(FN-4400): complete Step 5/6 wiring trims and prompt-size logging

Browse Source 
Fusion-Task-Id: FN-4400
Fusion-Task-Lineage: 0d4f9c48-5b8b-49eb-9cf3-d1d585ffe7fc
This commit is contained in:
Fusion
2026-05-14 03:49:49 -07:00
committed by gsxdsm
parent e77cd39b67
commit 7329c7179a
1 changed files with 20 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
packages/engine/src/agent-heartbeat.ts
View File

@@ -30,9 +30,9 @@ import {
buildPluginPromptSection,
resolveAgentHeartbeatProcedure,
} from "./agent-instructions.js";
import { resolveHeartbeatScopeDisciplineMode, selectHeartbeatProcedure } from "./heartbeat-procedure-resolver.js";
import { resolveHeartbeatPromptTemplate, resolveHeartbeatScopeDisciplineMode, selectHeartbeatProcedure } from "./heartbeat-procedure-resolver.js";
import { buildPromptLayers, collapsePromptLayers } from "./prompt-layers.js";
import { heartbeatLog, formatError } from "./logger.js";
import { createLogger, heartbeatLog, formatError } from "./logger.js";
import { acquireTaskWorktree } from "./worktree-acquisition.js";
import { createRunAuditor, type EngineRunContext } from "./run-audit.js";
import { promptWithFallback } from "./pi.js";
@@ -40,6 +40,9 @@ import { createResolvedAgentSession, extractRuntimeHint, resolveHeartbeatSession
import type { AgentActionGateContext } from "./agent-action-gate.js";
import { buildSessionSkillContextSync } from "./session-skill-context.js";
import type { AgentReflectionService } from "./agent-reflection.js";
import { trimPromptMd, trimTaskDescription, trimTriggeringComments } from "./heartbeat-prompt-trim.js";
const promptSizeLog = createLogger("prompt-size");
function adjustHeartbeatMemoryPrimer(basePrompt: string, mode: AgentMemoryInclusionMode): string {
if (mode === "full") return basePrompt;
@@ -2377,6 +2380,7 @@ export class HeartbeatMonitor {
);
}
const heartbeatScopeDiscipline = resolveHeartbeatScopeDisciplineMode(heartbeatModelSettings, agent);
const promptTemplate = resolveHeartbeatPromptTemplate(heartbeatModelSettings, agent);
const resolvedProcedureTemplate = selectHeartbeatProcedure(heartbeatScopeDiscipline, isNoTaskRun, {
task: {
strict: HEARTBEAT_PROCEDURE_STRICT,
@@ -2576,10 +2580,10 @@ export class HeartbeatMonitor {
heartbeatProcedureText,
"",
"Task description:",
taskDetail!.description,
trimTaskDescription(taskDetail!.description, promptTemplate),
"",
taskDetail!.prompt ? `PROMPT.md:\n${taskDetail!.prompt}` : "No PROMPT.md available.",
...triggeringCommentLines,
taskDetail!.prompt ? `PROMPT.md:\n${trimPromptMd(taskDetail!.prompt, promptTemplate)}` : "No PROMPT.md available.",
...trimTriggeringComments(triggeringCommentLines, promptTemplate),
...pendingMessagesLines,
...pendingRoomMessagesLines,
...(reportsHealthSection ? ["", reportsHealthSection] : []),
@@ -2599,8 +2603,19 @@ export class HeartbeatMonitor {
contextSnapshot: {
...(run.contextSnapshot ?? {}),
heartbeatScopeDiscipline,
heartbeatPromptTemplate: promptTemplate,
},
};
promptSizeLog.log("prompt-size", {
agentId: agent.id,
role: agent.role,
runId: run.id,
template: promptTemplate,
systemChars: systemPromptFinal.length,
execChars: executionPrompt.length,
totalChars: systemPromptFinal.length + executionPrompt.length,
isNoTaskRun,
});
await this.store.saveRun(runWithPrompts);
// Update local run reference so completeRun merges correctly
Object.assign(run, {