fix: Remove kAModel #1306 (#1713)

### What problem does this PR solve?

fix: Remove kAModel #1306
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
balibabu 2024-07-26 10:23:51 +08:00 committed by GitHub
parent c943517932
commit 3f495b2d22
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 25 additions and 39 deletions

View File

@ -236,4 +236,22 @@ export const useTestChunkRetrieval = () => {
return testChunk;
};
export const useTestNextChunkRetrieval = () => {
const {
data,
isPending: loading,
mutateAsync,
} = useMutation({
mutationKey: ['testChunk'],
mutationFn: async (canvasIds: string[]) => {
const { data } = await kbService.retrieval_test({ canvasIds });
if (data.retcode === 0) {
}
return data?.data ?? [];
},
});
return { data, loading, testChunk: mutateAsync };
};
//#endregion

View File

@ -2,22 +2,24 @@ import { ReactComponent as ConfigurationIcon } from '@/assets/svg/knowledge-conf
import { ReactComponent as DatasetIcon } from '@/assets/svg/knowledge-dataset.svg';
import { ReactComponent as TestingIcon } from '@/assets/svg/knowledge-testing.svg';
import { useFetchKnowledgeBaseConfiguration } from '@/hooks/knowledge-hooks';
import { useSecondPathName } from '@/hooks/route-hook';
import {
useGetKnowledgeSearchParams,
useSecondPathName,
} from '@/hooks/route-hook';
import { getWidth } from '@/utils';
import { Avatar, Menu, MenuProps, Space } from 'antd';
import classNames from 'classnames';
import { useCallback, useEffect, useMemo, useState } from 'react';
import { useTranslation } from 'react-i18next';
import { useNavigate, useSelector } from 'umi';
import { useNavigate } from 'umi';
import { KnowledgeRouteKey } from '../../constant';
import styles from './index.less';
const KnowledgeSidebar = () => {
const kAModel = useSelector((state: any) => state.kAModel);
const { id } = kAModel;
let navigate = useNavigate();
const activeKey = useSecondPathName();
const { knowledgeId } = useGetKnowledgeSearchParams();
const [windowWidth, setWindowWidth] = useState(getWidth());
const [collapsed, setCollapsed] = useState(false);
@ -25,7 +27,7 @@ const KnowledgeSidebar = () => {
const { data: knowledgeDetails } = useFetchKnowledgeBaseConfiguration();
const handleSelect: MenuProps['onSelect'] = (e) => {
navigate(`/knowledge/${e.key}?id=${id}`);
navigate(`/knowledge/${e.key}?id=${knowledgeId}`);
};
type MenuItem = Required<MenuProps>['items'][number];

View File

@ -1,32 +0,0 @@
import { DvaModel } from 'umi';
export interface kAModelState {
isShowPSwModal: boolean;
tenantIfo: any;
id: string;
doc_id: string;
}
const model: DvaModel<kAModelState> = {
namespace: 'kAModel',
state: {
isShowPSwModal: false,
tenantIfo: {},
id: '',
doc_id: '',
},
reducers: {
updateState(state, { payload }) {
return {
...state,
...payload,
};
},
},
subscriptions: {
setup({ dispatch, history }) {
history.listen((location) => {});
},
},
effects: {},
};
export default model;

2
web/typings.d.ts vendored
View File

@ -1,7 +1,6 @@
import { ChunkModelState } from '@/pages/add-knowledge/components/knowledge-chunk/model';
import { KFModelState } from '@/pages/add-knowledge/components/knowledge-file/model';
import { TestingModelState } from '@/pages/add-knowledge/components/knowledge-testing/model';
import { kAModelState } from '@/pages/add-knowledge/model';
import { ChatModelState } from '@/pages/chat/model';
declare module 'lodash';
@ -14,7 +13,6 @@ function useSelector<TState = RootState, TSelected = unknown>(
export interface RootState {
chatModel: ChatModelState;
kFModel: KFModelState;
kAModel: kAModelState;
chunkModel: ChunkModelState;
testingModel: TestingModelState;
}