From 2fa3bfe019546df595d522076357c35f6fa13f37 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.

(improvement)(headless)Add `timeFormat` to dimension metadata of the Text2SQL prompt. #1621.
---
 .../supersonic/headless/chat/parser/llm/PromptHelper.java      | 3 +++
 1 file changed, 3 insertions(+)

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..ce88d2f11 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,6 +114,9 @@ public class PromptHelper {
                                 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() + "'");