96 lines
2.9 KiB
TypeScript
96 lines
2.9 KiB
TypeScript
/*
|
|
* Copyright 2025 coze-dev Authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
import { useState, type FC } from 'react';
|
|
|
|
import { type ReactElement } from 'react-markdown/lib/react-markdown';
|
|
import { I18n } from '@coze-arch/i18n';
|
|
import { type ButtonProps } from '@coze-arch/coze-design';
|
|
import { IconMemoryDownMenu } from '@coze-arch/bot-icons';
|
|
import { DataErrorBoundary, DataNamespace } from '@coze-data/reporter';
|
|
import { BotE2e } from '@coze-data/e2e';
|
|
import {
|
|
MemoryDebugDropdown,
|
|
useMemoryDebugModal,
|
|
type MemoryDebugDropdownMenuItem,
|
|
type MemoryModule,
|
|
useSendTeaEventForMemoryDebug,
|
|
} from '@coze-data/database';
|
|
import { OperateTypeEnum, ToolPane } from '@coze-agent-ide/debug-tool-list';
|
|
|
|
export interface MemoryToolPaneProps {
|
|
menuList: MemoryDebugDropdownMenuItem[];
|
|
}
|
|
|
|
export const MemoryToolPane: FC<MemoryToolPaneProps> = ({ menuList }) => {
|
|
const isStore = false;
|
|
|
|
const sendTeaEventForMemoryDebug = useSendTeaEventForMemoryDebug({
|
|
isStore,
|
|
});
|
|
|
|
const [curMemoryModule, setCurMemoryModule] = useState<MemoryModule>();
|
|
|
|
const defaultModule = menuList[0]?.name;
|
|
|
|
const { open, node: memoryModal } = useMemoryDebugModal({
|
|
memoryModule: curMemoryModule || defaultModule,
|
|
menuList,
|
|
setMemoryModule: setCurMemoryModule,
|
|
isStore,
|
|
});
|
|
|
|
return (
|
|
<DataErrorBoundary namespace={DataNamespace.MEMORY}>
|
|
{memoryModal}
|
|
{
|
|
(
|
|
<ToolPane
|
|
visible={menuList.length > 0}
|
|
itemKey={`key_${I18n.t('database_memory_menu')}`}
|
|
operateType={OperateTypeEnum.DROPDOWN}
|
|
title={I18n.t('database_memory_menu')}
|
|
icon={<IconMemoryDownMenu />}
|
|
onEntryButtonClick={() => {
|
|
sendTeaEventForMemoryDebug(defaultModule);
|
|
setCurMemoryModule(defaultModule);
|
|
open();
|
|
}}
|
|
dropdownProps={{
|
|
showTick: true,
|
|
clickToHide: true,
|
|
render: (
|
|
<MemoryDebugDropdown
|
|
menuList={menuList}
|
|
onClickItem={memoryModule => {
|
|
setCurMemoryModule(memoryModule);
|
|
open();
|
|
}}
|
|
/>
|
|
),
|
|
}}
|
|
buttonProps={
|
|
{
|
|
'data-testid': BotE2e.BotMemoryDebugBtn,
|
|
} as unknown as ButtonProps
|
|
}
|
|
/>
|
|
) as ReactElement
|
|
}
|
|
</DataErrorBoundary>
|
|
);
|
|
};
|