fix: unblock discord listener concurrency
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
import type { Api, Model, OpenAICompletionsCompat } from "@mariozechner/pi-ai";
|
||||
import type { Api, Model } from "@mariozechner/pi-ai";
|
||||
|
||||
function isOpenAiCompletionsModel(model: Model<Api>): model is Model<"openai-completions"> {
|
||||
return model.api === "openai-completions";
|
||||
@@ -10,7 +10,7 @@ export function normalizeModelCompat(model: Model<Api>): Model<Api> {
|
||||
if (!isZai || !isOpenAiCompletionsModel(model)) return model;
|
||||
|
||||
const openaiModel = model as Model<"openai-completions">;
|
||||
const compat = openaiModel.compat as OpenAICompletionsCompat | undefined;
|
||||
const compat = openaiModel.compat ?? undefined;
|
||||
if (compat?.supportsDeveloperRole === false) return model;
|
||||
|
||||
openaiModel.compat = compat
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
import type { Guild } from "@buape/carbon";
|
||||
import { describe, expect, it } from "vitest";
|
||||
import { describe, expect, it, vi } from "vitest";
|
||||
import {
|
||||
allowListMatches,
|
||||
buildDiscordMediaPayload,
|
||||
@@ -17,6 +17,7 @@ import {
|
||||
sanitizeDiscordThreadName,
|
||||
shouldEmitDiscordReactionNotification,
|
||||
} from "./monitor.js";
|
||||
import { DiscordMessageListener } from "./monitor/listeners.js";
|
||||
|
||||
const fakeGuild = (id: string, name: string) => ({ id, name }) as Guild;
|
||||
|
||||
@@ -48,6 +49,85 @@ describe("registerDiscordListener", () => {
|
||||
});
|
||||
});
|
||||
|
||||
describe("DiscordMessageListener", () => {
|
||||
it("returns before the handler finishes", async () => {
|
||||
let handlerResolved = false;
|
||||
let resolveHandler: (() => void) | null = null;
|
||||
const handlerPromise = new Promise<void>((resolve) => {
|
||||
resolveHandler = () => {
|
||||
handlerResolved = true;
|
||||
resolve();
|
||||
};
|
||||
});
|
||||
const handler = vi.fn(() => handlerPromise);
|
||||
const listener = new DiscordMessageListener(handler);
|
||||
|
||||
await listener.handle(
|
||||
{} as unknown as import("./monitor/listeners.js").DiscordMessageEvent,
|
||||
{} as unknown as import("@buape/carbon").Client,
|
||||
);
|
||||
|
||||
expect(handler).toHaveBeenCalledOnce();
|
||||
expect(handlerResolved).toBe(false);
|
||||
|
||||
resolveHandler?.();
|
||||
await handlerPromise;
|
||||
});
|
||||
|
||||
it("logs handler failures", async () => {
|
||||
const logger = {
|
||||
warn: vi.fn(),
|
||||
error: vi.fn(),
|
||||
} as unknown as ReturnType<typeof import("../logging/subsystem.js").createSubsystemLogger>;
|
||||
const handler = vi.fn(async () => {
|
||||
throw new Error("boom");
|
||||
});
|
||||
const listener = new DiscordMessageListener(handler, logger);
|
||||
|
||||
await listener.handle(
|
||||
{} as unknown as import("./monitor/listeners.js").DiscordMessageEvent,
|
||||
{} as unknown as import("@buape/carbon").Client,
|
||||
);
|
||||
await new Promise((resolve) => setTimeout(resolve, 0));
|
||||
|
||||
expect(logger.error).toHaveBeenCalledWith(expect.stringContaining("discord handler failed"));
|
||||
});
|
||||
|
||||
it("logs slow handlers after the threshold", async () => {
|
||||
vi.useFakeTimers();
|
||||
vi.setSystemTime(0);
|
||||
|
||||
try {
|
||||
let resolveHandler: (() => void) | null = null;
|
||||
const handlerPromise = new Promise<void>((resolve) => {
|
||||
resolveHandler = resolve;
|
||||
});
|
||||
const handler = vi.fn(() => handlerPromise);
|
||||
const logger = {
|
||||
warn: vi.fn(),
|
||||
error: vi.fn(),
|
||||
} as unknown as ReturnType<typeof import("../logging/subsystem.js").createSubsystemLogger>;
|
||||
const listener = new DiscordMessageListener(handler, logger);
|
||||
|
||||
await listener.handle(
|
||||
{} as unknown as import("./monitor/listeners.js").DiscordMessageEvent,
|
||||
{} as unknown as import("@buape/carbon").Client,
|
||||
);
|
||||
|
||||
vi.setSystemTime(31_000);
|
||||
resolveHandler?.();
|
||||
await handlerPromise;
|
||||
await Promise.resolve();
|
||||
|
||||
expect(logger.warn).toHaveBeenCalled();
|
||||
const [, meta] = logger.warn.mock.calls[0] ?? [];
|
||||
expect(meta?.durationMs).toBeGreaterThanOrEqual(30_000);
|
||||
} finally {
|
||||
vi.useRealTimers();
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
describe("discord allowlist helpers", () => {
|
||||
it("normalizes slugs", () => {
|
||||
expect(normalizeDiscordSlug("Friends of Clawd")).toBe("friends-of-clawd");
|
||||
|
||||
@@ -73,16 +73,20 @@ export class DiscordMessageListener extends MessageCreateListener {
|
||||
|
||||
async handle(data: DiscordMessageEvent, client: Client) {
|
||||
const startedAt = Date.now();
|
||||
try {
|
||||
await this.handler(data, client);
|
||||
} finally {
|
||||
logSlowDiscordListener({
|
||||
logger: this.logger,
|
||||
listener: this.constructor.name,
|
||||
event: this.type,
|
||||
durationMs: Date.now() - startedAt,
|
||||
const task = Promise.resolve(this.handler(data, client));
|
||||
void task
|
||||
.catch((err) => {
|
||||
const logger = this.logger ?? discordEventQueueLog;
|
||||
logger.error(danger(`discord handler failed: ${String(err)}`));
|
||||
})
|
||||
.finally(() => {
|
||||
logSlowDiscordListener({
|
||||
logger: this.logger,
|
||||
listener: this.constructor.name,
|
||||
event: this.type,
|
||||
durationMs: Date.now() - startedAt,
|
||||
});
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
import { Editor, type EditorTheme, type TUI, Key, matchesKey } from "@mariozechner/pi-tui";
|
||||
import { Editor, type EditorTheme, Key, matchesKey } from "@mariozechner/pi-tui";
|
||||
|
||||
export class CustomEditor extends Editor {
|
||||
onEscape?: () => void;
|
||||
@@ -12,8 +12,8 @@ export class CustomEditor extends Editor {
|
||||
onShiftTab?: () => void;
|
||||
onAltEnter?: () => void;
|
||||
|
||||
constructor(tui: TUI, theme: EditorTheme) {
|
||||
super(tui, theme);
|
||||
constructor(theme: EditorTheme) {
|
||||
super(theme);
|
||||
}
|
||||
handleInput(data: string): void {
|
||||
if (matchesKey(data, Key.alt("enter")) && this.onAltEnter) {
|
||||
|
||||
@@ -193,7 +193,7 @@ export async function runTui(opts: TuiOptions) {
|
||||
const statusContainer = new Container();
|
||||
const footer = new Text("", 1, 0);
|
||||
const chatLog = new ChatLog();
|
||||
const editor = new CustomEditor(tui, editorTheme);
|
||||
const editor = new CustomEditor(editorTheme);
|
||||
const root = new Container();
|
||||
root.addChild(header);
|
||||
root.addChild(chatLog);
|
||||
|
||||
Reference in New Issue
Block a user