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

[improvement](Headless) Adding change event to tags and integrating tag into embeddings (#792)

Browse Source
This commit is contained in:
lexluo09
2024-03-06 16:13:20 +08:00
committed by GitHub
parent 475e071341
commit 3cccac58a1
8 changed files with 53 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
chat/core/src/main/java/com/tencent/supersonic/chat/core/parser/sql/llm/LLMSqlParser.java
View File

@@ -28,7 +28,7 @@ public class LLMSqlParser implements SemanticParser {
return;
}
try {
//2.get modelId from queryCtx and chatCtx.
//2.get dataSetId from queryCtx and chatCtx.
Long dataSetId = requestService.getDataSetId(queryCtx);
if (dataSetId == null) {
return;

4
chat/core/src/main/java/com/tencent/supersonic/chat/core/query/semantic/DataSetSchemaBuilder.java
View File

@@ -77,9 +77,7 @@ public class DataSetSchemaBuilder {
private static Set<SchemaElement> getTags(DataSetSchemaResp resp) {
Set<SchemaElement> tags = new HashSet<>();
List<TagResp> tagResps = resp.getTags();
for (TagResp tagResp : tagResps) {
for (TagResp tagResp : resp.getTags()) {
SchemaElement element = SchemaElement.builder()
.dataSet(resp.getId())
.model(tagResp.getModelId())