use data field

This commit is contained in:
Michael Poluektov 2024-07-04 12:21:09 +01:00
parent 4e433d9015
commit 0527755600

View File

@ -133,29 +133,24 @@
let message = history.messages[data.message_id]; let message = history.messages[data.message_id];
const type = data?.data?.type ?? null; const type = data?.data?.type ?? null;
if (type === "status") { const payload = data?.data?.data ?? null;
const status = { if (!type || !payload) {
done: data?.data?.done ?? null, console.log("Data and type fields must be provided.", data);
description: data?.data?.status ?? null return;
};
if (message.statusHistory) {
message.statusHistory.push(status);
} else {
message.statusHistory = [status];
} }
} else if (type === "citation") { const status_keys = ["done", "description"];
console.log(data); const citation_keys = ["document", "metadata", "source"];
const citation = { if (type === "status" && status_keys.every(key => key in payload)) {
document: data?.data?.document ?? null, if (message.statusHistory) {
metadata: data?.data?.metadata ?? null, message.statusHistory.push(payload);
source: data?.data?.source ?? null
};
if (message.citations) {
message.citations.push(citation);
} else { } else {
message.citations = [citation]; message.statusHistory = [payload];
}
} else if (type === "citation" && citation_keys.every(key => key in payload)) {
if (message.citations) {
message.citations.push(payload);
} else {
message.citations = [payload];
} }
} else { } else {
console.log("Unknown message type", data); console.log("Unknown message type", data);