feat(web): forgetting engine config

This commit is contained in:
zhaoying
2026-01-04 11:22:06 +08:00
parent b6cfd55aad
commit 02c8fd0e3f
4 changed files with 174 additions and 44 deletions

View File

@@ -157,11 +157,11 @@ export const deleteMemoryConfig = (config_id: number) => {
} }
// 遗忘引擎-获取配置 // 遗忘引擎-获取配置
export const getMemoryForgetConfig = (config_id: number | string) => { export const getMemoryForgetConfig = (config_id: number | string) => {
return request.get('/memory-storage/read_config_forget', { config_id }) return request.get('/memory/forget/read_config', { config_id })
} }
// 遗忘引擎-更新配置 // 遗忘引擎-更新配置
export const updateMemoryForgetConfig = (values: ForgetConfigForm) => { export const updateMemoryForgetConfig = (values: ForgetConfigForm) => {
return request.post('/memory-storage/update_config_forget', values) return request.post('/memory/forget/update_config', values)
} }
// 记忆萃取引擎-获取配置 // 记忆萃取引擎-获取配置
export const getMemoryExtractionConfig = (config_id: number | string) => { export const getMemoryExtractionConfig = (config_id: number | string) => {

View File

@@ -3,6 +3,39 @@ import { useTranslation } from 'react-i18next'
import ReactEcharts from 'echarts-for-react'; import ReactEcharts from 'echarts-for-react';
import type { ConfigForm } from '../types' import type { ConfigForm } from '../types'
// 定义当前数据类型
type CurrentDataItem = {
name: string;
data: number[];
config: ConfigForm | {};
type: string;
smooth: boolean;
lineStyle: { width: number };
showSymbol: boolean;
label: { show: boolean; position: string };
emphasis: { focus: string };
};
// 定义图表系列数据类型
type SeriesDataItem = {
name: string;
data: number[];
config: ChartConfig;
type: string;
smooth: boolean;
lineStyle: { width: number };
showSymbol: boolean;
label: { show: boolean; position: string };
emphasis: { focus: string };
};
// 定义简化的配置类型,只包含图表计算需要的属性
interface ChartConfig {
lambda_mem: number;
lambda_time: number;
offset: number;
}
interface LineCardProps { interface LineCardProps {
config: ConfigForm config: ConfigForm
} }
@@ -29,14 +62,14 @@ const Colors = ['#155EEF', '#4DA8FF', '#FFB048']
const LineChart: FC<LineCardProps> = ({ config }) => { const LineChart: FC<LineCardProps> = ({ config }) => {
const { t } = useTranslation() const { t } = useTranslation()
const chartRef = useRef<ReactEcharts>(null); const chartRef = useRef<ReactEcharts>(null);
const debounceRef = useRef() const debounceRef = useRef<number>()
const resizeScheduledRef = useRef(false) const resizeScheduledRef = useRef(false)
const xAxisData = [1, 3, 6, 9, 12, 15, 18, 21, 24, 27, 30, 33, 36, 39, 42, 45, 48, 51, 54, 57, 60] const xAxisData = [1, 3, 6, 9, 12, 15, 18, 21, 24, 27, 30, 33, 36, 39, 42, 45, 48, 51, 54, 57, 60]
const [initialData, setInitialData] = useState([]) const [initialData, setInitialData] = useState<SeriesDataItem[]>([])
const [currentData, setCurrentData] = useState({ const [currentData, setCurrentData] = useState<CurrentDataItem>({
...SeriesConfig, ...SeriesConfig,
name: `${t('forgettingEngine.currentConfig')}(λ_time=${config?.lambda_mem})`, name: `${t('forgettingEngine.currentConfig')}(λ_time=${config?.lambda_mem})`,
data: [], data: [] as number[],
config: {} config: {}
}) })
const seriesData = useMemo(() => [ const seriesData = useMemo(() => [
@@ -44,19 +77,26 @@ const LineChart: FC<LineCardProps> = ({ config }) => {
...SeriesConfig, ...SeriesConfig,
name: `${t('forgettingEngine.quicklyForget')}(λ_time=0.3)`, name: `${t('forgettingEngine.quicklyForget')}(λ_time=0.3)`,
data: [], data: [],
config: {lambda_mem: 0.3, lambda_time: 1, offset: 0.05} config: {lambda_mem: 0.3, lambda_time: 1, offset: 0.05} as ChartConfig
}, },
{ {
...SeriesConfig, ...SeriesConfig,
name: `${t('forgettingEngine.slowForgetting')}(λ_time=1)`, name: `${t('forgettingEngine.slowForgetting')}(λ_time=1)`,
data: [], data: [],
config: {lambda_mem: 1, lambda_time: 0.3, offset: 0.2} config: {lambda_mem: 1, lambda_time: 0.3, offset: 0.2} as ChartConfig
} }
], [t]) ], [t])
useEffect(() => { useEffect(() => {
getInitData() getInitData()
}, []) }, [t])
useEffect(() => {
// 语言切换时重新生成数据
if (config) {
getCaculateData(config)
}
}, [t, config])
useEffect(() => { useEffect(() => {
const handleResize = () => { const handleResize = () => {
@@ -101,14 +141,14 @@ const LineChart: FC<LineCardProps> = ({ config }) => {
setInitialData(list) setInitialData(list)
}, [seriesData]) }, [seriesData])
const calculateSeriesData = useCallback((days: number, data: ConfigForm) => { const calculateSeriesData = useCallback((days: number, data: ChartConfig | ConfigForm) => {
const offset = Number(data.offset) const offset = Number(data.offset)
const lambda_time = Number(data.lambda_time) const lambda_time = Number(data.lambda_time)
const lambda_mem = Number(data.lambda_mem) const lambda_mem = Number(data.lambda_mem)
// R = offset + (1 - offset) × e^(-λtime × t / (λmem × S)) // R = offset + (1 - offset) × e^(-λtime × t / (λmem × S))
return +(offset + (1 - offset) * Math.exp(-lambda_time * days / lambda_mem)).toFixed(4) return +(offset + (1 - offset) * Math.exp(-lambda_time * days / lambda_mem)).toFixed(4)
}, []) }, [])
const formatData = useCallback((data: ConfigForm) => { const formatData = useCallback((data: ChartConfig | ConfigForm) => {
return xAxisData.map(days => Number(calculateSeriesData(days, data))) return xAxisData.map(days => Number(calculateSeriesData(days, data)))
}, [calculateSeriesData]) }, [calculateSeriesData])

View File

@@ -1,5 +1,5 @@
import React, { useState, useEffect } from 'react'; import React, { useState, useEffect } from 'react';
import { Row, Col, Form, Slider, Button, Space, message } from 'antd'; import { Row, Col, Form, Slider, Button, Space, message, Switch } from 'antd';
import { useParams } from 'react-router-dom'; import { useParams } from 'react-router-dom';
import { useTranslation } from 'react-i18next'; import { useTranslation } from 'react-i18next';
import RbCard from '@/components/RbCard/Card'; import RbCard from '@/components/RbCard/Card';
@@ -18,19 +18,72 @@ const configList = [
{ {
key: 'forgettingRate', key: 'forgettingRate',
name: 'lambda_mem', name: 'lambda_mem',
range: [0, 1], range: [0.01, 1],
type: 'decimal', type: 'decimal',
}, },
{ {
key: 'offset', key: 'offset',
name: 'offset', name: 'offset',
range: [0, 1],
type: 'decimal', type: 'decimal',
} },
{
key: 'decay_constant',
name: 'decay_constant',
range: [0, 1],
type: 'decimal',
hiddenDesc: true,
},
{
key: 'max_history_length',
name: 'max_history_length',
type: 'decimal',
step: 1,
range: [10, 1000],
hiddenDesc: true,
},
{
key: 'forgetting_threshold',
name: 'forgetting_threshold',
type: 'decimal',
range: [0, 1],
hiddenDesc: true,
},
{
key: 'min_days_since_access',
name: 'min_days_since_access',
type: 'decimal',
step: 1,
range: [1, 365],
hiddenDesc: true,
},
{
key: 'enable_llm_summary',
name: 'enable_llm_summary',
type: 'button',
hiddenDesc: true,
},
{
key: 'max_merge_batch_size',
name: 'max_merge_batch_size',
type: 'decimal',
step: 1,
range: [1, 1000],
hiddenDesc: true,
},
{
key: 'forgetting_interval_hours',
name: 'forgetting_interval_hours',
type: 'decimal',
step: 1,
range: [1, 168],
hiddenDesc: true,
},
] ]
const ForgettingEngine: React.FC = () => { const ForgettingEngine: React.FC = () => {
const { t } = useTranslation(); const { t } = useTranslation();
const params = useParams(); const { id } = useParams();
const [configData, setConfigData] = useState<ConfigForm>(); const [configData, setConfigData] = useState<ConfigForm>();
const [form] = Form.useForm<ConfigForm>(); const [form] = Form.useForm<ConfigForm>();
const [messageApi, contextHolder] = message.useMessage(); const [messageApi, contextHolder] = message.useMessage();
@@ -43,7 +96,7 @@ const ForgettingEngine: React.FC = () => {
}, []) }, [])
const getConfigData = () => { const getConfigData = () => {
getMemoryForgetConfig(params.id) getMemoryForgetConfig(id as string)
.then((res) => { .then((res) => {
const response = res as ConfigForm const response = res as ConfigForm
const initialValues = { const initialValues = {
@@ -60,12 +113,12 @@ const ForgettingEngine: React.FC = () => {
}) })
} }
const handleReset = () => { const handleReset = () => {
form.setFieldsValue(configData); form.setFieldsValue(configData || {});
} }
const handleSave = () => { const handleSave = () => {
setLoading(true) setLoading(true)
updateMemoryForgetConfig({ updateMemoryForgetConfig({
config_id: params.id, config_id: id,
...values ...values
}) })
.then(() => { .then(() => {
@@ -83,7 +136,7 @@ const ForgettingEngine: React.FC = () => {
<RbCard <RbCard
title={ title={
<div className="rb:flex rb:items-center"> <div className="rb:flex rb:items-center">
<img src={strategyImpactSimulator} className="rb:w-[20px] rb:h-[20px] rb:mr-[8px]" /> <img src={strategyImpactSimulator} className="rb:w-5 rb:h-5 rb:mr-2" />
{t('forgettingEngine.forgettingEngineConfigParams')} {t('forgettingEngine.forgettingEngineConfigParams')}
</div> </div>
} }
@@ -99,28 +152,60 @@ const ForgettingEngine: React.FC = () => {
}} }}
> >
<Space size={24} direction="vertical" style={{ width: '100%' }}> <Space size={24} direction="vertical" style={{ width: '100%' }}>
{configList.map(config => ( {configList.map(config => {
<div key={config.key}> if (config.type === 'button') {
<div className="rb:text-[14px] rb:font-medium rb:leading-[20px] rb:mb-[8px]"> return (
{t(`forgettingEngine.${config.key}`)} <div key={config.key} className="rb:mb-2">
<div className="rb:flex rb:items-center rb:justify-between">
<div>
<span className="rb:text-[14px] rb:font-medium rb:leading-5">{t(`forgettingEngine.${config.key}`)}</span>
</div>
<Form.Item
name={config.name}
valuePropName="checked"
className="rb:ml-2 rb:mb-0!"
>
<Switch />
</Form.Item>
</div>
<div className="rb:flex rb:text-[12px] rb:items-center rb:justify-between rb:text-[#5B6167] rb:leading-5">
<Space size={4}>
{config.range && <span>{t(`forgettingEngine.range`)}: {config.range?.join('-')}</span>}
{config.type && <span>{t(`forgettingEngine.type`)}: {config.type}</span>}
</Space>
</div>
</div>
)
}
return (
<div key={config.key}>
<div className="rb:text-[14px] rb:font-medium rb:leading-5 rb:mb-2">
{t(`forgettingEngine.${config.key}`)}
</div>
{!config.hiddenDesc && <div className="rb:mt-1 rb:text-[12px] rb:text-[#5B6167] rb:font-regular rb:leading-4 ">
{t(`forgettingEngine.${config.key}Desc`)}
</div>}
<Form.Item
name={config.name}
>
{config.type === 'decimal'
? <Slider tooltip={{ open: false }} max={config.range?.[1] || 1} min={config.range?.[0] || 0} step={config.step ?? 0.01} style={{ margin: '0' }} />
: config.type === 'button'
? <Switch />
: null
}
</Form.Item>
<div className="rb:flex rb:text-[12px] rb:items-center rb:justify-between rb:text-[#5B6167] rb:leading-5 rb:-mt-6.5">
<Space size={4}>
{config.range && <span>{t(`forgettingEngine.range`)}: {config.range?.join('-')}</span>}
{config.type && <span>{t(`forgettingEngine.type`)}: {config.type}</span>}
</Space>
<>{t('forgettingEngine.CurrentValue')}: {values?.[config.name] || 0}</>
</div>
</div> </div>
<div className="rb:mt-[4px] rb:text-[12px] rb:text-[#5B6167] rb:font-regular rb:leading-[16px] "> )
{t(`forgettingEngine.${config.key}Desc`)} })}
</div>
<Form.Item
name={config.name}
>
<Slider tooltip={{open: false}} max={config.range?.[1] || 1} min={config.range?.[0] || 0} step={0.01} style={{ margin: '0' }} />
</Form.Item>
<div className="rb:flex rb:text-[12px] rb:items-center rb:justify-between rb:text-[#5B6167] rb:leading-[20px] rb:mt-[-26px]">
<Space size={4}>
{config.range && <span>{t(`forgettingEngine.range`)}: {config.range?.join('-')}</span>}
{config.type && <span>{t(`forgettingEngine.type`)}: {config.type}</span>}
</Space>
<>{t('forgettingEngine.CurrentValue')}: {values?.[config.name] || 0}</>
</div>
</div>
))}
<Row gutter={16}> <Row gutter={16}>
<Col span={12}> <Col span={12}>
<Button block onClick={handleReset}>{t('common.reset')}</Button> <Button block onClick={handleReset}>{t('common.reset')}</Button>
@@ -135,7 +220,6 @@ const ForgettingEngine: React.FC = () => {
</Col> </Col>
<Col span={15}> <Col span={15}>
<RbCard <RbCard
className='rb:h-full!'
> >
<LineChart <LineChart
config={values} config={values}

View File

@@ -40,11 +40,17 @@ export interface CurveRecord {
export interface ConfigForm { export interface ConfigForm {
config_id?: string; config_id?: string;
statement_granularity?: string;
include_dialogue_context?: boolean;
max_context?: string;
lambda_time: string | number; lambda_time: string | number;
lambda_mem: string | number; lambda_mem: string | number;
offset: string | number; offset: string | number;
decay_constant: string | number;
max_history_length: string | number;
forgetting_threshold: string | number;
min_days_since_access: string | number;
enable_llm_summary: boolean;
max_merge_batch_size: string | number;
forgetting_interval_hours: string | number;
[key: string]: any; [key: string]: any;
} }