From 840277f5846ab13eaec0f3848ebd86d3a4ade410 Mon Sep 17 00:00:00 2001
From: Yidadaa <yidadaa@qq.com>
Date: Tue, 15 Aug 2023 22:42:55 +0800
Subject: [PATCH] fix: #2566 click avatar to edit context messages

---
 app/components/chat.tsx | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/app/components/chat.tsx b/app/components/chat.tsx
index f661d0a4..65620858 100644
--- a/app/components/chat.tsx
+++ b/app/components/chat.tsx
@@ -1123,9 +1123,9 @@ function _Chat() {
                               10,
                             );
                             chatStore.updateCurrentSession((session) => {
-                              const m = session.messages.find(
-                                (m) => m.id === message.id,
-                              );
+                              const m = session.mask.context
+                                .concat(session.messages)
+                                .find((m) => m.id === message.id);
                               if (m) {
                                 m.content = newMessage;
                               }