From 13f76c974ea4361da5ef40a8245e1fd078d79065 Mon Sep 17 00:00:00 2001
From: yokuminto <yokuminto@qq.com>
Date: Sat, 22 Feb 2025 16:01:46 +0800
Subject: [PATCH] reasoning or reasoning_content

---
 public/index.html           |  2 +-
 public/scripts/openai.js    | 10 +++++++++-
 public/scripts/reasoning.js |  5 +++++
 3 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/public/index.html b/public/index.html
index 123c4032d..d4fc97b5c 100644
--- a/public/index.html
+++ b/public/index.html
@@ -2000,7 +2000,7 @@
                                             </span>
                                         </div>
                                     </div>
-                                    <div class="range-block" data-source="deepseek,openrouter">
+                                    <div class="range-block" data-source="deepseek,openrouter,custom">
                                         <label for="openai_show_thoughts" class="checkbox_label widthFreeExpand">
                                             <input id="openai_show_thoughts" type="checkbox" />
                                             <span>
diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index 71b59f93a..860cc251c 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -2167,7 +2167,15 @@ function getStreamingReply(data, state) {
             state.reasoning += (data.choices?.filter(x => x?.delta?.reasoning)?.[0]?.delta?.reasoning || '');
         }
         return data.choices?.[0]?.delta?.content ?? data.choices?.[0]?.message?.content ?? data.choices?.[0]?.text ?? '';
-    } else {
+    } else if(oai_settings.chat_completion_source === chat_completion_sources.CUSTOM){
+        if (oai_settings.show_thoughts) {
+            state.reasoning +=
+            data.choices?.filter(x => x?.delta?.reasoning_content)?.[0]?.delta?.reasoning_content??
+            data.choices?.filter(x => x?.delta?.reasoning)?.[0]?.delta?.reasoning??
+            '';
+        }
+        return data.choices?.[0]?.delta?.content ?? data.choices?.[0]?.message?.content ?? data.choices?.[0]?.text ?? '';
+    }else{
         return data.choices?.[0]?.delta?.content ?? data.choices?.[0]?.message?.content ?? data.choices?.[0]?.text ?? '';
     }
 }
diff --git a/public/scripts/reasoning.js b/public/scripts/reasoning.js
index 2d4711db2..02a2f34dc 100644
--- a/public/scripts/reasoning.js
+++ b/public/scripts/reasoning.js
@@ -76,6 +76,11 @@ export function extractReasoningFromData(data) {
                     return data?.choices?.[0]?.message?.reasoning ?? '';
                 case chat_completion_sources.MAKERSUITE:
                     return data?.responseContent?.parts?.filter(part => part.thought)?.map(part => part.text)?.join('\n\n') ?? '';
+                case chat_completion_sources.CUSTOM:{
+                    return data?.choices?.[0]?.message?.reasoning_content
+                    ?? data?.choices?.[0]?.message?.reasoning
+                    ?? '';
+                }
             }
             break;
     }