mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-17 05:00:16 +00:00
mom: sync user messages from log.jsonl to context.jsonl before each agent run
- Ensures channel chatter and backfilled messages are in context - Syncs on session creation and before each subsequent run - User messages formatted as '[username]: text' in context
This commit is contained in:
parent
3f6db8e99c
commit
ca23ade9f8
2 changed files with 103 additions and 6 deletions
|
|
@ -254,8 +254,8 @@ function formatToolArgsForSlack(_toolName: string, args: Record<string, unknown>
|
||||||
return lines.join("\n");
|
return lines.join("\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
// Cache for AgentSession per channel
|
// Cache for AgentSession and SessionManager per channel
|
||||||
const channelSessions = new Map<string, AgentSession>();
|
const channelSessions = new Map<string, { session: AgentSession; sessionManager: MomSessionManager }>();
|
||||||
|
|
||||||
export function createAgentRunner(sandboxConfig: SandboxConfig): AgentRunner {
|
export function createAgentRunner(sandboxConfig: SandboxConfig): AgentRunner {
|
||||||
let currentSession: AgentSession | null = null;
|
let currentSession: AgentSession | null = null;
|
||||||
|
|
@ -293,11 +293,13 @@ export function createAgentRunner(sandboxConfig: SandboxConfig): AgentRunner {
|
||||||
const tools = createMomTools(executor);
|
const tools = createMomTools(executor);
|
||||||
|
|
||||||
// Get or create AgentSession for this channel
|
// Get or create AgentSession for this channel
|
||||||
let session = channelSessions.get(channelId);
|
const cached = channelSessions.get(channelId);
|
||||||
|
let session: AgentSession;
|
||||||
|
let sessionManager: MomSessionManager;
|
||||||
|
|
||||||
if (!session) {
|
if (!cached) {
|
||||||
// Create session manager and settings manager
|
// Create session manager and settings manager
|
||||||
const sessionManager = new MomSessionManager(channelDir);
|
sessionManager = new MomSessionManager(channelDir);
|
||||||
const settingsManager = new MomSettingsManager(join(channelDir, ".."));
|
const settingsManager = new MomSettingsManager(join(channelDir, ".."));
|
||||||
|
|
||||||
// Create agent with proper message transformer for compaction
|
// Create agent with proper message transformer for compaction
|
||||||
|
|
@ -328,10 +330,16 @@ export function createAgentRunner(sandboxConfig: SandboxConfig): AgentRunner {
|
||||||
settingsManager: settingsManager as any, // Type compatibility
|
settingsManager: settingsManager as any, // Type compatibility
|
||||||
});
|
});
|
||||||
|
|
||||||
channelSessions.set(channelId, session);
|
channelSessions.set(channelId, { session, sessionManager });
|
||||||
} else {
|
} else {
|
||||||
|
session = cached.session;
|
||||||
|
sessionManager = cached.sessionManager;
|
||||||
|
|
||||||
// Update system prompt for existing session (memory may have changed)
|
// Update system prompt for existing session (memory may have changed)
|
||||||
session.agent.setSystemPrompt(systemPrompt);
|
session.agent.setSystemPrompt(systemPrompt);
|
||||||
|
|
||||||
|
// Sync any new messages from log.jsonl (e.g., messages that arrived while processing)
|
||||||
|
sessionManager.syncFromLog();
|
||||||
}
|
}
|
||||||
|
|
||||||
currentSession = session;
|
currentSession = session;
|
||||||
|
|
|
||||||
|
|
@ -46,6 +46,7 @@ function uuidv4(): string {
|
||||||
export class MomSessionManager {
|
export class MomSessionManager {
|
||||||
private sessionId: string;
|
private sessionId: string;
|
||||||
private contextFile: string;
|
private contextFile: string;
|
||||||
|
private logFile: string;
|
||||||
private channelDir: string;
|
private channelDir: string;
|
||||||
private sessionInitialized: boolean = false;
|
private sessionInitialized: boolean = false;
|
||||||
private inMemoryEntries: SessionEntry[] = [];
|
private inMemoryEntries: SessionEntry[] = [];
|
||||||
|
|
@ -54,6 +55,7 @@ export class MomSessionManager {
|
||||||
constructor(channelDir: string) {
|
constructor(channelDir: string) {
|
||||||
this.channelDir = channelDir;
|
this.channelDir = channelDir;
|
||||||
this.contextFile = join(channelDir, "context.jsonl");
|
this.contextFile = join(channelDir, "context.jsonl");
|
||||||
|
this.logFile = join(channelDir, "log.jsonl");
|
||||||
|
|
||||||
// Ensure channel directory exists
|
// Ensure channel directory exists
|
||||||
if (!existsSync(channelDir)) {
|
if (!existsSync(channelDir)) {
|
||||||
|
|
@ -68,6 +70,93 @@ export class MomSessionManager {
|
||||||
} else {
|
} else {
|
||||||
this.sessionId = uuidv4();
|
this.sessionId = uuidv4();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Sync missing messages from log.jsonl to context.jsonl
|
||||||
|
this.syncFromLog();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Sync user messages from log.jsonl that aren't in context.jsonl.
|
||||||
|
*
|
||||||
|
* log.jsonl and context.jsonl must have the same user messages.
|
||||||
|
* This handles:
|
||||||
|
* - Backfilled messages (mom was offline)
|
||||||
|
* - Messages that arrived while mom was processing a previous turn
|
||||||
|
* - Channel chatter between @mentions
|
||||||
|
*
|
||||||
|
* Channel chatter is formatted as "[username]: message" to distinguish from direct @mentions.
|
||||||
|
*
|
||||||
|
* Called automatically on construction and should be called before each agent run.
|
||||||
|
*/
|
||||||
|
syncFromLog(): void {
|
||||||
|
if (!existsSync(this.logFile)) return;
|
||||||
|
|
||||||
|
// Get timestamps of messages already in context
|
||||||
|
const contextTimestamps = new Set<string>();
|
||||||
|
for (const entry of this.inMemoryEntries) {
|
||||||
|
if (entry.type === "message") {
|
||||||
|
contextTimestamps.add(entry.timestamp);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read log.jsonl and find user messages not in context
|
||||||
|
const logContent = readFileSync(this.logFile, "utf-8");
|
||||||
|
const logLines = logContent.trim().split("\n").filter(Boolean);
|
||||||
|
|
||||||
|
interface LogMessage {
|
||||||
|
date?: string;
|
||||||
|
ts?: string;
|
||||||
|
user?: string;
|
||||||
|
userName?: string;
|
||||||
|
text?: string;
|
||||||
|
isBot?: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
const newMessages: Array<{ timestamp: string; message: AppMessage }> = [];
|
||||||
|
|
||||||
|
for (const line of logLines) {
|
||||||
|
try {
|
||||||
|
const logMsg: LogMessage = JSON.parse(line);
|
||||||
|
|
||||||
|
const ts = logMsg.date || logMsg.ts;
|
||||||
|
if (!ts) continue;
|
||||||
|
|
||||||
|
// Skip if already in context
|
||||||
|
if (contextTimestamps.has(ts)) continue;
|
||||||
|
|
||||||
|
// Skip bot messages - added through agent flow
|
||||||
|
if (logMsg.isBot) continue;
|
||||||
|
|
||||||
|
const msgTime = new Date(ts).getTime() || Date.now();
|
||||||
|
const userMessage: AppMessage = {
|
||||||
|
role: "user",
|
||||||
|
content: `[${logMsg.userName || logMsg.user || "unknown"}]: ${logMsg.text || ""}`,
|
||||||
|
timestamp: msgTime,
|
||||||
|
};
|
||||||
|
|
||||||
|
newMessages.push({ timestamp: ts, message: userMessage });
|
||||||
|
} catch {
|
||||||
|
// Skip malformed lines
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (newMessages.length === 0) return;
|
||||||
|
|
||||||
|
// Sort by timestamp and add to context
|
||||||
|
newMessages.sort((a, b) => new Date(a.timestamp).getTime() - new Date(b.timestamp).getTime());
|
||||||
|
|
||||||
|
for (const { timestamp, message } of newMessages) {
|
||||||
|
const entry: SessionMessageEntry = {
|
||||||
|
type: "message",
|
||||||
|
timestamp,
|
||||||
|
message,
|
||||||
|
};
|
||||||
|
|
||||||
|
this.inMemoryEntries.push(entry);
|
||||||
|
appendFileSync(this.contextFile, JSON.stringify(entry) + "\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log(`[mom] Synced ${newMessages.length} messages from log.jsonl to context.jsonl`);
|
||||||
}
|
}
|
||||||
|
|
||||||
private extractSessionId(): string | null {
|
private extractSessionId(): string | null {
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue