137 lines
4.1 KiB
TypeScript
137 lines
4.1 KiB
TypeScript
import type { ClawdbotConfig } from "../config/config.js";
|
|
import type { MsgContext } from "../auto-reply/templating.js";
|
|
import { applyTemplate } from "../auto-reply/templating.js";
|
|
import type { LinkModelConfig, LinkToolsConfig } from "../config/types.tools.js";
|
|
import { logVerbose, shouldLogVerbose } from "../globals.js";
|
|
import { runExec } from "../process/exec.js";
|
|
import { CLI_OUTPUT_MAX_BUFFER } from "../media-understanding/defaults.js";
|
|
import { resolveTimeoutMs } from "../media-understanding/resolve.js";
|
|
import {
|
|
normalizeMediaUnderstandingChatType,
|
|
resolveMediaUnderstandingScope,
|
|
} from "../media-understanding/scope.js";
|
|
import { DEFAULT_LINK_TIMEOUT_SECONDS } from "./defaults.js";
|
|
import { extractLinksFromMessage } from "./detect.js";
|
|
|
|
export type LinkUnderstandingResult = {
|
|
urls: string[];
|
|
outputs: string[];
|
|
};
|
|
|
|
function resolveScopeDecision(params: {
|
|
config?: LinkToolsConfig;
|
|
ctx: MsgContext;
|
|
}): "allow" | "deny" {
|
|
return resolveMediaUnderstandingScope({
|
|
scope: params.config?.scope,
|
|
sessionKey: params.ctx.SessionKey,
|
|
channel: params.ctx.Surface ?? params.ctx.Provider,
|
|
chatType: normalizeMediaUnderstandingChatType(params.ctx.ChatType),
|
|
});
|
|
}
|
|
|
|
function resolveTimeoutMsFromConfig(params: {
|
|
config?: LinkToolsConfig;
|
|
entry: LinkModelConfig;
|
|
}): number {
|
|
const configured = params.entry.timeoutSeconds ?? params.config?.timeoutSeconds;
|
|
return resolveTimeoutMs(configured, DEFAULT_LINK_TIMEOUT_SECONDS);
|
|
}
|
|
|
|
async function runCliEntry(params: {
|
|
entry: LinkModelConfig;
|
|
ctx: MsgContext;
|
|
url: string;
|
|
config?: LinkToolsConfig;
|
|
}): Promise<string | null> {
|
|
if ((params.entry.type ?? "cli") !== "cli") return null;
|
|
const command = params.entry.command.trim();
|
|
if (!command) return null;
|
|
const args = params.entry.args ?? [];
|
|
const timeoutMs = resolveTimeoutMsFromConfig({ config: params.config, entry: params.entry });
|
|
const templCtx = {
|
|
...params.ctx,
|
|
LinkUrl: params.url,
|
|
};
|
|
const argv = [command, ...args].map((part, index) =>
|
|
index === 0 ? part : applyTemplate(part, templCtx),
|
|
);
|
|
|
|
if (shouldLogVerbose()) {
|
|
logVerbose(`Link understanding via CLI: ${argv.join(" ")}`);
|
|
}
|
|
|
|
const { stdout } = await runExec(argv[0], argv.slice(1), {
|
|
timeoutMs,
|
|
maxBuffer: CLI_OUTPUT_MAX_BUFFER,
|
|
});
|
|
const trimmed = stdout.trim();
|
|
return trimmed || null;
|
|
}
|
|
|
|
async function runLinkEntries(params: {
|
|
entries: LinkModelConfig[];
|
|
ctx: MsgContext;
|
|
url: string;
|
|
config?: LinkToolsConfig;
|
|
}): Promise<string | null> {
|
|
let lastError: unknown;
|
|
for (const entry of params.entries) {
|
|
try {
|
|
const output = await runCliEntry({
|
|
entry,
|
|
ctx: params.ctx,
|
|
url: params.url,
|
|
config: params.config,
|
|
});
|
|
if (output) return output;
|
|
} catch (err) {
|
|
lastError = err;
|
|
if (shouldLogVerbose()) {
|
|
logVerbose(`Link understanding failed for ${params.url}: ${String(err)}`);
|
|
}
|
|
}
|
|
}
|
|
if (lastError && shouldLogVerbose()) {
|
|
logVerbose(`Link understanding exhausted for ${params.url}`);
|
|
}
|
|
return null;
|
|
}
|
|
|
|
export async function runLinkUnderstanding(params: {
|
|
cfg: ClawdbotConfig;
|
|
ctx: MsgContext;
|
|
message?: string;
|
|
}): Promise<LinkUnderstandingResult> {
|
|
const config = params.cfg.tools?.links;
|
|
if (!config || config.enabled === false) return { urls: [], outputs: [] };
|
|
|
|
const scopeDecision = resolveScopeDecision({ config, ctx: params.ctx });
|
|
if (scopeDecision === "deny") {
|
|
if (shouldLogVerbose()) {
|
|
logVerbose("Link understanding disabled by scope policy.");
|
|
}
|
|
return { urls: [], outputs: [] };
|
|
}
|
|
|
|
const message = params.message ?? params.ctx.CommandBody ?? params.ctx.RawBody ?? params.ctx.Body;
|
|
const links = extractLinksFromMessage(message ?? "", { maxLinks: config?.maxLinks });
|
|
if (links.length === 0) return { urls: [], outputs: [] };
|
|
|
|
const entries = config?.models ?? [];
|
|
if (entries.length === 0) return { urls: links, outputs: [] };
|
|
|
|
const outputs: string[] = [];
|
|
for (const url of links) {
|
|
const output = await runLinkEntries({
|
|
entries,
|
|
ctx: params.ctx,
|
|
url,
|
|
config,
|
|
});
|
|
if (output) outputs.push(output);
|
|
}
|
|
|
|
return { urls: links, outputs };
|
|
}
|