fix: useData

This commit is contained in:
Adam
2025-11-26 06:03:13 -06:00
parent cd4fabd11b
commit 30d6a26e3e
3 changed files with 13 additions and 11 deletions

View File

@@ -7,7 +7,7 @@ import type { AssistantMessage as AssistantMessageType, ToolPart } from "@openco
export function MessageProgress(props: { assistantMessages: () => AssistantMessageType[]; done?: boolean }) {
const data = useData()
const sanitizer = createMemo(() => (data.directory ? new RegExp(`${data.directory}/`, "g") : undefined))
const parts = createMemo(() => props.assistantMessages().flatMap((m) => data.part[m.id]))
const parts = createMemo(() => props.assistantMessages().flatMap((m) => data.store.part[m.id]))
const done = createMemo(() => props.done ?? false)
const currentTask = createMemo(
() =>
@@ -27,8 +27,10 @@ export function MessageProgress(props: { assistantMessages: () => AssistantMessa
let resolved = parts()
const task = currentTask()
if (task && task.state && "metadata" in task.state && task.state.metadata?.sessionId) {
const messages = data.message[task.state.metadata.sessionId as string]?.filter((m) => m.role === "assistant")
resolved = messages?.flatMap((m) => data.part[m.id]) ?? parts()
const messages = data.store.message[task.state.metadata.sessionId as string]?.filter(
(m) => m.role === "assistant",
)
resolved = messages?.flatMap((m) => data.store.part[m.id]) ?? parts()
}
return resolved
})
@@ -149,7 +151,7 @@ export function MessageProgress(props: { assistantMessages: () => AssistantMessa
{(p) => {
const part = p() as ToolPart
const message = createMemo(() =>
data.message[part.sessionID].find((m) => m.id === part.messageID),
data.store.message[part.sessionID].find((m) => m.id === part.messageID),
)
return (
<div data-slot="message-progress-item">