@bojackduy/opencode-telescope 0.1.14 → 0.1.15

package/package.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://json.schemastore.org/package.json",
   "name": "@bojackduy/opencode-telescope",
-  "version": "0.1.14",
+  "version": "0.1.15",
   "description": "Fuzzy search across all OpenCode conversations — grep session and chat history, find code snippets, and jump to any chat instantly",
   "type": "module",
   "license": "MIT",

package/search.ts

@@ -11,6 +11,7 @@ export type SearchResult = {
   sessionTitle: string
   directory: string
   role: "user" | "assistant"
+  partType: "text" | "reasoning" | "tool" | "patch"
   timeCreated: number
   snippet: string
   matchStart: number
@@ -67,10 +68,15 @@ type Row = {
   session_title: string | null
   directory: string
   role: SearchRole
+  part_type?: SearchResult["partType"]
   time_created: number
   text: string
 }
 
+type SourceRow = Omit<Row, "text"> & {
+  data: string
+}
+
 type ConversationRow = {
   id: string
   message_id: string
@@ -318,6 +324,7 @@ export function rowToSearchResult(row: Row, query: string): SearchResult | undef
     sessionTitle: row.session_title || "Untitled session",
     directory: row.directory,
     role: row.role,
+    partType: row.part_type ?? "text",
     timeCreated: row.time_created,
     snippet: makeSnippet(text, query),
     matchStart: match.start,
@@ -337,7 +344,7 @@ export function rowToSearchResult(row: Row, query: string): SearchResult | undef
 
 export function extractSearchText(data: string) {
   try {
-    return extractFromValue(JSON.parse(data)).replace(/\s+/g, " ").trim()
+    return searchablePartText(JSON.parse(data)).replace(/\s+/g, " ").trim()
   } catch {
     return data.replace(/\s+/g, " ").trim()
   }
@@ -569,7 +576,7 @@ function indexedTextRows(db: Database, dbPath: string, limit: number, query: str
     debug.time("query:fts:exec")
     const rows = index.query<Row, (string | number)[]>(`
       SELECT id, message_id, session_id, session_title, directory, role,
-             CAST(time_created AS INTEGER) AS time_created, text
+             part_type, CAST(time_created AS INTEGER) AS time_created, text
       FROM document_fts
       WHERE ${conditions.join(" AND ")}
       ORDER BY bm25(document_fts), CAST(time_created AS INTEGER) DESC
@@ -586,7 +593,7 @@ function indexedTextRows(db: Database, dbPath: string, limit: number, query: str
 function visibleTextRows(db: Database, limit: number, query?: string, directory?: string, offset?: number, role?: SearchRole) {
   const offsetClause = offset ? "OFFSET ?" : ""
   const conditions: string[] = [
-    "json_extract(p.data, '$.type') = 'text'",
+    "json_extract(p.data, '$.type') IN ('text', 'reasoning', 'tool', 'patch')",
     role ? "json_extract(m.data, '$.role') = ?" : "json_extract(m.data, '$.role') IN ('user', 'assistant')",
   ]
   const params: (string | number)[] = []
@@ -600,7 +607,7 @@ function visibleTextRows(db: Database, limit: number, query?: string, directory?
 
   const tokens = query ? query.trim().split(/\s+/).filter(Boolean) : []
   for (const token of tokens) {
-    conditions.push("json_extract(p.data, '$.text') LIKE ?")
+    conditions.push("p.data LIKE ?")
     params.push(`%${token}%`)
   }
 
@@ -610,8 +617,9 @@ function visibleTextRows(db: Database, limit: number, query?: string, directory?
   const sql = `
     SELECT p.id, p.message_id, p.session_id, s.title AS session_title, s.directory,
            json_extract(m.data, '$.role') AS role,
+           json_extract(p.data, '$.type') AS part_type,
            p.time_created,
-           json_extract(p.data, '$.text') AS text
+           p.data
     FROM part p
     JOIN message m ON m.id = p.message_id
     JOIN session s ON s.id = p.session_id
@@ -620,9 +628,9 @@ function visibleTextRows(db: Database, limit: number, query?: string, directory?
     LIMIT ? ${offsetClause}
   `
   debug.time("query:sql:exec")
-  const rows = db.query<Row, (string | number)[]>(sql).all(...params as any[])
+  const rows = db.query<SourceRow, (string | number)[]>(sql).all(...params as any[])
   debug.timeEnd("query:sql:exec")
-  return rows
+  return rows.flatMap(sourceRowToSearchRow)
 }
 
 function ensureSearchIndex(source: Database, sourcePath: string) {
@@ -638,12 +646,15 @@ function ensureSearchIndex(source: Database, sourcePath: string) {
   const currentDataVersion = getMeta(_indexDb, "source_data_version")
   const currentMtimeMs = getMeta(_indexDb, "source_mtime_ms")
   const currentPath = getMeta(_indexDb, "source_path")
-  if (currentPath !== sourcePath || currentDataVersion !== String(state.dataVersion) || currentMtimeMs !== String(state.mtimeMs)) {
+  const currentIndexVersion = getMeta(_indexDb, "index_version")
+  if (currentPath !== sourcePath || currentDataVersion !== String(state.dataVersion) || currentMtimeMs !== String(state.mtimeMs) || currentIndexVersion !== SEARCH_INDEX_VERSION) {
     rebuildSearchIndex(source, _indexDb, sourcePath, state)
   }
   return _indexDb
 }
 
+const SEARCH_INDEX_VERSION = "3"
+
 function searchIndexPath(sourcePath: string) {
   const parsed = path.parse(sourcePath)
   return path.join(parsed.dir, `${parsed.name}-search.db`)
@@ -662,11 +673,31 @@ function migrateSearchIndex(db: Database) {
       session_title,
       directory UNINDEXED,
       role UNINDEXED,
+      part_type UNINDEXED,
       time_created UNINDEXED,
       text,
       tokenize='unicode61'
     );
   `)
+
+  const hasPartType = db.query<{ name: string }, []>("PRAGMA table_info(document_fts)").all().some((column) => column.name === "part_type")
+  if (!hasPartType) {
+    db.exec("DROP TABLE document_fts")
+    db.exec(`
+      CREATE VIRTUAL TABLE document_fts USING fts5(
+        id UNINDEXED,
+        message_id UNINDEXED,
+        session_id UNINDEXED,
+        session_title,
+        directory UNINDEXED,
+        role UNINDEXED,
+        part_type UNINDEXED,
+        time_created UNINDEXED,
+        text,
+        tokenize='unicode61'
+      );
+    `)
+  }
 }
 
 function sourceState(db: Database, sourcePath: string) {
@@ -677,27 +708,27 @@ function sourceState(db: Database, sourcePath: string) {
 
 function rebuildSearchIndex(source: Database, index: Database, sourcePath: string, state: { dataVersion: number; mtimeMs: number }) {
   debug.time("fts:rebuild")
-  const rows = source.query<Row, []>(`
+  const rows = source.query<SourceRow, []>(`
     SELECT p.id, p.message_id, p.session_id, s.title AS session_title, s.directory,
-           json_extract(m.data, '$.role') AS role,
-           p.time_created,
-           json_extract(p.data, '$.text') AS text
+            json_extract(m.data, '$.role') AS role,
+            json_extract(p.data, '$.type') AS part_type,
+            p.time_created,
+            p.data
     FROM part p
     JOIN message m ON m.id = p.message_id
     JOIN session s ON s.id = p.session_id
-    WHERE json_extract(p.data, '$.type') = 'text'
+    WHERE json_extract(p.data, '$.type') IN ('text', 'reasoning', 'tool', 'patch')
       AND json_extract(m.data, '$.role') IN ('user', 'assistant')
-      AND json_extract(p.data, '$.text') IS NOT NULL
     ORDER BY p.time_created DESC
   `).all()
-  const insert = index.query<Row, [string, string, string, string, string, string, number, string]>(`
-    INSERT INTO document_fts(id, message_id, session_id, session_title, directory, role, time_created, text)
-    VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+  const insert = index.query<Row, [string, string, string, string, string, string, SearchResult["partType"], number, string]>(`
+    INSERT INTO document_fts(id, message_id, session_id, session_title, directory, role, part_type, time_created, text)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
   `)
   index.exec("BEGIN IMMEDIATE")
   try {
     index.exec("DELETE FROM document_fts")
-    for (const row of rows) {
+    for (const row of rows.flatMap(sourceRowToSearchRow)) {
       insert.run(
         row.id,
         row.message_id,
@@ -705,6 +736,7 @@ function rebuildSearchIndex(source: Database, index: Database, sourcePath: strin
         row.session_title ?? "Untitled session",
         row.directory,
         row.role,
+        row.part_type ?? "text",
         row.time_created,
         row.text,
       )
@@ -712,6 +744,7 @@ function rebuildSearchIndex(source: Database, index: Database, sourcePath: strin
     setMeta(index, "source_path", sourcePath)
     setMeta(index, "source_data_version", String(state.dataVersion))
     setMeta(index, "source_mtime_ms", String(state.mtimeMs))
+    setMeta(index, "index_version", SEARCH_INDEX_VERSION)
     index.exec("COMMIT")
   } catch (err) {
     index.exec("ROLLBACK")
@@ -783,6 +816,29 @@ function extractFromValue(value: unknown): string {
     .join("\n")
 }
 
+function sourceRowToSearchRow(row: SourceRow): Row[] {
+  const text = extractSearchText(row.data)
+  if (!text) return []
+  return [{ ...row, text }]
+}
+
+function searchablePartText(value: unknown): string {
+  if (!value || typeof value !== "object" || Array.isArray(value)) return extractFromValue(value)
+  const record = value as Record<string, unknown>
+  const type = record.type
+
+  if (type === "text" || type === "reasoning") return typeof record.text === "string" ? record.text : extractFromValue(record)
+  if (type === "patch") return extractFromValue(record.files)
+
+  if (type === "tool") {
+    const state = record.state && typeof record.state === "object" && !Array.isArray(record.state) ? record.state as Record<string, unknown> : undefined
+    if (record.tool !== "apply_patch") return ""
+    return extractFromValue(state?.input)
+  }
+
+  return extractFromValue(record)
+}
+
 function truncate(value: string, length: number) {
   if (value.length <= length) return value
   return `${value.slice(0, length - 3)}...`

package/telescope.tsx

@@ -46,14 +46,14 @@ export const Telescope = (props: { api: TuiPluginApi; config: TelescopeConfig; o
   const [loadingPreviewMore, setLoadingPreviewMore] = createSignal(false)
   const [prefetchingPreviewBefore, setPrefetchingPreviewBefore] = createSignal(false)
   const [prefetchingPreviewAfter, setPrefetchingPreviewAfter] = createSignal(false)
-  const MIN_SEARCH_BATCH_SIZE = 25
-  const MIN_RECENT_BATCH_SIZE = 15
+  const [previewEdgeLoadingReady, setPreviewEdgeLoadingReady] = createSignal(false)
+  const RESULT_BATCH_SIZE = 50
+  const RESULT_PREFETCH_AHEAD_ROWS = 25
   const RESULT_OVERSCAN_MULTIPLIER = 2
-  const RESULT_BATCH_VIEWPORTS = 4
-  const RESULT_PREFETCH_VIEWPORTS = 10
   const RESULT_CACHE_BEHIND_VIEWPORTS = 6
-  const INITIAL_PREVIEW_BEFORE = 20
-  const INITIAL_PREVIEW_AFTER = 30
+  const INITIAL_PREVIEW_BEFORE = 6
+  const INITIAL_PREVIEW_AFTER = 12
+  const INITIAL_PREVIEW_DELAY_MS = 75
   const PREVIEW_PAGE_SIZE = 20
   const PREVIEW_PREFETCH_VIEWPORTS = 0.5
   let input: InputRenderable | undefined
@@ -91,18 +91,19 @@ export const Telescope = (props: { api: TuiPluginApi; config: TelescopeConfig; o
   let resultPreviousTimer: ReturnType<typeof setTimeout> | undefined
   let previewBeforeTimer: ReturnType<typeof setTimeout> | undefined
   let previewAfterTimer: ReturnType<typeof setTimeout> | undefined
-  let pendingPreviewBefore: { previousContentHeight: number; preserveScroll: boolean; visibleLoad: boolean } | undefined
-  let pendingPreviewAfterVisible = false
+  let pendingPreviewBefore: { itemId: string; previousContentHeight: number; preserveScroll: boolean; visibleLoad: boolean } | undefined
+  let pendingPreviewAfter: { itemId: string; visibleLoad: boolean } | undefined
   const cancelPreviewPrefetch = () => {
     if (previewBeforeTimer) clearTimeout(previewBeforeTimer)
     if (previewAfterTimer) clearTimeout(previewAfterTimer)
     previewBeforeTimer = undefined
     previewAfterTimer = undefined
     pendingPreviewBefore = undefined
-    pendingPreviewAfterVisible = false
+    pendingPreviewAfter = undefined
     setPrefetchingPreviewBefore(false)
     setPrefetchingPreviewAfter(false)
     setLoadingPreviewMore(false)
+    setPreviewEdgeLoadingReady(false)
   }
   onCleanup(() => {
     if (resultPrefetchTimer) clearTimeout(resultPrefetchTimer)
@@ -115,9 +116,9 @@ export const Telescope = (props: { api: TuiPluginApi; config: TelescopeConfig; o
     const viewportHeight = resultScroll?.height || Math.max(8, height() - 7)
     return Math.max(5, Math.floor(viewportHeight / resultRowHeight()))
   }
-  const searchBatchSize = () => Math.max(MIN_SEARCH_BATCH_SIZE, visibleResultRows() * RESULT_BATCH_VIEWPORTS)
-  const recentBatchSize = () => Math.max(MIN_RECENT_BATCH_SIZE, visibleResultRows() * RESULT_BATCH_VIEWPORTS)
-  const resultPrefetchThreshold = () => visibleResultRows() * RESULT_PREFETCH_VIEWPORTS
+  const searchBatchSize = () => RESULT_BATCH_SIZE
+  const recentBatchSize = () => RESULT_BATCH_SIZE
+  const resultPrefetchThreshold = () => RESULT_PREFETCH_AHEAD_ROWS
   const resultPrefetchState = () => {
     const cachedEnd = resultBaseOffset() + results().length
     const rowsAhead = Math.max(0, cachedEnd - selected() - 1)
@@ -437,14 +438,20 @@ export const Telescope = (props: { api: TuiPluginApi; config: TelescopeConfig; o
     return lastChild ? lastChild.y + lastChild.height : 0
   }
 
-  const loadPreviewBefore = (previousContentHeight = previewContentHeight(), preserveScroll = true, visibleLoad = false) => {
+  const resetPreviewScroll = () => {
+    if (!previewScroll || previewScroll.y === 0) return
+    previewScroll.scrollBy(-previewScroll.y)
+  }
+
+  const loadPreviewBefore = (itemId: string, previousContentHeight = previewContentHeight(), preserveScroll = true, visibleLoad = false) => {
     const item = selectedResult()
     const first = previewParts()[0]
-    if (!item || !first || loadingPreviewMore() || prefetchingPreviewBefore()) return
+    if (!item || item.id !== itemId || !first || loadingPreviewMore() || prefetchingPreviewBefore()) return
     visibleLoad ? setLoadingPreviewMore(true) : setPrefetchingPreviewBefore(true)
     debug.time("preview:load-before")
     try {
       const page = loadConversationBefore(item, { id: first.id, timeCreated: first.timeCreated }, { limit: PREVIEW_PAGE_SIZE, dbPath: dbPath() })
+      if (selectedResult()?.id !== itemId) return
       debug.log("preview:load-before", {
         item: item.id,
         added: page.parts.length,
@@ -458,6 +465,7 @@ export const Telescope = (props: { api: TuiPluginApi; config: TelescopeConfig; o
         setPreviewParts((prev) => [...page.parts, ...prev])
         if (preserveScroll) {
           setTimeout(() => {
+            if (selectedResult()?.id !== itemId) return
             const delta = previewContentHeight() - previousContentHeight
             if (delta > 0) previewScroll?.scrollBy(delta)
           }, 1)
@@ -472,14 +480,15 @@ export const Telescope = (props: { api: TuiPluginApi; config: TelescopeConfig; o
     }
   }
 
-  const loadPreviewAfter = (visibleLoad = false) => {
+  const loadPreviewAfter = (itemId: string, visibleLoad = false) => {
     const item = selectedResult()
     const last = previewParts().at(-1)
-    if (!item || !last || loadingPreviewMore() || prefetchingPreviewAfter()) return
+    if (!item || item.id !== itemId || !last || loadingPreviewMore() || prefetchingPreviewAfter()) return
     visibleLoad ? setLoadingPreviewMore(true) : setPrefetchingPreviewAfter(true)
     debug.time("preview:load-after")
     try {
       const page = loadConversationAfter(item, { id: last.id, timeCreated: last.timeCreated }, { limit: PREVIEW_PAGE_SIZE, dbPath: dbPath() })
+      if (selectedResult()?.id !== itemId) return
       debug.log("preview:load-after", {
         item: item.id,
         added: page.parts.length,
@@ -499,7 +508,8 @@ export const Telescope = (props: { api: TuiPluginApi; config: TelescopeConfig; o
   }
 
   const schedulePreviewBefore = (previousContentHeight = previewContentHeight(), preserveScroll = true, visibleLoad = false) => {
-    if (!hasMorePreviewBefore() || loadingPreviewMore() || prefetchingPreviewBefore()) return
+    const item = selectedResult()
+    if (!item || !hasMorePreviewBefore() || loadingPreviewMore() || prefetchingPreviewBefore()) return
     if (previewBeforeTimer) {
       if (pendingPreviewBefore) {
         pendingPreviewBefore.preserveScroll = pendingPreviewBefore.preserveScroll && preserveScroll
@@ -507,26 +517,27 @@ export const Telescope = (props: { api: TuiPluginApi; config: TelescopeConfig; o
       }
       return
     }
-    pendingPreviewBefore = { previousContentHeight, preserveScroll, visibleLoad }
-    debug.log("preview:prefetch-before-scheduled", { preserveScroll, visibleLoad })
+    pendingPreviewBefore = { itemId: item.id, previousContentHeight, preserveScroll, visibleLoad }
+    debug.log("preview:prefetch-before-scheduled", { item: item.id, preserveScroll, visibleLoad })
     previewBeforeTimer = setTimeout(() => {
       const pending = pendingPreviewBefore
       previewBeforeTimer = undefined
       pendingPreviewBefore = undefined
-      if (pending) loadPreviewBefore(pending.previousContentHeight, pending.preserveScroll, pending.visibleLoad)
+      if (pending && selectedResult()?.id === pending.itemId) loadPreviewBefore(pending.itemId, pending.previousContentHeight, pending.preserveScroll, pending.visibleLoad)
     }, 1)
   }
 
   const schedulePreviewAfter = (visibleLoad = false) => {
-    if (!hasMorePreviewAfter() || loadingPreviewMore() || prefetchingPreviewAfter()) return
-    pendingPreviewAfterVisible = pendingPreviewAfterVisible || visibleLoad
+    const item = selectedResult()
+    if (!item || !hasMorePreviewAfter() || loadingPreviewMore() || prefetchingPreviewAfter()) return
+    pendingPreviewAfter = { itemId: item.id, visibleLoad: (pendingPreviewAfter?.visibleLoad || visibleLoad) }
     if (previewAfterTimer) return
-    debug.log("preview:prefetch-after-scheduled", { visibleLoad })
+    debug.log("preview:prefetch-after-scheduled", { item: item.id, visibleLoad })
     previewAfterTimer = setTimeout(() => {
-      const pendingVisibleLoad = pendingPreviewAfterVisible
+      const pending = pendingPreviewAfter
       previewAfterTimer = undefined
-      pendingPreviewAfterVisible = false
-      loadPreviewAfter(pendingVisibleLoad)
+      pendingPreviewAfter = undefined
+      if (pending && selectedResult()?.id === pending.itemId) loadPreviewAfter(pending.itemId, pending.visibleLoad)
     }, 1)
   }
 
@@ -543,33 +554,62 @@ export const Telescope = (props: { api: TuiPluginApi; config: TelescopeConfig; o
     if (item.id === lastPreviewItemId) return
     lastPreviewItemId = item.id
     cancelPreviewPrefetch()
+    resetPreviewScroll()
+    solidBatch(() => {
+      setPreviewParts([])
+      setHasMorePreviewBefore(false)
+      setHasMorePreviewAfter(false)
+      setPreviewEdgeLoadingReady(false)
+    })
     debug.log("preview:new-item", item.sessionTitle?.slice(0, 40) ?? item.id.slice(-8))
     const db = dbPath()
-    debug.time("preview:load")
-    try {
-      const page = loadConversationAround(item, { before: INITIAL_PREVIEW_BEFORE, after: INITIAL_PREVIEW_AFTER, dbPath: db })
-      debug.log("preview:init", {
-        item: item.id,
-        session: item.sessionID,
-        parts: page.parts.length,
-        hasMoreBefore: page.hasMoreBefore,
-        hasMoreAfter: page.hasMoreAfter,
-        first: page.parts[0]?.id,
-        last: page.parts.at(-1)?.id,
-      })
-      setPreviewParts(page.parts)
-      setHasMorePreviewBefore(page.hasMoreBefore)
-      setHasMorePreviewAfter(page.hasMoreAfter)
-    } catch {}
+    const itemId = item.id
+    debug.log("preview:schedule", { item: item.id, delayMs: INITIAL_PREVIEW_DELAY_MS, before: INITIAL_PREVIEW_BEFORE, after: INITIAL_PREVIEW_AFTER })
     debug.timeEnd("nav:total")
-    debug.timeEnd("preview:load")
+    const timer = setTimeout(() => {
+      if (selectedResult()?.id !== itemId) {
+        debug.log("preview:load:cancelled", { item: itemId })
+        return
+      }
+      debug.time("preview:load")
+      try {
+        const page = loadConversationAround(item, { before: INITIAL_PREVIEW_BEFORE, after: INITIAL_PREVIEW_AFTER, dbPath: db })
+        if (selectedResult()?.id !== itemId) {
+          debug.log("preview:load:cancelled", { item: itemId, afterLoad: true })
+          return
+        }
+        debug.log("preview:init", {
+          item: item.id,
+          session: item.sessionID,
+          parts: page.parts.length,
+          hasMoreBefore: page.hasMoreBefore,
+          hasMoreAfter: page.hasMoreAfter,
+          first: page.parts[0]?.id,
+          last: page.parts.at(-1)?.id,
+        })
+        solidBatch(() => {
+          setPreviewParts(page.parts)
+          setHasMorePreviewBefore(page.hasMoreBefore)
+          setHasMorePreviewAfter(page.hasMoreAfter)
+        })
+      } catch {
+      } finally {
+        debug.timeEnd("preview:load")
+      }
+    }, INITIAL_PREVIEW_DELAY_MS)
+    onCleanup(() => {
+      clearTimeout(timer)
+      if (selectedResult()?.id !== itemId) {
+        debug.log("preview:load:cancelled", { item: itemId, cleanup: true })
+      }
+    })
   })
 
   createEffect(() => {
     const item = selectedResult()
     if (!item) return
     const interval = setInterval(() => {
-      if (loadingPreviewMore() || prefetchingPreviewBefore() || prefetchingPreviewAfter()) return
+      if (!previewEdgeLoadingReady() || loadingPreviewMore() || prefetchingPreviewBefore() || prefetchingPreviewAfter()) return
       const scroll = previewScroll
       const children = scroll?.getChildren()
       if (!scroll || !children || children.length === 0) return
@@ -610,8 +650,19 @@ export const Telescope = (props: { api: TuiPluginApi; config: TelescopeConfig; o
     if (!item) return
     if (item.id === scrolledItem) return
     scrolledItem = item.id
-    const timer = setTimeout(() => scrollPreviewToTarget(previewScroll, messageTargetID(item)), 1)
-    onCleanup(() => clearTimeout(timer))
+    const itemId = item.id
+    let readyTimer: ReturnType<typeof setTimeout> | undefined
+    const timer = setTimeout(() => {
+      if (selectedResult()?.id !== itemId) return
+      scrollPreviewToTarget(previewScroll, messageTargetID(item))
+      readyTimer = setTimeout(() => {
+        if (selectedResult()?.id === itemId) setPreviewEdgeLoadingReady(true)
+      }, 100)
+    }, 1)
+    onCleanup(() => {
+      clearTimeout(timer)
+      if (readyTimer) clearTimeout(readyTimer)
+    })
   })
 
   const open = () => {

package/ui/render-target.ts

@@ -6,6 +6,8 @@ export function previewScrollAmount(scroll: ScrollBoxRenderable | undefined) {
 }
 
 export function messageTargetID(item: SearchResult) {
+  if (item.partType === "tool") return `tool-inline-${item.messageID}-${item.id}`
+  if (item.partType === "reasoning") return `text-${item.messageID}-${item.id}`
   if (item.role === "assistant") return `text-${item.messageID}-${item.id}`
   return item.messageID
 }