diff --git a/chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/parse/ErrorMsgRewriteProcessor.java b/chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/parse/ErrorMsgRewriteProcessor.java index 88c174de0..ec3f3b671 100644 --- a/chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/parse/ErrorMsgRewriteProcessor.java +++ b/chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/parse/ErrorMsgRewriteProcessor.java @@ -4,6 +4,7 @@ import com.tencent.supersonic.chat.server.pojo.ParseContext; import com.tencent.supersonic.common.pojo.ChatApp; import com.tencent.supersonic.common.pojo.enums.AppModule; import com.tencent.supersonic.common.util.ChatAppManager; +import com.tencent.supersonic.headless.api.pojo.response.ParseResp; import com.tencent.supersonic.headless.server.utils.ModelConfigHelper; import dev.langchain4j.data.message.AiMessage; import dev.langchain4j.model.chat.ChatLanguageModel; @@ -71,6 +72,7 @@ public class ErrorMsgRewriteProcessor implements ParseResultProcessor { Response response = chatLanguageModel.generate(prompt.toUserMessage()); String rewrittenMsg = response.content().text(); parseContext.getResponse().setErrorMsg(rewrittenMsg); + parseContext.getResponse().setState(ParseResp.ParseState.FAILED); keyPipelineLog.info("ErrorMessageProcessor modelReq:\n{} \nmodelResp:\n{}", prompt.text(), rewrittenMsg); }