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

mom: add working indicator and improve stop command

Browse source 
Working Indicator:
- Add '...' to channel messages while mom is processing
- Automatically removed when work completes or stops
- Applies to working message, not status messages

Improved Stop Command:
- Posts separate 'Stopping...' message that updates to 'Stopped'
- Original working message continues updating with tool results
- Clean separation between status and work output
- Properly handles abort during multi-step operations

Clean Stop Reason Handling:
- Agent run() now returns { stopReason } instead of throwing
- Handle 'aborted', 'error', 'stop', 'length', 'toolUse' cases cleanly
- No more exception-based control flow
- Track stopReason from assistant message.stopReason field

New SlackContext Methods:
- replaceMessage() - replace message text instead of appending
- setWorking() - add/remove working indicator
- Improved context tracking for stop command updates
This commit is contained in:
Mario Zechner 2025-11-26 20:49:02 +01:00
parent bfe7df6a49
commit 0c6c0f34dd
4 changed files with 104 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

46
packages/mom/src/main.ts
View file

@ -68,7 +68,7 @@ if (!MOM_SLACK_APP_TOKEN || !MOM_SLACK_BOT_TOKEN || (!ANTHROPIC_API_KEY && !ANTH
await validateSandbox(sandbox);
// Track active agent runs per channel
const activeRuns = new Map<string, AgentRunner>();
const activeRuns = new Map<string, { runner: AgentRunner; context: SlackContext; stopContext?: SlackContext }>();
async function handleMessage(ctx: SlackContext, _source: "channel" | "dm"): Promise<void> {
const channelId = ctx.message.channel;
@ -82,11 +82,15 @@ async function handleMessage(ctx: SlackContext, _source: "channel" | "dm"): Prom
// Check for stop command
if (messageText === "stop") {
const runner = activeRuns.get(channelId);
if (runner) {
const active = activeRuns.get(channelId);
if (active) {
log.logStopRequest(logCtx);
runner.abort();
// Post a NEW message saying "Stopping..."
await ctx.respond("_Stopping..._");
// Store this context to update it to "Stopped" later
active.stopContext = ctx;
// Abort the runner
active.runner.abort();
} else {
await ctx.respond("_Nothing running._");
}
@ -103,23 +107,31 @@ async function handleMessage(ctx: SlackContext, _source: "channel" | "dm"): Prom
const channelDir = join(workingDir, channelId);
const runner = createAgentRunner(sandbox);
activeRuns.set(channelId, runner);
activeRuns.set(channelId, { runner, context: ctx });
await ctx.setTyping(true);
try {
await runner.run(ctx, channelDir, ctx.store);
} catch (error) {
// Don't report abort errors
const msg = error instanceof Error ? error.message : String(error);
if (msg.includes("aborted") || msg.includes("Aborted")) {
// Already said "Stopping..." - nothing more to say
} else {
log.logAgentError(logCtx, msg);
await ctx.respond(`❌ Error: ${msg}`);
await ctx.setWorking(true);
const result = await runner.run(ctx, channelDir, ctx.store);
// Remove working indicator
await ctx.setWorking(false);
// Handle different stop reasons
const active = activeRuns.get(channelId);
if (result.stopReason === "aborted") {
// Replace the STOP message with "Stopped"
if (active?.stopContext) {
await active.stopContext.setWorking(false);
await active.stopContext.replaceMessage("_Stopped_");
}
} finally {
activeRuns.delete(channelId);
} else if (result.stopReason === "error") {
// Agent encountered an error
log.logAgentError(logCtx, "Agent stopped with error");
}
// "stop", "length", "toolUse" are normal completions - nothing extra to do
activeRuns.delete(channelId);
}
const bot = new MomBot(