taodeng/memory-continuity
1
0
Fork 0
mirror of https://github.com/dtzp555-max/memory-continuity.git synced 2026-07-19 09:42:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: session log token estimation handles array content, fix TOCTOU race

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
taodeng
2026-03-31 14:31:18 +10:00
co-authored by Claude Sonnet 4.6
parent 9d4968c44b
commit a0cd296ec7
1 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
index.js
+11 -10
View File
@@ -210,9 +210,13 @@ function writeSessionLog(workspaceDir, messages, config = {}) {
// Count messages by role // Count messages by role
const userCount = messages.filter(m => m?.role === "user").length; const userCount = messages.filter(m => m?.role === "user").length;
const assistantCount = messages.filter(m => m?.role === "assistant").length; const assistantCount = messages.filter(m => m?.role === "assistant").length;
const totalTokens = estimateTokens( const contentToString = (m) => {
messages.map(m => typeof m?.content === "string" ? m.content : "").join("") if (typeof m?.content === "string") return m.content;
); if (Array.isArray(m?.content))
return m.content.filter(b => b?.type === "text").map(b => b.text).join(" ");
return "";
};
const totalTokens = estimateTokens(messages.map(contentToString).join(" "));
// Build log entry // Build log entry
const entry = [ const entry = [
@@ -223,13 +227,10 @@ function writeSessionLog(workspaceDir, messages, config = {}) {
"", "",
].join("\n"); ].join("\n");
// Append to daily log (create with header if new) // Append-safe: use appendFileSync for both new and existing files
if (!fs.existsSync(logFile)) { const isNew = !fs.existsSync(logFile);
const header = `# Session Log — ${dateStr}\n\n`; const prefix = isNew ? `# Session Log — ${dateStr}\n\n` : "";
fs.writeFileSync(logFile, header + entry, "utf8"); fs.appendFileSync(logFile, prefix + entry, "utf8");
} else {
fs.appendFileSync(logFile, entry, "utf8");
}
} }
function extractStateFromMessages(messages) { function extractStateFromMessages(messages) {