From 754752c80ab37117e0dce4ce3cb5235c15b34a32 Mon Sep 17 00:00:00 2001
From: splitclover <splitclover@proton.me>
Date: Fri, 9 Aug 2024 23:49:18 +0200
Subject: [PATCH] Added chat completion import/export events

---
 public/script.js         | 2 ++
 public/scripts/openai.js | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/public/script.js b/public/script.js
index fe353fd80..594324c37 100644
--- a/public/script.js
+++ b/public/script.js
@@ -425,6 +425,8 @@ export const event_types = {
     CHATCOMPLETION_MODEL_CHANGED: 'chatcompletion_model_changed',
     OAI_PRESET_CHANGED_BEFORE: 'oai_preset_changed_before',
     OAI_PRESET_CHANGED_AFTER: 'oai_preset_changed_after',
+    OAI_PRESET_EXPORT_READY: 'oai_preset_export_ready',
+    OAI_PRESET_IMPORT_READY: 'oai_preset_import_ready',
     WORLDINFO_SETTINGS_UPDATED: 'worldinfo_settings_updated',
     WORLDINFO_UPDATED: 'worldinfo_updated',
     CHARACTER_EDITED: 'character_edited',
diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index b8c0efd8d..e29f8db2d 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -3597,6 +3597,8 @@ async function onPresetImportFileChange(e) {
         }
     }
 
+    eventSource.emit(event_types.OAI_PRESET_IMPORT_READY, { data: presetBody, presetName: name });
+
     const savePresetSettings = await fetch(`/api/presets/save-openai?name=${name}`, {
         method: 'POST',
         headers: getRequestHeaders(),
@@ -3652,6 +3654,7 @@ async function onExportPresetClick() {
         sensitiveFields.forEach(field => delete preset[field]);
     }
 
+    eventSource.emit(event_types.OAI_PRESET_EXPORT_READY, preset);
     const presetJsonString = JSON.stringify(preset, null, 4);
     const presetFileName = `${oai_settings.preset_settings_openai}.json`;
     download(presetJsonString, presetFileName, 'application/json');