From fa4ce8e102d675fd281d9e44983065c026fdcae1 Mon Sep 17 00:00:00 2001
From: jerryjzhang <jerryjzhang@tencent.com>
Date: Thu, 12 Sep 2024 09:59:04 +0800
Subject: [PATCH] (improvement)(headless)Add `timeFormat` to dimension metadata
 of the Text2SQL prompt.

---
 .../supersonic/headless/chat/parser/llm/PromptHelper.java   | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java
index 3ad844a36..014fd7ca7 100644
--- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java
+++ b/headless/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());