123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392 |
- import {
- GetDBGPTsListResponse,
- PostAgentHubUpdateParams,
- PostAgentMyPluginResponse,
- PostAgentPluginResponse,
- PostAgentQueryParams,
- PostDbgptMyQueryParams,
- } from '@/types/agent';
- import { GetAppInfoParams, IApp } from '@/types/app';
- import {
- ChatHistoryResponse,
- DialogueListResponse,
- FeedBack,
- IChatDialogueSchema,
- IDB,
- NewDialogueParam,
- SceneResponse,
- UserParam,
- UserParamResponse,
- } from '@/types/chat';
- import {
- ChatFeedBackSchema,
- DbListResponse,
- DbSupportTypeResponse,
- PostDbParams,
- PostDbRefreshParams,
- } from '@/types/db';
- import {
- GetEditorSQLRoundRequest,
- GetEditorySqlParams,
- PostEditorChartRunParams,
- PostEditorChartRunResponse,
- PostEditorSQLRunParams,
- PostSQLEditorSubmitParams,
- } from '@/types/editor';
- import {
- AddKnowledgeParams,
- ArgumentsParams,
- ChunkListParams,
- DocumentParams,
- GraphVisResult,
- IArguments,
- IChunkList,
- IChunkStrategyResponse,
- IDocumentResponse,
- ISpace,
- ISyncBatchParameter,
- ISyncBatchResponse,
- SpaceConfig,
- } from '@/types/knowledge';
- import { BaseModelParams, IModelData, StartModelParams, SupportModel } from '@/types/model';
- import { AxiosRequestConfig } from 'axios';
- import { GET, POST } from '.';
- /** App */
- export const postScenes = () => {
- return POST<null, Array<SceneResponse>>('/api/v1/chat/dialogue/scenes');
- };
- export const newDialogue = (data: NewDialogueParam) => {
- return POST<NewDialogueParam, IChatDialogueSchema>(
- `/api/v1/chat/dialogue/new?chat_mode=${data.chat_mode}&model_name=${data.model}`,
- data,
- );
- };
- export const addUser = (data: UserParam) => {
- return POST<UserParam, UserParamResponse>('/api/v1/user/add', data);
- };
- /** Database Page */
- export const getDbList = () => {
- return GET<null, DbListResponse>('/api/v1/chat/db/list');
- };
- export const getDbSupportType = () => {
- return GET<null, DbSupportTypeResponse>('/api/v1/chat/db/support/type');
- };
- export const postDbDelete = (dbName: string) => {
- return POST(`/api/v1/chat/db/delete?db_name=${dbName}`);
- };
- export const postDbEdit = (data: PostDbParams) => {
- return POST<PostDbParams, null>('/api/v1/chat/db/edit', data);
- };
- export const postDbAdd = (data: PostDbParams) => {
- return POST<PostDbParams, null>('/api/v1/chat/db/add', data);
- };
- export const postDbTestConnect = (data: PostDbParams) => {
- return POST<PostDbParams, null>('/api/v1/chat/db/test/connect', data);
- };
- export const postDbRefresh = (data: PostDbRefreshParams) => {
- return POST<PostDbRefreshParams, boolean>('/api/v1/chat/db/refresh', data);
- };
- /** Chat Page */
- export const getDialogueList = () => {
- return GET<null, DialogueListResponse>('/api/v1/chat/dialogue/list');
- };
- export const getUsableModels = () => {
- return GET<null, Array<string>>('/api/v1/model/types');
- };
- export const postChatModeParamsList = (chatMode: string) => {
- return POST<null, IDB[]>(`/api/v1/chat/mode/params/list?chat_mode=${chatMode}`);
- };
- export const postChatModeParamsInfoList = (chatMode: string) => {
- return POST<null, Record<string, string>>(`/api/v1/chat/mode/params/info?chat_mode=${chatMode}`);
- };
- export const getChatHistory = (convId: string) => {
- return GET<null, ChatHistoryResponse>(`/api/v1/chat/dialogue/messages/history?con_uid=${convId}`);
- };
- export const postChatModeParamsFileLoad = ({
- convUid,
- chatMode,
- data,
- config,
- model,
- userName,
- sysCode,
- }: {
- convUid: string;
- chatMode: string;
- data: FormData;
- model: string;
- userName?: string;
- sysCode?: string;
- config?: Omit<AxiosRequestConfig, 'headers'>;
- }) => {
- return POST<FormData, any>(
- `/api/v1/resource/file/upload?conv_uid=${convUid}&chat_mode=${chatMode}&model_name=${model}&user_name=${userName}&sys_code=${sysCode}`,
- data,
- {
- headers: {
- 'Content-Type': 'multipart/form-data',
- },
- ...config,
- },
- );
- };
- export const clearChatHistory = (conUid: string) => {
- return POST<null, Record<string, string>>(`/api/v1/chat/dialogue/clear?con_uid=${conUid}`);
- };
- /** Menu */
- export const delDialogue = (conv_uid: string) => {
- return POST(`/api/v1/chat/dialogue/delete?con_uid=${conv_uid}`);
- };
- /** Editor */
- export const getEditorSqlRounds = (id: string) => {
- return GET<null, GetEditorSQLRoundRequest>(`/api/v1/editor/sql/rounds?con_uid=${id}`);
- };
- export const postEditorSqlRun = (data: PostEditorSQLRunParams) => {
- return POST<PostEditorSQLRunParams>(`/api/v1/editor/sql/run`, data);
- };
- export const postEditorChartRun = (data: PostEditorChartRunParams) => {
- return POST<PostEditorChartRunParams, PostEditorChartRunResponse>(`/api/v1/editor/chart/run`, data);
- };
- export const postSqlEditorSubmit = (data: PostSQLEditorSubmitParams) => {
- return POST<PostSQLEditorSubmitParams>(`/api/v1/sql/editor/submit`, data);
- };
- export const getEditorSql = (id: string, round: string | number) => {
- return POST<GetEditorySqlParams, string | Array<any>>('/api/v1/editor/sql', {
- con_uid: id,
- round,
- });
- };
- /** knowledge */
- export const getArguments = (knowledgeName: string) => {
- return POST<any, IArguments>(`/knowledge/${knowledgeName}/arguments`, {});
- };
- export const saveArguments = (knowledgeName: string, data: ArgumentsParams) => {
- return POST<ArgumentsParams, IArguments>(`/knowledge/${knowledgeName}/argument/save`, data);
- };
- export const getSpaceList = (data?: any) => {
- return POST<any, Array<ISpace>>('/knowledge/space/list', data);
- };
- export const getDocumentList = (spaceName: string, data: Record<string, number | Array<number>>) => {
- return POST<Record<string, number | Array<number>>, IDocumentResponse>(`/knowledge/${spaceName}/document/list`, data);
- };
- export const getGraphVis = (spaceName: string, data: { limit: number }) => {
- return POST<Record<string, number>, GraphVisResult>(`/knowledge/${spaceName}/graphvis`, data);
- };
- export const addDocument = (knowledgeName: string, data: DocumentParams) => {
- return POST<DocumentParams, number>(`/knowledge/${knowledgeName}/document/add`, data);
- };
- export const addSpace = (data: AddKnowledgeParams) => {
- return POST<AddKnowledgeParams, number>(`/knowledge/space/add`, data);
- };
- export const getChunkStrategies = () => {
- return GET<null, Array<IChunkStrategyResponse>>('/knowledge/document/chunkstrategies');
- };
- export const syncDocument = (spaceName: string, data: Record<string, Array<number>>) => {
- return POST<Record<string, Array<number>>, string | null>(`/knowledge/${spaceName}/document/sync`, data);
- };
- export const syncBatchDocument = (spaceName: string, data: Array<ISyncBatchParameter>) => {
- return POST<Array<ISyncBatchParameter>, ISyncBatchResponse>(`/knowledge/${spaceName}/document/sync_batch`, data);
- };
- export const uploadDocument = (knowLedgeName: string, data: FormData) => {
- return POST<FormData, number>(`/knowledge/${knowLedgeName}/document/upload`, data);
- };
- export const getChunkList = (spaceName: string, data: ChunkListParams) => {
- return POST<ChunkListParams, IChunkList>(`/knowledge/${spaceName}/chunk/list`, data);
- };
- export const delDocument = (spaceName: string, data: Record<string, number>) => {
- return POST<Record<string, number>, null>(`/knowledge/${spaceName}/document/delete`, data);
- };
- export const delSpace = (data: Record<string, string>) => {
- return POST<Record<string, string>, null>(`/knowledge/space/delete`, data);
- };
- /** models */
- export const getModelList = () => {
- return GET<null, Array<IModelData>>('/api/v1/worker/model/list');
- };
- export const stopModel = (data: BaseModelParams) => {
- return POST<BaseModelParams, boolean>('/api/v1/worker/model/stop', data);
- };
- export const startModel = (data: StartModelParams) => {
- return POST<StartModelParams, boolean>('/api/v1/worker/model/start', data);
- };
- export const getSupportModels = () => {
- return GET<null, Array<SupportModel>>('/api/v1/worker/model/params');
- };
- /** Agent */
- export const postAgentQuery = (data: PostAgentQueryParams) => {
- return POST<PostAgentQueryParams, PostAgentPluginResponse>('/api/v1/agent/query', data);
- };
- export const postDbgptsQuery = (data: PostAgentQueryParams) => {
- return POST<PostAgentQueryParams, PostAgentPluginResponse>(
- `/api/v1/serve/dbgpts/hub/query_page?page=${data?.page_index}&page_size=${data?.page_size}`,
- data,
- );
- };
- export const postAgentHubUpdate = (data?: PostAgentHubUpdateParams) => {
- return POST<PostAgentHubUpdateParams>(
- '/api/v1/agent/hub/update',
- data ?? { channel: '', url: '', branch: '', authorization: '' },
- );
- };
- export const postDbgptsHubUpdate = (data?: PostAgentHubUpdateParams) => {
- return POST<PostAgentHubUpdateParams>(
- '/api/v1/serve/dbgpts/hub/source/refresh',
- data ?? { channel: '', url: '', branch: '', authorization: '' },
- );
- };
- export const postAgentMy = (user?: string) => {
- return POST<undefined, PostAgentMyPluginResponse>('/api/v1/agent/my', undefined, { params: { user } });
- };
- export const postDbgptsMy = (data?: PostDbgptMyQueryParams) => {
- return POST<PostDbgptMyQueryParams, PostAgentMyPluginResponse>(
- `/api/v1/serve/dbgpts/my/query_page?page=${data?.page_index}&page_size=${data?.page_size}`,
- data,
- );
- };
- export const postAgentInstall = (pluginName: string, user?: string) => {
- return POST('/api/v1/agent/install', undefined, {
- params: { plugin_name: pluginName, user },
- timeout: 60000,
- });
- };
- export const postDbgptsInstall = (data: object, user?: string) => {
- return POST('/api/v1/serve/dbgpts/hub/install', data, {
- params: { user },
- timeout: 60000,
- });
- };
- export const postAgentUninstall = (pluginName: string, user?: string) => {
- return POST('/api/v1/agent/uninstall', undefined, {
- params: { plugin_name: pluginName, user },
- timeout: 60000,
- });
- };
- export const postDbgptsUninstall = (data: { name: string; type: string }, user?: string) => {
- return POST('/api/v1/serve/dbgpts/my/uninstall', undefined, {
- params: { ...data, user },
- timeout: 60000,
- });
- };
- export const postAgentUpload = (user = '', data: FormData, config?: Omit<AxiosRequestConfig, 'headers'>) => {
- return POST<FormData>('/api/v1/personal/agent/upload', data, {
- params: { user },
- headers: {
- 'Content-Type': 'multipart/form-data',
- },
- ...config,
- });
- };
- export const getDbgptsList = () => {
- return GET<undefined, GetDBGPTsListResponse>('/api/v1/dbgpts/list');
- };
- /** chat feedback **/
- export const getChatFeedBackSelect = () => {
- return GET<null, FeedBack>(`/api/v1/feedback/select`, undefined);
- };
- export const getChatFeedBackItme = (conv_uid: string, conv_index: number) => {
- return GET<null, Record<string, string>>(
- `/api/v1/feedback/find?conv_uid=${conv_uid}&conv_index=${conv_index}`,
- undefined,
- );
- };
- export const postChatFeedBackForm = ({
- data,
- config,
- }: {
- data: ChatFeedBackSchema;
- config?: Omit<AxiosRequestConfig, 'headers'>;
- }) => {
- return POST<ChatFeedBackSchema, any>(`/api/v1/feedback/commit`, data, {
- headers: {
- 'Content-Type': 'application/json',
- },
- ...config,
- });
- };
- /** prompt */
- /** app */
- export const collectApp = (data: Record<string, string>) => {
- return POST<Record<string, string>, []>('/api/v1/app/collect', data);
- };
- export const unCollectApp = (data: Record<string, string>) => {
- return POST<Record<string, string>, []>('/api/v1/app/uncollect', data);
- };
- export const getResourceType = () => {
- return GET<null, string[]>('/api/v1/resource-type/list');
- };
- export const publishApp = (app_code: string) => {
- return POST<Record<string, any>, []>('/api/v1/app/publish', { app_code });
- };
- export const unPublishApp = (app_code: string) => {
- return POST<Record<string, any>, []>('/api/v1/app/unpublish', { app_code });
- };
- export const addOmcDB = (params: Record<string, string>) => {
- return POST<Record<string, any>, []>('/api/v1/chat/db/add', params);
- };
- export const getAppInfo = (data: GetAppInfoParams) => {
- return GET<GetAppInfoParams, IApp>('/api/v1/app/info', data);
- };
- export const getSupportDBList = (db_name = '') => {
- return GET<null, Record<string, any>>(`/api/v1/permission/db/list?db_name=${db_name}`);
- };
- export const recommendApps = (data: Record<string, string>) => {
- return POST<Record<string, string>, []>('/api/v1/app/hot/list', data);
- };
- export const flowSearch = (data: Record<string, string>) => {
- return POST<Record<string, string>, []>('/api/v1/serve/awel/flows', data);
- };
- export const modelSearch = (data: Record<string, string>) => {
- return POST<Record<string, string>, []>('/api/controller/models', data);
- };
- export const getKnowledgeAdmins = (spaceId: string) => {
- return GET<string, Record<string, any>>(`/knowledge/users/list?space_id=${spaceId}`);
- };
- export const updateKnowledgeAdmins = (data: Record<string, string>) => {
- return POST<Record<string, any>, any[]>(`/knowledge/users/update`, data);
- };
- /** AWEL Flow */
- /** app */
- export const delApp = (data: Record<string, string>) => {
- return POST<Record<string, string>, []>('/api/v1/app/remove', data);
- };
- export const getSpaceConfig = () => {
- return GET<string, SpaceConfig>(`/knowledge/space/config`);
- };
|