test-chat-sdk 0.0.0

package/src/hooks/useComposing.ts

@@ -0,0 +1,31 @@
+import { type MutableRefObject, useEffect, useState } from 'react';
+
+const isRefObject = <T>(value: any): value is MutableRefObject<T> => {
+  return value !== null && typeof value === 'object' && 'current' in value;
+};
+
+export const useComposing = (element?: HTMLElement | null | MutableRefObject<HTMLElement>) => {
+  const [isComposing, setIsComposing] = useState(false);
+
+  useEffect(() => {
+    const handleCompositionStart = (): void => {
+      setIsComposing(true);
+    };
+    const handleCompositionEnd = (): void => {
+      setIsComposing(false);
+    };
+
+    const dom = isRefObject(element) ? element.current : element;
+    const target = dom || window;
+
+    target.addEventListener('compositionstart', handleCompositionStart);
+    target.addEventListener('compositionend', handleCompositionEnd);
+
+    return () => {
+      target.removeEventListener('compositionstart', handleCompositionStart);
+      target.removeEventListener('compositionend', handleCompositionEnd);
+    };
+  }, [element]);
+
+  return { isComposing, setIsComposing };
+};

package/src/hooks/useExportByEcharts.ts

@@ -0,0 +1,41 @@
+import { message } from 'antd';
+import { ECharts } from 'echarts';
+
+export interface ExportByEchartsProps {
+  instanceRef: React.MutableRefObject<ECharts | undefined>;
+  question: string;
+  options?: Parameters<ECharts['getConnectedDataURL']>[0];
+}
+
+export const useExportByEcharts = ({ instanceRef, question, options }: ExportByEchartsProps) => {
+  const handleSaveAsImage = () => {
+    if (instanceRef.current) {
+      return instanceRef.current.getConnectedDataURL({
+        type: 'png',
+        pixelRatio: 2,
+        backgroundColor: '#fff',
+        excludeComponents: ['toolbox'],
+        ...options,
+      });
+    }
+  };
+
+  const downloadImage = (url: string) => {
+    const a = document.createElement('a');
+    a.href = url;
+    a.download = `${question}.png`;
+    a.click();
+  };
+
+  const downloadChartAsImage = () => {
+    const url = handleSaveAsImage();
+    if (url) {
+      downloadImage(url);
+      message.success('导出图片成功');
+    } else {
+      message.error('该条消息暂不支持导出图片');
+    }
+  };
+
+  return { downloadChartAsImage };
+};

package/src/hooks/useMethodRegister.ts

@@ -0,0 +1,25 @@
+import { useCallback, useRef } from 'react';
+
+export const useMethodRegister = (fallback?: (...args: any[]) => any) => {
+  const methodStore = useRef<Map<string, (...args: any[]) => any>>(new Map());
+
+  const register = useCallback<(key: string, method: (...args: any[]) => any) => any>(
+    (key, method) => {
+      methodStore.current.set(key, method);
+    },
+    [methodStore]
+  );
+
+  const call = useCallback<(key: string, ...args: any[]) => any>(
+    (key, ...args) => {
+      const method = methodStore.current.get(key);
+      if (method) {
+        return method(...args);
+      }
+      return fallback?.(...args);
+    },
+    [methodStore]
+  );
+
+  return { register, call };
+};

package/src/index.tsx

@@ -0,0 +1,44 @@
+import './styles/index.less';
+
+// import ReactDOM from 'react-dom/client';
+// import Chat from './demo/Chat';
+// import ChatDemo from './demo/ChatDemo';
+// import CopilotDemo from './demo/CopilotDemo';
+// const root = ReactDOM.createRoot(document.getElementById('root') as HTMLElement);
+// root.render(<ChatDemo />);
+
+export { default as Chat } from './Chat';
+
+export { default as Copilot } from './Copilot';
+
+export { default as ChatMsg } from './components/ChatMsg';
+
+export { default as ChatItem } from './components/ChatItem';
+
+export { default as ShowCase } from './ShowCase';
+
+export type {
+  SearchRecommendItem,
+  FieldType,
+  ModelInfoType,
+  EntityInfoType,
+  DateInfoType,
+  ChatContextType,
+  MsgValidTypeEnum,
+  MsgDataType,
+  PluginResonseType,
+  ColumnType,
+  SuggestionItemType,
+  SuggestionType,
+  SuggestionDataType,
+  FilterItemType,
+  HistoryType,
+  HistoryMsgItemType,
+  SendMsgParamsType,
+} from './common/type';
+
+export { searchRecommend } from './service';
+
+export { saveConversation, getAllConversations } from './Chat/service';
+
+export { setToken } from './utils/utils';

package/src/service/axiosInstance.ts

@@ -0,0 +1,58 @@
+// 引入axios库
+import axios, { AxiosInstance } from 'axios';
+import { getToken } from '../utils/utils';
+
+// 创建axios实例
+const axiosInstance: AxiosInstance = axios.create({
+  // 设置基本URL，所有请求都会使用这个URL作为前缀
+  baseURL: 'http://192.168.50.202:9082',
+  // 设置请求超时时间（毫秒）
+  timeout: 120000,
+  // 设置请求头
+  headers: {
+    'Content-Type': 'application/json',
+  },
+});
+
+// 请求拦截器
+axiosInstance.interceptors.request.use(
+  (config: any) => {
+    const token = getToken();
+    if (token && config?.headers) {
+      config.headers.Authorization = `Bearer ${token}`;
+    }
+    return config;
+  },
+  (error) => {
+    // 请求错误时的处理
+    return Promise.reject(error);
+  }
+);
+
+// 响应拦截器
+axiosInstance.interceptors.response.use(
+  (response: any) => {
+    const redirect = response.headers['redirect'];
+    if (redirect === 'REDIRECT') {
+      let win: any = window;
+      while (win !== win.top) {
+        win = win.top;
+      }
+      const contextpath = response.headers['contextpath'];
+      win.location.href =
+        contextpath?.substring(0, contextpath?.indexOf('&')) +
+        `&redirect_uri=${encodeURIComponent(`http://${win.location.host}`)}`;
+    }
+    return response.data;
+  },
+  (error) => {
+    // 对响应错误进行处理
+    if (error.response && error.response.status === 401) {
+      // 如果响应状态码为401，表示未授权，可以在这里处理重新登录等操作
+      console.log('Unauthorized, please log in again.');
+    }
+    return Promise.reject(error);
+  }
+);
+
+export default axiosInstance;

package/src/service/index.ts

@@ -0,0 +1,174 @@
+import axios from './axiosInstance';
+import {
+  ChatContextType,
+  HistoryMsgItemType,
+  HistoryType,
+  MsgDataType,
+  ParseDataType,
+  SearchRecommendItem,
+} from '../common/type';
+import { isMobile, getToken } from '../utils/utils';
+
+const DEFAULT_CHAT_ID = 0;
+
+const prefix = isMobile ? '/openapi' : '/api';
+
+export function searchRecommend(
+  queryText: string,
+  chatId?: number,
+  modelId?: number,
+  agentId?: number
+) {
+  return axios.post<SearchRecommendItem[]>(`${prefix}/chat/query/search`, {
+    queryText,
+    chatId: chatId || DEFAULT_CHAT_ID,
+    modelId,
+    agentId,
+  });
+}
+
+export function chatQuery(queryText: string, chatId?: number, modelId?: number, filters?: any[]) {
+  return axios.post<MsgDataType>(`${prefix}/chat/query/query`, {
+    queryText,
+    chatId: chatId || DEFAULT_CHAT_ID,
+    modelId,
+    queryFilters: filters
+      ? {
+          filters,
+        }
+      : undefined,
+  });
+}
+
+export function chatParse({
+  queryText,
+  chatId,
+  modelId,
+  agentId,
+  parseId,
+  queryId,
+  filters,
+  parseInfo,
+  dataSetIds,
+}: {
+  queryText: string;
+  chatId?: number;
+  modelId?: number;
+  agentId?: number;
+  queryId?: number;
+  parseId?: number;
+  filters?: any[];
+  parseInfo?: ChatContextType;
+  dataSetIds?: number[];
+}) {
+  return axios.post<ParseDataType>(`${prefix}/chat/query/parse`, {
+    queryText,
+    chatId: chatId || DEFAULT_CHAT_ID,
+    dataSetId: dataSetIds && dataSetIds.length > 0 ? dataSetIds[0] : modelId,
+    agentId,
+    parseId,
+    queryId,
+    selectedParse: parseInfo,
+    dataSetIds,
+    queryFilters: filters
+      ? {
+          filters,
+        }
+      : undefined,
+  });
+}
+
+export function chatExecute(
+  queryText: string,
+  chatId: number,
+  parseInfo: ChatContextType,
+  agentId?: number
+) {
+  return axios.post<MsgDataType>(`${prefix}/chat/query/execute`, {
+    queryText,
+    agentId,
+    chatId: chatId || DEFAULT_CHAT_ID,
+    queryId: parseInfo.queryId,
+    parseId: parseInfo.id,
+    dataSetIds: parseInfo.dataSetIds,
+  });
+}
+
+// 流式总结功能已移除
+// export function streamSummary(queryId: number, parseId: number, agentId: number, dataSetIds?: number[]) {
+//   const token = getToken();
+//   const tokenParam = token ? `&token=${encodeURIComponent(token)}` : '';
+//   const dataSetIdsParam = dataSetIds && dataSetIds.length > 0 ? `&dataSetIds=${dataSetIds.join(',')}` : '';
+//   return new EventSource(
+//     `${prefix}/chat/query/stream/summarize?queryId=${queryId}&parseId=${parseId}&agentId=${agentId}${dataSetIdsParam}${tokenParam}`
+//   );
+// }
+
+export function switchEntity(entityId: string, modelId?: number, chatId?: number) {
+  return axios.post<any>(`${prefix}/chat/query/switchQuery`, {
+    queryText: entityId,
+    modelId,
+    chatId: chatId || DEFAULT_CHAT_ID,
+  });
+}
+
+export function queryData(chatContext: Partial<ChatContextType>) {
+  return axios.post<MsgDataType>(`${prefix}/chat/query/queryData`, chatContext);
+}
+
+export function getHistoryMsg(
+  current: number,
+  chatId: number = DEFAULT_CHAT_ID,
+  pageSize: number = 10
+) {
+  return axios.post<HistoryType>(`${prefix}/chat/manage/pageQueryInfo?chatId=${chatId}`, {
+    current,
+    pageSize,
+  });
+}
+
+export function querySimilarQuestions(queryId: number) {
+  return axios.get<HistoryMsgItemType>(`${prefix}/chat/manage/getChatQuery/${queryId}`);
+}
+
+export function deleteQuery(queryId: number) {
+  return axios.delete<any>(`${prefix}/chat/manage/${queryId}`);
+}
+
+export function queryEntities(entityId: string | number, modelId: number) {
+  return axios.post<any>(`${prefix}/chat/query/choice`, {
+    entityId,
+    modelId,
+  });
+}
+
+export function updateQAFeedback(questionId: number, score: number) {
+  return axios.post<any>(
+    `${prefix}/chat/manage/updateQAFeedback?id=${questionId}&score=${score}&feedback=`
+  );
+}
+
+export function queryDimensionValues(
+  modelId: number,
+  bizName: string,
+  agentId: number,
+  elementID: number,
+  value: string
+) {
+  return axios.post<any>(`${prefix}/chat/query/queryDimensionValue`, {
+    modelId,
+    bizName,
+    agentId,
+    elementID,
+    value,
+  });
+}
+
+export function getDataSetDetail(dataSetId: number) {
+  return axios.get<any>(`${prefix}/semantic/dataSet/${dataSetId}`);
+}
+
+export function getDataSetList(domainId?: number) {
+  const params = domainId ? { domainId } : {};
+  return axios.get<any>(`${prefix}/semantic/dataSet/getDataSetList`, { params });
+}

package/src/setupProxy.js

@@ -0,0 +1,18 @@
+const { createProxyMiddleware } = require('http-proxy-middleware');
+
+module.exports = function (app) {
+  app.use(
+    '/api',
+    createProxyMiddleware({
+      target: 'http://localhost:9080',
+      changeOrigin: true,
+    })
+  );
+  app.use(
+    '/openapi',
+    createProxyMiddleware({
+      target: 'http://localhost:9080',
+      changeOrigin: true,
+    })
+  );
+};

package/src/setupTests.ts

@@ -0,0 +1,5 @@
+// jest-dom adds custom jest matchers for asserting on DOM nodes.
+// allows you to do things like:
+// expect(element).toHaveTextContent(/react/i)
+// learn more: https://github.com/testing-library/jest-dom
+import '@testing-library/jest-dom';

package/src/styles/global.less

@@ -0,0 +1,52 @@
+// @import '~antd/dist/antd.css';
+
+@import './index.less';
+
+@prefix-cls: ~'@{supersonic-chat-prefix}';
+
+.@{prefix-cls} {
+  &-dimension,
+  &-metric {
+    position: relative;
+
+    &::after {
+      position: absolute;
+      right: 0.5px;
+      bottom: -2px;
+      left: 0.5px;
+      height: 2px;
+      margin: 0 1px;
+      content: '';
+    }
+  }
+
+  &-dimension {
+    &::after {
+      background: var(--chat-blue);
+    }
+  }
+
+  &-metric {
+    &::after {
+      background: var(--primary-green);
+    }
+  }
+
+  &-table-row {
+    cursor: pointer;
+  }
+
+  &-even-row {
+    background-color: #fbfbfb;
+  }
+
+  &-no-border-table {
+    .ant-table-cell {
+      border: none !important;
+    }
+
+    .ant-table-tbody > tr.ant-table-row:hover > td {
+      background-color: #efefef !important;
+    }
+  } 
+}

package/src/styles/index.less

@@ -0,0 +1,39 @@
+@import './reboot.less';
+
+@import './variables.less';
+
+@import './global.less';
+
+@import '../components/ChatMsg/style.less';
+
+@import '../components/ChatMsg/Bar/style.less';
+
+@import '../components/ChatMsg/Table/style.less';
+
+@import '../components/ChatMsg/Message/style.less';
+
+@import '../components/ChatMsg/MetricCard/style.less';
+
+@import "../components/ChatMsg/MetricTrend/style.less";
+
+@import "../components/ChatMsg/Pie/style.less";
+
+@import "../components/ChatMsg/ApplyAuth/style.less";
+
+@import "../components/ChatMsg/NoPermissionChart/style.less";
+
+@import "../components/ChatMsg/FilterSection/style.less";
+
+@import "../components/ChatMsg/DateOptions/style.less";
+
+@import "../components/ChatMsg/Text/style.less";
+
+@import '../components/ChatItem/style.less';
+
+@import "../components/Tools/style.less";
+
+@import "../components/RecommendOptions/style.less";
+
+@import "../components/DrillDownDimensions/style.less";
+
+@import "../components/MetricOptions/style.less";

package/src/styles/reboot.less

@@ -0,0 +1,14 @@
+body {
+  margin: 0;
+  font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', 'Oxygen',
+    'Ubuntu', 'Cantarell', 'Fira Sans', 'Droid Sans', 'Helvetica Neue',
+    sans-serif;
+  -webkit-font-smoothing: antialiased;
+  -moz-osx-font-smoothing: grayscale;
+  font-size: 14px;
+}
+
+code {
+  font-family: source-code-pro, Menlo, Monaco, Consolas, 'Courier New',
+    monospace;
+}

package/src/styles/variables.less

@@ -0,0 +1,80 @@
+@supersonic-chat-prefix: ~'ss-chat';
+
+:root:root {
+  --primary: 180deg 4%;
+  --primary-color: #f87653;
+  --blue: #296df3;
+  --green: #00d59c;
+  --deep-blue: #446dff;
+  --chat-blue: #1b4aef;
+  --wy-color: #c20c0c;
+  --detail-width: 1300px;
+  --primary-1: #fff0f0;
+  --primary-2: #ffc7c7;
+  --primary-3: #ff9ea1;
+  --primary-4: #ff757e;
+  --primary-5: #ff4d5e;
+  --primary-6: #ff2441;
+  --primary-7: #d91434;
+  --primary-8: rgba(255, 36, 66, 0.1);
+  --body-background: #f7fafa;
+  --deep-background: #f0f0f0;
+  --light-background: #f5f5f5;
+  --component-background: #fff;
+  --header-color: #edf2f2;
+  --text-color: #181a1a;
+  --text-color-secondary: #3d4242;
+  --text-color-third: #626a6a;
+  --text-color-fourth: #889191;
+  --text-color-fifth: #afb6b6;
+  --text-color-six: #a3a4a6;
+  --text-color-fifth-4: hsla(180, 5%, 70%, 0.4);
+  --tooltip-max-width: 350px;
+  --info-color: #ff2442;
+  --success-color: #52c41a;
+  --processing-color: #ff2442;
+  --error-color: #ff4d4f;
+  --highlight-color: #ff4d4f;
+  --newrank-color: #ff7800;
+  --warning-color: #faad14;
+  --normal-color: #d9d9d9;
+  --white: #fff;
+  --white-30: hsla(0, 0%, 100%, 0.3);
+  --black: #000;
+  --disabled-color: #afb6b6;
+  --disabled-bg: #eceeee;
+  --border-color-base: #e1e6e6;
+  --chat-border-color-base: #d5d7db;
+  --light-blue-background: rgba(58, 100, 255, 0.1);
+  --link-color: #3a64ff;
+  --link-hover-color: #638aff;
+  --link-active-color: #2748d9;
+  --link-bg-color: rgba(58, 100, 255, 0.1);
+  --text-accent-color: #3a64ff;
+  --primary-green: #31c462;
+  --link-hover-bg-color: rgba(58, 100, 255, 0.06);
+  --success-2: rgba(82, 196, 26, 0.2);
+  --success-pink: #ff8193;
+  --disabled-bg-3: hsla(180, 6%, 93%, 0.3);
+  --tooltip-bg: #fff;
+  --record-btn: #00b354;
+  --record-btn-bg: rgba(0, 179, 84, 0.1);
+  --record-btn-bg-3: rgba(0, 179, 84, 0.3);
+  --border-color-base-bg-5: hsla(180, 9%, 89%, 0.5);
+  --user-gao-color: #fcad36;
+  --user-hao-color: #ec6f6f;
+  --user-all-color: #252526;
+  --nr-menu-highlight-color: #ff2442;
+  --nr-menu-icon-hover-color: #ff2442;
+  --nr-sider-background: #fff;
+  --nr-menu-bg: #fff;
+  --nr-sider-fixed-zindex: 12;
+  --nr-header-fixed-zindex: 11;
+  --newrank-color-bg: rgba(255, 120, 0, 0.1);
+  --newrank-color-bg-3: rgba(255, 120, 0, 0.3);
+  --warning-05: rgba(250, 173, 20, 0.05);
+  --bridge-account-color: #ff2442;
+  --bridge-agency-color: #3a64ff;
+  --bridge-free-color: #ff7800;
+  --bridge-medium-color: #00b354;
+}