From 1d40c9aeacbb60d4a67e4ac091113af52e917481 Mon Sep 17 00:00:00 2001
From: Kwaroran <amasio1234@proton.me>
Date: Sun, 23 Mar 2025 16:36:46 +0900
Subject: [PATCH] Enhance reverse_proxy handling to support custom model
 parameters

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

diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 8502f2d9..7b3c61da 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -1170,8 +1170,23 @@ async function requestOpenAI(arg:RequestDataArgumentExtended):Promise<requestDat
         }
     }
 
-    if(aiModel === 'reverse_proxy'){
-        const additionalParams = db.additionalParams
+    if(aiModel === 'reverse_proxy' || aiModel.startsWith('xcustom:::')){
+        let additionalParams = aiModel === 'reverse_proxy' ? db.additionalParams : []
+
+        if(aiModel.startsWith('xcustom:::')){
+            const found = db.customModels.find(m => m.id === aiModel)
+            const params = found?.params
+            if(params){
+                const lines = params.split('\n')
+                for(const line of lines){
+                    const split = line.split('=')
+                    if(split.length >= 2){
+                        additionalParams.push([split[0], split.slice(1).join('=')])
+                    }
+                }
+            }
+        }
+
         for(let i=0;i<additionalParams.length;i++){
             let key = additionalParams[i][0]
             let value = additionalParams[i][1]