test-chat-sdk 0.0.0

package/src/common/type.ts

@@ -0,0 +1,270 @@
+import { Dayjs } from 'dayjs';
+import { ChatContextTypeQueryTypeEnum } from './constants';
+
+export type SearchRecommendItem = {
+  complete: boolean;
+  modelId: number;
+  modelName: string;
+  recommend: string;
+  subRecommend: string;
+  schemaElementType: string;
+};
+
+export type FieldType = {
+  bizName: string;
+  itemId: number;
+  id: number;
+  name: string;
+  status: number;
+  model: number;
+  type: string;
+  value: string;
+};
+
+export type ModelInfoType = {
+  bizName: string;
+  itemId: number;
+  name: string;
+  primaryEntityBizName: string;
+  value: string;
+  words: string[];
+};
+
+export type EntityInfoType = {
+  dataSetInfo: ModelInfoType;
+  dimensions: FieldType[];
+  metrics: FieldType[];
+  entityId: number;
+};
+
+export type DateInfoType = {
+  dateList: any[];
+  dateMode: string;
+  period: string;
+  startDate: string;
+  endDate: string;
+  text: string;
+  unit: number;
+};
+
+export type FilterItemType = {
+  elementID: number;
+  name: string;
+  bizName: string;
+  operator?: string;
+  type?: string;
+  value: any;
+  entityName?: string;
+};
+
+export type ModelType = {
+  alias: string;
+  bizName: string;
+  id: number;
+  model: number;
+  name: string;
+  modelNames: string[];
+  type: string;
+  useCnt: number;
+};
+
+export type EntityDimensionType = {
+  bizName: string;
+  itemId: number;
+  name: string;
+  value: string;
+};
+
+export type SqlInfoType = {
+  parsedS2SQL: string;
+  correctedS2SQL: string;
+  querySQL: string;
+  correctedQuerySQL?: string;
+};
+
+export type ChatContextType = {
+  id: number;
+  queryId?: number;
+  aggType: string;
+  modelId: number;
+  modelName: string;
+  dataSet: ModelType;
+  dataSetIds?: number[];
+  dateInfo: DateInfoType;
+  dimensions: FieldType[];
+  metrics: FieldType[];
+  entity: { alias: string[]; id: number };
+  entityInfo: EntityInfoType;
+  elementMatches: any[];
+  nativeQuery: boolean;
+  queryMode: string;
+  queryType: ChatContextTypeQueryTypeEnum;
+  dimensionFilters: FilterItemType[];
+  properties: any;
+  sqlInfo: SqlInfoType;
+  textInfo: string;
+};
+
+export enum MsgValidTypeEnum {
+  NORMAL = 0,
+  SEARCH_EXCEPTION = 1,
+  EMPTY = 2,
+  INVALID = 3,
+}
+
+export type PluginResonseType = {
+  description: string;
+  webPage: { url: string; paramOptions: any; params: any; valueParams: any };
+  pluginId: number;
+  pluginType: string;
+  name: string;
+};
+
+export type MetricInfoType = {
+  date: string;
+  name: string;
+  statistics: any;
+  value: string;
+};
+
+export type AggregateInfoType = {
+  metricInfos: MetricInfoType[];
+};
+
+export type MsgDataType = {
+  id: number;
+  question: string;
+  aggregateInfo: AggregateInfoType;
+  chatContext: ChatContextType;
+  entityInfo: EntityInfoType;
+  queryAuthorization: any;
+  queryColumns: ColumnType[];
+  queryResults: any[];
+  queryId: number;
+  queryMode: string;
+  queryState: string;
+  queryText: string;
+  response: PluginResonseType;
+  parseInfos?: ChatContextType[];
+  queryTimeCost?: number;
+  similarQueries: SimilarQuestionType[];
+  recommendedDimensions: DrillDownDimensionType[];
+  textResult: string;
+  errorMsg: string;
+  textSummary: string;
+};
+
+export enum ParseStateEnum {
+  COMPLETED = 'COMPLETED',
+  PENDING = 'PENDING',
+  FAILED = 'FAILED',
+}
+
+export type ParseDataType = {
+  chatId: number;
+  queryId: number;
+  queryText: string;
+  state: ParseStateEnum;
+  selectedParses: ChatContextType[];
+  candidateParses: ChatContextType[];
+  similarSolvedQuery: SimilarQuestionType[];
+};
+
+export type QueryDataType = {
+  chatContext: ChatContextType;
+  aggregateInfo: AggregateInfoType;
+  queryColumns: ColumnType[];
+  queryResults: any[];
+};
+
+export type ColumnType = {
+  authorized: boolean;
+  name: string;
+  nameEn: string;
+  bizName: string;
+  showType: string;
+  type: string;
+  dataFormatType: string;
+  dataFormat: {
+    decimalPlaces: number;
+    needMultiply100: boolean;
+  };
+};
+
+export enum SemanticTypeEnum {
+  DOMAIN = 'DOMAIN',
+  DIMENSION = 'DIMENSION',
+  METRIC = 'METRIC',
+  VALUE = 'VALUE',
+}
+
+export const SEMANTIC_TYPE_MAP = {
+  [SemanticTypeEnum.DOMAIN]: '数据模型',
+  [SemanticTypeEnum.DIMENSION]: '维度',
+  [SemanticTypeEnum.METRIC]: '指标',
+  [SemanticTypeEnum.VALUE]: '维度值',
+};
+
+export type SuggestionItemType = {
+  model: number;
+  name: string;
+  bizName: string;
+};
+
+export type SuggestionType = {
+  dimensions: SuggestionItemType[];
+  metrics: SuggestionItemType[];
+};
+
+export type SuggestionDataType = {
+  currentAggregateType: string;
+  columns: ColumnType[];
+  mainEntity: EntityInfoType;
+  suggestions: SuggestionType;
+};
+
+export type HistoryMsgItemType = {
+  questionId: number;
+  queryText: string;
+  parseInfos: ChatContextType[];
+  parseTimeCost: ParseTimeCostType;
+  queryResult: MsgDataType;
+  chatId: number;
+  createTime: string;
+  feedback: string;
+  score: number;
+  similarQueries: SimilarQuestionType[];
+};
+
+export type HistoryType = {
+  hasNextPage: boolean;
+  list: HistoryMsgItemType[];
+};
+
+export type DrillDownDimensionType = {
+  id: number;
+  model: number;
+  name: string;
+  bizName: string;
+};
+
+export type SendMsgParamsType = {
+  msg: string;
+  agentId: number;
+  modelId: number;
+  filters?: FilterItemType[];
+};
+
+export type SimilarQuestionType = {
+  queryId: number;
+  parseId: number;
+  queryText: string;
+};
+
+export type ParseTimeCostType = {
+  parseStartTime: number;
+  parseTime: number;
+  sqlTime: number;
+};
+
+export type RangeValue = [Dayjs, Dayjs];

package/src/components/ChatItem/ExecuteItem.tsx

@@ -0,0 +1,210 @@
+import { Space, Spin, Switch, Tooltip, message } from 'antd';
+import { CheckCircleFilled, InfoCircleOutlined } from '@ant-design/icons';
+import { PREFIX_CLS, MsgContentTypeEnum } from '../../common/constants';
+import { MsgDataType } from '../../common/type';
+import ChatMsg from '../ChatMsg';
+import WebPage from '../ChatMsg/WebPage';
+import Loading from './Loading';
+import { Prism as SyntaxHighlighter } from 'react-syntax-highlighter';
+import { solarizedlight } from 'react-syntax-highlighter/dist/esm/styles/prism';
+import React, { ReactNode, useState, useEffect } from 'react';
+import ReactMarkdown from 'react-markdown';
+
+type Props = {
+  queryId?: number;
+  parseId?: number;
+  question: string;
+  queryMode?: string;
+  executeLoading: boolean;
+  entitySwitchLoading?: boolean;
+  chartIndex: number;
+  executeTip?: string;
+  executeErrorMsg?: string;
+  executeItemNode?: ReactNode;
+  renderCustomExecuteNode?: boolean;
+  data?: MsgDataType;
+  triggerResize?: boolean;
+  isDeveloper?: boolean;
+  isSimpleMode?: boolean;
+  enableStreamingSummary?: boolean;
+  onStreamingSummaryUpdate?: (summary: string) => void;
+  agentId?: number;
+  dataSetIds?: number[];
+};
+
+const ExecuteItem: React.FC<Props> = ({
+  queryId,
+  parseId,
+  question,
+  queryMode,
+  executeLoading,
+  entitySwitchLoading = false,
+  chartIndex,
+  executeTip,
+  executeErrorMsg,
+  executeItemNode,
+  renderCustomExecuteNode,
+  data,
+  triggerResize,
+  isDeveloper,
+  isSimpleMode,
+  enableStreamingSummary,
+  onStreamingSummaryUpdate,
+  agentId,
+  dataSetIds,
+}) => {
+  const prefixCls = `${PREFIX_CLS}-item`;
+  const [showMsgContentTable, setShowMsgContentTable] = useState<boolean>(false);
+  const [msgContentType, setMsgContentType] = useState<MsgContentTypeEnum>();
+  const [showErrMsg, setShowErrMsg] = useState<boolean>(false);
+  const [isStreaming, setIsStreaming] = useState(false);
+  const [summaryBuilder, setSummaryBuilder] = useState('');
+  const titlePrefix = queryMode === 'PLAIN_TEXT' || queryMode === 'WEB_SERVICE' ? '问答' : '数据';
+
+  const getNodeTip = (title: ReactNode, tip?: string | ReactNode) => {
+    return (
+      <>
+        <div className={`${prefixCls}-title-bar`}>
+          <CheckCircleFilled className={`${prefixCls}-step-icon`} />
+          <div className={`${prefixCls}-step-title`}>
+            {title}
+            {!tip && <Loading />}
+          </div>
+        </div>
+        {tip && <div className={`${prefixCls}-content-container`}>{tip}</div>}
+      </>
+    );
+  };
+
+  // 流式总结功能已移除
+  useEffect(() => {
+    // 流式总结功能已移除，不再处理
+  }, [enableStreamingSummary, queryId, parseId, agentId]);
+
+  if (executeLoading) {
+    return getNodeTip(`${titlePrefix}查询中`);
+  }
+
+  const handleCopy = (_: string, result: any) => {
+    result ? message.success('复制SQL成功', 1) : message.error('复制SQL失败', 1);
+  };
+
+  if (executeTip) {
+    return getNodeTip(
+      <>
+        <span>{titlePrefix}查询失败</span>
+        {executeErrorMsg && (
+          <Space>
+            <InfoCircleOutlined style={{ marginLeft: 5, color: 'red' }} />
+            <a
+              onClick={() => {
+                setShowErrMsg(!showErrMsg);
+              }}
+            >
+              {!showErrMsg ? '查看' : '收起'}
+            </a>
+          </Space>
+        )}
+        {!!data?.queryTimeCost && isDeveloper && (
+          <span className={`${prefixCls}-title-tip`}>(耗时: {data.queryTimeCost}ms)</span>
+        )}
+      </>,
+
+      <>
+        {showErrMsg && (
+          <SyntaxHighlighter className={`${prefixCls}-code`} language="sql" style={solarizedlight}>
+            {executeErrorMsg}
+          </SyntaxHighlighter>
+        )}
+      </>
+    );
+  }
+
+  if (!data) {
+    return null;
+  }
+
+  const getSummaryNode = () => {
+    // 流式总结功能已移除，直接显示最终的总结
+    if (data.textSummary) {
+      return (
+        <div className={`${prefixCls}-summary`}>
+          <ReactMarkdown>{data.textSummary}</ReactMarkdown>
+        </div>
+      );
+    }
+
+    return null;
+  };
+
+  return (
+    <>
+      {!isSimpleMode && (
+        <div className={`${prefixCls}-title-bar`}>
+          <CheckCircleFilled className={`${prefixCls}-step-icon`} />
+          <div
+            className={`${prefixCls}-step-title ${prefixCls}-execute-title-bar`}
+            style={{ width: '100%' }}
+          >
+            <div>
+              {titlePrefix}查询
+              {!!data?.queryTimeCost && isDeveloper && (
+                <span className={`${prefixCls}-title-tip`}>(耗时: {data.queryTimeCost}ms)</span>
+              )}
+            </div>
+            <div>
+              {[
+                MsgContentTypeEnum.METRIC_TREND,
+                MsgContentTypeEnum.METRIC_BAR,
+                MsgContentTypeEnum.METRIC_PIE,
+              ].includes(msgContentType as MsgContentTypeEnum) && (
+                <Switch
+                  checkedChildren="表格"
+                  unCheckedChildren="表格"
+                  onChange={checked => {
+                    setShowMsgContentTable(checked);
+                  }}
+                />
+              )}
+            </div>
+          </div>
+        </div>
+      )}
+
+      <div
+        className={`${prefixCls}-content-container ${
+          isSimpleMode ? `${prefixCls}-content-container-simple` : ''
+        }`}
+        style={{ borderLeft: queryMode === 'PLAIN_TEXT' ? 'none' : undefined }}
+      >
+        <Spin spinning={entitySwitchLoading}>
+          {data.queryAuthorization?.message && (
+            <div className={`${prefixCls}-auth-tip`}>提示：{data.queryAuthorization.message}</div>
+          )}
+          {getSummaryNode()}
+
+          {renderCustomExecuteNode && executeItemNode ? (
+            executeItemNode
+          ) : data?.queryMode === 'PLAIN_TEXT' || data?.queryMode === 'WEB_SERVICE' ? (
+            data?.textResult
+          ) : data?.queryMode === 'WEB_PAGE' ? (
+            <WebPage id={queryId!} data={data} />
+          ) : (
+            <ChatMsg
+              isSimpleMode={isSimpleMode}
+              forceShowTable={showMsgContentTable}
+              queryId={queryId}
+              question={question}
+              data={data}
+              chartIndex={chartIndex}
+              triggerResize={triggerResize}
+              onMsgContentTypeChange={setMsgContentType}
+            />
+          )}
+        </Spin>
+      </div>
+    </>
+  );
+};
+
+export default ExecuteItem;