@dcrays/dcgchat-test 0.4.18 → 0.4.20

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dcrays/dcgchat-test",
-  "version": "0.4.18",
+  "version": "0.4.20",
   "type": "module",
   "description": "OpenClaw channel plugin for 书灵墨宝 (WebSocket)",
   "main": "index.ts",

package/src/bot.ts

@@ -14,7 +14,7 @@ import { resolveAccount, sendDcgchatMedia } from './channel.js'
 import { generateSignUrl } from './request/api.js'
 import { sendChunk, sendFinal, sendText as sendTextMsg, sendError, wsSendRaw, sendText } from './transport.js'
 import { dcgLogger } from './utils/log.js'
-import { channelInfo, systemCommand, interruptCommand, ENV, ignoreToolCommand } from './utils/constant.js'
+import { channelInfo, systemCommand, stopCommand, ENV, ignoreToolCommand } from './utils/constant.js'
 import { sendGatewayRpc } from './gateway/socket.js'
 import { clearParamsMessage, getEffectiveMsgParams, setParamsMessage } from './utils/params.js'
 import { getDescendantSessionKeysForRequester, resetSubagentStateForRequesterSession, waitUntilSubagentsIdle } from './tool.js'
@@ -33,12 +33,21 @@ const mediaMaxBytes = 300 * 1024 * 1024
 /** 当前会话最近一次 agent run 的 runId（供观测；/stop 时对子会话与主会话使用不传 runId 的 chat.abort 以终止该键下全部活跃 run） */
 const activeRunIdBySessionKey = new Map<string, string>()
 
+/** 本通道用 dispatchReplyFromConfig 在进程内跑 agent，网关 chat.abort 往往碰不到该 run；靠 AbortSignal 才能真正停工具与模型调用。 */
+const dispatchAbortBySessionKey = new Map<string, AbortController>()
+
 /**
  * 用户在该 sessionKey 上触发打断后，旧 run 的流式/投递不再下发；与 sessionKey 一一对应，支持多会话。
  * 清除时机：① 下一条非打断用户消息开始处理时；② 旧 run 收尾到 mobook 段时若仍抑制则跳过并发后删除。
  */
 const sessionStreamSuppressed = new Set<string>()
 
+/**
+ * 每次 /stop 递增。用于丢弃「已被后续打断」的那条入站处理在 dispatch 结束后的尾部逻辑
+ *（否则旧 handler 会删掉 sessionStreamSuppressed、再发 sendFinal(end)/recordInboundSession，导致旧回复复活、新消息异常）。
+ */
+const replyAbortEpochBySessionKey = new Map<string, number>()
+
 /** 各 sessionKey 当前轮回复的流式分片序号（仅统计实际下发的文本 chunk）；每轮非打断消息开始时置 0 */
 const streamChunkIdxBySessionKey = new Map<string, number>()
 
@@ -167,6 +176,7 @@ export async function handleDcgchatMessage(msg: InboundMessage, accountId: strin
 
   const effectiveAgentId = embeddedAgentId ?? route.agentId
   const dcgSessionKey = getSessionKey(msg.content, account.accountId)
+  const replyAbortEpochAtStart = replyAbortEpochBySessionKey.get(dcgSessionKey) ?? 0
 
   const mergedParams = {
     userId: msg._userId,
@@ -174,7 +184,7 @@ export async function handleDcgchatMessage(msg: InboundMessage, accountId: strin
     sessionId: conversationId,
     messageId: msg.content.message_id,
     domainId: msg.content.domain_id,
-    appId: msg.content.app_id,
+    appId: config.channels?.["dcgchat-test"]?.appId || 100,
     botId: msg.content.bot_id ?? '',
     agentId: msg.content.agent_id ?? '',
     sessionKey: dcgSessionKey,
@@ -297,10 +307,14 @@ export async function handleDcgchatMessage(msg: InboundMessage, accountId: strin
       }
     })
 
+    let dispatchAbort: AbortController | undefined
     try {
-      if (!interruptCommand.includes(text?.trim())) {
+      if (!stopCommand.includes(text?.trim())) {
         sessionStreamSuppressed.delete(dcgSessionKey)
         streamChunkIdxBySessionKey.set(dcgSessionKey, 0)
+        dispatchAbortBySessionKey.get(dcgSessionKey)?.abort()
+        dispatchAbort = new AbortController()
+        dispatchAbortBySessionKey.set(dcgSessionKey, dispatchAbort)
       }
 
       if (systemCommand.includes(text?.trim())) {
@@ -315,6 +329,7 @@ export async function handleDcgchatMessage(msg: InboundMessage, accountId: strin
               dispatcher,
               replyOptions: {
                 ...replyOptions,
+                abortSignal: dispatchAbort!.signal,
                 onModelSelected: prefixContext.onModelSelected,
                 onAgentRunStart: (runId) => {
                   activeRunIdBySessionKey.set(dcgSessionKey, runId)
@@ -322,12 +337,22 @@ export async function handleDcgchatMessage(msg: InboundMessage, accountId: strin
               }
             })
         })
-      } else if (interruptCommand.includes(text?.trim())) {
+      } else if (stopCommand.includes(text?.trim())) {
         dcgLogger(`interrupt command: ${text}`)
+        const inFlightAbort = dispatchAbortBySessionKey.get(dcgSessionKey)
+        if (inFlightAbort) {
+          dcgLogger(`interrupt: AbortController.abort() in-process run sessionKey=${dcgSessionKey}`)
+          inFlightAbort.abort()
+          dispatchAbortBySessionKey.delete(dcgSessionKey)
+        }
         const ctxForAbort = priorOutboundCtx.messageId?.trim() || priorOutboundCtx.sessionId?.trim() ? priorOutboundCtx : outboundCtx
         sendFinal(ctxForAbort.messageId?.trim() ? ctxForAbort : { ...ctxForAbort, messageId: `${Date.now()}` }, 'abort')
         sessionStreamSuppressed.add(dcgSessionKey)
-        // 网关侧彻底中止：子 agent 使用独立 sessionKey，仅 abort 主会话不会停子 run；自深到浅 abort 后代，再 abort 主会话；不传 runId 以终止该键下全部活跃 chat run
+        replyAbortEpochBySessionKey.set(
+          dcgSessionKey,
+          (replyAbortEpochBySessionKey.get(dcgSessionKey) ?? 0) + 1
+        )
+        // 网关侧：子 agent 独立 sessionKey，自深到浅 abort 后再 abort 主会话；主会话有 runId 时带上以便网关侧若有登记可一并取消
         const descendantKeys = getDescendantSessionKeysForRequester(dcgSessionKey)
         const abortSubKeys = [...descendantKeys].reverse()
         if (abortSubKeys.length > 0) {
@@ -341,7 +366,13 @@ export async function handleDcgchatMessage(msg: InboundMessage, accountId: strin
           }
         }
         try {
-          await sendGatewayRpc({ method: 'chat.abort', params: { sessionKey: dcgSessionKey } })
+          const mainRunId = activeRunIdBySessionKey.get(dcgSessionKey)
+          await sendGatewayRpc({
+            method: 'chat.abort',
+            params: mainRunId
+              ? { sessionKey: dcgSessionKey, runId: mainRunId }
+              : { sessionKey: dcgSessionKey }
+          })
         } catch (e) {
           dcgLogger(`chat.abort main ${dcgSessionKey}: ${String(e)}`, 'error')
         }
@@ -381,6 +412,7 @@ export async function handleDcgchatMessage(msg: InboundMessage, accountId: strin
               dispatcher,
               replyOptions: {
                 ...replyOptions,
+                abortSignal: dispatchAbort!.signal,
                 onModelSelected: prefixContext.onModelSelected,
                 onAgentRunStart: (runId) => {
                   activeRunIdBySessionKey.set(dcgSessionKey, runId)
@@ -425,9 +457,21 @@ export async function handleDcgchatMessage(msg: InboundMessage, accountId: strin
       }
     } catch (err: unknown) {
       dcgLogger(` dispatchReplyFromConfig error: ${String(err)}`, 'error')
+    } finally {
+      if (dispatchAbort && dispatchAbortBySessionKey.get(dcgSessionKey) === dispatchAbort) {
+        dispatchAbortBySessionKey.delete(dcgSessionKey)
+      }
+    }
+
+    const replyAbortEpochNow = replyAbortEpochBySessionKey.get(dcgSessionKey) ?? 0
+    if (replyAbortEpochNow !== replyAbortEpochAtStart) {
+      dcgLogger(
+        `skip post-reply tail: sessionKey=${dcgSessionKey} (abort epoch ${replyAbortEpochAtStart}→${replyAbortEpochNow}, stale handler after interrupt)`
+      )
+      return
     }
 
-    if (![...systemCommand, ...interruptCommand].includes(text?.trim())) {
+    if (![...systemCommand, ...stopCommand].includes(text?.trim())) {
       if (sessionStreamSuppressed.has(dcgSessionKey)) {
         sessionStreamSuppressed.delete(dcgSessionKey)
       }

package/src/channel.ts

@@ -107,39 +107,48 @@ export async function sendDcgchatMedia(opts: DcgchatMediaSendOptions): Promise<v
   const baseCtx = getOutboundMsgParams(sessionKey)
   const msgCtx = opts.messageId?.trim() ? { ...baseCtx, messageId: opts.messageId.trim() } : baseCtx
   if (!isWsOpen()) {
-    dcgLogger(`outbound media skipped -> ws not open: ${opts.mediaUrl ?? ''}`)
+    dcgLogger(`outbound media skipped -> ws not isWsOpen failed open: ${opts.mediaUrl ?? ''}`)
     return
   }
 
   const mediaUrl = opts.mediaUrl
-  const dedupeId = msgCtx.messageId
-  if (mediaUrl && dedupeId && hasSentMediaKey(dedupeId, mediaUrl)) {
-    dcgLogger(`dcgchat: sendMedia skipped (duplicate in session): ${mediaUrl}`)
-    return
-  }
-  if (mediaUrl && dedupeId) {
-    addSentMediaKey(dedupeId, mediaUrl)
+
+  if (mediaUrl && msgCtx.sessionId) {
+    if (hasSentMediaKey(msgCtx.sessionId, mediaUrl)) {
+      dcgLogger(`dcgchat: sendMedia skipped (hasSentMediaKey): ${mediaUrl} sessionId=${msgCtx.sessionId} sessionKey=${sessionKey}`)
+      return
+    }
+    addSentMediaKey(msgCtx.sessionId, mediaUrl)
   }
 
+  const { sessionId, agentId } = getInfoBySessionKey(sessionKey)
+  if (!msgCtx.sessionId) {
+    msgCtx.sessionId = sessionId
+  }
+  if (!mediaUrl || !msgCtx.sessionId) {
+    dcgLogger(`dcgchat: sendMedia skipped (duplicate in session): ${mediaUrl} sessionId=${msgCtx.sessionId} sessionKey=${sessionKey}`)
+    return
+  }
   const fileName = mediaUrl?.split(/[\\/]/).pop() || ''
-
+  const notMessageId = `${msgCtx?.messageId}`?.length !== 13 || !msgCtx?.messageId
   try {
     const botToken = msgCtx.botToken ?? getOpenClawConfig()?.channels?.["dcgchat-test"]?.botToken ?? ''
     const url = opts.mediaUrl ? await ossUpload(opts.mediaUrl, botToken, 1) : ''
-    if (!msgCtx.sessionId) {
-      const { sessionId, agentId } = getInfoBySessionKey(sessionKey)
-      msgCtx.sessionId = sessionId
+    if (!msgCtx.agentId) {
       msgCtx.agentId = agentId
     }
     wsSendRaw(msgCtx, {
       response: opts.text ?? '',
+      is_finish: notMessageId ? -1 : 0,
+      message_tags: notMessageId ? { mark: 'empty_text_file' } : {},
       files: [{ url, name: fileName }]
     })
     dcgLogger(`dcgchat: sendMedia session=${sessionKey}, file=${fileName}`)
   } catch (error) {
     wsSendRaw(msgCtx, {
       response: opts.text ?? '',
-      message_tags: { source: 'file' },
+      message_tags: notMessageId ? { mark: 'empty_text_file' } : {},
+      is_finish: notMessageId ? -1 : 0,
       files: [{ url: opts.mediaUrl ?? '', name: fileName }]
     })
     dcgLogger(`dcgchat: error sendMedia session=${sessionKey}: ${String(error)}`, 'error')
@@ -287,24 +296,10 @@ export const dcgchatPlugin: ChannelPlugin<ResolvedDcgchatAccount> = {
         const outboundCtx = getOutboundMsgParams(to)
         const content: Record<string, unknown> = { response: ctx.text }
         if (!isCron) {
-          // messageId = getCronMessageId(to) || `${Date.now()}`
-          // const { sessionId, agentId } = getInfoBySessionKey(to)
-          // content.is_finish = -1
-          // content.message_tags = { source: 'cron' }
-          // const merged = mergeDefaultParams({
-          //   agentId: agentId,
-          //   sessionId: `${sessionId}`,
-          //   messageId: messageId,
-          //   real_mobook: !sessionId ? 1 : ''
-          // })
-          // wsSendRaw(merged, content)
-          // } else {
           if (outboundCtx?.sessionId) {
             messageId = outboundCtx?.messageId || `${Date.now()}`
             const newCtx = { ...outboundCtx, messageId }
             wsSendRaw(newCtx, content)
-          } else {
-            dcgLogger(`channel sendText to ${ctx.to} -> sessionId not found`, 'error')
           }
         }
       }
@@ -323,8 +318,8 @@ export const dcgchatPlugin: ChannelPlugin<ResolvedDcgchatAccount> = {
       const cronMsgId = getCronMessageId(to)
       const fallbackMessageId = `${Date.now()}`
       const messageId = cronMsgId || (isCron ? fallbackMessageId : msgCtx?.messageId || fallbackMessageId)
-
-      if (!outboundCtx?.sessionId) {
+      const { sessionId } = getInfoBySessionKey(to)
+      if (!sessionId) {
         dcgLogger(`channel sendMedia to ${ctx.to} -> sessionId not found`, 'error')
         return {
           channel: "dcgchat-test",

package/src/cron.ts

@@ -135,7 +135,7 @@ export const onRunCronJob = async (jobId: string, messageId: string) => {
   )
   sendMessageToGateway(JSON.stringify({ method: 'cron.run', params: { id: jobId, mode: 'force' } }))
 }
-export const finishedDcgchatCron = async (jobId: string, summary: string) => {
+export const finishedDcgchatCron = async (jobId: string, summary: string, hasFileOutput?: boolean) => {
   const id = jobId?.trim()
   if (!id) {
     dcgLogger('finishedDcgchatCron: empty jobId', 'error')
@@ -159,7 +159,11 @@ export const finishedDcgchatCron = async (jobId: string, summary: string) => {
     messageId: messageId || `${Date.now()}`,
     real_mobook: !sessionId ? 1 : ''
   })
-  wsSendRaw(merged, { response: summary, message_tags: { source: 'cron' }, is_finish: -1 })
+  const message_tags = { source: 'cron' } as Record<string, string | boolean>
+  if (hasFileOutput) {
+    message_tags.hasFile = true
+  }
+  wsSendRaw(merged, { response: summary, message_tags, is_finish: -1 })
   setTimeout(() => {
     sendFinal(merged, 'cron send')
   }, 200)

package/src/cronToolCall.ts

@@ -117,33 +117,44 @@ function needsBestEffort(delivery: CronDelivery): boolean {
 }
 
 /**
- * 深拷贝 params 并注入 bestEffort: true
+ * 深拷贝 params，在 delivery 上写入 dcg 路由（to / accountId / sessionKey）。
+ * bestEffort + 默认 channel 仅在 needsBestEffort 为真时写入（announce 且无 channel）。
+ *
+ * 说明：原先仅在 needsBestEffort 为真时才改 delivery，若 jobs 里已有 channel 会整段跳过，
+ * 导致 `delivery.to` 永远不会被本钩子写入；运行时若缺 to 就会一直 not-delivered。
  */
-function injectBestEffort(params: Record<string, unknown>, sk: string): Record<string, unknown> {
+function patchCronDeliveryInParams(
+  params: Record<string, unknown>,
+  sk: string,
+  deliverySnapshot: CronDelivery
+): Record<string, unknown> | null {
   const newParams = JSON.parse(JSON.stringify(params)) as Record<string, unknown>
-  const { agentId, sessionId } = formatterSessionKey(sk)
-  // 顶层 delivery
+  const { agentId } = formatterSessionKey(sk)
+  const announceNoChannel = needsBestEffort(deliverySnapshot)
+
+  const apply = (d: CronDelivery) => {
+    d.to = `dcg-cron:${sk}`
+    if (agentId) d.accountId = agentId
+    if (announceNoChannel) {
+      d.bestEffort = true
+      d.channel = "dcgchat-test"
+    }
+  }
+
   if (newParams.delivery && typeof newParams.delivery === 'object') {
-    ;(newParams.delivery as CronDelivery).bestEffort = true
-    ;(newParams.delivery as CronDelivery).to = `dcg-cron:${sk}`
-    ;(newParams.delivery as CronDelivery).accountId = agentId
-    ;(newParams.delivery as CronDelivery).channel = "dcgchat-test"
+    apply(newParams.delivery as CronDelivery)
     newParams.sessionKey = sk
     return newParams
   }
 
-  // job.delivery
   const job = newParams.job as Record<string, unknown> | undefined
   if (job?.delivery && typeof job.delivery === 'object') {
-    ;(job.delivery as CronDelivery).bestEffort = true
-    ;(newParams.delivery as CronDelivery).to = `dcg-cron:${sk}`
-    ;(newParams.delivery as CronDelivery).accountId = agentId
-    ;(newParams.delivery as CronDelivery).channel = "dcgchat-test"
+    apply(job.delivery as CronDelivery)
     newParams.sessionKey = sk
     return newParams
   }
 
-  return newParams
+  return null
 }
 
 export function cronToolCall(event: { toolName: any; params: any; toolCallId: any }, sk: string) {
@@ -154,23 +165,27 @@ export function cronToolCall(event: { toolName: any; params: any; toolCallId: an
     const delivery = extractDelivery(params)
     if (!delivery) {
       dcgLogger(`[${LOG_TAG}] cron call (${toolCallId}) has no delivery config, skip.`)
-      return params
+      return undefined
+    }
+    const newParams = patchCronDeliveryInParams(params, sk, delivery)
+    if (!newParams) {
+      dcgLogger(`[${LOG_TAG}] cron call (${toolCallId}) could not locate delivery on params clone, skip.`)
+      return undefined
     }
-    if (!needsBestEffort(delivery)) {
+
+    const patched = newParams.delivery ?? (newParams.job as Record<string, unknown>)?.delivery
+    if (needsBestEffort(delivery)) {
       dcgLogger(
-        `[${LOG_TAG}] cron call (${toolCallId}) delivery does not need bestEffort ` +
-          `(mode=${String(delivery.mode)}, channel=${String(delivery.channel)}, bestEffort=${String(delivery.bestEffort)}), skip.`
+        `[${LOG_TAG}] cron call (${toolCallId}) patched delivery (announce, no channel): bestEffort=true, ` +
+          `delivery=${JSON.stringify(patched)}`
+      )
+    } else {
+      dcgLogger(
+        `[${LOG_TAG}] cron call (${toolCallId}) patched delivery.to / accountId (sessionKey=${sk}), ` +
+          `delivery=${JSON.stringify(patched)}`
       )
-      return params
     }
 
-    // ★ 核心：注入 bestEffort: true
-    const newParams = injectBestEffort(params, sk)
-    dcgLogger(
-      `[${LOG_TAG}] cron call (${toolCallId}) injected bestEffort=true ` +
-        `(mode=announce, no channel). delivery=${JSON.stringify(newParams.delivery || (newParams.job as Record<string, unknown>)?.delivery)}`
-    )
-
     return { params: newParams }
   } else if (toolName === 'exec') {
     if (params.command.indexOf('cron create') > -1 || params.command.indexOf('cron add') > -1) {
@@ -179,9 +194,9 @@ export function cronToolCall(event: { toolName: any; params: any; toolCallId: an
         params.command.replace('--json', '') + `  --session-key ${sk}  --channel ${"dcgchat-test"}  --to dcg-cron:${sk} --json`
       return { params: newParams }
     } else {
-      return params
+      return undefined
     }
   }
 
-  return params
+  return undefined
 }

package/src/tool.ts

@@ -325,6 +325,16 @@ function resolveHookSessionKey(
   return (args?.sessionKey || '').trim()
 }
 
+/** 定时触发时会话往往非 running，但仍需跑 before_tool_call 以注入 sessionKey / delivery（见 cronToolCall） */
+function shouldRunBeforeToolCallWithoutRunningSession(event: { toolName?: string; params?: { command?: string } }): boolean {
+  if (event?.toolName === 'cron') return true
+  const cmd = event?.params?.command
+  if (event?.toolName === 'exec' && typeof cmd === 'string') {
+    return cmd.includes('cron create') || cmd.includes('cron add')
+  }
+  return false
+}
+
 function trackSubagentLifecycle(eventName: string, event: any, args: any): void {
   if (eventName === 'subagent_spawned') {
     const runId = typeof event?.runId === 'string' ? event.runId : ''
@@ -351,7 +361,10 @@ export function monitoringToolMessage(api: OpenClawPluginApi) {
       trackSubagentLifecycle(item.event, event, args)
       const sk = resolveHookSessionKey(item.event, args ?? {})
       if (sk) {
-        if (isSessionActiveForTool(sk)) {
+        const toolHooksOk =
+          isSessionActiveForTool(sk) ||
+          (item.event === 'before_tool_call' && shouldRunBeforeToolCallWithoutRunningSession(event))
+        if (toolHooksOk) {
           if (['after_tool_call', 'before_tool_call'].includes(item.event)) {
             const { result: _result, ...rest } = event
             dcgLogger(`工具调用结果： ~ event:${item.event} ~ params:${JSON.stringify(rest)}`)

package/src/transport.ts

@@ -1,4 +1,4 @@
-import { getWsConnection } from './utils/global.js'
+import { clearSentMediaKeys, getWsConnection } from './utils/global.js'
 import { dcgLogger } from './utils/log.js'
 import type { IMsgParams } from './types.js'
 import { getEffectiveMsgParams, getParamsDefaults } from './utils/params.js'
@@ -170,6 +170,7 @@ export function sendChunk(text: string, ctx: IMsgParams, chunkIdx: number): bool
 
 export function sendFinal(ctx: IMsgParams, tag: string): boolean {
   dcgLogger(` message handling complete state: to=${ctx.sessionId} final tag:${tag}`)
+  clearSentMediaKeys(ctx.sessionId)
   return wsSend(ctx, { response: '', state: 'final' })
 }
 

package/src/utils/constant.ts

@@ -2,6 +2,6 @@ export const ENV: 'production' | 'test' | 'develop' = 'test'
 
 
 export const systemCommand = ['/new', '/status']
-export const interruptCommand = ['/stop']
+export const stopCommand = ['/stop']
 
-export const ignoreToolCommand = ['/search', '/abort', '/queue interrupt', ...systemCommand, ...interruptCommand]
+export const ignoreToolCommand = ['/search', '/abort', '/queue interrupt', ...systemCommand, ...stopCommand]

package/src/utils/gatewayMsgHanlder.ts

@@ -36,7 +36,12 @@ export function handleGatewayEventMessage(msg: { event?: string; payload?: Recor
         sendDcgchatCron(p?.jobId as string)
       }
       if (p?.action === 'finished' && p?.status === 'ok') {
-        finishedDcgchatCron(p?.jobId as string, p?.summary as string)
+        const hasFileOutput = p.delivered === true && p.deliveryStatus === 'delivered'
+        let summary = p?.summary as string
+        if (summary.indexOf('HEARTBEAT_OK') >= 0 && summary !== 'HEARTBEAT_OK') {
+          summary = summary.replace('HEARTBEAT_OK', '')
+        }
+        finishedDcgchatCron(p?.jobId as string, summary, hasFileOutput)
       }
     }
   } catch (error) {

package/src/utils/global.ts

@@ -82,11 +82,11 @@ const getMediaKey = (url: string) => url.split(/[\\/]/).pop() ?? url
 /** 已发送媒体去重：外层 messageId → 内层该会话下已发送的媒体 key（文件名） */
 const sentMediaKeysBySession = new Map<string, Set<string>>()
 
-function getSessionMediaSet(messageId: string): Set<string> {
-  let set = sentMediaKeysBySession.get(messageId)
+function getSessionMediaSet(sessionId: string): Set<string> {
+  let set = sentMediaKeysBySession.get(sessionId)
   if (!set) {
     set = new Set<string>()
-    sentMediaKeysBySession.set(messageId, set)
+    sentMediaKeysBySession.set(sessionId, set)
   }
   return set
 }
@@ -95,14 +95,14 @@ export function addSentMediaKey(messageId: string, url: string) {
   getSessionMediaSet(messageId).add(getMediaKey(url))
 }
 
-export function hasSentMediaKey(messageId: string, url: string): boolean {
-  return sentMediaKeysBySession.get(messageId)?.has(getMediaKey(url)) ?? false
+export function hasSentMediaKey(sessionId: string, url: string): boolean {
+  return sentMediaKeysBySession.get(sessionId)?.has(getMediaKey(url)) ?? false
 }
 
 /** 不传 messageId 时清空全部会话；传入则只清空该会话 */
-export function clearSentMediaKeys(messageId?: string) {
-  if (messageId) {
-    sentMediaKeysBySession.delete(messageId)
+export function clearSentMediaKeys(sessionId?: string) {
+  if (sessionId) {
+    sentMediaKeysBySession.delete(sessionId)
   } else {
     sentMediaKeysBySession.clear()
   }