From 4e433d9015b2d744bc0efdc504d2a8865f0bc5e1 Mon Sep 17 00:00:00 2001
From: Michael Poluektov <michael.poluektov@gmail.com>
Date: Wed, 3 Jul 2024 18:18:33 +0100
Subject: [PATCH] wip: citations via __event_emitter__

---
 src/lib/components/chat/Chat.svelte | 32 ++++++++++++++++++++++-------
 1 file changed, 25 insertions(+), 7 deletions(-)

diff --git a/src/lib/components/chat/Chat.svelte b/src/lib/components/chat/Chat.svelte
index 3d03246b7..87bd9b4de 100644
--- a/src/lib/components/chat/Chat.svelte
+++ b/src/lib/components/chat/Chat.svelte
@@ -132,15 +132,33 @@
 			console.log(data);
 			let message = history.messages[data.message_id];
 
-			const status = {
-				done: data?.data?.done ?? null,
-				description: data?.data?.status ?? null
-			};
+			const type = data?.data?.type ?? null;
+			if (type === "status") {
+				const status = {
+					done: data?.data?.done ?? null,
+					description: data?.data?.status ?? null
+				};
 
-			if (message.statusHistory) {
-				message.statusHistory.push(status);
+				if (message.statusHistory) {
+					message.statusHistory.push(status);
+				} else {
+					message.statusHistory = [status];
+				}
+			} else if (type === "citation") {
+				console.log(data);
+				const citation = {
+					document: data?.data?.document ?? null,
+					metadata: data?.data?.metadata ?? null,
+					source: data?.data?.source ?? null
+				};
+
+				if (message.citations) {
+					message.citations.push(citation);
+				} else {
+					message.citations = [citation];
+				}
 			} else {
-				message.statusHistory = [status];
+				console.log("Unknown message type", data);
 			}
 
 			messages = messages;