aa49d2a360
Fix role toggle functionality where switching message roles (assistant/system) did not update the UI immediately and required page refresh to see changes. Changes: - Add message.role comparison in OptimizedMessageContent memo function - Add message.role comparison in OptimizedMessageActions memo function The issue was caused by React.memo optimization that wasn't tracking role changes, preventing re-renders when only the message role property changed. Now role switches are reflected immediately in both message content display and action button states. Fixes: Role switching requires page refresh to display correctly
58 lines
2.6 KiB
JavaScript
58 lines
2.6 KiB
JavaScript
import React from 'react';
|
|
import MessageContent from './MessageContent';
|
|
import MessageActions from './MessageActions';
|
|
import SettingsPanel from './SettingsPanel';
|
|
import DebugPanel from './DebugPanel';
|
|
|
|
// 优化的消息内容组件
|
|
export const OptimizedMessageContent = React.memo(MessageContent, (prevProps, nextProps) => {
|
|
// 只有这些属性变化时才重新渲染
|
|
return (
|
|
prevProps.message.id === nextProps.message.id &&
|
|
prevProps.message.content === nextProps.message.content &&
|
|
prevProps.message.status === nextProps.message.status &&
|
|
prevProps.message.role === nextProps.message.role &&
|
|
prevProps.message.isReasoningExpanded === nextProps.message.isReasoningExpanded &&
|
|
prevProps.isEditing === nextProps.isEditing &&
|
|
prevProps.editValue === nextProps.editValue &&
|
|
prevProps.styleState.isMobile === nextProps.styleState.isMobile
|
|
);
|
|
});
|
|
|
|
// 优化的消息操作组件
|
|
export const OptimizedMessageActions = React.memo(MessageActions, (prevProps, nextProps) => {
|
|
return (
|
|
prevProps.message.id === nextProps.message.id &&
|
|
prevProps.message.role === nextProps.message.role &&
|
|
prevProps.isAnyMessageGenerating === nextProps.isAnyMessageGenerating &&
|
|
prevProps.isEditing === nextProps.isEditing
|
|
);
|
|
});
|
|
|
|
// 优化的设置面板组件
|
|
export const OptimizedSettingsPanel = React.memo(SettingsPanel, (prevProps, nextProps) => {
|
|
return (
|
|
JSON.stringify(prevProps.inputs) === JSON.stringify(nextProps.inputs) &&
|
|
JSON.stringify(prevProps.parameterEnabled) === JSON.stringify(nextProps.parameterEnabled) &&
|
|
JSON.stringify(prevProps.models) === JSON.stringify(nextProps.models) &&
|
|
JSON.stringify(prevProps.groups) === JSON.stringify(nextProps.groups) &&
|
|
prevProps.customRequestMode === nextProps.customRequestMode &&
|
|
prevProps.customRequestBody === nextProps.customRequestBody &&
|
|
prevProps.showDebugPanel === nextProps.showDebugPanel &&
|
|
prevProps.showSettings === nextProps.showSettings &&
|
|
JSON.stringify(prevProps.previewPayload) === JSON.stringify(nextProps.previewPayload) &&
|
|
JSON.stringify(prevProps.messages) === JSON.stringify(nextProps.messages)
|
|
);
|
|
});
|
|
|
|
// 优化的调试面板组件
|
|
export const OptimizedDebugPanel = React.memo(DebugPanel, (prevProps, nextProps) => {
|
|
return (
|
|
prevProps.show === nextProps.show &&
|
|
prevProps.activeTab === nextProps.activeTab &&
|
|
JSON.stringify(prevProps.debugData) === JSON.stringify(nextProps.debugData) &&
|
|
JSON.stringify(prevProps.previewPayload) === JSON.stringify(nextProps.previewPayload) &&
|
|
prevProps.customRequestMode === nextProps.customRequestMode &&
|
|
prevProps.showDebugPanel === nextProps.showDebugPanel
|
|
);
|
|
});
|