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

Release v0.18.5

Browse source
This commit is contained in:
Mario Zechner 2025-12-12 10:00:57 +01:00
parent 29cbf16218
commit 44e9b1c8e9
14 changed files with 274 additions and 99 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -42,7 +42,11 @@ export interface PendingMessage {
}
export interface AgentRunner {
run(ctx: SlackContext, store: ChannelStore, pendingMessages?: PendingMessage[]): Promise<{ stopReason: string }>;
run(
ctx: SlackContext,
store: ChannelStore,
pendingMessages?: PendingMessage[],
): Promise<{ stopReason: string; errorMessage?: string }>;
abort(): void;
}
@ -346,6 +350,7 @@ function createRunner(sandboxConfig: SandboxConfig, channelId: string, channelDi
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
},
stopReason: "stop",
errorMessage: undefined as string | undefined,
};
// Subscribe to events ONCE
@ -412,6 +417,9 @@ function createRunner(sandboxConfig: SandboxConfig, channelId: string, channelDi
if (assistantMsg.stopReason) {
runState.stopReason = assistantMsg.stopReason;
}
if (assistantMsg.errorMessage) {
runState.errorMessage = assistantMsg.errorMessage;
}
if (assistantMsg.usage) {
runState.totalUsage.input += assistantMsg.usage.input;
@ -492,7 +500,7 @@ function createRunner(sandboxConfig: SandboxConfig, channelId: string, channelDi
ctx: SlackContext,
_store: ChannelStore,
_pendingMessages?: PendingMessage[],
): Promise<{ stopReason: string }> {
): Promise<{ stopReason: string; errorMessage?: string }> {
// Ensure channel directory exists
await mkdir(channelDir, { recursive: true });
@ -538,6 +546,7 @@ function createRunner(sandboxConfig: SandboxConfig, channelId: string, channelDi
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
};
runState.stopReason = "stop";
runState.errorMessage = undefined;
// Create queue for this run
let queueChain = Promise.resolve();
@ -595,25 +604,36 @@ function createRunner(sandboxConfig: SandboxConfig, channelId: string, channelDi
// Wait for queued messages
await queueChain;
// Final message update
const messages = session.messages;
const lastAssistant = messages.filter((m) => m.role === "assistant").pop();
const finalText =
lastAssistant?.content
.filter((c): c is { type: "text"; text: string } => c.type === "text")
.map((c) => c.text)
.join("\n") || "";
if (finalText.trim()) {
// Handle error case - update main message and post error to thread
if (runState.stopReason === "error" && runState.errorMessage) {
try {
const mainText =
finalText.length > SLACK_MAX_LENGTH
? finalText.substring(0, SLACK_MAX_LENGTH - 50) + "\n\n_(see thread for full response)_"
: finalText;
await ctx.replaceMessage(mainText);
await ctx.replaceMessage("_Sorry, something went wrong_");
await ctx.respondInThread(`_Error: ${runState.errorMessage}_`);
} catch (err) {
const errMsg = err instanceof Error ? err.message : String(err);
log.logWarning("Failed to replace message with final text", errMsg);
log.logWarning("Failed to post error message", errMsg);
}
} else {
// Final message update
const messages = session.messages;
const lastAssistant = messages.filter((m) => m.role === "assistant").pop();
const finalText =
lastAssistant?.content
.filter((c): c is { type: "text"; text: string } => c.type === "text")
.map((c) => c.text)
.join("\n") || "";
if (finalText.trim()) {
try {
const mainText =
finalText.length > SLACK_MAX_LENGTH
? finalText.substring(0, SLACK_MAX_LENGTH - 50) + "\n\n_(see thread for full response)_"
: finalText;
await ctx.replaceMessage(mainText);
} catch (err) {
const errMsg = err instanceof Error ? err.message : String(err);
log.logWarning("Failed to replace message with final text", errMsg);
}
}
}
@ -644,7 +664,7 @@ function createRunner(sandboxConfig: SandboxConfig, channelId: string, channelDi
runState.logCtx = null;
runState.queue = null;
return { stopReason: runState.stopReason };
return { stopReason: runState.stopReason, errorMessage: runState.errorMessage };
},
abort(): void {