topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-11 03:58:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(fix)(chat)Remove spaces when obtaining a MemoryReviewResult (#1594)

Browse Source
This commit is contained in:
xiuzhu9527
2024-08-26 10:27:08 +08:00
committed by GitHub
parent 3915d90eec
commit 5606633481
2 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/enums/MemoryReviewResult.java
View File

@@ -1,9 +1,21 @@
package com.tencent.supersonic.chat.api.pojo.enums; package com.tencent.supersonic.chat.api.pojo.enums;
import com.tencent.supersonic.common.pojo.exception.InvalidArgumentException;
import org.apache.commons.lang3.StringUtils;
public enum MemoryReviewResult { public enum MemoryReviewResult {
POSITIVE, POSITIVE,
NEGATIVE NEGATIVE;
public static MemoryReviewResult getMemoryReviewResult(String value) {
String validValue = StringUtils.trim(value);
for (MemoryReviewResult reviewRet : MemoryReviewResult.values()) {
if (StringUtils.equalsIgnoreCase(reviewRet.name(), validValue)) {
return reviewRet;
}
}
throw new InvalidArgumentException("Invalid MemoryReviewResult type:[" + value + "]");
}
} }

2
chat/server/src/main/java/com/tencent/supersonic/chat/server/memory/MemoryReviewTask.java
View File

@@ -67,7 +67,7 @@ public class MemoryReviewTask {
Matcher matcher = OUTPUT_PATTERN.matcher(response); Matcher matcher = OUTPUT_PATTERN.matcher(response);
if (matcher.find()) { if (matcher.find()) {
m.setLlmReviewRet(MemoryReviewResult.valueOf(matcher.group(1))); m.setLlmReviewRet(MemoryReviewResult.getMemoryReviewResult(matcher.group(1)));
m.setLlmReviewCmt(matcher.group(2)); m.setLlmReviewCmt(matcher.group(2));
// directly enable memory if the LLM determines it positive // directly enable memory if the LLM determines it positive
if (MemoryReviewResult.POSITIVE.equals(m.getLlmReviewRet())) { if (MemoryReviewResult.POSITIVE.equals(m.getLlmReviewRet())) {