Stabilize web timeline height estimates

2026-04-24 11:55:43 +07:00
parent 480b18c500
commit 0c3d15ffb9
5 changed files with 201 additions and 7 deletions
@@ -0,0 +1,47 @@
+import { beforeEach, describe, expect, it } from "vitest";
+import {
+  clearAssistantMessageHeightEstimateCache,
+  estimateAssistantMessageHeightFromCache,
+  setAssistantMarkdownBlockHeight,
+} from "./assistant-message-height-estimate";
+import {
+  clearAssistantImageMetadataCache,
+  setAssistantImageMetadata,
+} from "./assistant-image-metadata";
+
+describe("assistant message height estimate", () => {
+  beforeEach(() => {
+    clearAssistantMessageHeightEstimateCache();
+    clearAssistantImageMetadataCache();
+  });
+
+  it("estimates assistant message height from measured markdown block heights", () => {
+    setAssistantMarkdownBlockHeight({
+      block: "First paragraph",
+      width: 804,
+      height: 18.2,
+    });
+    setAssistantMarkdownBlockHeight({
+      block: "Second paragraph",
+      width: 804,
+      height: 41.1,
+    });
+
+    expect(estimateAssistantMessageHeightFromCache("First paragraph\n\nSecond paragraph")).toBe(97);
+  });
+
+  it("falls back to image metadata when markdown blocks are not measured", () => {
+    setAssistantImageMetadata(
+      {
+        source: "https://example.com/landscape.png",
+      },
+      { width: 1200, height: 800 },
+    );
+
+    expect(
+      estimateAssistantMessageHeightFromCache(
+        "Here is the screenshot\n\n![Screenshot](https://example.com/landscape.png)",
+      ),
+    ).toBeGreaterThan(220);
+  });
+});