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

…Text2SQL prompt. (improvement)(headless)Add `timeFormat` to dimension metadata of the Text2SQL prompt. #1621.
tencentmusic · Sep 12, 2024 · 2fa3bfe · 2fa3bfe
1 parent ff0833d
commit 2fa3bfe
Showing 1 changed file with 3 additions and 0 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,6 +114,9 @@ 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() + "'");