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

(improvement)(Headless) Add terms management to help the model learn private domain knowledge (#979)

Browse Source
This commit is contained in:
LXW
2024-05-11 14:26:14 +08:00
committed by GitHub
parent 942fd9beed
commit 7f52215ff8
23 changed files with 338 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/Term.java Normal file
View File

@@ -0,0 +1,15 @@
package com.tencent.supersonic.headless.api.pojo;
import lombok.Data;
import java.util.List;
@Data
public class Term {
private String name;
private String description;
private List<String> similarTerms;
}

19
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/request/TermSetReq.java Normal file
View File

@@ -0,0 +1,19 @@
package com.tencent.supersonic.headless.api.pojo.request;
import com.google.common.collect.Lists;
import com.tencent.supersonic.common.pojo.RecordInfo;
import com.tencent.supersonic.headless.api.pojo.Term;
import lombok.Data;
import javax.validation.constraints.NotNull;
import java.util.List;
@Data
public class TermSetReq extends RecordInfo {
@NotNull(message = "主题域ID不可为空")
private Long domainId;
private List<Term> terms = Lists.newArrayList();
}

3
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/response/MapInfoResp.java
View File

@@ -1,6 +1,7 @@
package com.tencent.supersonic.headless.api.pojo.response;
import com.tencent.supersonic.headless.api.pojo.SchemaElementMatch;
import com.tencent.supersonic.headless.api.pojo.Term;
import lombok.Data;
import java.util.List;
@@ -15,4 +16,6 @@ public class MapInfoResp {
private Map<String, List<SchemaElementMatch>> topFields;
private Map<String, List<Term>> terms;
}

24
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/response/TermSetResp.java Normal file
View File

@@ -0,0 +1,24 @@
package com.tencent.supersonic.headless.api.pojo.response;
import com.google.common.collect.Lists;
import com.tencent.supersonic.common.pojo.RecordInfo;
import com.tencent.supersonic.headless.api.pojo.Term;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.util.List;
@Data
@AllArgsConstructor
@NoArgsConstructor
public class TermSetResp extends RecordInfo {
private Long domainId;
private List<Term> terms = Lists.newArrayList();
public TermSetResp(Long domainId) {
this.domainId = domainId;
}
}