feat(plugin): add sender normalization, TTL, and one-shot decisions
This commit is contained in:
@@ -1,8 +1,14 @@
|
|||||||
import type { OpenClawPluginApi } from "openclaw/plugin-sdk";
|
import type { OpenClawPluginApi } from "openclaw/plugin-sdk";
|
||||||
import { evaluateDecision, type Decision, type WhisperGateConfig } from "./rules.js";
|
import { evaluateDecision, type Decision, type WhisperGateConfig } from "./rules.js";
|
||||||
|
|
||||||
const sessionDecision = new Map<string, Decision>();
|
type DecisionRecord = {
|
||||||
|
decision: Decision;
|
||||||
|
createdAt: number;
|
||||||
|
};
|
||||||
|
|
||||||
|
const sessionDecision = new Map<string, DecisionRecord>();
|
||||||
const MAX_SESSION_DECISIONS = 2000;
|
const MAX_SESSION_DECISIONS = 2000;
|
||||||
|
const DECISION_TTL_MS = 5 * 60 * 1000;
|
||||||
|
|
||||||
function normalizeChannel(ctx: Record<string, unknown>): string {
|
function normalizeChannel(ctx: Record<string, unknown>): string {
|
||||||
const candidates = [ctx.commandSource, ctx.messageProvider, ctx.channelId, ctx.channel];
|
const candidates = [ctx.commandSource, ctx.messageProvider, ctx.channelId, ctx.channel];
|
||||||
@@ -12,7 +18,27 @@ function normalizeChannel(ctx: Record<string, unknown>): string {
|
|||||||
return "";
|
return "";
|
||||||
}
|
}
|
||||||
|
|
||||||
function pruneDecisionMap() {
|
function normalizeSender(event: Record<string, unknown>, ctx: Record<string, unknown>): string | undefined {
|
||||||
|
const direct = [ctx.senderId, ctx.from, event.from];
|
||||||
|
for (const v of direct) {
|
||||||
|
if (typeof v === "string" && v.trim()) return v.trim();
|
||||||
|
}
|
||||||
|
|
||||||
|
const meta = (event.metadata || ctx.metadata) as Record<string, unknown> | undefined;
|
||||||
|
if (!meta) return undefined;
|
||||||
|
const metaCandidates = [meta.senderId, meta.sender_id, meta.userId, meta.user_id];
|
||||||
|
for (const v of metaCandidates) {
|
||||||
|
if (typeof v === "string" && v.trim()) return v.trim();
|
||||||
|
}
|
||||||
|
|
||||||
|
return undefined;
|
||||||
|
}
|
||||||
|
|
||||||
|
function pruneDecisionMap(now = Date.now()) {
|
||||||
|
for (const [k, v] of sessionDecision.entries()) {
|
||||||
|
if (now - v.createdAt > DECISION_TTL_MS) sessionDecision.delete(k);
|
||||||
|
}
|
||||||
|
|
||||||
if (sessionDecision.size <= MAX_SESSION_DECISIONS) return;
|
if (sessionDecision.size <= MAX_SESSION_DECISIONS) return;
|
||||||
const keys = sessionDecision.keys();
|
const keys = sessionDecision.keys();
|
||||||
while (sessionDecision.size > MAX_SESSION_DECISIONS) {
|
while (sessionDecision.size > MAX_SESSION_DECISIONS) {
|
||||||
@@ -35,20 +61,16 @@ export default {
|
|||||||
const sessionKey = typeof c.sessionKey === "string" ? c.sessionKey : undefined;
|
const sessionKey = typeof c.sessionKey === "string" ? c.sessionKey : undefined;
|
||||||
if (!sessionKey) return;
|
if (!sessionKey) return;
|
||||||
|
|
||||||
const senderId =
|
const senderId = normalizeSender(e, c);
|
||||||
typeof c.senderId === "string"
|
|
||||||
? c.senderId
|
|
||||||
: typeof e.from === "string"
|
|
||||||
? e.from
|
|
||||||
: undefined;
|
|
||||||
|
|
||||||
const content = typeof e.content === "string" ? e.content : "";
|
const content = typeof e.content === "string" ? e.content : "";
|
||||||
const channel = normalizeChannel(c);
|
const channel = normalizeChannel(c);
|
||||||
|
|
||||||
const decision = evaluateDecision({ config, channel, senderId, content });
|
const decision = evaluateDecision({ config, channel, senderId, content });
|
||||||
sessionDecision.set(sessionKey, decision);
|
sessionDecision.set(sessionKey, { decision, createdAt: Date.now() });
|
||||||
pruneDecisionMap();
|
pruneDecisionMap();
|
||||||
api.logger.debug?.(`whispergate: session=${sessionKey} decision=${decision.reason}`);
|
api.logger.debug?.(
|
||||||
|
`whispergate: session=${sessionKey} sender=${senderId ?? "unknown"} channel=${channel || "unknown"} decision=${decision.reason}`,
|
||||||
|
);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
api.logger.warn(`whispergate: message hook failed: ${String(err)}`);
|
api.logger.warn(`whispergate: message hook failed: ${String(err)}`);
|
||||||
}
|
}
|
||||||
@@ -57,11 +79,20 @@ export default {
|
|||||||
api.on("before_model_resolve", async (_event, ctx) => {
|
api.on("before_model_resolve", async (_event, ctx) => {
|
||||||
const key = ctx.sessionKey;
|
const key = ctx.sessionKey;
|
||||||
if (!key) return;
|
if (!key) return;
|
||||||
const decision = sessionDecision.get(key);
|
|
||||||
if (!decision?.shouldUseNoReply) return;
|
const rec = sessionDecision.get(key);
|
||||||
|
if (!rec) return;
|
||||||
|
if (Date.now() - rec.createdAt > DECISION_TTL_MS) {
|
||||||
|
sessionDecision.delete(key);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// one-shot decision per inbound turn
|
||||||
|
sessionDecision.delete(key);
|
||||||
|
if (!rec.decision.shouldUseNoReply) return;
|
||||||
|
|
||||||
api.logger.info(
|
api.logger.info(
|
||||||
`whispergate: override model for session=${key}, provider=${config.noReplyProvider}, model=${config.noReplyModel}, reason=${decision.reason}`,
|
`whispergate: override model for session=${key}, provider=${config.noReplyProvider}, model=${config.noReplyModel}, reason=${rec.decision.reason}`,
|
||||||
);
|
);
|
||||||
|
|
||||||
return {
|
return {
|
||||||
|
|||||||
Reference in New Issue
Block a user