115 lines
4.2 KiB
TypeScript
115 lines
4.2 KiB
TypeScript
import {
|
|
abortEmbeddedPiRun,
|
|
compactEmbeddedPiSession,
|
|
isEmbeddedPiRunActive,
|
|
waitForEmbeddedPiRunEnd,
|
|
} from "../../agents/pi-embedded.js";
|
|
import type { ClawdbotConfig } from "../../config/config.js";
|
|
import { resolveSessionFilePath } from "../../config/sessions.js";
|
|
import { logVerbose } from "../../globals.js";
|
|
import { enqueueSystemEvent } from "../../infra/system-events.js";
|
|
import { formatContextUsageShort, formatTokenCount } from "../status.js";
|
|
import type { CommandHandler } from "./commands-types.js";
|
|
import { stripMentions, stripStructuralPrefixes } from "./mentions.js";
|
|
import { incrementCompactionCount } from "./session-updates.js";
|
|
|
|
function extractCompactInstructions(params: {
|
|
rawBody?: string;
|
|
ctx: import("../templating.js").MsgContext;
|
|
cfg: ClawdbotConfig;
|
|
agentId?: string;
|
|
isGroup: boolean;
|
|
}): string | undefined {
|
|
const raw = stripStructuralPrefixes(params.rawBody ?? "");
|
|
const stripped = params.isGroup
|
|
? stripMentions(raw, params.ctx, params.cfg, params.agentId)
|
|
: raw;
|
|
const trimmed = stripped.trim();
|
|
if (!trimmed) return undefined;
|
|
const lowered = trimmed.toLowerCase();
|
|
const prefix = lowered.startsWith("/compact") ? "/compact" : null;
|
|
if (!prefix) return undefined;
|
|
let rest = trimmed.slice(prefix.length).trimStart();
|
|
if (rest.startsWith(":")) rest = rest.slice(1).trimStart();
|
|
return rest.length ? rest : undefined;
|
|
}
|
|
|
|
export const handleCompactCommand: CommandHandler = async (params) => {
|
|
const compactRequested =
|
|
params.command.commandBodyNormalized === "/compact" ||
|
|
params.command.commandBodyNormalized.startsWith("/compact ");
|
|
if (!compactRequested) return null;
|
|
if (!params.command.isAuthorizedSender) {
|
|
logVerbose(
|
|
`Ignoring /compact from unauthorized sender: ${params.command.senderId || "<unknown>"}`,
|
|
);
|
|
return { shouldContinue: false };
|
|
}
|
|
if (!params.sessionEntry?.sessionId) {
|
|
return {
|
|
shouldContinue: false,
|
|
reply: { text: "⚙️ Compaction unavailable (missing session id)." },
|
|
};
|
|
}
|
|
const sessionId = params.sessionEntry.sessionId;
|
|
if (isEmbeddedPiRunActive(sessionId)) {
|
|
abortEmbeddedPiRun(sessionId);
|
|
await waitForEmbeddedPiRunEnd(sessionId, 15_000);
|
|
}
|
|
const customInstructions = extractCompactInstructions({
|
|
rawBody: params.ctx.CommandBody ?? params.ctx.RawBody ?? params.ctx.Body,
|
|
ctx: params.ctx,
|
|
cfg: params.cfg,
|
|
agentId: params.agentId,
|
|
isGroup: params.isGroup,
|
|
});
|
|
const result = await compactEmbeddedPiSession({
|
|
sessionId,
|
|
sessionKey: params.sessionKey,
|
|
messageChannel: params.command.channel,
|
|
sessionFile: resolveSessionFilePath(sessionId, params.sessionEntry),
|
|
workspaceDir: params.workspaceDir,
|
|
config: params.cfg,
|
|
skillsSnapshot: params.sessionEntry.skillsSnapshot,
|
|
provider: params.provider,
|
|
model: params.model,
|
|
thinkLevel: params.resolvedThinkLevel ?? (await params.resolveDefaultThinkingLevel()),
|
|
bashElevated: {
|
|
enabled: false,
|
|
allowed: false,
|
|
defaultLevel: "off",
|
|
},
|
|
customInstructions,
|
|
ownerNumbers: params.command.ownerList.length > 0 ? params.command.ownerList : undefined,
|
|
});
|
|
|
|
const totalTokens =
|
|
params.sessionEntry.totalTokens ??
|
|
(params.sessionEntry.inputTokens ?? 0) + (params.sessionEntry.outputTokens ?? 0);
|
|
const contextSummary = formatContextUsageShort(
|
|
totalTokens > 0 ? totalTokens : null,
|
|
params.contextTokens ?? params.sessionEntry.contextTokens ?? null,
|
|
);
|
|
const compactLabel = result.ok
|
|
? result.compacted
|
|
? result.result?.tokensBefore
|
|
? `Compacted (${formatTokenCount(result.result.tokensBefore)} before)`
|
|
: "Compacted"
|
|
: "Compaction skipped"
|
|
: "Compaction failed";
|
|
if (result.ok && result.compacted) {
|
|
await incrementCompactionCount({
|
|
sessionEntry: params.sessionEntry,
|
|
sessionStore: params.sessionStore,
|
|
sessionKey: params.sessionKey,
|
|
storePath: params.storePath,
|
|
});
|
|
}
|
|
const reason = result.reason?.trim();
|
|
const line = reason
|
|
? `${compactLabel}: ${reason} • ${contextSummary}`
|
|
: `${compactLabel} • ${contextSummary}`;
|
|
enqueueSystemEvent(line, { sessionKey: params.sessionKey });
|
|
return { shouldContinue: false, reply: { text: `⚙️ ${line}` } };
|
|
};
|