Browse Source

simplify citation API

Michael Poluektov 10 months ago
parent
commit
55b7c30028
1 changed files with 8 additions and 3 deletions
  1. 8 3
      src/lib/components/chat/Chat.svelte

+ 8 - 3
src/lib/components/chat/Chat.svelte

@@ -139,7 +139,7 @@
 				return;
 				return;
 			}
 			}
 			const status_keys = ["done", "description"];
 			const status_keys = ["done", "description"];
-			const citation_keys = ["document", "metadata", "source"];
+			const citation_keys = ["document", "url", "title"];
 			if (type === "status" && status_keys.every(key => key in payload)) {
 			if (type === "status" && status_keys.every(key => key in payload)) {
 				if (message.statusHistory) {
 				if (message.statusHistory) {
 					message.statusHistory.push(payload);
 					message.statusHistory.push(payload);
@@ -147,10 +147,15 @@
 					message.statusHistory = [payload];
 					message.statusHistory = [payload];
 				}
 				}
 			} else if (type === "citation" && citation_keys.every(key => key in payload)) {
 			} else if (type === "citation" && citation_keys.every(key => key in payload)) {
+				const citation = {
+					document: [payload.document],
+					metadata: [{source: payload.url}],
+					source: {name: payload.title}
+				};
 				if (message.citations) {
 				if (message.citations) {
-					message.citations.push(payload);
+					message.citations.push(citation);
 				} else {
 				} else {
-					message.citations = [payload];
+					message.citations = [citation];
 				}
 				}
 			} else {
 			} else {
 				console.log("Unknown message type", data);
 				console.log("Unknown message type", data);