File size: 3,450 Bytes
eb3f11e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | /** Diagnostic helpers for embedded-agent compaction. */
import type { ApiRegistry } from "@openclaw/ai";
import type { OpenClawConfig } from "../../config/types.openclaw.js";
import { generateSecureToken } from "../../infra/secure-random.js";
import type { ProviderRuntimeModel } from "../../plugins/provider-runtime-model.types.js";
import { isRealConversationMessage } from "../compaction-real-conversation.js";
import { registerProviderStreamForModel } from "../provider-stream.js";
import type { AgentMessage } from "../runtime/index.js";
import { estimateTokens } from "../sessions/index.js";
import type { CompactionMessageMetrics } from "./compact.types.js";
export function createDirectCompactionDiagId(): string {
return `cmp-${Date.now().toString(36)}-${generateSecureToken(4)}`;
}
export function resolveCompactionProviderStream(params: {
effectiveModel: ProviderRuntimeModel;
config?: OpenClawConfig;
agentDir: string;
effectiveWorkspace: string;
apiRegistry: ApiRegistry;
}) {
return registerProviderStreamForModel({
model: params.effectiveModel,
cfg: params.config,
agentDir: params.agentDir,
workspaceDir: params.effectiveWorkspace,
apiRegistry: params.apiRegistry,
});
}
export function normalizeObservedTokenCount(value: unknown): number | undefined {
return typeof value === "number" && Number.isFinite(value) && value > 0
? Math.floor(value)
: undefined;
}
function getMessageTextChars(msg: AgentMessage): number {
const content = (msg as { content?: unknown }).content;
if (typeof content === "string") {
return content.length;
}
return Array.isArray(content)
? content.reduce((total, block) => {
const text =
block && typeof block === "object" ? (block as { text?: unknown }).text : undefined;
return total + (typeof text === "string" ? text.length : 0);
}, 0)
: 0;
}
function resolveMessageToolLabel(msg: AgentMessage): string | undefined {
const candidate =
(msg as { toolName?: unknown }).toolName ??
(msg as { name?: unknown }).name ??
(msg as { tool?: unknown }).tool;
return typeof candidate === "string" && candidate.trim().length > 0 ? candidate : undefined;
}
export function summarizeCompactionMessages(messages: AgentMessage[]): CompactionMessageMetrics {
let historyTextChars = 0;
let toolResultChars = 0;
const contributors: Array<{ role: string; chars: number; tool?: string }> = [];
let estTokens = 0;
let tokenEstimationFailed = false;
for (const msg of messages) {
const role = typeof msg.role === "string" ? msg.role : "unknown";
const chars = getMessageTextChars(msg);
historyTextChars += chars;
if (role === "toolResult") {
toolResultChars += chars;
}
contributors.push({ role, chars, tool: resolveMessageToolLabel(msg) });
if (!tokenEstimationFailed) {
try {
estTokens += estimateTokens(msg);
} catch {
tokenEstimationFailed = true;
}
}
}
return {
messages: messages.length,
historyTextChars,
toolResultChars,
estTokens: tokenEstimationFailed ? undefined : estTokens,
contributors: contributors.toSorted((left, right) => right.chars - left.chars).slice(0, 3),
};
}
export function containsRealConversationMessages(messages: AgentMessage[]): boolean {
return messages.some((message, index, allMessages) =>
isRealConversationMessage(message, allMessages, index),
);
}
|