From d417bec3ca8fb5a9207a69d0eead04cea52658e7 Mon Sep 17 00:00:00 2001 From: Zhang Rui Date: Wed, 18 Mar 2026 16:33:52 +0800 Subject: [PATCH] fix(wecom): improve WS image message handling by ensuring proper error response and initializing mediaRefs --- pkg/channels/wecom/aibot_ws.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pkg/channels/wecom/aibot_ws.go b/pkg/channels/wecom/aibot_ws.go index c748ac4b1..584b65cd9 100644 --- a/pkg/channels/wecom/aibot_ws.go +++ b/pkg/channels/wecom/aibot_ws.go @@ -662,14 +662,15 @@ func (c *WeComAIBotWSChannel) handleWSImageMessage(reqID string, msg WeComAIBotW ctx, cancel := context.WithTimeout(c.ctx, wsImageDownloadTimeout) defer cancel() - var mediaRefs []string + mediaRefs := make([]string, 0, 1) ref, err := c.storeWSImage(ctx, chatID, msg.MsgID, msg.Image.URL, msg.Image.AESKey) if err != nil { logger.WarnCF("wecom_aibot", "Failed to download/store WS image", map[string]any{"error": err.Error(), "url": msg.Image.URL}) - } else { - mediaRefs = append(mediaRefs, ref) + c.wsSendStreamFinish(reqID, wsGenerateID(), "Image message could not be processed.") + return } + mediaRefs = append(mediaRefs, ref) c.dispatchWSAgentTask(reqID, msg, "[image]", mediaRefs) }