fix(ui): stabilize virtual message list measurements

2026-03-07 21:08:06 +00:00
parent 0d215342e3
commit c64a9a03f9
4 changed files with 327 additions and 39 deletions
--- a/packages/ui/src/components/message-section.tsx
+++ b/packages/ui/src/components/message-section.tsx
@@ -1066,7 +1066,7 @@ export default function MessageSection(props: MessageSectionProps) {
              </Show>
            </>
          )}
-          renderItem={(messageId, index) => (
+          renderItem={(messageId, index, options) => (
            <MessageBlock
              messageId={messageId}
              instanceId={props.instanceId}
@@ -1085,7 +1085,11 @@ export default function MessageSection(props: MessageSectionProps) {
              onRevert={props.onRevert}
              onDeleteMessagesUpTo={props.onDeleteMessagesUpTo}
              onFork={props.onFork}
-              onContentRendered={handleContentRendered}
+              onContentRendered={() => {
+                options.notifyItemRendered()
+                handleContentRendered()
+              }}
+              onMeasureElementChange={options.registerMeasureElement}
            />
          )}
          renderOverlay={() => (