fix(ui): preserve upstream behavior after performance rebase

packages/ui/src/components/chat/markdown/detachedMarkdownDomCache.test.ts

@@ -5,7 +5,7 @@ import { DetachedMarkdownDomCache, type DetachedMarkdownDom } from './detachedMa
 
 Object.assign(globalThis, { document: new Window().document });
 
-const keyFor = ({ scope, id, locale }: DetachedMarkdownDom) => ({ scope, id, locale });
+const keyFor = ({ scope, id, locale, directory }: DetachedMarkdownDom) => ({ scope, id, locale, directory });
 
 const createEntry = (
   document: Document,
@@ -21,6 +21,7 @@ const createEntry = (
     scope: `runtime:${sessionId}`,
     id: `${messageId}:${partId}`,
     locale: 'en',
+    directory: '/repo-a',
     fragment,
   };
 };
@@ -51,11 +52,13 @@ describe('DetachedMarkdownDomCache', () => {
       scope: 'runtime:session-a',
       id: 'message-2:part',
       locale: 'en',
+      directory: '/repo-a',
     })).toBeNull();
     expect(cache.take({
       scope: 'runtime:session-c',
       id: 'message-5:part',
       locale: 'en',
+      directory: '/repo-a',
     })).not.toBeNull();
   });
 
@@ -75,6 +78,15 @@ describe('DetachedMarkdownDomCache', () => {
     expect(cache.take(keyFor(replacement))?.firstChild).toBe(replacementNode);
   });
 
+  test('does not restore file-link DOM under another directory', () => {
+    const cache = new DetachedMarkdownDomCache({ maxSessions: 2, maxEntriesPerSession: 2 });
+    const entry = createEntry(document, 'session', 'message', 'part');
+
+    cache.store(entry);
+
+    expect(cache.take({ ...keyFor(entry), directory: '/repo-b' })).toBeNull();
+  });
+
   test('refreshes session LRU and clears all entries', () => {
     const cache = new DetachedMarkdownDomCache({ maxSessions: 2, maxEntriesPerSession: 2 });
     const sessionA = createEntry(document, 'session-a', 'message-a', 'part');

packages/ui/src/components/chat/markdown/detachedMarkdownDomCache.ts

@@ -2,6 +2,7 @@ export type DetachedMarkdownDomKey = {
   scope: string;
   id: string;
   locale: string;
+  directory: string;
 };
 
 export type DetachedMarkdownDom = DetachedMarkdownDomKey & {
@@ -79,7 +80,7 @@ export class DetachedMarkdownDomCache {
     // A fragment is a move-only resource; taking it removes cache ownership.
     session.delete(entryKey);
     if (session.size === 0) this.sessions.delete(sessionKey);
-    if (entry.locale !== key.locale) return null;
+    if (entry.locale !== key.locale || entry.directory !== key.directory) return null;
     return entry.fragment;
   }