mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-16 04:01:56 +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");
|
||||
}
|
||||
|
||||
// Cache for AgentSession per channel
|
||||
const channelSessions = new Map<string, AgentSession>();
|
||||
// Cache for AgentSession and SessionManager per channel
|
||||
const channelSessions = new Map<string, { session: AgentSession; sessionManager: MomSessionManager }>();
|
||||
|
||||
export function createAgentRunner(sandboxConfig: SandboxConfig): AgentRunner {
|
||||
let currentSession: AgentSession | null = null;
|
||||
|
|
@ -293,11 +293,13 @@ export function createAgentRunner(sandboxConfig: SandboxConfig): AgentRunner {
|
|||
const tools = createMomTools(executor);
|
||||
|
||||
// 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
|
||||
const sessionManager = new MomSessionManager(channelDir);
|
||||
sessionManager = new MomSessionManager(channelDir);
|
||||
const settingsManager = new MomSettingsManager(join(channelDir, ".."));
|
||||
|
||||
// Create agent with proper message transformer for compaction
|
||||
|
|
@ -328,10 +330,16 @@ export function createAgentRunner(sandboxConfig: SandboxConfig): AgentRunner {
|
|||
settingsManager: settingsManager as any, // Type compatibility
|
||||
});
|
||||
|
||||
channelSessions.set(channelId, session);
|
||||
channelSessions.set(channelId, { session, sessionManager });
|
||||
} else {
|
||||
session = cached.session;
|
||||
sessionManager = cached.sessionManager;
|
||||
|
||||
// Update system prompt for existing session (memory may have changed)
|
||||
session.agent.setSystemPrompt(systemPrompt);
|
||||
|
||||
// Sync any new messages from log.jsonl (e.g., messages that arrived while processing)
|
||||
sessionManager.syncFromLog();
|
||||
}
|
||||
|
||||
currentSession = session;
|
||||
|
|
|
|||
|
|
@ -46,6 +46,7 @@ function uuidv4(): string {
|
|||
export class MomSessionManager {
|
||||
private sessionId: string;
|
||||
private contextFile: string;
|
||||
private logFile: string;
|
||||
private channelDir: string;
|
||||
private sessionInitialized: boolean = false;
|
||||
private inMemoryEntries: SessionEntry[] = [];
|
||||
|
|
@ -54,6 +55,7 @@ export class MomSessionManager {
|
|||
constructor(channelDir: string) {
|
||||
this.channelDir = channelDir;
|
||||
this.contextFile = join(channelDir, "context.jsonl");
|
||||
this.logFile = join(channelDir, "log.jsonl");
|
||||
|
||||
// Ensure channel directory exists
|
||||
if (!existsSync(channelDir)) {
|
||||
|
|
@ -68,6 +70,93 @@ export class MomSessionManager {
|
|||
} else {
|
||||
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 {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue