From bea14f22c1091659e17ad1ef13af4acd55eb336e Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Fri, 15 Nov 2024 12:44:50 +0900
Subject: [PATCH] Add JSON schema for Claude

---
 src/ts/process/request.ts | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 6dd4420d..f082df33 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -2230,9 +2230,16 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                                                         break
                                                     }
                                                     text += "Error:" + JSON.parse(e.data).error?.message
-                                                    controller.enqueue({
-                                                        "0": text
-                                                    })
+                                                    if(db.extractJson && db.jsonSchemaEnabled){
+                                                        controller.enqueue({
+                                                            "0": extractJSON(text, db.jsonSchema)
+                                                        })
+                                                    }
+                                                    else{
+                                                        controller.enqueue({
+                                                            "0": text
+                                                        })
+                                                    }
                                                 }
                                                 break
                                             }
@@ -2302,6 +2309,12 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                         result: JSON.stringify(res.data)
                     }
                 }
+                if(db.extractJson && db.jsonSchemaEnabled){
+                    return {
+                        type: 'success',
+                        result: extractJSON(resText, db.jsonSchema)
+                    }
+                }
                 return {
                     type: 'success',
                     result: resText