(improvement)(headless)Add timeFormat to dimension metadata of the …

…Text2SQL prompt.
tencentmusic · Sep 12, 2024 · fa4ce8e · fa4ce8e
1 parent ff0833d
commit fa4ce8e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...less/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java b/...less/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java
@@ -114,9 +114,11 @@ public String buildSchemaStr(LLMReq llmReq) {
                                 dimension.getAlias().stream().forEach(a -> alias.append(a + ","));
                                 dimensionStr.append(" ALIAS '" + alias + "'");
                             }
+                            if (StringUtils.isNotEmpty(dimension.getTimeFormat())) {
+                                dimensionStr.append(" FORMAT '" + dimension.getTimeFormat() + "'");
+                            }
                             if (StringUtils.isNotEmpty(dimension.getDescription())) {
-                                dimensionStr.append(
-                                        " COMMENT '" + dimension.getDescription() + "'");
+                                dimensionStr.append(" COMMENT '" + dimension.getDescription() + "'");
                             }
                             dimensionStr.append(">");
                             dimensions.add(dimensionStr.toString());