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)(headless) 修复LLM问题变成术语的问题 (#1617)

Browse Source 
Co-authored-by: wangyudong
This commit is contained in:
yudong
2024-09-02 21:41:19 +08:00
committed by GitHub
parent 26863769b5
commit 14177e265f
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/mapper/TermDescMapper.java
View File

@@ -23,11 +23,15 @@ public class TermDescMapper extends BaseMapper {
chatQueryContext.setOriQueryText(chatQueryContext.getQueryText()); chatQueryContext.setOriQueryText(chatQueryContext.getQueryText());
} }
for (SchemaElement schemaElement : termDescriptionToMap) { for (SchemaElement schemaElement : termDescriptionToMap) {
if (schemaElement.isDescriptionMapped()) {
continue;
}
if (chatQueryContext.getQueryText().equals(schemaElement.getDescription())) { if (chatQueryContext.getQueryText().equals(schemaElement.getDescription())) {
schemaElement.setDescriptionMapped(true); schemaElement.setDescriptionMapped(true);
continue; continue;
} }
chatQueryContext.setQueryText(schemaElement.getDescription()); chatQueryContext.setQueryText(schemaElement.getDescription());
break;
} }
if (CollectionUtils.isEmpty(chatQueryContext.getMapInfo().getTermDescriptionToMap())) { if (CollectionUtils.isEmpty(chatQueryContext.getMapInfo().getTermDescriptionToMap())) {
chatQueryContext.setQueryText(chatQueryContext.getOriQueryText()); chatQueryContext.setQueryText(chatQueryContext.getOriQueryText());

2
headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ChatWorkflowEngine.java
View File

@@ -47,6 +47,8 @@ public class ChatWorkflowEngine {
parseResult.setState(ParseResp.ParseState.FAILED); parseResult.setState(ParseResp.ParseState.FAILED);
parseResult.setErrorMsg("No semantic entities can be mapped against user question."); parseResult.setErrorMsg("No semantic entities can be mapped against user question.");
queryCtx.setChatWorkflowState(ChatWorkflowState.FINISHED); queryCtx.setChatWorkflowState(ChatWorkflowState.FINISHED);
} else if (queryCtx.getMapInfo().needContinueMap()) {
queryCtx.setChatWorkflowState(ChatWorkflowState.MAPPING);
} else { } else {
queryCtx.setChatWorkflowState(ChatWorkflowState.PARSING); queryCtx.setChatWorkflowState(ChatWorkflowState.PARSING);
} }