diff --git a/jeecg-boot/jeecg-boot-module/jeecg-boot-module-airag/src/main/java/org/jeecg/modules/airag/llm/controller/AiragModelController.java b/jeecg-boot/jeecg-boot-module/jeecg-boot-module-airag/src/main/java/org/jeecg/modules/airag/llm/controller/AiragModelController.java index 393733a2b..ee805607e 100644 --- a/jeecg-boot/jeecg-boot-module/jeecg-boot-module-airag/src/main/java/org/jeecg/modules/airag/llm/controller/AiragModelController.java +++ b/jeecg-boot/jeecg-boot-module/jeecg-boot-module-airag/src/main/java/org/jeecg/modules/airag/llm/controller/AiragModelController.java @@ -185,7 +185,7 @@ public class AiragModelController extends JeecgController entry : MODEL_ERROR_MAP.entrySet()) { String key = entry.getKey(); String value = entry.getValue(); - if (errMsg.contains(key)) { + if (exceptionMsg.contains(key)) { errMsg = value; break; } @@ -454,7 +459,7 @@ public class AIChatHandler implements IAIChatHandler { for (Map.Entry entry : MODEL_ERROR_MAP.entrySet()) { String key = entry.getKey(); String value = entry.getValue(); - if (errMsg.contains(key)) { + if (e.getMessage().contains(key)) { errMsg = value; break; }