From c6b70260cc8e586a7730108bd5f009387382e2d8 Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Wed, 11 Sep 2024 23:05:31 +0900
Subject: [PATCH] Fix JSON Schema export

---
 src/ts/storage/database.ts | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/ts/storage/database.ts b/src/ts/storage/database.ts
index 848a192e..2c66c76c 100644
--- a/src/ts/storage/database.ts
+++ b/src/ts/storage/database.ts
@@ -998,6 +998,10 @@ export interface botPreset{
     top_k?:number
     instructChatTemplate?:string
     JinjaTemplate?:string
+    jsonSchemaEnabled?:boolean
+    jsonSchema?:string
+    strictJsonSchema?:boolean
+    extractJson?:string
 }
 
 
@@ -1282,7 +1286,11 @@ export function saveCurrentPreset(){
         moduleIntergration: db.moduleIntergration ?? "",
         top_k: db.top_k,
         instructChatTemplate: db.instructChatTemplate,
-        JinjaTemplate: db.JinjaTemplate ?? ''
+        JinjaTemplate: db.JinjaTemplate ?? '',
+        jsonSchemaEnabled:db.jsonSchemaEnabled??false,
+        jsonSchema:db.jsonSchema ?? '',
+        strictJsonSchema:db.strictJsonSchema ?? true,
+        extractJson:db.extractJson ?? '',
     }
     db.botPresets = pres
     setDatabase(db)
@@ -1371,6 +1379,10 @@ export function setPreset(db:Database, newPres: botPreset){
     db.top_k = newPres.top_k ?? db.top_k
     db.instructChatTemplate = newPres.instructChatTemplate ?? db.instructChatTemplate
     db.JinjaTemplate = newPres.JinjaTemplate ?? db.JinjaTemplate
+    db.jsonSchemaEnabled = newPres.jsonSchemaEnabled ?? false
+    db.jsonSchema = newPres.jsonSchema ?? ''
+    db.strictJsonSchema = newPres.strictJsonSchema ?? true
+    db.extractJson = newPres.extractJson ?? ''
     return db
 }