From da0ac7b26c9d621b09b45523d92606bbcd92ca63 Mon Sep 17 00:00:00 2001
From: williamhliu <137068196+williamhliu@users.noreply.github.com>
Date: Sat, 10 Aug 2024 10:17:41 +0800
Subject: [PATCH] (improvement)(chat-sdk) add question in export log and hide
 date range picker when dateInfo is null (#1550)

---
 .../src/components/ChatItem/ParseTip.tsx      | 58 ++++++++++---------
 .../src/components/ChatItem/SqlItem.tsx       |  7 +++
 .../src/components/ChatItem/index.tsx         |  1 +
 3 files changed, 38 insertions(+), 28 deletions(-)

diff --git a/webapp/packages/chat-sdk/src/components/ChatItem/ParseTip.tsx b/webapp/packages/chat-sdk/src/components/ChatItem/ParseTip.tsx
index 1264268e5..617561bb5 100644
--- a/webapp/packages/chat-sdk/src/components/ChatItem/ParseTip.tsx
+++ b/webapp/packages/chat-sdk/src/components/ChatItem/ParseTip.tsx
@@ -252,34 +252,36 @@ const ParseTip: React.FC<Props> = ({
     });
     return (
       <div className={`${prefixCls}-tip-item-filter-content`}>
-        <div className={tipItemOptionClass}>
-          <span className={`${prefixCls}-tip-item-filter-name`}>数据时间：</span>
-          {nativeQuery ? (
-            <span className={itemValueClass}>
-              {startDate === endDate ? startDate : `${startDate} ~ ${endDate}`}
-            </span>
-          ) : (
-            <RangePicker
-              value={[dayjs(startDate), dayjs(endDate)]}
-              onChange={onDateInfoChange}
-              format="YYYY-MM-DD"
-              renderExtraFooter={() => (
-                <Row gutter={[28, 28]}>
-                  {Object.keys(ranges).map(key => (
-                    <Col key={key}>
-                      <Button
-                        size="small"
-                        onClick={() => handlePresetClick(ranges[key as RangeKeys])}
-                      >
-                        {key}
-                      </Button>
-                    </Col>
-                  ))}
-                </Row>
-              )}
-            />
-          )}
-        </div>
+        {!!dateInfo && (
+          <div className={tipItemOptionClass}>
+            <span className={`${prefixCls}-tip-item-filter-name`}>数据时间：</span>
+            {nativeQuery ? (
+              <span className={itemValueClass}>
+                {startDate === endDate ? startDate : `${startDate} ~ ${endDate}`}
+              </span>
+            ) : (
+              <RangePicker
+                value={[dayjs(startDate), dayjs(endDate)]}
+                onChange={onDateInfoChange}
+                format="YYYY-MM-DD"
+                renderExtraFooter={() => (
+                  <Row gutter={[28, 28]}>
+                    {Object.keys(ranges).map(key => (
+                      <Col key={key}>
+                        <Button
+                          size="small"
+                          onClick={() => handlePresetClick(ranges[key as RangeKeys])}
+                        >
+                          {key}
+                        </Button>
+                      </Col>
+                    ))}
+                  </Row>
+                )}
+              />
+            )}
+          </div>
+        )}
         {filters?.map((filter: any, index: number) => (
           <FilterItem
             modelId={modelId!}
diff --git a/webapp/packages/chat-sdk/src/components/ChatItem/SqlItem.tsx b/webapp/packages/chat-sdk/src/components/ChatItem/SqlItem.tsx
index 31abccc0c..cc8fa04ce 100644
--- a/webapp/packages/chat-sdk/src/components/ChatItem/SqlItem.tsx
+++ b/webapp/packages/chat-sdk/src/components/ChatItem/SqlItem.tsx
@@ -12,6 +12,7 @@ import { exportTextFile } from '../../utils/utils';
 type Props = {
   agentId?: number;
   queryId: number;
+  question: string;
   llmReq?: any;
   llmResp?: any;
   integrateSystem?: string;
@@ -23,6 +24,7 @@ type Props = {
 const SqlItem: React.FC<Props> = ({
   agentId,
   queryId,
+  question,
   llmReq,
   llmResp,
   integrateSystem,
@@ -126,6 +128,11 @@ ${format(sqlInfo.querySQL)}
 
   const onExportLog = () => {
     let text = '';
+    if (question) {
+      text += `
+问题：${question}
+`;
+    }
     if (llmReq) {
       text += getSchemaMapText();
     }
diff --git a/webapp/packages/chat-sdk/src/components/ChatItem/index.tsx b/webapp/packages/chat-sdk/src/components/ChatItem/index.tsx
index a4e24cac6..1e8e4079c 100644
--- a/webapp/packages/chat-sdk/src/components/ChatItem/index.tsx
+++ b/webapp/packages/chat-sdk/src/components/ChatItem/index.tsx
@@ -353,6 +353,7 @@ const ChatItem: React.FC<Props> = ({
                 <SqlItem
                   agentId={agentId}
                   queryId={parseInfo.queryId}
+                  question={msg}
                   llmReq={llmReq}
                   llmResp={llmResp}
                   integrateSystem={integrateSystem}