openclaw/src/auto-reply/reply/get-reply.ts

519 lines
15 KiB
TypeScript

import {
resolveAgentDir,
resolveAgentWorkspaceDir,
resolveSessionAgentId,
resolveAgentSkillsFilter,
} from "../../agents/agent-scope.js";
import { resolveModelRefFromString } from "../../agents/model-selection.js";
import { resolveAgentTimeoutMs } from "../../agents/timeout.js";
import { DEFAULT_AGENT_WORKSPACE_DIR, ensureAgentWorkspace } from "../../agents/workspace.js";
import { resolveChannelModelOverride } from "../../channels/model-overrides.js";
import { type OpenClawConfig, loadConfig } from "../../config/config.js";
import { applyMergePatch } from "../../config/merge-patch.js";
import { defaultRuntime } from "../../runtime.js";
import { normalizeStringEntries } from "../../shared/string-normalization.js";
import { resolveCommandAuthorization } from "../command-auth.js";
import type { MsgContext } from "../templating.js";
import { SILENT_REPLY_TOKEN } from "../tokens.js";
import type { GetReplyOptions, ReplyPayload } from "../types.js";
import { resolveDefaultModel } from "./directive-handling.defaults.js";
import { resolveReplyDirectives } from "./get-reply-directives.js";
import { handleInlineActions } from "./get-reply-inline-actions.js";
import { runPreparedReply } from "./get-reply-run.js";
import { finalizeInboundContext } from "./inbound-context.js";
import { emitPreAgentMessageHooks } from "./message-preprocess-hooks.js";
import { initSessionState } from "./session.js";
import { createTypingController } from "./typing.js";
type ResetCommandAction = "new" | "reset";
let sessionResetModelRuntimePromise: Promise<
typeof import("./session-reset-model.runtime.js")
> | null = null;
let stageSandboxMediaRuntimePromise: Promise<
typeof import("./stage-sandbox-media.runtime.js")
> | null = null;
function loadSessionResetModelRuntime() {
sessionResetModelRuntimePromise ??= import("./session-reset-model.runtime.js");
return sessionResetModelRuntimePromise;
}
function loadStageSandboxMediaRuntime() {
stageSandboxMediaRuntimePromise ??= import("./stage-sandbox-media.runtime.js");
return stageSandboxMediaRuntimePromise;
}
let hookRunnerGlobalPromise: Promise<typeof import("../../plugins/hook-runner-global.js")> | null =
null;
let originRoutingPromise: Promise<typeof import("./origin-routing.js")> | null = null;
function loadHookRunnerGlobal() {
hookRunnerGlobalPromise ??= import("../../plugins/hook-runner-global.js");
return hookRunnerGlobalPromise;
}
function loadOriginRouting() {
originRoutingPromise ??= import("./origin-routing.js");
return originRoutingPromise;
}
function mergeSkillFilters(channelFilter?: string[], agentFilter?: string[]): string[] | undefined {
const normalize = (list?: string[]) => {
if (!Array.isArray(list)) {
return undefined;
}
return normalizeStringEntries(list);
};
const channel = normalize(channelFilter);
const agent = normalize(agentFilter);
if (!channel && !agent) {
return undefined;
}
if (!channel) {
return agent;
}
if (!agent) {
return channel;
}
if (channel.length === 0 || agent.length === 0) {
return [];
}
const agentSet = new Set(agent);
return channel.filter((name) => agentSet.has(name));
}
function hasInboundMedia(ctx: MsgContext): boolean {
return Boolean(
ctx.StickerMediaIncluded ||
ctx.Sticker ||
ctx.MediaPath?.trim() ||
ctx.MediaUrl?.trim() ||
ctx.MediaPaths?.some((value) => value?.trim()) ||
ctx.MediaUrls?.some((value) => value?.trim()) ||
ctx.MediaTypes?.length,
);
}
function hasLinkCandidate(ctx: MsgContext): boolean {
const message = ctx.BodyForCommands ?? ctx.CommandBody ?? ctx.RawBody ?? ctx.Body;
if (!message) {
return false;
}
return /\bhttps?:\/\/\S+/i.test(message);
}
async function applyMediaUnderstandingIfNeeded(params: {
ctx: MsgContext;
cfg: OpenClawConfig;
agentDir?: string;
activeModel: { provider: string; model: string };
}): Promise<boolean> {
if (!hasInboundMedia(params.ctx)) {
return false;
}
const { applyMediaUnderstanding } = await import("../../media-understanding/apply.runtime.js");
await applyMediaUnderstanding(params);
return true;
}
async function applyLinkUnderstandingIfNeeded(params: {
ctx: MsgContext;
cfg: OpenClawConfig;
}): Promise<boolean> {
if (!hasLinkCandidate(params.ctx)) {
return false;
}
const { applyLinkUnderstanding } = await import("../../link-understanding/apply.runtime.js");
await applyLinkUnderstanding(params);
return true;
}
export async function getReplyFromConfig(
ctx: MsgContext,
opts?: GetReplyOptions,
configOverride?: OpenClawConfig,
): Promise<ReplyPayload | ReplyPayload[] | undefined> {
const isFastTestEnv = process.env.OPENCLAW_TEST_FAST === "1";
const cfg =
configOverride == null
? loadConfig()
: (applyMergePatch(loadConfig(), configOverride) as OpenClawConfig);
const targetSessionKey =
ctx.CommandSource === "native" ? ctx.CommandTargetSessionKey?.trim() : undefined;
const agentSessionKey = targetSessionKey || ctx.SessionKey;
const agentId = resolveSessionAgentId({
sessionKey: agentSessionKey,
config: cfg,
});
const mergedSkillFilter = mergeSkillFilters(
opts?.skillFilter,
resolveAgentSkillsFilter(cfg, agentId),
);
const resolvedOpts =
mergedSkillFilter !== undefined ? { ...opts, skillFilter: mergedSkillFilter } : opts;
const agentCfg = cfg.agents?.defaults;
const sessionCfg = cfg.session;
const { defaultProvider, defaultModel, aliasIndex } = resolveDefaultModel({
cfg,
agentId,
});
let provider = defaultProvider;
let model = defaultModel;
let hasResolvedHeartbeatModelOverride = false;
if (opts?.isHeartbeat) {
// Prefer the resolved per-agent heartbeat model passed from the heartbeat runner,
// fall back to the global defaults heartbeat model for backward compatibility.
const heartbeatRaw =
opts.heartbeatModelOverride?.trim() ?? agentCfg?.heartbeat?.model?.trim() ?? "";
const heartbeatRef = heartbeatRaw
? resolveModelRefFromString({
raw: heartbeatRaw,
defaultProvider,
aliasIndex,
})
: null;
if (heartbeatRef) {
provider = heartbeatRef.ref.provider;
model = heartbeatRef.ref.model;
hasResolvedHeartbeatModelOverride = true;
}
}
const workspaceDirRaw = resolveAgentWorkspaceDir(cfg, agentId) ?? DEFAULT_AGENT_WORKSPACE_DIR;
const workspace = await ensureAgentWorkspace({
dir: workspaceDirRaw,
ensureBootstrapFiles: !agentCfg?.skipBootstrap && !isFastTestEnv,
});
const workspaceDir = workspace.dir;
const agentDir = resolveAgentDir(cfg, agentId);
const timeoutMs = resolveAgentTimeoutMs({ cfg, overrideSeconds: opts?.timeoutOverrideSeconds });
const configuredTypingSeconds =
agentCfg?.typingIntervalSeconds ?? sessionCfg?.typingIntervalSeconds;
const typingIntervalSeconds =
typeof configuredTypingSeconds === "number" ? configuredTypingSeconds : 6;
const typing = createTypingController({
onReplyStart: opts?.onReplyStart,
onCleanup: opts?.onTypingCleanup,
typingIntervalSeconds,
silentToken: SILENT_REPLY_TOKEN,
log: defaultRuntime.log,
});
opts?.onTypingController?.(typing);
const finalized = finalizeInboundContext(ctx);
if (!isFastTestEnv) {
await applyMediaUnderstandingIfNeeded({
ctx: finalized,
cfg,
agentDir,
activeModel: { provider, model },
});
await applyLinkUnderstandingIfNeeded({
ctx: finalized,
cfg,
});
}
emitPreAgentMessageHooks({
ctx: finalized,
cfg,
isFastTestEnv,
});
const commandAuthorized = finalized.CommandAuthorized;
resolveCommandAuthorization({
ctx: finalized,
cfg,
commandAuthorized,
});
const sessionState = await initSessionState({
ctx: finalized,
cfg,
commandAuthorized,
});
let {
sessionCtx,
sessionEntry,
previousSessionEntry,
sessionStore,
sessionKey,
sessionId,
isNewSession,
resetTriggered,
systemSent,
abortedLastRun,
storePath,
sessionScope,
groupResolution,
isGroup,
triggerBodyNormalized,
bodyStripped,
} = sessionState;
if (resetTriggered && bodyStripped?.trim()) {
const { applyResetModelOverride } = await loadSessionResetModelRuntime();
await applyResetModelOverride({
cfg,
agentId,
resetTriggered,
bodyStripped,
sessionCtx,
ctx: finalized,
sessionEntry,
sessionStore,
sessionKey,
storePath,
defaultProvider,
defaultModel,
aliasIndex,
});
}
const channelModelOverride = resolveChannelModelOverride({
cfg,
channel:
groupResolution?.channel ??
sessionEntry.channel ??
sessionEntry.origin?.provider ??
(typeof finalized.OriginatingChannel === "string"
? finalized.OriginatingChannel
: undefined) ??
finalized.Provider,
groupId: groupResolution?.id ?? sessionEntry.groupId,
groupChatType: sessionEntry.chatType ?? sessionCtx.ChatType ?? finalized.ChatType,
groupChannel: sessionEntry.groupChannel ?? sessionCtx.GroupChannel ?? finalized.GroupChannel,
groupSubject: sessionEntry.subject ?? sessionCtx.GroupSubject ?? finalized.GroupSubject,
parentSessionKey: sessionCtx.ParentSessionKey,
});
const hasSessionModelOverride = Boolean(
sessionEntry.modelOverride?.trim() || sessionEntry.providerOverride?.trim(),
);
if (!hasResolvedHeartbeatModelOverride && !hasSessionModelOverride && channelModelOverride) {
const resolved = resolveModelRefFromString({
raw: channelModelOverride.model,
defaultProvider,
aliasIndex,
});
if (resolved) {
provider = resolved.ref.provider;
model = resolved.ref.model;
}
}
const directiveResult = await resolveReplyDirectives({
ctx: finalized,
cfg,
agentId,
agentDir,
workspaceDir,
agentCfg,
sessionCtx,
sessionEntry,
sessionStore,
sessionKey,
storePath,
sessionScope,
groupResolution,
isGroup,
triggerBodyNormalized,
commandAuthorized,
defaultProvider,
defaultModel,
aliasIndex,
provider,
model,
hasResolvedHeartbeatModelOverride,
typing,
opts: resolvedOpts,
skillFilter: mergedSkillFilter,
});
if (directiveResult.kind === "reply") {
return directiveResult.reply;
}
let {
commandSource,
command,
allowTextCommands,
skillCommands,
directives,
cleanedBody,
elevatedEnabled,
elevatedAllowed,
elevatedFailures,
defaultActivation,
resolvedThinkLevel,
resolvedVerboseLevel,
resolvedReasoningLevel,
resolvedElevatedLevel,
execOverrides,
blockStreamingEnabled,
blockReplyChunking,
resolvedBlockStreamingBreak,
provider: resolvedProvider,
model: resolvedModel,
modelState,
contextTokens,
inlineStatusRequested,
directiveAck,
perMessageQueueMode,
perMessageQueueOptions,
} = directiveResult.result;
provider = resolvedProvider;
model = resolvedModel;
const maybeEmitMissingResetHooks = async () => {
if (!resetTriggered || !command.isAuthorizedSender || command.resetHookTriggered) {
return;
}
const resetMatch = command.commandBodyNormalized.match(/^\/(new|reset)(?:\s|$)/);
if (!resetMatch) {
return;
}
const { emitResetCommandHooks } = await import("./commands-core.runtime.js");
const action: ResetCommandAction = resetMatch[1] === "reset" ? "reset" : "new";
await emitResetCommandHooks({
action,
ctx,
cfg,
command,
sessionKey,
sessionEntry,
previousSessionEntry,
workspaceDir,
});
};
const inlineActionResult = await handleInlineActions({
ctx,
sessionCtx,
cfg,
agentId,
agentDir,
sessionEntry,
previousSessionEntry,
sessionStore,
sessionKey,
storePath,
sessionScope,
workspaceDir,
isGroup,
opts: resolvedOpts,
typing,
allowTextCommands,
inlineStatusRequested,
command,
skillCommands,
directives,
cleanedBody,
elevatedEnabled,
elevatedAllowed,
elevatedFailures,
defaultActivation: () => defaultActivation,
resolvedThinkLevel,
resolvedVerboseLevel,
resolvedReasoningLevel,
resolvedElevatedLevel,
blockReplyChunking,
resolvedBlockStreamingBreak,
resolveDefaultThinkingLevel: modelState.resolveDefaultThinkingLevel,
provider,
model,
contextTokens,
directiveAck,
abortedLastRun,
skillFilter: mergedSkillFilter,
});
if (inlineActionResult.kind === "reply") {
await maybeEmitMissingResetHooks();
return inlineActionResult.reply;
}
await maybeEmitMissingResetHooks();
directives = inlineActionResult.directives;
abortedLastRun = inlineActionResult.abortedLastRun ?? abortedLastRun;
// Allow plugins to intercept and return a synthetic reply before the LLM runs.
const { getGlobalHookRunner } = await loadHookRunnerGlobal();
const hookRunner = getGlobalHookRunner();
if (hookRunner?.hasHooks("before_agent_reply")) {
const { resolveOriginMessageProvider } = await loadOriginRouting();
const hookMessageProvider = resolveOriginMessageProvider({
originatingChannel: sessionCtx.OriginatingChannel,
provider: sessionCtx.Provider,
});
const hookResult = await hookRunner.runBeforeAgentReply(
{ cleanedBody },
{
agentId,
sessionKey: agentSessionKey,
sessionId,
workspaceDir,
messageProvider: hookMessageProvider,
trigger: opts?.isHeartbeat ? "heartbeat" : "user",
channelId: hookMessageProvider,
},
);
if (hookResult?.handled) {
return hookResult.reply ?? { text: SILENT_REPLY_TOKEN };
}
}
if (sessionKey && hasInboundMedia(ctx)) {
const { stageSandboxMedia } = await loadStageSandboxMediaRuntime();
await stageSandboxMedia({
ctx,
sessionCtx,
cfg,
sessionKey,
workspaceDir,
});
}
return runPreparedReply({
ctx,
sessionCtx,
cfg,
agentId,
agentDir,
agentCfg,
sessionCfg,
commandAuthorized,
command,
commandSource,
allowTextCommands,
directives,
defaultActivation,
resolvedThinkLevel,
resolvedVerboseLevel,
resolvedReasoningLevel,
resolvedElevatedLevel,
execOverrides,
elevatedEnabled,
elevatedAllowed,
blockStreamingEnabled,
blockReplyChunking,
resolvedBlockStreamingBreak,
modelState,
provider,
model,
perMessageQueueMode,
perMessageQueueOptions,
typing,
opts: resolvedOpts,
defaultProvider,
defaultModel,
timeoutMs,
isNewSession,
resetTriggered,
systemSent,
sessionEntry,
sessionStore,
sessionKey,
sessionId,
storePath,
workspaceDir,
abortedLastRun,
});
}