topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-10 11:07:06 +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())) {
return;
}
SchemaElement element = semanticParseInfo.getMetrics().iterator().next();
List<SchemaElement> dimensionRecommended = getDimensions(element.getId(), element.getDataSetId());
Long dataSetId = semanticParseInfo.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);
}

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.
PromptConfig promptConfig = llmReq.getPromptConfig();
String prompTemplate = INSTRUCTION;
String promptTemplate = INSTRUCTION;
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