ai巡检

ruoyi-ui-app/api/diseaseTreatment/aiOnlineConsult.js

@@ -106,7 +106,9 @@ function parseChatJsonPayload(payload) {
     const json = JSON.parse(payload)
     const sessionId = extractLlmSessionIdFromJson(json)
     const delta = extractDeltaFromChatJson(json)
-    return { sessionId, delta }
+    const model =
+      json.model != null && json.model !== '' ? String(json.model) : null
+    return { sessionId, delta, model }
   } catch (e) {
     return null
   }
@@ -162,6 +164,12 @@ function consumeSseLines(state, lines, callbacks) {
         callbacks.onSessionId(parsed.sessionId)
       }
     }
+    if (parsed.model) {
+      state.model = parsed.model
+      if (callbacks.onModel) {
+        callbacks.onModel(parsed.model)
+      }
+    }
     if (parsed.delta) {
       const before = state.fullText
       appendStreamDelta(state, parsed.delta)
@@ -192,6 +200,12 @@ function appendSseChunk(state, chunk, callbacks) {
           callbacks.onSessionId(parsed.sessionId)
         }
       }
+      if (parsed.model) {
+        state.model = parsed.model
+        if (callbacks.onModel) {
+          callbacks.onModel(parsed.model)
+        }
+      }
       return
     }
   }
@@ -250,14 +264,15 @@ function streamRequestHeaders() {
 }
 
 function createStreamState(options) {
-  const { onDelta, onSessionId } = options
+  const { onDelta, onSessionId, onModel } = options
   return {
     startMs: Date.now(),
     buffer: '',
     fullText: '',
     sessionId: null,
+    model: null,
     chunkReceived: false,
-    callbacks: { onDelta, onSessionId }
+    callbacks: { onDelta, onSessionId, onModel }
   }
 }
 
@@ -266,6 +281,7 @@ function finishStreamState(state) {
   return {
     content: state.fullText,
     sessionId: state.sessionId,
+    model: state.model,
     durationMs: Date.now() - state.startMs
   }
 }
@@ -362,8 +378,8 @@ function sendChatMessageStream(body, options) {
 
 /**
  * 聊天 POST /v1/chat/completions（默认 stream: true）
- * options: onDelta, onSessionId, onReadyTask(task), signal(H5)
- * 流式 resolve：{ content, sessionId, durationMs }，durationMs 为整段流式请求耗时（毫秒）
+ * options: onDelta, onSessionId, onModel, onReadyTask(task), signal(H5)
+ * 流式 resolve：{ content, sessionId, model, durationMs }，durationMs 为整段流式请求耗时（毫秒）
  */
 export function sendChatMessage(data, options = {}) {
   const body = { stream: true, ...(data || {}) }

ruoyi-ui-app/package-a/ai-assistant/index.vue

@@ -1110,13 +1110,14 @@ export default {
         if (norm.sendTime) row.sendTime = norm.sendTime
       }
     },
-    queuePendingPersist(consultSessionId, persistBody, aiReplyContent, llmSessionId, costTime) {
+    queuePendingPersist(consultSessionId, persistBody, aiReplyContent, llmSessionId, costTime, aiCategory) {
       this.pendingPersist = {
         consultSessionId,
         persistBody: { ...persistBody },
         aiReplyContent,
         llmSessionId: llmSessionId != null ? llmSessionId : null,
-        costTime: costTime != null ? costTime : null
+        costTime: costTime != null ? costTime : null,
+        aiCategory: aiCategory != null && aiCategory !== '' ? aiCategory : null
       }
     },
     flushPendingPersist() {
@@ -1128,14 +1129,15 @@ export default {
         ...pending.persistBody,
         aiReplyContent: pending.aiReplyContent,
         realSessionId: pending.llmSessionId != null ? pending.llmSessionId : null,
-        costTime: pending.costTime != null ? pending.costTime : undefined
+        costTime: pending.costTime != null ? pending.costTime : undefined,
+        aiCategory: pending.aiCategory != null ? pending.aiCategory : undefined
       })
         .then(() => {
           this.pendingPersist = null
         })
         .catch(() => {})
     },
-    persistConsultExchange(persistBody, aiReplyContent, costTime) {
+    persistConsultExchange(persistBody, aiReplyContent, costTime, aiCategory) {
       const reply = (aiReplyContent || '').trim()
       const consultSessionId = this.consultSessionIdForApi()
       if (!consultSessionId || !persistBody || !reply) {
@@ -1146,7 +1148,8 @@ export default {
         ...persistBody,
         aiReplyContent: reply,
         realSessionId: llmSessionId || null,
-        costTime: costTime != null ? costTime : undefined
+        costTime: costTime != null ? costTime : undefined,
+        aiCategory: aiCategory != null && aiCategory !== '' ? aiCategory : undefined
       })
         .then((res) => {
           this.mergePersistResult(res.data, reply)
@@ -1160,7 +1163,7 @@ export default {
           return true
         })
         .catch((err) => {
-          this.queuePendingPersist(consultSessionId, persistBody, reply, llmSessionId, costTime)
+          this.queuePendingPersist(consultSessionId, persistBody, reply, llmSessionId, costTime, aiCategory)
           const msg = (err && err.message) || this.$t('aiOnlineConsult.saveSessionFailed')
           uni.showToast({ title: msg, icon: 'none' })
           return false
@@ -1246,6 +1249,7 @@ export default {
       let aiReplyContent = ''
       let aiMsg = null
       let costTime = null
+      let aiCategory = null
       this.abortStreamRequest()
       this.freezeFeedScrollTopForStream()
       this.sending = true
@@ -1258,6 +1262,9 @@ export default {
           onSessionId: (llmSid) => {
             this.rememberLlmSessionId(llmSid)
           },
+          onModel: (model) => {
+            aiCategory = model
+          },
           onDelta: (fullText) => {
             aiReplyContent = fullText
             if (!aiMsg) {
@@ -1275,6 +1282,9 @@ export default {
           this.rememberLlmSessionId(llmSid)
         }
         costTime = streamResult.durationMs != null ? streamResult.durationMs : null
+        if (streamResult.model) {
+          aiCategory = streamResult.model
+        }
         this.clearStreamRevealTimer()
         aiReplyContent = (streamResult.content || aiReplyContent || '').trim() || this.$t('aiOnlineConsult.noContent')
         if (!aiMsg) {
@@ -1308,7 +1318,12 @@ export default {
         }
         this.$nextTick(() => this.scheduleScrollToBottom())
       }
-      await this.persistConsultExchange(persistBody, (aiReplyContent || '').trim(), costTime)
+      await this.persistConsultExchange(
+        persistBody,
+        (aiReplyContent || '').trim(),
+        costTime,
+        aiCategory
+      )
     },
     validateMediaFile(filePath, fileName, kind) {
       const rule = MEDIA_RULES[kind]

ruoyi-ui/src/api/diseaseTreatment/aiOnlineConsult.js

@@ -96,7 +96,9 @@ function parseChatJsonPayload(payload) {
     const json = JSON.parse(payload)
     const sessionId = extractLlmSessionIdFromJson(json)
     const delta = extractDeltaFromChatJson(json)
-    return { sessionId, delta }
+    const model =
+      json.model != null && json.model !== "" ? String(json.model) : null
+    return { sessionId, delta, model }
   } catch (e) {
     return null
   }
@@ -151,6 +153,12 @@ function consumeSseLines(state, lines, callbacks) {
         callbacks.onSessionId(parsed.sessionId)
       }
     }
+    if (parsed.model) {
+      state.model = parsed.model
+      if (callbacks.onModel) {
+        callbacks.onModel(parsed.model)
+      }
+    }
     if (parsed.delta) {
       const before = state.fullText
       appendStreamDelta(state, parsed.delta)
@@ -181,6 +189,12 @@ function appendSseChunk(state, chunk, callbacks) {
           callbacks.onSessionId(parsed.sessionId)
         }
       }
+      if (parsed.model) {
+        state.model = parsed.model
+        if (callbacks.onModel) {
+          callbacks.onModel(parsed.model)
+        }
+      }
       return
     }
   }
@@ -220,9 +234,9 @@ function streamRequestHeaders() {
 /** 流式：fetch + ReadableStream 逐块解析（axios XHR 常会缓冲到结束才触发 progress） */
 async function sendChatMessageStream(data, options) {
   const startMs = Date.now()
-  const { onDelta, onSessionId, signal } = options
-  const state = { buffer: "", fullText: "", sessionId: null }
-  const callbacks = { onDelta, onSessionId }
+  const { onDelta, onSessionId, onModel, signal } = options
+  const state = { buffer: "", fullText: "", sessionId: null, model: null }
+  const callbacks = { onDelta, onSessionId, onModel }
 
   const res = await fetch(chatCompletionsUrl(), {
     method: "POST",
@@ -260,6 +274,7 @@ async function sendChatMessageStream(data, options) {
   return {
     content: state.fullText,
     sessionId: state.sessionId,
+    model: state.model,
     durationMs: Date.now() - startMs
   }
 }
@@ -267,8 +282,8 @@ async function sendChatMessageStream(data, options) {
 /**
  * 聊天 POST /v1/chat/completions（默认 stream: true）
  * - stream: false → JSON（走 request 默认响应）
- * - 其余情况 → SSE，options 可传 onDelta / onSessionId / signal
- * - 流式 resolve：{ content, sessionId, durationMs }，durationMs 为整段流式请求耗时（毫秒）
+ * - 其余情况 → SSE，options 可传 onDelta / onSessionId / onModel / signal
+ * - 流式 resolve：{ content, sessionId, model, durationMs }，durationMs 为整段流式请求耗时（毫秒）
  */
 export function sendChatMessage(data, options = {}) {
   const body = { stream: true, ...(data || {}) }

ruoyi-ui/src/views/diseaseTreatment/onlineConsult/ai/index.vue

@@ -317,10 +317,10 @@ const DEFAULT_SESSION_TITLE = "新会话"
 const LLM_SESSION_STORAGE_PREFIX = "ai_consult_llm_session_"
 
 const MODEL_OPTION_DEFS = [
-  { value: "auto", labelKey: "modelAuto", shortKey: "modelAutoShort", descKey: "modelAutoDesc", icon: "el-icon-cpu" },
-  { value: "yak-disease", labelKey: "modelDisease", shortKey: "modelDiseaseShort", descKey: "modelDiseaseDesc", icon: "el-icon-s-flag" },
+  // { value: "auto", labelKey: "modelAuto", shortKey: "modelAutoShort", descKey: "modelAutoDesc", icon: "el-icon-cpu" },
   { value: "yak-general", labelKey: "modelGeneral", shortKey: "modelGeneralShort", descKey: "modelGeneralDesc", icon: "el-icon-chat-dot-round" },
   { value: "yak-feeding", labelKey: "modelFeeding", shortKey: "modelFeedingShort", descKey: "modelFeedingDesc", icon: "el-icon-s-goods" },
+  { value: "yak-disease", labelKey: "modelDisease", shortKey: "modelDiseaseShort", descKey: "modelDiseaseDesc", icon: "el-icon-s-flag" },
   { value: "yak-growth", labelKey: "modelGrowth", shortKey: "modelGrowth", descKey: "modelGrowthDesc", icon: "el-icon-s-flag" },
 ]
 
@@ -1017,13 +1017,14 @@ export default {
         }
       }
     },
-    queuePendingPersist(consultSessionId, persistBody, aiReplyContent, llmSessionId, costTime) {
+    queuePendingPersist(consultSessionId, persistBody, aiReplyContent, llmSessionId, costTime, aiCategory) {
       this.pendingPersist = {
         consultSessionId,
         persistBody: { ...persistBody },
         aiReplyContent,
         llmSessionId: llmSessionId != null ? llmSessionId : null,
-        costTime: costTime != null ? costTime : null
+        costTime: costTime != null ? costTime : null,
+        aiCategory: aiCategory != null && aiCategory !== "" ? aiCategory : null
       }
     },
     async flushPendingPersist() {
@@ -1036,14 +1037,15 @@ export default {
           ...pending.persistBody,
           aiReplyContent: pending.aiReplyContent,
           realSessionId: pending.llmSessionId != null ? pending.llmSessionId : null,
-          costTime: pending.costTime != null ? pending.costTime : undefined
+          costTime: pending.costTime != null ? pending.costTime : undefined,
+          aiCategory: pending.aiCategory != null ? pending.aiCategory : undefined
         })
         this.pendingPersist = null
       } catch (err) {
         // 离开页面时仍失败则保留队列，下次进入同页可再试（仅内存，刷新会丢）
       }
     },
-    async persistConsultExchange(persistBody, aiReplyContent, costTime) {
+    async persistConsultExchange(persistBody, aiReplyContent, costTime, aiCategory) {
       const reply = (aiReplyContent || "").trim()
       if (!this.activeSessionId || !persistBody || !reply) {
         return true
@@ -1054,7 +1056,8 @@ export default {
         ...persistBody,
         aiReplyContent: reply,
         realSessionId: llmSessionId || null,
-        costTime: costTime != null ? costTime : undefined
+        costTime: costTime != null ? costTime : undefined,
+        aiCategory: aiCategory != null && aiCategory !== "" ? aiCategory : undefined
       }
       try {
         const res = await sendAiConsultMessage(consultSessionId, saveBody)
@@ -1069,7 +1072,7 @@ export default {
         }
         return true
       } catch (err) {
-        this.queuePendingPersist(consultSessionId, persistBody, reply, llmSessionId, costTime)
+        this.queuePendingPersist(consultSessionId, persistBody, reply, llmSessionId, costTime, aiCategory)
         const msg = (err && err.message) || this.dtT("saveSessionFailed")
         this.$modal.msgWarning(msg)
         return false
@@ -1231,6 +1234,7 @@ export default {
       let aiReplyContent = ""
       let aiMsg = null
       let costTime = null
+      let aiCategory = null
       this.abortStreamRequest()
       const abortController = new AbortController()
       this.streamAbortController = abortController
@@ -1241,6 +1245,9 @@ export default {
           onSessionId: (llmSid) => {
             this.rememberLlmSessionId(session, llmSid)
           },
+          onModel: (model) => {
+            aiCategory = model
+          },
           onDelta: (fullText) => {
             aiReplyContent = fullText
             if (!aiMsg) {
@@ -1257,6 +1264,9 @@ export default {
           this.rememberLlmSessionId(session, llmSid)
         }
         costTime = streamResult.durationMs != null ? streamResult.durationMs : null
+        if (streamResult.model) {
+          aiCategory = streamResult.model
+        }
         this.clearStreamRevealTimer()
         aiReplyContent = (streamResult.content || aiReplyContent || "").trim() || this.dtT("noContent")
         if (!aiMsg) {
@@ -1287,7 +1297,12 @@ export default {
         this.sending = false
         this.$nextTick(() => this.scrollChatBottom())
       }
-      await this.persistConsultExchange(persistBody, (aiReplyContent || "").trim(), costTime)
+      await this.persistConsultExchange(
+        persistBody,
+        (aiReplyContent || "").trim(),
+        costTime,
+        aiCategory
+      )
     },
     extOf(fileName) {
       if (!fileName || fileName.lastIndexOf(".") < 0) {