fix: handle failed AI message persistence per updated Go contract

Backend now returns 201 with status:"failed" and a reserved all-zero
message_id when an AI message is saved without a valid message_id,
instead of a request error. Surface this as an error bubble and skip
calling traceability with the reserved id.

docs/API_CONTRACT_BE_GOLANG.md

@@ -1,4 +1,6 @@
-# Frontend API Contract
+# Backend Go API Contract
+
+**Last updated**: 2026-07-09
 
 Dokumen ini merangkum endpoint Orchestration Agent Service yang dipakai oleh frontend. Fokus flow:
 
@@ -802,11 +804,14 @@ Errors:
 
 ```json
 {
-  "id": "message-id",
+  "id": "uuid-dari-db",
+  "message_id": "uuid-dari-db",
   "analysis_id": "analysis-id",
   "user_id": "user-id",
   "role": "user",
   "content": "Apa penyebab revenue turun di Q3?",
+  "status": "success",
+  "note": "",
   "created_at": "2026-06-30T08:00:00Z"
 }
 ```
@@ -816,11 +821,18 @@ Errors:
 - `user`: pertanyaan atau instruksi dari user
 - `ai`: jawaban dari AI Agent Service
 
+`status` hanya:
+
+- `success`: pesan tersimpan sesuai konten yang dikirim. Untuk `role=user`, `message_id` selalu system-generated (default), berapa pun nilai `message_id` yang dikirim frontend (jika ada) — nilai kiriman itu diabaikan. Untuk `role=ai`, `success` berarti `message_id` yang dikirim valid dan dipakai apa adanya.
+- `failed`: khusus `role=ai` yang dikirim tanpa `message_id` valid (lihat catatan di bawah). `content` diganti pesan fallback hardcoded dan `message_id` diganti default reserved value.
+
+`note` hanya terisi (non-empty) ketika `status=failed`, berisi penjelasan singkat penyebab kegagalan. Saat `status=success`, field ini kosong/absent.
+
 ### `POST /api/v1/analyses/{id}/messages`
 
 Merekam tepat satu pesan conversation ke `analyses_messages`.
 
-Request untuk pertanyaan user:
+Request untuk pertanyaan user — **jangan kirim `message_id`**; jika dikirim tetap akan diabaikan sepenuhnya dan diganti dengan nilai default (system-generated):
 
 ```json
 {
@@ -829,16 +841,17 @@ Request untuk pertanyaan user:
 }
 ```
 
-Request untuk jawaban AI:
+Request untuk jawaban AI — **`message_id` wajib diisi untuk `role=ai`**:
 
 ```json
 {
   "role": "ai",
-  "content": "Revenue turun karena penurunan transaksi enterprise dan kenaikan churn di wilayah barat."
+  "content": "Revenue turun karena penurunan transaksi enterprise dan kenaikan churn di wilayah barat.",
+  "message_id": "uuid-yang-dibuat-frontend-atau-agent-service"
 }
 ```
 
-Success `201`:
+Success `201` (role=user, atau role=ai dengan `message_id` valid):
 
 ```json
 {
@@ -852,13 +865,38 @@ Success `201`:
       "user_id": "user-id",
       "role": "user",
       "content": "Apa penyebab revenue turun di Q3?",
+      "status": "success",
       "created_at": "2026-06-30T08:00:00Z"
     }
   }
 }
 ```
 
-> `message_id` bernilai sama dengan `id` — disertakan agar frontend dapat langsung menggunakannya untuk memanggil endpoint observability (`GET /api/v1/observability?analysis_id=...&message_id=...`) tanpa mapping tambahan.
+> `message_id` bernilai sama dengan `id` untuk pesan sukses (baik `role=user` maupun `role=ai` dengan `message_id` valid) — disertakan agar frontend dapat langsung menggunakannya untuk memanggil endpoint observability (`GET /api/v1/observability?analysis_id=...&message_id=...`) tanpa mapping tambahan.
+
+**Kasus `role=ai` tanpa `message_id`** (missing atau blank): endpoint **tetap** mengembalikan `201` (bukan error) karena kegagalan ini dianggap outcome data-level, bukan request error. Response menunjukkan `status: "failed"`:
+
+```json
+{
+  "status": "success",
+  "message": "Message created",
+  "data": {
+    "message": {
+      "id": "uuid-dari-db",
+      "message_id": "00000000-0000-0000-0000-000000000000",
+      "analysis_id": "analysis-id",
+      "user_id": "user-id",
+      "role": "ai",
+      "content": "Maaf, terjadi kesalahan saat memproses respons AI. Silakan coba lagi.",
+      "status": "failed",
+      "note": "message_id is required for AI-role messages but was not provided",
+      "created_at": "2026-06-30T08:01:00Z"
+    }
+  }
+}
+```
+
+`message_id: "00000000-0000-0000-0000-000000000000"` adalah nilai reserved yang dipakai berulang untuk setiap pesan `ai` yang gagal (bukan di-generate baru setiap kali) — jangan gunakan nilai ini untuk memanggil endpoint observability, karena tidak menunjuk ke jawaban AI yang valid. Chat tetap bisa dilanjutkan setelah kegagalan ini; tidak ada retry otomatis dari backend.
 
 Errors:
 
@@ -891,14 +929,26 @@ Success `200`:
         "user_id": "user-id",
         "role": "ai",
         "content": "Revenue turun karena penurunan transaksi enterprise.",
+        "status": "success",
         "created_at": "2026-06-30T08:01:00Z"
+      },
+      {
+        "id": "uuid-dari-db-2",
+        "message_id": "00000000-0000-0000-0000-000000000000",
+        "analysis_id": "analysis-id",
+        "user_id": "user-id",
+        "role": "ai",
+        "content": "Maaf, terjadi kesalahan saat memproses respons AI. Silakan coba lagi.",
+        "status": "failed",
+        "note": "message_id is required for AI-role messages but was not provided",
+        "created_at": "2026-06-30T08:02:00Z"
       }
     ]
   }
 }
 ```
 
-> Setiap item menyertakan `message_id` (sama dengan `id`) untuk digunakan frontend pada pemanggilan `GET /api/v1/observability?analysis_id=...&message_id=...` setelah page refresh.
+> Setiap item menyertakan `message_id`, `status`, dan (jika `failed`) `note`. Untuk pesan `status=success`, `message_id` sama dengan `id` dan dapat dipakai memanggil `GET /api/v1/observability?analysis_id=...&message_id=...` setelah page refresh. Untuk pesan `status=failed`, gunakan field ini hanya untuk ditampilkan sebagai error bubble di UI — jangan dipakai memanggil observability endpoint.
 
 Errors: `400`, `401`, `404`.
 
@@ -942,7 +992,7 @@ GET  /api/v1/data-catalog/{user_id}
 POST /api/v1/analyses with business_questions
 call AI Agent Service outside this service
 POST /api/v1/analyses/{analysis_id}/messages with role=user and content=user_question
-POST /api/v1/analyses/{analysis_id}/messages with role=ai and content=agent_answer
+POST /api/v1/analyses/{analysis_id}/messages with role=ai, content=agent_answer, message_id=agent_message_id
 GET  /api/v1/analyses/{analysis_id}/messages
 ```
 
@@ -954,4 +1004,13 @@ GET  /api/v1/analyses/{analysis_id}/messages
 - Perubahan `data_bind` akan rebuild analysis-level catalog. Frontend juga dapat memanggil endpoint rebuild analysis catalog secara eksplisit jika diperlukan.
 - Setelah refresh token sukses, selalu replace refresh token lama dengan refresh token baru.
 - Untuk endpoint yang membutuhkan `user_id`, gunakan `data.user.id` dari login dan pastikan sama dengan token aktif.
-- Untuk binding analysis, pakai source yang sudah berhasil diupload/diproses atau database client yang sudah diingest.
+- Untuk binding analysis, pakai source yang sudah berhasil diupload/diproses atau database client yang sudah diingest.
+- Saat merekam jawaban AI (`role=ai`), selalu sertakan `message_id` dari AI Agent Service. Jika `message_id` tidak dikirim atau kosong, endpoint tetap sukses (`201`) tetapi menyimpan pesan dengan `status: "failed"`, `content` fallback hardcoded, dan `message_id` reserved (`00000000-0000-0000-0000-000000000000`) — frontend harus memeriksa `status` pada response, bukan hanya HTTP status code, untuk mendeteksi kegagalan ini.
+- Saat merekam pertanyaan user (`role=user`), **jangan** mengirim `message_id` sama sekali — backend selalu mengabaikannya dan memakai nilai default (system-generated) untuk `role=user`, tidak peduli apa yang dikirim frontend.
+
+
+
+
+
+
+

src/app/components/KnowledgeManagement.tsx

@@ -305,8 +305,10 @@ export default function KnowledgeManagement({
     if (!userId) return;
     setIngesting(clientId);
     try {
-      const res = await ingestDatabaseClient(clientId, userId);
-      const tableCount = res.tables?.length ?? 0;
+      await ingestDatabaseClient(clientId, userId);
+      const catalog = await getDataCatalog(userId);
+      const source = catalog.sources.find((s) => s.source_id === clientId);
+      const tableCount = source?.table_count ?? 0;
       toast.success(`Ingested ${tableCount} table${tableCount !== 1 ? "s" : ""} successfully`);
       const now = new Date().toISOString();
       setIngestTimestamps((prev) => {

src/app/components/analysis/AnalysisShell.tsx

@@ -12,6 +12,7 @@ import {
   getDocuments,
   listAnalyses,
   updateAnalysisDataBind,
+  RESERVED_FAILED_MESSAGE_ID,
   type Analysis,
   type AnalysisMessage,
   type DataBindItem,
@@ -29,13 +30,17 @@ import type { StreamState, UiMessage } from "./types";
 import { cx, makeLocalId } from "./utils";
 
 function toUiMessage(message: AnalysisMessage): UiMessage {
+  const failed = message.status === "failed";
   return {
     id: message.id,
     role: message.role,
     content: message.content,
     created_at: message.created_at,
-    status: "complete",
+    status: failed ? "error" : "complete",
+    statusText: failed ? message.note : undefined,
     messageId: message.message_id ?? undefined,
+    persistStatus: message.status,
+    persistNote: message.note,
   };
 }
 
@@ -153,6 +158,7 @@ export function AnalysisShell() {
           message.role === "ai" &&
           message.status === "complete" &&
           Boolean(message.messageId) &&
+          message.messageId !== RESERVED_FAILED_MESSAGE_ID &&
           !message.traceability &&
           !message.traceabilityLoading &&
           !message.traceabilityError
@@ -229,14 +235,19 @@ export function AnalysisShell() {
         const saved = await createAnalysisMessage(activeAnalysis.id, { role: "ai", content: completedContent, message_id: doneMessageId });
         traceTargetId = saved.id;
         const savedUi = toUiMessage(saved);
-        resolvedMessageId = savedUi.messageId ?? doneMessageId;
+        const persistFailed = saved.status === "failed";
+        resolvedMessageId = persistFailed ? undefined : savedUi.messageId ?? doneMessageId;
         setMessages((prev) =>
           prev.map((msg) =>
             msg.id === aiId
-              ? { ...savedUi, sources: msg.sources, traceabilityLoading: Boolean(savedUi.messageId) }
+              ? { ...savedUi, sources: msg.sources, traceabilityLoading: Boolean(resolvedMessageId) }
               : msg
           )
         );
+        if (persistFailed) {
+          toast.error(saved.note || "Failed to save AI response");
+          return;
+        }
       }
 
       await fetchTraceability(activeAnalysis.id, resolvedMessageId, traceTargetId);

src/app/components/analysis/MarkdownContent.tsx

@@ -26,8 +26,8 @@ function normalizeReadableMarkdown(content: string) {
     return repairMarkdownMarkers(line)
       .replace(/([.!?])\s*-\s*(?=[A-Z0-9*])/g, "$1\n\n- ")
       .replace(/(^|\n)-(?=\*\*|[A-Z0-9])/g, "$1- ")
-      .replace(/([,;:])(?=\S)/g, "$1 ")
-      .replace(/([.!?])(?=[A-Z0-9])/g, "$1 ")
+      .replace(/(?<!\d)([,;:])(?=\S)/g, "$1 ")
+      .replace(/(?<!\d)([.!?])(?=[A-Z])/g, "$1 ")
       .replace(/[ \t]{2,}/g, " ");
   });
 
@@ -79,7 +79,7 @@ export function MarkdownContent({ content }: { content: string }) {
   const readableContent = normalizeReadableMarkdown(content);
 
   return (
-    <div className="agent-markdown max-w-none break-words text-sm leading-7 text-slate-800 [overflow-wrap:anywhere]">
+    <div className="agent-markdown max-w-none text-sm leading-7 text-slate-800 [overflow-wrap:anywhere]">
       <ReactMarkdown remarkPlugins={[remarkGfm, remarkMath]} rehypePlugins={[rehypeKatex]} components={markdownComponents}>
         {readableContent}
       </ReactMarkdown>

src/app/components/analysis/ReportSidebar.tsx

@@ -72,7 +72,7 @@ export function ReportSidebar({ analysis, userId, onCollapse }: ReportSidebarPro
       }
     };
     loadDetail();
-  }, [analysis?.id, selectedVersion]);
+  }, [selectedVersion]);
 
   const handleGenerate = async () => {
     if (!analysis?.id || !userId) return;

src/app/components/analysis/types.ts

@@ -12,6 +12,8 @@ export interface UiMessage {
   statusText?: string;
   sources?: unknown[];
   messageId?: string;
+  persistStatus?: "success" | "failed";
+  persistNote?: string;
   traceability?: Observability | null;
   traceabilityLoading?: boolean;
   traceabilityError?: string | null;

src/services/orchestrationApi.ts

@@ -185,9 +185,13 @@ export interface AnalysisMessage {
   role: "user" | "ai";
   content: string;
   message_id?: string | null;
+  status?: "success" | "failed";
+  note?: string;
   created_at: string;
 }
 
+export const RESERVED_FAILED_MESSAGE_ID = "00000000-0000-0000-0000-000000000000";
+
 export interface CreateAnalysisPayload {
   analysis_title: string;
   objective: string;
@@ -250,25 +254,35 @@ async function parseError(res: Response): Promise<ApiError> {
   return new ApiError(message, res.status, body);
 }
 
+let _refreshPromise: Promise<UserSession | null> | null = null;
+
 async function refreshWithStoredSession(): Promise<UserSession | null> {
-  const current = readStoredSession();
-  if (!current?.refresh_token) return null;
+  if (_refreshPromise) return _refreshPromise;
 
-  const res = await fetch(`${ORCHESTRATION_BASE_URL}/api/refresh`, {
-    method: "POST",
-    headers: { "Content-Type": "application/json" },
-    body: JSON.stringify({ refresh_token: current.refresh_token }),
-  });
+  _refreshPromise = (async () => {
+    const current = readStoredSession();
+    if (!current?.refresh_token) return null;
 
-  if (!res.ok) {
-    clearStoredSession();
-    return null;
-  }
+    const res = await fetch(`${ORCHESTRATION_BASE_URL}/api/refresh`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ refresh_token: current.refresh_token }),
+    });
+
+    if (!res.ok) {
+      clearStoredSession();
+      return null;
+    }
+
+    const envelope = (await res.json()) as RefreshResponse;
+    const next = sessionFromAuthData(envelope.data, current);
+    writeStoredSession(next);
+    return next;
+  })().finally(() => {
+    _refreshPromise = null;
+  });
 
-  const envelope = (await res.json()) as RefreshResponse;
-  const next = sessionFromAuthData(envelope.data, current);
-  writeStoredSession(next);
-  return next;
+  return _refreshPromise;
 }
 
 async function orchestrationRequest<T>(path: string, options: RequestInit = {}, retry = true): Promise<T> {