feat(web): forgetting engine config
This commit is contained in:
@@ -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) => {
|
||||||
|
|||||||
@@ -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])
|
||||||
|
|
||||||
|
|||||||
@@ -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}
|
||||||
|
|||||||
@@ -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;
|
||||||
}
|
}
|
||||||
Reference in New Issue
Block a user