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

(improvement)(Headless) Add STRICT, MODERATE, and LOOSE modes in the mapper phase. (#900)

Browse Source
This commit is contained in:
lexluo09
2024-04-09 16:27:57 +08:00
committed by GitHub
parent 43b9d5545d
commit d6b81249d7
13 changed files with 79 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/enums/MapModeEnum.java
View File

@@ -1,7 +1,12 @@
package com.tencent.supersonic.headless.api.pojo.enums;
public enum MapModeEnum {
STRICT,
MODERATE,
LOOSE;
STRICT(0),
MODERATE(2),
LOOSE(4);
public int threshold;
MapModeEnum(Integer threshold) {
this.threshold = threshold;
}
}

2
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/request/QueryMapReq.java
View File

@@ -14,5 +14,5 @@ public class QueryMapReq {
private List<String> dataSetNames;
private User user;
private Integer topN = 10;
private MapModeEnum mapModeEnum;
private MapModeEnum mapModeEnum = MapModeEnum.STRICT;
}

2
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/request/QueryReq.java
View File

@@ -4,6 +4,7 @@ import com.google.common.collect.Sets;
import com.tencent.supersonic.auth.api.authentication.pojo.User;
import com.tencent.supersonic.common.pojo.enums.Text2SQLType;
import com.tencent.supersonic.headless.api.pojo.SchemaMapInfo;
import com.tencent.supersonic.headless.api.pojo.enums.MapModeEnum;
import lombok.Data;
import java.util.Set;
@@ -17,5 +18,6 @@ public class QueryReq {
private QueryFilters queryFilters;
private boolean saveAnswer = true;
private Text2SQLType text2SQLType = Text2SQLType.RULE_AND_LLM;
private MapModeEnum mapModeEnum = MapModeEnum.STRICT;
private SchemaMapInfo mapInfo = new SchemaMapInfo();
}