Quellcode durchsuchen

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)
tags/v0.9.0
balibabu vor 1 Jahr
Ursprung
Commit
3f495b2d22
Es ist kein Account mit der E-Mail-Adresse des Committers verbunden

+ 18
- 0
web/src/hooks/knowledge-hooks.ts Datei anzeigen

@@ -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

+ 7
- 5
web/src/pages/add-knowledge/components/knowledge-sidebar/index.tsx Datei anzeigen

@@ -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];

+ 0
- 32
web/src/pages/add-knowledge/model.ts Datei anzeigen

@@ -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;

+ 0
- 2
web/typings.d.ts Datei anzeigen

@@ -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;
}

Laden…
Abbrechen
Speichern