From 2285755b34d9e4b7763fbd93e4f118faebdfa74f Mon Sep 17 00:00:00 2001
From: hubian <908234780@qq.com>
Date: Sun, 26 Apr 2026 11:27:25 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E6=B7=B1=E5=BA=A6?=
 =?UTF-8?q?=E6=80=9D=E8=80=83=E5=92=8C=E8=81=94=E7=BD=91=E6=90=9C=E7=B4=A2?=
 =?UTF-8?q?=E6=8C=89=E9=92=AE=EF=BC=8C=E6=94=AF=E6=8C=81=E6=80=9D=E8=80=83?=
 =?UTF-8?q?=E5=86=85=E5=AE=B9=E6=B5=81=E5=BC=8F=E8=BE=93=E5=87=BA=E5=92=8C?=
 =?UTF-8?q?=E6=8A=98=E5=8F=A0=E6=98=BE=E7=A4=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 www/app.js     | 120 ++++++++++++++++++++++++++++++++++++++++++++++---
 www/index.html |   6 +--
 www/style.css  |  96 +++++++++++++++++++++++++++++++++++++++
 3 files changed, 213 insertions(+), 9 deletions(-)

diff --git a/www/app.js b/www/app.js
index 395d19c..c56bc29 100644
--- a/www/app.js
+++ b/www/app.js
@@ -5,6 +5,7 @@ const CONFIG = {
   apiUrl: 'https://open.bigmodel.cn/api/paas/v4/chat/completions',
   apiKey: '2259e33a1357460abe17919aaf81e73d.K44a8LPQTmFM5PKm',
   model: 'glm-4.5-air',
+  thinkingModel: 'glm-4-flash-thinking',  // 思考模型
   maxTokens: 2048
 };
 
@@ -13,6 +14,10 @@ let conversations = [];  // 对话列表
 let currentConversation = null;  // 当前对话
 let isLoading = false;
 
+// 功能开关
+let enableThinking = false;  // 深度思考
+let enableSearch = false;    // 联网搜索
+
 // DOM 元素（初始为 null，在 openConversation 时重新获取）
 let appContainer = null;
 let messagesContainer = null;
@@ -20,6 +25,8 @@ let messagesDiv = null;
 let userInput = null;
 let sendBtn = null;
 let welcome = null;
+let thinkingBtn = null;
+let searchBtn = null;
 
 // 初始化
 document.addEventListener('DOMContentLoaded', () => {
@@ -171,6 +178,18 @@ function openConversation(id) {
         <div class="messages" id="messages"></div>
       </div>
 
+      <!-- 功能开关栏 -->
+      <div class="feature-bar">
+        <button class="feature-btn thinking-btn ${enableThinking ? 'active' : ''}" id="thinkingBtn">
+          <svg viewBox="0 0 24 24" width="16" height="16"><path fill="currentColor" d="M12 2C6.48 2 2 6.48 2 12s4.48 10 10 10 10-4.48 10-10S17.52 2 12 2zm-2 15l-5-5 1.41-1.41L10 14.17l7.59-7.59L19 8l-9 9z"/></svg>
+          <span>深度思考</span>
+        </button>
+        <button class="feature-btn search-btn ${enableSearch ? 'active' : ''}" id="searchBtn">
+          <svg viewBox="0 0 24 24" width="16" height="16"><path fill="currentColor" d="M15.5 14h-.79l-.28-.27C15.41 12.59 16 11.11 16 9.5 16 5.91 13.09 3 9.5 3S3 5.91 3 9.5 5.91 16 9.5 16c1.61 0 3.09-.59 4.23-1.57l.27.28v.79l5 4.99L20.49 19l-4.99-5zm-6 0C7.01 14 5 11.99 5 9.5S7.01 7 9.5 7 14 9.01 14 9.5 11.99 14 9.5 14z"/></svg>
+          <span>联网搜索</span>
+        </button>
+      </div>
+
       <div class="input-area">
         <button class="attach-btn" id="attachBtn" title="上传文件">
           <svg viewBox="0 0 24 24" width="24" height="24"><path fill="currentColor" d="M19 13h-6v6h-2v-6H5v-2h6V5h2v6h6v2z"/></svg>
@@ -211,6 +230,8 @@ function openConversation(id) {
   userInput = document.getElementById('userInput');
   sendBtn = document.getElementById('sendBtn');
   welcome = document.getElementById('welcome');
+  thinkingBtn = document.getElementById('thinkingBtn');
+  searchBtn = document.getElementById('searchBtn');
   
   // 绑定按钮事件
   const backBtn = document.getElementById('backBtn');
@@ -219,6 +240,31 @@ function openConversation(id) {
   const clearBtn = document.getElementById('clearBtn');
   if (clearBtn) clearBtn.addEventListener('click', clearCurrentChat);
   
+  // 绑定功能开关按钮事件
+  if (thinkingBtn) {
+    thinkingBtn.addEventListener('click', () => {
+      enableThinking = !enableThinking;
+      thinkingBtn.classList.toggle('active', enableThinking);
+      // 如果开启深度思考，关闭联网搜索（智谱思考模型不支持联网）
+      if (enableThinking && enableSearch) {
+        enableSearch = false;
+        searchBtn.classList.remove('active');
+      }
+    });
+  }
+  
+  if (searchBtn) {
+    searchBtn.addEventListener('click', () => {
+      enableSearch = !enableSearch;
+      searchBtn.classList.toggle('active', enableSearch);
+      // 如果开启联网搜索，关闭深度思考
+      if (enableSearch && enableThinking) {
+        enableThinking = false;
+        thinkingBtn.classList.remove('active');
+      }
+    });
+  }
+  
   // 绑定输入事件
   userInput.addEventListener('keydown', handleKeyDown);
   userInput.addEventListener('input', (e) => autoResize(e.target));
@@ -349,14 +395,24 @@ async function streamGenerate(userMsgIndex) {
   sendBtn.disabled = true;
 
   const aiMessageIndex = currentConversation.messages.length;
-  currentConversation.messages.push({ role: 'assistant', content: '' });
+  currentConversation.messages.push({ 
+    role: 'assistant', 
+    content: '',
+    thinking: ''  // 思考内容
+  });
   renderMessages();
   
   const lastMessageEl = messagesDiv.lastElementChild;
   const contentEl = lastMessageEl.querySelector('.message-content');
+  const thinkingEl = lastMessageEl.querySelector('.thinking-content');
+  
   contentEl.innerHTML = '<span class="streaming-cursor">▌</span>';
+  if (thinkingEl) thinkingEl.innerHTML = '<span class="streaming-cursor">▌</span>';
 
   try {
+    // 根据开关选择模型
+    const model = enableThinking ? CONFIG.thinkingModel : CONFIG.model;
+    
     const response = await fetch(CONFIG.apiUrl, {
       method: 'POST',
       headers: {
@@ -364,7 +420,7 @@ async function streamGenerate(userMsgIndex) {
         'Authorization': `Bearer ${CONFIG.apiKey}`
       },
       body: JSON.stringify({
-        model: CONFIG.model,
+        model: model,
         messages: currentConversation.messages.slice(0, aiMessageIndex).map(m => ({
           role: m.role,
           content: m.content
@@ -381,6 +437,7 @@ async function streamGenerate(userMsgIndex) {
     const reader = response.body.getReader();
     const decoder = new TextDecoder();
     let buffer = '';
+    let thinkingComplete = false;  // 思考是否完成
 
     while (true) {
       const { done, value } = await reader.read();
@@ -397,16 +454,47 @@ async function streamGenerate(userMsgIndex) {
           
           try {
             const data = JSON.parse(jsonStr);
-            if (data.choices && data.choices[0]?.delta?.content) {
-              currentConversation.messages[aiMessageIndex].content += data.choices[0].delta.content;
-              contentEl.innerHTML = renderMarkdown(currentConversation.messages[aiMessageIndex].content) + '<span class="streaming-cursor">▌</span>';
-              scrollToBottom();
+            const delta = data.choices?.[0]?.delta;
+            
+            if (delta) {
+              // 处理思考内容
+              if (delta.reasoning_content || delta.thinking) {
+                const thinkingChunk = delta.reasoning_content || delta.thinking;
+                currentConversation.messages[aiMessageIndex].thinking += thinkingChunk;
+                if (thinkingEl) {
+                  thinkingEl.innerHTML = renderMarkdown(currentConversation.messages[aiMessageIndex].thinking) + '<span class="streaming-cursor">▌</span>';
+                  // 思考时展开
+                  const thinkingBlock = lastMessageEl.querySelector('.thinking-block');
+                  if (thinkingBlock) thinkingBlock.classList.add('expanded');
+                }
+                scrollToBottom();
+              }
+              
+              // 处理正式回复内容
+              if (delta.content) {
+                // 如果开始输出正式内容，说明思考完成
+                if (currentConversation.messages[aiMessageIndex].thinking && !thinkingComplete) {
+                  thinkingComplete = true;
+                  // 折叠思考内容
+                  const thinkingBlock = lastMessageEl.querySelector('.thinking-block');
+                  if (thinkingBlock) thinkingBlock.classList.remove('expanded');
+                  if (thinkingEl) thinkingEl.innerHTML = renderMarkdown(currentConversation.messages[aiMessageIndex].thinking);
+                }
+                
+                currentConversation.messages[aiMessageIndex].content += delta.content;
+                contentEl.innerHTML = renderMarkdown(currentConversation.messages[aiMessageIndex].content) + '<span class="streaming-cursor">▌</span>';
+                scrollToBottom();
+              }
             }
           } catch (e) {}
         }
       }
     }
 
+    // 最终渲染
+    if (thinkingEl && currentConversation.messages[aiMessageIndex].thinking) {
+      thinkingEl.innerHTML = renderMarkdown(currentConversation.messages[aiMessageIndex].thinking);
+    }
     contentEl.innerHTML = renderMarkdown(currentConversation.messages[aiMessageIndex].content);
 
   } catch (error) {
@@ -541,6 +629,20 @@ function renderMessages() {
       contentHtml = renderMarkdown(msg.content);
     }
     
+    // 思考内容块（仅AI消息）
+    let thinkingHtml = '';
+    if (!isUser && msg.thinking) {
+      thinkingHtml = `
+        <div class="thinking-block" onclick="toggleThinking(this)">
+          <div class="thinking-header">
+            <svg viewBox="0 0 24 24" width="16" height="16"><path fill="currentColor" d="M12 2C6.48 2 2 6.48 2 12s4.48 10 10 10 10-4.48 10-10S17.52 2 12 2zm-2 15l-5-5 1.41-1.41L10 14.17l7.59-7.59L19 8l-9 9z"/></svg>
+            <span>思考过程</span>
+            <svg class="thinking-arrow" viewBox="0 0 24 24" width="14" height="14"><path fill="currentColor" d="M7 10l5 5 5-5z"/></svg>
+          </div>
+          <div class="thinking-content">${renderMarkdown(msg.thinking)}</div>
+        </div>`;
+    }
+    
     const copyIcon = `<svg viewBox="0 0 24 24" width="16" height="16"><path fill="currentColor" d="M16 1H4c-1.1 0-2 .9-2 2v14h2V3h12V1zm3 4H8c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h11c1.1 0 2-.9 2-2V7c0-1.1-.9-2-2-2zm0 16H8V7h11v14z"/></svg>`;
     
     const actions = isUser 
@@ -564,6 +666,7 @@ function renderMessages() {
       <div class="message ${msg.role}" data-index="${index}">
         <div class="message-avatar">${avatar}</div>
         <div class="message-body">
+          ${thinkingHtml}
           <div class="message-content">${contentHtml}</div>
           ${actions}
         </div>
@@ -585,6 +688,11 @@ function renderMessages() {
   scrollToBottom();
 }
 
+// 折叠/展开思考内容
+function toggleThinking(block) {
+  block.classList.toggle('expanded');
+}
+
 // ==================== 工具函数 ====================
 
 // 渲染 Markdown
diff --git a/www/index.html b/www/index.html
index ec0bab9..caf840d 100644
--- a/www/index.html
+++ b/www/index.html
@@ -8,12 +8,12 @@
   <meta http-equiv="Pragma" content="no-cache">
   <meta http-equiv="Expires" content="0">
   <title>AI助手</title>
-  <link rel="stylesheet" href="style.css?v=2.1.2">
+  <link rel="stylesheet" href="style.css?v=2.2.0">
   <link rel="manifest" href="manifest.json">
 </head>
 <body>
   <div id="app"></div>
-  <script src="marked.min.js?v=2.1.2"></script>
-  <script src="app.js?v=2.1.2"></script>
+  <script src="marked.min.js?v=2.2.0"></script>
+  <script src="app.js?v=2.2.0"></script>
 </body>
 </html>
\ No newline at end of file
diff --git a/www/style.css b/www/style.css
index 72ea0db..1d0a5c4 100644
--- a/www/style.css
+++ b/www/style.css
@@ -582,6 +582,102 @@ body {
   font-weight: 600;
 }
 
+/* 功能开关栏 */
+.feature-bar {
+  display: flex;
+  gap: 8px;
+  padding: 8px 16px;
+  background: rgba(255,255,255,0.95);
+  border-top: 1px solid var(--border-color);
+}
+
+.feature-btn {
+  display: flex;
+  align-items: center;
+  gap: 4px;
+  padding: 6px 12px;
+  background: white;
+  border: 1px solid var(--border-color);
+  border-radius: 20px;
+  font-size: 13px;
+  color: var(--text-light);
+  cursor: pointer;
+  transition: all 0.2s;
+}
+
+.feature-btn:hover {
+  border-color: var(--primary);
+  color: var(--primary);
+}
+
+.feature-btn.active {
+  background: linear-gradient(135deg, var(--primary) 0%, #764ba2 100%);
+  border-color: transparent;
+  color: white;
+}
+
+.feature-btn svg {
+  flex-shrink: 0;
+}
+
+/* 思考内容块 */
+.thinking-block {
+  margin-bottom: 12px;
+  background: rgba(102, 126, 234, 0.08);
+  border: 1px solid rgba(102, 126, 234, 0.2);
+  border-radius: 10px;
+  overflow: hidden;
+  cursor: pointer;
+  transition: all 0.2s;
+}
+
+.thinking-block:hover {
+  border-color: rgba(102, 126, 234, 0.4);
+}
+
+.thinking-header {
+  display: flex;
+  align-items: center;
+  gap: 6px;
+  padding: 8px 12px;
+  background: rgba(102, 126, 234, 0.12);
+  font-size: 13px;
+  color: var(--primary);
+  font-weight: 500;
+}
+
+.thinking-header svg:first-child {
+  color: var(--primary);
+}
+
+.thinking-arrow {
+  margin-left: auto;
+  transition: transform 0.2s;
+}
+
+.thinking-block.expanded .thinking-arrow {
+  transform: rotate(180deg);
+}
+
+.thinking-content {
+  padding: 0 12px;
+  max-height: 0;
+  overflow: hidden;
+  transition: max-height 0.3s ease, padding 0.3s ease;
+  color: var(--text-light);
+  font-size: 14px;
+  line-height: 1.6;
+}
+
+.thinking-block.expanded .thinking-content {
+  padding: 12px;
+  max-height: 500px;
+}
+
+.thinking-content p {
+  margin: 8px 0;
+}
+
 /* 输入区域 */
 .input-area {
   display: flex;