| import { parseSSE } from "./sse"; |
|
|
| export interface Source { |
| index: number; |
| filename: string; |
| page: number | null; |
| line_bboxes: number[][]; |
| text: string; |
| score: number; |
| } |
|
|
| export interface ChatTurn { |
| role: "user" | "assistant"; |
| content: string; |
| } |
|
|
| export interface ChatRequest { |
| question: string; |
| report_name?: string | null; |
| report_names?: string[] | null; |
| top_k?: number; |
| history?: ChatTurn[]; |
| } |
|
|
| export interface ChatCallbacks { |
| onSources?: (sources: Source[]) => void; |
| onToken?: (text: string) => void; |
| onDone?: (numSources: number) => void; |
| onError?: (message: string) => void; |
| } |
|
|
| export async function getReports(): Promise<string[]> { |
| const r = await fetch("/api/reports"); |
| if (!r.ok) throw new Error("Failed to load reports"); |
| return (await r.json()).reports as string[]; |
| } |
|
|
| export interface Health { |
| status: "ok" | "degraded"; |
| num_documents: number; |
| detail?: string | null; |
| } |
|
|
| export async function getHealth(): Promise<Health> { |
| const r = await fetch("/api/health"); |
| return r.json(); |
| } |
|
|
| |
| export async function streamChat( |
| req: ChatRequest, |
| cb: ChatCallbacks, |
| signal?: AbortSignal, |
| ): Promise<void> { |
| const resp = await fetch("/api/chat", { |
| method: "POST", |
| headers: { "Content-Type": "application/json" }, |
| body: JSON.stringify(req), |
| signal, |
| }); |
|
|
| if (!resp.ok) { |
| const detail = await resp.text().catch(() => ""); |
| cb.onError?.(`Request failed (${resp.status}). ${detail}`); |
| return; |
| } |
|
|
| for await (const evt of parseSSE(resp)) { |
| const data = safeParse(evt.data); |
| if (evt.event === "sources") cb.onSources?.(data.sources ?? []); |
| else if (evt.event === "token") cb.onToken?.(data.text ?? ""); |
| else if (evt.event === "done") cb.onDone?.(data.num_sources ?? 0); |
| else if (evt.event === "error") cb.onError?.(data.message ?? "Unknown error"); |
| } |
| } |
|
|
| export interface UploadProgress { |
| stage: string; |
| page?: number; |
| total?: number; |
| num_chunks?: number; |
| vectors_added?: number; |
| } |
|
|
| |
| export async function uploadAndTrack( |
| file: File, |
| onProgress: (p: UploadProgress) => void, |
| onComplete: (stats: Record<string, unknown>) => void, |
| onError: (message: string) => void, |
| ): Promise<void> { |
| const form = new FormData(); |
| form.append("file", file); |
|
|
| const start = await fetch("/api/upload", { method: "POST", body: form }); |
| if (!start.ok) { |
| onError(`Upload failed (${start.status}).`); |
| return; |
| } |
| const { job_id } = await start.json(); |
|
|
| const events = await fetch(`/api/upload/${job_id}/events`); |
| if (!events.ok) { |
| onError("Failed to open progress stream."); |
| return; |
| } |
|
|
| for await (const evt of parseSSE(events)) { |
| const data = safeParse(evt.data); |
| if (evt.event === "progress") onProgress(data as UploadProgress); |
| else if (evt.event === "complete") onComplete(data); |
| else if (evt.event === "error") onError(data.message ?? "Processing failed"); |
| } |
| } |
|
|
| export function pdfUrl(filename: string): string { |
| return `/api/pdf/${encodeURIComponent(filename)}`; |
| } |
|
|
| function safeParse(s: string): any { |
| try { |
| return JSON.parse(s); |
| } catch { |
| return {}; |
| } |
| } |
|
|
