Update web/src/lib/api.ts

web/src/lib/api.ts

@@ -3,9 +3,19 @@
 // POST /api/login, /api/chat, /api/upload, /api/export, /api/summary, /api/feedback
 // GET  /api/memoryline
 
-export type LearningMode = "general" | "concept" | "socratic" | "exam" | "assignment" | "summary";
+export type LearningMode =
+  | "general"
+  | "concept"
+  | "socratic"
+  | "exam"
+  | "assignment"
+  | "summary";
 export type LanguagePref = "Auto" | "English" | "中文";
-export type DocType = "Syllabus" | "Lecture Slides / PPT" | "Literature Review / Paper" | "Other Course Document";
+export type DocType =
+  | "Syllabus"
+  | "Lecture Slides / PPT"
+  | "Literature Review / Paper"
+  | "Other Course Document";
 
 const DEFAULT_TIMEOUT_MS = 20000;
 
@@ -15,7 +25,11 @@ function getBaseUrl() {
   return v && v.trim() ? v.trim() : "";
 }
 
-async function fetchWithTimeout(input: RequestInfo, init?: RequestInit, timeoutMs = DEFAULT_TIMEOUT_MS) {
+async function fetchWithTimeout(
+  input: RequestInfo,
+  init?: RequestInit,
+  timeoutMs = DEFAULT_TIMEOUT_MS
+) {
   const controller = new AbortController();
   const id = setTimeout(() => controller.abort(), timeoutMs);
   try {
@@ -35,7 +49,7 @@ async function parseJsonSafe(res: Response) {
 }
 
 function errMsg(data: any, fallback: string) {
-  return (data && (data.error || data.detail || data.message))
+  return data && (data.error || data.detail || data.message)
     ? String(data.error || data.detail || data.message)
     : fallback;
 }
@@ -71,20 +85,53 @@ export async function apiLogin(payload: ApiLoginReq): Promise<ApiLoginResp> {
 export type ApiChatReq = {
   user_id: string;
   message: string;
-  learning_mode: string;        // backend expects string (not strict union)
+  learning_mode: string; // backend expects string (not strict union)
   language_preference?: string; // "Auto" | "English" | "中文"
-  doc_type?: string;            // "Syllabus" | "Lecture Slides / PPT" | ...
+  doc_type?: string; // "Syllabus" | "Lecture Slides / PPT" | ...
 };
 
-export type ApiChatRef = { source_file?: string; section?: string };
+// ✅ allow backend to return either object refs or preformatted string refs
+export type ApiChatRefObj = { source_file?: string; section?: string };
+export type ApiChatRefRaw = ApiChatRefObj | string;
+
+// ✅ normalize ANY ref format into {source_file, section} so App can map reliably
+function normalizeRefs(raw: any): ApiChatRefObj[] {
+  const arr: any[] = Array.isArray(raw) ? raw : [];
+  return arr
+    .map((x) => {
+      // Case A: already object
+      if (x && typeof x === "object" && !Array.isArray(x)) {
+        const a = x.source_file != null ? String(x.source_file) : "";
+        const b = x.section != null ? String(x.section) : "";
+        return { source_file: a || undefined, section: b || undefined };
+      }
+
+      // Case B: string like "file.pdf — p3#1" (or just "file.pdf")
+      if (typeof x === "string") {
+        const s = x.trim();
+        if (!s) return null;
+        const parts = s.split("—").map((p) => p.trim()).filter(Boolean);
+        if (parts.length >= 2) {
+          return { source_file: parts[0], section: parts.slice(1).join(" — ") };
+        }
+        return { source_file: s, section: undefined };
+      }
+
+      return null;
+    })
+    .filter(Boolean) as ApiChatRefObj[];
+}
 
 export type ApiChatResp = {
   reply: string;
   session_status_md: string;
-  refs: ApiChatRef[];
+
+  // ✅ after normalization, always object array
+  refs: ApiChatRefObj[];
+
   latency_ms: number;
 
-  // ✅ NEW: optional tracing run id returned by backend
+  // optional tracing run id returned by backend
   run_id?: string | null;
 };
 
@@ -108,7 +155,16 @@ export async function apiChat(payload: ApiChatReq): Promise<ApiChatResp> {
   if (!res.ok) throw new Error(errMsg(data, `apiChat failed (${res.status})`));
 
   // backend returns { reply, session_status_md, refs, latency_ms, run_id? }
-  return data as ApiChatResp;
+  // but refs may be ApiChatRefObj[] OR string[]
+  const out = data as any;
+
+  return {
+    reply: String(out?.reply ?? ""),
+    session_status_md: String(out?.session_status_md ?? ""),
+    refs: normalizeRefs(out?.refs ?? out?.references),
+    latency_ms: Number(out?.latency_ms ?? 0),
+    run_id: out?.run_id ?? null,
+  };
 }
 
 // --------------------
@@ -117,21 +173,21 @@ export async function apiChat(payload: ApiChatReq): Promise<ApiChatResp> {
 export type ApiQuizStartReq = {
   user_id: string;
   language_preference?: string; // "Auto" | "English" | "中文"
-  doc_type?: string;            // default: "Literature Review / Paper" (backend default ok)
-  learning_mode?: string;       // default: "quiz"
+  doc_type?: string; // default: "Literature Review / Paper" (backend default ok)
+  learning_mode?: string; // default: "quiz"
 };
 
 export type ApiQuizStartResp = {
   reply: string;
   session_status_md: string;
-  refs: ApiChatRef[];
+  refs: ApiChatRefObj[];
   latency_ms: number;
-
-  // ✅ NEW: optional tracing run id returned by backend (if enabled)
   run_id?: string | null;
 };
 
-export async function apiQuizStart(payload: ApiQuizStartReq): Promise<ApiQuizStartResp> {
+export async function apiQuizStart(
+  payload: ApiQuizStartReq
+): Promise<ApiQuizStartResp> {
   const base = getBaseUrl();
   const res = await fetchWithTimeout(
     `${base}/api/quiz/start`,
@@ -150,7 +206,16 @@ export async function apiQuizStart(payload: ApiQuizStartReq): Promise<ApiQuizSta
 
   const data = await parseJsonSafe(res);
   if (!res.ok) throw new Error(errMsg(data, `apiQuizStart failed (${res.status})`));
-  return data as ApiQuizStartResp;
+
+  const out = data as any;
+
+  return {
+    reply: String(out?.reply ?? ""),
+    session_status_md: String(out?.session_status_md ?? ""),
+    refs: normalizeRefs(out?.refs ?? out?.references),
+    latency_ms: Number(out?.latency_ms ?? 0),
+    run_id: out?.run_id ?? null,
+  };
 }
 
 // --------------------
@@ -163,14 +228,22 @@ export type ApiUploadResp = {
   error?: string;
 };
 
-export async function apiUpload(args: { user_id: string; doc_type: string; file: File }): Promise<ApiUploadResp> {
+export async function apiUpload(args: {
+  user_id: string;
+  doc_type: string;
+  file: File;
+}): Promise<ApiUploadResp> {
   const base = getBaseUrl();
   const fd = new FormData();
   fd.append("user_id", args.user_id);
   fd.append("doc_type", args.doc_type);
   fd.append("file", args.file);
 
-  const res = await fetchWithTimeout(`${base}/api/upload`, { method: "POST", body: fd }, 120000);
+  const res = await fetchWithTimeout(
+    `${base}/api/upload`,
+    { method: "POST", body: fd },
+    120000
+  );
   const data = await parseJsonSafe(res);
   if (!res.ok) throw new Error(errMsg(data, `apiUpload failed (${res.status})`));
   return data as ApiUploadResp;
@@ -179,7 +252,10 @@ export async function apiUpload(args: { user_id: string; doc_type: string; file:
 // --------------------
 // /api/export
 // --------------------
-export async function apiExport(payload: { user_id: string; learning_mode: string }): Promise<{ markdown: string }> {
+export async function apiExport(payload: {
+  user_id: string;
+  learning_mode: string;
+}): Promise<{ markdown: string }> {
   const base = getBaseUrl();
   const res = await fetchWithTimeout(`${base}/api/export`, {
     method: "POST",
@@ -219,7 +295,7 @@ export type ApiFeedbackReq = {
   user_id: string;
   rating: "helpful" | "not_helpful";
 
-  // ✅ NEW: run id so backend can attach feedback to tracing run
+  // run id so backend can attach feedback to tracing run
   run_id?: string | null;
 
   assistant_message_id?: string;
@@ -236,7 +312,9 @@ export type ApiFeedbackReq = {
   timestamp_ms?: number;
 };
 
-export async function apiFeedback(payload: ApiFeedbackReq): Promise<{ ok: boolean }> {
+export async function apiFeedback(
+  payload: ApiFeedbackReq
+): Promise<{ ok: boolean }> {
   const base = getBaseUrl();
   const res = await fetchWithTimeout(`${base}/api/feedback`, {
     method: "POST",
@@ -252,7 +330,9 @@ export async function apiFeedback(payload: ApiFeedbackReq): Promise<{ ok: boolea
 // --------------------
 // /api/memoryline
 // --------------------
-export async function apiMemoryline(user_id: string): Promise<{ next_review_label: string; progress_pct: number }> {
+export async function apiMemoryline(
+  user_id: string
+): Promise<{ next_review_label: string; progress_pct: number }> {
   const base = getBaseUrl();
   const res = await fetchWithTimeout(
     `${base}/api/memoryline?user_id=${encodeURIComponent(user_id)}`,