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)(chat) Fixed the NullPointerException issue in DimensionRecommendProcessor (#1629)

Browse Source
This commit is contained in:
lexluo09
2024-09-04 22:46:38 +08:00
committed by GitHub
parent 5eb0a7a5c5
commit 1b6277e53f
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/execute/DimensionRecommendProcessor.java
View File

@@ -34,8 +34,13 @@ public class DimensionRecommendProcessor implements ExecuteResultProcessor {
|| CollectionUtils.isEmpty(semanticParseInfo.getMetrics())) { || CollectionUtils.isEmpty(semanticParseInfo.getMetrics())) {
return; return;
} }
SchemaElement element = semanticParseInfo.getMetrics().iterator().next(); Long dataSetId = semanticParseInfo.getDataSetId();
List<SchemaElement> dimensionRecommended = getDimensions(element.getId(), element.getDataSetId()); Optional<SchemaElement> firstMetric = semanticParseInfo.getMetrics().stream().findFirst();
if (!firstMetric.isPresent()) {
return;
}
SchemaElement element = firstMetric.get();
List<SchemaElement> dimensionRecommended = getDimensions(element.getId(), dataSetId);
queryResult.setRecommendedDimensions(dimensionRecommended); queryResult.setRecommendedDimensions(dimensionRecommended);
} }

6
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/OnePassSCSqlGenStrategy.java
View File

@@ -101,11 +101,11 @@ public class OnePassSCSqlGenStrategy extends SqlGenStrategy {
// use custom prompt template if provided. // use custom prompt template if provided.
PromptConfig promptConfig = llmReq.getPromptConfig(); PromptConfig promptConfig = llmReq.getPromptConfig();
String prompTemplate = INSTRUCTION; String promptTemplate = INSTRUCTION;
if (promptConfig != null && StringUtils.isNotBlank(promptConfig.getPromptTemplate())) { if (promptConfig != null && StringUtils.isNotBlank(promptConfig.getPromptTemplate())) {
prompTemplate = promptConfig.getPromptTemplate(); promptTemplate = promptConfig.getPromptTemplate();
} }
return PromptTemplate.from(prompTemplate).apply(variable); return PromptTemplate.from(promptTemplate).apply(variable);
} }
@Override @Override