From fde905aac1812b84066ff72e5f2f90b56d4c3a59 Mon Sep 17 00:00:00 2001
From: Sean Hatfield <seanhatfield5@gmail.com>
Date: Thu, 29 Feb 2024 17:04:59 -0800
Subject: [PATCH] [FEAT] JSON export append all metadata fields to workspace
chats (#845)
have JSON export append all metadata fields
---
server/utils/helpers/chat/convertTo.js | 10 +++-------
1 file changed, 3 insertions(+), 7 deletions(-)
diff --git a/server/utils/helpers/chat/convertTo.js b/server/utils/helpers/chat/convertTo.js
index 119c155ae..7d2b5f691 100644
--- a/server/utils/helpers/chat/convertTo.js
+++ b/server/utils/helpers/chat/convertTo.js
@@ -21,12 +21,8 @@ async function convertToCSV(preparedData) {
return rows.join("\n");
}
-async function convertToJSON(workspaceChatsMap) {
- const allMessages = [].concat.apply(
- [],
- Object.values(workspaceChatsMap).map((workspace) => workspace.messages)
- );
- return JSON.stringify(allMessages, null, 4);
+async function convertToJSON(preparedData) {
+ return JSON.stringify(preparedData, null, 4);
}
// ref: https://raw.githubusercontent.com/gururise/AlpacaDataCleaned/main/alpaca_data.json
@@ -48,7 +44,7 @@ async function prepareWorkspaceChatsForExport(format = "jsonl") {
id: "asc",
});
- if (format === "csv") {
+ if (format === "csv" || format === "json") {
const preparedData = chats.map((chat) => {
const responseJson = JSON.parse(chat.response);
return {