-
- {sources.map((s) => {
- const host = sourceHost(s.url)
- return (
-
-
-
+ {sources.map((source) => {
+ const host = source.url ? sourceHost(source.url) : null
+ const content = (
+ <>
+ {host ? (
- - - {s.title?.trim() || host} - - - {host} - + ) : ( +
+ )} + + + {source.title} - + + {source.subtitle || host || "Saved memory"} + + + + ) + return ( + -
+ {source.url ? (
+
+ {content}
+
+ ) : (
+ + {content} ++ )}
)
})}
@@ -1263,6 +1303,89 @@ function ToolCallDisplay({ part }: { part: ToolCallDisplayPart }) {
)
}
+function SaveResearchToMemoryButton({
+ runId,
+ apiBase,
+ className,
+}: {
+ runId: string
+ apiBase: string
+ className?: string
+}) {
+ const [saveState, setSaveState] = useState<
+ "idle" | "saving" | "saved" | "error"
+ >("idle")
+ const { data } = useQuery({
+ queryKey: ["nova-research-memory-status", runId],
+ queryFn: async () => {
+ const response = await fetch(`${apiBase}/chat/research/${runId}`, {
+ credentials: "include",
+ })
+ if (!response.ok) throw new Error("Could not load research status")
+ return (await response.json()) as {
+ run?: { reportDocumentId?: string | null }
+ }
+ },
+ staleTime: 30 * 1000,
+ })
+ const saved = saveState === "saved" || Boolean(data?.run?.reportDocumentId)
+ const saving = saveState === "saving"
+
+ const save = async () => {
+ if (saved || saving) return
+ setSaveState("saving")
+ try {
+ const response = await fetch(
+ `${apiBase}/chat/research/${runId}/save-to-memory`,
+ { method: "POST", credentials: "include" },
+ )
+ const result = (await response.json().catch(() => null)) as {
+ error?: string
+ saved?: boolean
+ } | null
+ if (!response.ok || !result?.saved) {
+ throw new Error(result?.error || "Could not save this report")
+ }
+ setSaveState("saved")
+ } catch {
+ setSaveState("error")
+ }
+ }
+
+ const label = saved
+ ? "Saved to memory"
+ : saveState === "error"
+ ? "Try saving again"
+ : "Save to memory"
+
+ return (
+
+ )
+}
+
interface AgentMessageProps {
message: UIMessage
index: number
@@ -1299,15 +1422,47 @@ export function AgentMessage({
.filter((part) => part.type === "text")
.map((part) => part.text)
.join(" ")
- const copyText = stripSourceMarkup(messageText)
+ const researchMetadata = (
+ message as UIMessage & {
+ metadata?: {
+ research?: {
+ runId?: string
+ title?: string
+ artifact?: string
+ durationMs?: number
+ sources?: NovaResearchSource[]
+ }
+ }
+ }
+ ).metadata?.research
+ const copyText = researchMetadata
+ ? normalizeResearchMarkdownForDisplay(stripSourceMarkup(messageText))
+ : stripSourceMarkup(messageText)
const memoryOutputs = useMemo(
() => extractMemoryToolOutputs(message),
[message],
)
- const citationIndex = useMemo(
- () => buildCitationIndex(memoryOutputs),
- [memoryOutputs],
- )
+ const citationIndex = useMemo(() => {
+ const index = buildCitationIndex(memoryOutputs)
+ for (const source of researchMetadata?.sources ?? []) {
+ if (
+ source.type !== "memory" ||
+ !source.sourceId ||
+ !isSafeSourceId(source.sourceId) ||
+ index.has(source.sourceId)
+ ) {
+ continue
+ }
+ index.set(source.sourceId, {
+ sourceId: source.sourceId,
+ documentId: source.documentId,
+ title: source.title ?? source.space,
+ type: "memory",
+ url: source.url,
+ })
+ }
+ return index
+ }, [memoryOutputs, researchMetadata?.sources])
const allowedSourceIds = useMemo(
() => new Set(citationIndex.keys()),
[citationIndex],
@@ -1342,6 +1497,39 @@ export function AgentMessage({
}
return out
}, [message.parts])
+ const actionSources = useMemo(() => {
+ const seen = new Set
- {researchMetadata?.runId ? (
-
- {
- parseSourceAnnotatedMarkdown(runText, allowedSourceIds)
- .markdown
- }
+ {researchMetadata
+ ? normalizeResearchMarkdownForDisplay(
+ parseSourceAnnotatedMarkdown(
+ runText,
+ allowedSourceIds,
+ ).markdown,
+ )
+ : parseSourceAnnotatedMarkdown(runText, allowedSourceIds)
+ .markdown}
)
@@ -1518,16 +1685,38 @@ export function AgentMessage({
onLike={onLike}
onDislike={onDislike}
/>
- {webSources.length > 0 && (
+ {actionSources.length > 0 && (
+}) {
+ const [questionIndex, setQuestionIndex] = useState(0)
+ const [answers, setAnswers] = useState>({})
+ const [otherByQuestion, setOtherByQuestion] = useState<
+ Record
+ >({})
+ const [submitting, setSubmitting] = useState(false)
+ const [error, setError] = useState(null)
+ const question = request.questions[questionIndex]
+ if (!question) return null
+
+ const answer = answers[question.id] ?? ""
+ const usingOther = otherByQuestion[question.id] === true
+ const isLast = questionIndex === request.questions.length - 1
+ const canContinue = answer.trim().length > 0
+
+ const submit = async () => {
+ if (!canContinue || submitting) return
+ setSubmitting(true)
+ setError(null)
+ try {
+ await onSubmit(
+ request.questions.map((item) => ({
+ questionId: item.id,
+ value: answers[item.id]?.trim() ?? "",
+ })),
+ )
+ } catch (submitError) {
+ setError(
+ submitError instanceof Error
+ ? submitError.message
+ : "Could not submit your answers.",
+ )
+ setSubmitting(false)
+ }
+ }
+
+ return (
+
-
- ) : null}
-
-
-
-
-
-
-
- Nova Research Report
-
-
- {researchMetadata.title || "Research complete"}
-
- + Research completed in{" "} + + {researchDuration} + +
+ ) : null} {message.parts.map((part, partIndex) => { if (part.type === "source-url") { @@ -1445,17 +1611,18 @@ export function AgentMessage({ return (
-
)}
+ {researchMetadata?.runId ? (
+ <>
+
+
+
+
+
+
+
+
+
+ {questionIndex + 1} / {request.questions.length}
+
+ + A few details first +
+ {request.intro ? ( +{request.intro}
+ ) : null} +
+ {request.questions.map((item, index) => (
+
+ ))}
+
+
+
+ {question.question} +
+
+ {question.options.map((option) => {
+ const selected = !usingOther && answer === option.label
+ return (
+
+ )
+ })}
+ {question.allowOther !== false ? (
+
+ ) : null}
+
+ {usingOther ? (
+