瀏覽代碼

🐛 fix(message): ensure retry uses current selected model instead of stale one

- Add onMessageReset reference comparison to OptimizedMessageActions memo
- Force component re-render when model selection changes
- Prevent stale closure issue in retry functionality
- Ensure first retry attempt uses newly selected model

Previously, when changing the model selection, the retry button would
still use the previous model due to React memo optimization preventing
re-renders. By comparing the onMessageReset callback reference, the
component now properly updates when the model changes, ensuring the
retry functionality immediately uses the currently selected model.
Apple\Apple 10 月之前
父節點
當前提交
2250f35a7e
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      web/src/components/playground/OptimizedComponents.js

+ 2 - 1
web/src/components/playground/OptimizedComponents.js

@@ -26,7 +26,8 @@ export const OptimizedMessageActions = React.memo(MessageActions, (prevProps, ne
     prevProps.message.id === nextProps.message.id &&
     prevProps.message.role === nextProps.message.role &&
     prevProps.isAnyMessageGenerating === nextProps.isAnyMessageGenerating &&
-    prevProps.isEditing === nextProps.isEditing
+    prevProps.isEditing === nextProps.isEditing &&
+    prevProps.onMessageReset === nextProps.onMessageReset
   );
 });