getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-21 13:00:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

mom: sync user messages from log.jsonl to context.jsonl before each agent run

Browse source 
- 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:
Mario Zechner 2025-12-11 13:20:40 +01:00
parent 3f6db8e99c
commit ca23ade9f8
2 changed files with 103 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

20
packages/mom/src/agent.ts
View file

@ -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;