feat(web): forgetting engine config

2026-01-04 11:22:06 +08:00
parent b6cfd55aad
commit 02c8fd0e3f
4 changed files with 174 additions and 44 deletions
--- a/web/src/api/memory.ts
+++ b/web/src/api/memory.ts
@@ -157,11 +157,11 @@ export const deleteMemoryConfig = (config_id: number) => {
 }
 // 遗忘引擎-获取配置
 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) => {
-  return request.post('/memory-storage/update_config_forget', values)
+  return request.post('/memory/forget/update_config', values)
 }
 // 记忆萃取引擎-获取配置
 export const getMemoryExtractionConfig = (config_id: number | string) => {
--- a/web/src/views/ForgettingEngine/components/LineChart.tsx
+++ b/web/src/views/ForgettingEngine/components/LineChart.tsx
@@ -3,6 +3,39 @@ import { useTranslation } from 'react-i18next'
 import ReactEcharts from 'echarts-for-react';
 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 {
  config: ConfigForm
 }
@@ -29,14 +62,14 @@ const Colors = ['#155EEF', '#4DA8FF', '#FFB048']
 const LineChart: FC<LineCardProps> = ({ config }) => {
  const { t } = useTranslation()
  const chartRef = useRef<ReactEcharts>(null);
-  const debounceRef = useRef()
+  const debounceRef = useRef<number>()
  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 [initialData, setInitialData] = useState([])
-  const [currentData, setCurrentData] = useState({
+  const [initialData, setInitialData] = useState<SeriesDataItem[]>([])
+  const [currentData, setCurrentData] = useState<CurrentDataItem>({
      ...SeriesConfig,
      name: `${t('forgettingEngine.currentConfig')}(λ_time=${config?.lambda_mem})`,
-      data: [],
+      data: [] as number[],
      config: {}
  })
  const seriesData = useMemo(() => [
@@ -44,19 +77,26 @@ const LineChart: FC<LineCardProps> = ({ config }) => {
      ...SeriesConfig,
      name: `${t('forgettingEngine.quicklyForget')}(λ_time=0.3)`,
      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,
      name: `${t('forgettingEngine.slowForgetting')}(λ_time=1)`,
      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])

  useEffect(() => {
    getInitData()
-  }, [])
+  }, [t])
+
+  useEffect(() => {
+    // 语言切换时重新生成数据
+    if (config) {
+      getCaculateData(config)
+    }
+  }, [t, config])

  useEffect(() => {
    const handleResize = () => {
@@ -101,14 +141,14 @@ const LineChart: FC<LineCardProps> = ({ config }) => {
    setInitialData(list)
  }, [seriesData])
  
-  const calculateSeriesData = useCallback((days: number, data: ConfigForm) => {
+  const calculateSeriesData = useCallback((days: number, data: ChartConfig | ConfigForm) => {
    const offset = Number(data.offset)
    const lambda_time = Number(data.lambda_time)
    const lambda_mem = Number(data.lambda_mem)
    // R = offset + (1 - offset) × e^(-λtime × t / (λmem × S))
    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)))
  }, [calculateSeriesData])

--- a/web/src/views/ForgettingEngine/index.tsx
+++ b/web/src/views/ForgettingEngine/index.tsx
@@ -1,5 +1,5 @@
 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 { useTranslation } from 'react-i18next';
 import RbCard from '@/components/RbCard/Card';
@@ -18,19 +18,72 @@ const configList = [
  {
    key: 'forgettingRate',
    name: 'lambda_mem',
-    range: [0, 1],
+    range: [0.01, 1],
    type: 'decimal',
  },
  {
    key: 'offset',
    name: 'offset',
+    range: [0, 1],
    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 { t } = useTranslation();
-  const params = useParams();
+  const { id } = useParams();
  const [configData, setConfigData] = useState<ConfigForm>();
  const [form] = Form.useForm<ConfigForm>();
  const [messageApi, contextHolder] = message.useMessage();
@@ -43,7 +96,7 @@ const ForgettingEngine: React.FC = () => {
  }, [])

  const getConfigData = () => {
-    getMemoryForgetConfig(params.id)
+    getMemoryForgetConfig(id as string)
      .then((res) => {
        const response = res as ConfigForm
        const initialValues = {
@@ -60,12 +113,12 @@ const ForgettingEngine: React.FC = () => {
      })
  }
  const handleReset = () => {
-    form.setFieldsValue(configData);
+    form.setFieldsValue(configData || {});
  }
  const handleSave = () => {
    setLoading(true)
    updateMemoryForgetConfig({
-      config_id: params.id,
+      config_id: id,
      ...values
    })
      .then(() => {
@@ -83,7 +136,7 @@ const ForgettingEngine: React.FC = () => {
        <RbCard 
          title={
            <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')}
            </div>
          }
@@ -99,28 +152,60 @@ const ForgettingEngine: React.FC = () => {
            }}
          >
            <Space size={24} direction="vertical" style={{ width: '100%' }}>
-              {configList.map(config => (
-                <div key={config.key}>
-                  <div className="rb:text-[14px] rb:font-medium rb:leading-[20px] rb:mb-[8px]">
-                    {t(`forgettingEngine.${config.key}`)}
+              {configList.map(config => {
+                if (config.type === 'button') {
+                  return (
+                    <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 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}>
                <Col span={12}>
                  <Button block onClick={handleReset}>{t('common.reset')}</Button>
@@ -135,7 +220,6 @@ const ForgettingEngine: React.FC = () => {
      </Col>
      <Col span={15}>
        <RbCard
-          className='rb:h-full!'
        >
          <LineChart
            config={values}
--- a/web/src/views/ForgettingEngine/types.ts
+++ b/web/src/views/ForgettingEngine/types.ts
@@ -40,11 +40,17 @@ export interface CurveRecord {

 export interface ConfigForm {
  config_id?: string;
-  statement_granularity?: string;
-  include_dialogue_context?: boolean;
-  max_context?: string;
  lambda_time: string | number;
  lambda_mem: 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;
 }