Skip to content

Commit 4d095f9

Browse files
(fix)(chat) Fix query dimension value error (#440)
Co-authored-by: jolunoluo
1 parent 7c7ccad commit 4d095f9

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

chat/core/src/main/java/com/tencent/supersonic/chat/service/impl/QueryServiceImpl.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,9 @@
1010
import com.tencent.supersonic.chat.api.component.SemanticQuery;
1111
import com.tencent.supersonic.chat.api.pojo.ChatContext;
1212
import com.tencent.supersonic.chat.api.pojo.QueryContext;
13+
import com.tencent.supersonic.chat.api.pojo.SchemaElement;
1314
import com.tencent.supersonic.chat.api.pojo.SemanticParseInfo;
15+
import com.tencent.supersonic.chat.api.pojo.SemanticSchema;
1416
import com.tencent.supersonic.chat.api.pojo.request.DimensionValueReq;
1517
import com.tencent.supersonic.chat.api.pojo.request.ExecuteQueryReq;
1618
import com.tencent.supersonic.chat.api.pojo.request.QueryDataReq;
@@ -580,8 +582,12 @@ private void validFilter(Set<QueryFilter> filters) {
580582
@Override
581583
public Object queryDimensionValue(DimensionValueReq dimensionValueReq, User user) throws Exception {
582584
QueryResultWithSchemaResp queryResultWithSchemaResp = new QueryResultWithSchemaResp();
585+
SemanticService semanticService = ContextUtils.getBean(SemanticService.class);
586+
SemanticSchema semanticSchema = semanticService.getSemanticSchema();
587+
SchemaElement schemaElement = semanticSchema.getDimensions(dimensionValueReq.getElementID());
583588
Set<Long> detectModelIds = new HashSet<>();
584-
detectModelIds.add(dimensionValueReq.getModelId());
589+
detectModelIds.add(schemaElement.getModel());
590+
dimensionValueReq.setModelId(schemaElement.getModel());
585591
List<String> dimensionValues = getDimensionValues(dimensionValueReq, detectModelIds);
586592
// if the search results is null,search dimensionValue from database
587593
if (CollectionUtils.isEmpty(dimensionValues)) {

0 commit comments

Comments
 (0)