supersonic/webapp/packages/chat-sdk/src/service/index.ts

import axios from './axiosInstance';
import { ChatContextType, HistoryType, MsgDataType, SearchRecommendItem } from '../common/type';
import { QueryDataType } from '../common/type';

const DEFAULT_CHAT_ID = 0;

const prefix = '/api';

export function searchRecommend(queryText: string, chatId?: number, domainId?: number) {
  return axios.post<Result<SearchRecommendItem[]>>(`${prefix}/chat/query/search`, {
    queryText,
    chatId: chatId || DEFAULT_CHAT_ID,
    domainId,
  });
}

export function chatQuery(queryText: string, chatId?: number, domainId?: number, isSaveQuestionAnswer?: boolean) {
  return axios.post<Result<MsgDataType>>(`${prefix}/chat/query/query`, {
    queryText,
    chatId: chatId || DEFAULT_CHAT_ID,
    domainId,
    isSaveQuestionAnswer
  });
}

export function queryData(chatContext: ChatContextType) {
  return axios.post<Result<QueryDataType>>(`${prefix}/chat/query/queryData`, chatContext);
}

export function queryContext(queryText: string, chatId?: number) {
  return axios.post<Result<ChatContextType>>(`${prefix}/chat/query/queryContext`, {
    queryText,
    chatId: chatId || DEFAULT_CHAT_ID,
  });
}

export function querySuggestionInfo(domainId: number) {
  return axios.get<Result<any>>(`${prefix}/chat/recommend/${domainId}`);
}

export function getHistoryMsg(current: number, chatId: number = DEFAULT_CHAT_ID, pageSize: number = 10) {
  return axios.post<Result<HistoryType>>(`${prefix}/chat/manage/pageQueryInfo?chatId=${chatId}`, {
    current,
    pageSize,
  });
}

export function queryMetricInfo(data: any) {
  return axios.get(`/semantic/metric/getMetric/${data.classId}/${data.uniqueId}`);
}

export function getRelatedDimensionFromStatInfo(data: any) {
  return axios.get(
    `/semantic/metric/getRelatedDimensionFromStatInfo/${data.classId}/${data.uniqueId}`,
  );
}

export function getMetricQueryInfo(data: any) {
  return axios.get<any>(
    `/openapi/bd-bi/api/polaris/intelligentQuery/getMetricQueryInfo/${data.classId}/${data.metricName}`
  );
}

export function saveConversation(chatName: string) {
  return axios.post<Result<any>>(`${prefix}/chat/manage/save?chatName=${chatName}`);
}

export function getAllConversations() {
  return axios.get<Result<any>>(`${prefix}/chat/manage/getAll`);
}

export function queryEntities(entityId: string | number, domainId: number) {
  return axios.post<Result<any>>(`${prefix}/chat/query/choice`, {
    entityId,
    domainId,
  });
}