LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-04-24 07:01:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(sessions): simplify provider normalizer matching

Browse Source
This commit is contained in:
Peter Steinberger
2026-03-08 01:26:43 +00:00
parent e381ab630e
commit 8cc477b873
2 changed files with 50 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/config/sessions/explicit-session-key-normalization.test.ts
View File

@@ -39,6 +39,19 @@ describe("normalizeExplicitSessionKey", () => {
).toBe("discord:direct:123456");
});
it("uses Provider when Surface is absent", () => {
expect(
normalizeExplicitSessionKey(
"agent:fina:discord:dm:123456",
makeCtx({
Provider: "Discord",
ChatType: "direct",
SenderId: "123456",
}),
),
).toBe("agent:fina:discord:direct:123456");
});
it("lowercases and passes through unknown providers unchanged", () => {
expect(
normalizeExplicitSessionKey(

59
src/config/sessions/explicit-session-key-normalization.ts
View File

@@ -2,34 +2,49 @@ import type { MsgContext } from "../../auto-reply/templating.js";
import { normalizeExplicitDiscordSessionKey } from "../../discord/session-key-normalization.js";
type ExplicitSessionKeyNormalizer = (sessionKey: string, ctx: MsgContext) => string;
const EXPLICIT_SESSION_KEY_NORMALIZERS: Record<string, ExplicitSessionKeyNormalizer> = {
discord: normalizeExplicitDiscordSessionKey,
type ExplicitSessionKeyNormalizerEntry = {
provider: string;
normalize: ExplicitSessionKeyNormalizer;
matches: (params: {
sessionKey: string;
provider?: string;
surface?: string;
from: string;
}) => boolean;
};
function resolveExplicitSessionKeyProvider(
const EXPLICIT_SESSION_KEY_NORMALIZERS: ExplicitSessionKeyNormalizerEntry[] = [
{
provider: "discord",
normalize: normalizeExplicitDiscordSessionKey,
matches: ({ sessionKey, provider, surface, from }) =>
surface === "discord" ||
provider === "discord" ||
from.startsWith("discord:") ||
sessionKey.startsWith("discord:") ||
sessionKey.includes(":discord:"),
},
];
function resolveExplicitSessionKeyNormalizer(
sessionKey: string,
ctx: Pick<MsgContext, "From" | "Provider" | "Surface">,
): string | undefined {
const explicitProvider = [ctx.Surface, ctx.Provider]
.map((entry) => entry?.trim().toLowerCase())
.find((entry) => entry && entry in EXPLICIT_SESSION_KEY_NORMALIZERS);
if (explicitProvider) {
return explicitProvider;
}
const from = (ctx.From ?? "").trim().toLowerCase();
if (from.startsWith("discord:")) {
return "discord";
}
if (sessionKey.startsWith("discord:") || sessionKey.includes(":discord:")) {
return "discord";
}
return undefined;
): ExplicitSessionKeyNormalizer | undefined {
const normalizedProvider = ctx.Provider?.trim().toLowerCase();
const normalizedSurface = ctx.Surface?.trim().toLowerCase();
const normalizedFrom = (ctx.From ?? "").trim().toLowerCase();
return EXPLICIT_SESSION_KEY_NORMALIZERS.find((entry) =>
entry.matches({
sessionKey,
provider: normalizedProvider,
surface: normalizedSurface,
from: normalizedFrom,
}),
)?.normalize;
}
export function normalizeExplicitSessionKey(sessionKey: string, ctx: MsgContext): string {
const normalized = sessionKey.trim().toLowerCase();
const provider = resolveExplicitSessionKeyProvider(normalized, ctx);
return provider ? EXPLICIT_SESSION_KEY_NORMALIZERS[provider](normalized, ctx) : normalized;
const normalize = resolveExplicitSessionKeyNormalizer(normalized, ctx);
return normalize ? normalize(normalized, ctx) : normalized;
}