handle the exception for web front page

api/handler/chatimpl/chat_handler.go

@@ -9,6 +9,10 @@ package chatimpl
 
 import (
 	"bytes"
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
 	"geekai/core"
 	"geekai/core/types"
 	"geekai/handler"
@@ -19,10 +23,6 @@ import (
 	"geekai/store/vo"
 	"geekai/utils"
 	"geekai/utils/resp"
-	"context"
-	"encoding/json"
-	"errors"
-	"fmt"
 	"net/http"
 	"net/url"
 	"regexp"
@@ -481,11 +481,11 @@ func (h *ChatHandler) StopGenerate(c *gin.Context) {
 func (h *ChatHandler) doRequest(ctx context.Context, req types.ApiRequest, session *types.ChatSession, apiKey *model.ApiKey) (*http.Response, error) {
 	// if the chat model bind a KEY, use it directly
 	if session.Model.KeyId > 0 {
-		h.DB.Debug().Where("id", session.Model.KeyId).Find(apiKey)
+		h.DB.Debug().Where("id", session.Model.KeyId).Where("enabled", true).Find(apiKey)
 	}
 	// use the last unused key
 	if apiKey.Id == 0 {
-		h.DB.Debug().Where("platform = ?", session.Model.Platform).Where("type = ?", "chat").Where("enabled = ?", true).Order("last_used_at ASC").First(apiKey)
+		h.DB.Debug().Where("platform", session.Model.Platform).Where("type", "chat").Where("enabled", true).Order("last_used_at ASC").First(apiKey)
 	}
 	if apiKey.Id == 0 {
 		return nil, errors.New("no available key, please import key")

api/handler/chatimpl/openai_handler.go

@@ -9,13 +9,13 @@ package chatimpl
 
 import (
 	"bufio"
+	"context"
+	"encoding/json"
+	"fmt"
 	"geekai/core/types"
 	"geekai/store/model"
 	"geekai/store/vo"
 	"geekai/utils"
-	"context"
-	"encoding/json"
-	"fmt"
 	"html/template"
 	"io"
 	"strings"
@@ -75,12 +75,16 @@ func (h *ChatHandler) sendOpenAiMessage(
 
 			var responseBody = types.ApiResponse{}
 			err = json.Unmarshal([]byte(line[6:]), &responseBody)
-			if err != nil || len(responseBody.Choices) == 0 { // 数据解析出错
+			if err != nil { // 数据解析出错
 				logger.Error(err, line)
 				utils.ReplyMessage(ws, ErrorMsg)
 				utils.ReplyMessage(ws, ErrImg)
 				break
 			}
+			if len(responseBody.Choices) == 0 { // Fixed: 兼容 Azure API 第一个输出空行
+				continue
+			}
+			
 			if responseBody.Choices[0].FinishReason == "stop" && len(contents) == 0 {
 				utils.ReplyMessage(ws, "抱歉😔😔😔，AI助手由于未知原因已经停止输出内容。")
 				break

api/handler/chatimpl/xunfei_handler.go

@@ -8,16 +8,16 @@ package chatimpl
 // * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 
 import (
-	"geekai/core/types"
-	"geekai/store/model"
-	"geekai/store/vo"
-	"geekai/utils"
 	"context"
 	"crypto/hmac"
 	"crypto/sha256"
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
+	"geekai/core/types"
+	"geekai/store/model"
+	"geekai/store/vo"
+	"geekai/utils"
 	"github.com/gorilla/websocket"
 	"gorm.io/gorm"
 	"html/template"
@@ -80,11 +80,11 @@ func (h *ChatHandler) sendXunFeiMessage(
 	var res *gorm.DB
 	// use the bind key
 	if session.Model.KeyId > 0 {
-		res = h.DB.Where("id", session.Model.KeyId).Find(&apiKey)
+		res = h.DB.Where("id", session.Model.KeyId).Where("enabled", true).Find(&apiKey)
 	}
 	// use the last unused key
 	if res.Error != nil {
-		res = h.DB.Where("platform = ?", session.Model.Platform).Where("type = ?", "chat").Where("enabled = ?", true).Order("last_used_at ASC").First(&apiKey)
+		res = h.DB.Where("platform", session.Model.Platform).Where("type", "chat").Where("enabled", true).Order("last_used_at ASC").First(&apiKey)
 	}
 	if res.Error != nil {
 		utils.ReplyMessage(ws, "抱歉😔😔😔，系统已经没有可用的 API KEY，请联系管理员！")

api/handler/markmap_handler.go

@@ -10,13 +10,13 @@ package handler
 import (
 	"bufio"
 	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
 	"geekai/core"
 	"geekai/core/types"
 	"geekai/store/model"
 	"geekai/utils"
-	"encoding/json"
-	"errors"
-	"fmt"
 	"github.com/gin-gonic/gin"
 	"github.com/gorilla/websocket"
 	"gorm.io/gorm"
@@ -213,7 +213,7 @@ func (h *MarkMapHandler) doRequest(req types.ApiRequest, chatModel model.ChatMod
 	// if the chat model bind a KEY, use it directly
 	var res *gorm.DB
 	if chatModel.KeyId > 0 {
-		res = h.DB.Where("id", chatModel.KeyId).Find(apiKey)
+		res = h.DB.Where("id", chatModel.KeyId).Where("enabled", true).Find(apiKey)
 	}
 	// use the last unused key
 	if apiKey.Id == 0 {

web/src/components/ChatPrompt.vue

@@ -57,6 +57,7 @@ export default defineComponent({
     if (!this.finalTokens) {
       httpPost("/api/chat/tokens", {text: this.content, model: this.model}).then(res => {
         this.finalTokens = res.data;
+      }).catch(() => {
       })
     }
   }

web/src/views/ChatPlus.vue

@@ -654,13 +654,13 @@ const connect = function (chat_id, role_id) {
   });
 
   _socket.addEventListener('message', event => {
+    try {
       if (event.data instanceof Blob) {
         const reader = new FileReader();
         reader.readAsText(event.data, "UTF-8");
         reader.onload = () => {
           const data = JSON.parse(String(reader.result));
           if (data.type === 'start') {
-          console.log(data)
             chatData.value.push({
               type: "reply",
               id: randString(32),
@@ -682,13 +682,18 @@ const connect = function (chat_id, role_id) {
 
             // 获取 token
             const reply = chatData.value[chatData.value.length - 1]
-          httpPost("/api/chat/tokens", {text: "", model: getModelValue(modelID.value), chat_id: chat_id}).then(res => {
+            httpPost("/api/chat/tokens", {
+              text: "",
+              model: getModelValue(modelID.value),
+              chat_id: chat_id
+            }).then(res => {
               reply['created_at'] = new Date().getTime();
               reply['tokens'] = res.data;
               // 将聊天框的滚动条滑动到最底部
               nextTick(() => {
                 document.getElementById('chat-box').scrollTo(0, document.getElementById('chat-box').scrollHeight)
               })
+            }).catch(() => {
             })
 
           } else {
@@ -704,6 +709,9 @@ const connect = function (chat_id, role_id) {
           })
         };
       }
+    } catch (e) {
+      console.error(e)
+    }
 
   });
 

web/src/views/MarkMap.vue

@@ -99,7 +99,6 @@
 import LoginDialog from "@/components/LoginDialog.vue";
 import {nextTick, onMounted, onUnmounted, ref} from 'vue';
 import {Markmap} from 'markmap-view';
-import {loadCSS, loadJS} from 'markmap-common';
 import {Transformer} from 'markmap-lib';
 import {checkSession} from "@/action/session";
 import {httpGet} from "@/utils/http";
@@ -129,9 +128,6 @@ const showLoginDialog = ref(false)
 const isLogin = ref(false)
 const loginUser = ref({power: 0})
 const transformer = new Transformer();
-const {scripts, styles} = transformer.getAssets()
-loadCSS(styles);
-loadJS(scripts);
 
 
 const svgRef = ref(null)
@@ -142,8 +138,12 @@ const loading = ref(false)
 
 onMounted(() => {
   initData()
+  try {
     markMap.value = Markmap.create(svgRef.value)
     update()
+  } catch (e) {
+    console.error(e)
+  }
 });
 
 const initData = () => {
@@ -168,9 +168,13 @@ const initData = () => {
 
 const update = () => {
 
+  try {
     const {root} = transformer.transform(processContent(text.value))
     markMap.value.setData(root)
     markMap.value.fit()
+  } catch (e) {
+    console.error(e)
+  }
 }
 
 const processContent = (text) => {