gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix 修正格式防止曹飞

Browse Source
This commit is contained in:
SengokuCola
2025-03-21 14:49:59 +08:00
parent 96637a4088
commit 432104f582
6 changed files with 22 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/plugins/chat/config.py
View File

@@ -321,9 +321,18 @@ class BotConfig:
) )
config.memory_compress_rate = memory_config.get("memory_compress_rate", config.memory_compress_rate) config.memory_compress_rate = memory_config.get("memory_compress_rate", config.memory_compress_rate)
if config.INNER_VERSION in SpecifierSet(">=0.0.11"): if config.INNER_VERSION in SpecifierSet(">=0.0.11"):
config.memory_build_distribution = memory_config.get("memory_build_distribution", config.memory_build_distribution) config.memory_build_distribution = memory_config.get(
config.build_memory_sample_num = memory_config.get("build_memory_sample_num", config.build_memory_sample_num) "memory_build_distribution",
config.build_memory_sample_length = memory_config.get("build_memory_sample_length", config.build_memory_sample_length) config.memory_build_distribution
)
config.build_memory_sample_num = memory_config.get(
"build_memory_sample_num",
config.build_memory_sample_num
)
config.build_memory_sample_length = memory_config.get(
"build_memory_sample_length",
config.build_memory_sample_length
)
def remote(parent: dict): def remote(parent: dict):

6
src/plugins/memory_system/memory.py
View File

@@ -239,7 +239,11 @@ class Hippocampus:
chat_samples = [] chat_samples = []
for timestamp in timestamps: for timestamp in timestamps:
messages = self.random_get_msg_snippet(timestamp, global_config.build_memory_sample_length, max_memorized_time_per_msg) messages = self.random_get_msg_snippet(
timestamp,
global_config.build_memory_sample_length,
max_memorized_time_per_msg
)
if messages: if messages:
time_diff = (datetime.datetime.now().timestamp() - timestamp) / 3600 time_diff = (datetime.datetime.now().timestamp() - timestamp) / 3600
logger.debug(f"成功抽取 {time_diff:.1f} 小时前的消息样本，共{len(messages)}条") logger.debug(f"成功抽取 {time_diff:.1f} 小时前的消息样本，共{len(messages)}条")

2
src/plugins/memory_system/memory_manual_build.py
View File

@@ -17,7 +17,7 @@ import jieba
root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../..")) root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../.."))
sys.path.append(root_path) sys.path.append(root_path)
from src.common.logger import get_module_logger from src.common.logger import get_module_logger # noqa: E402
from src.common.database import db # noqa E402 from src.common.database import db # noqa E402
from src.plugins.memory_system.offline_llm import LLMModel # noqa E402 from src.plugins.memory_system.offline_llm import LLMModel # noqa E402

2
src/plugins/memory_system/sample_distribution.py
View File

@@ -1,7 +1,5 @@
import numpy as np import numpy as np
import matplotlib.pyplot as plt
from scipy import stats from scipy import stats
import time
from datetime import datetime, timedelta from datetime import datetime, timedelta
class DistributionVisualizer: class DistributionVisualizer:

9
src/plugins/schedule/schedule_generator copy.py
View File

@@ -5,16 +5,15 @@ import os
import sys import sys
from typing import Dict, Union from typing import Dict, Union
from nonebot import get_driver
# 添加项目根目录到 Python 路径 # 添加项目根目录到 Python 路径
root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../..")) root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../.."))
sys.path.append(root_path) sys.path.append(root_path)
# from src.plugins.chat.config import global_config from src.common.database import db # noqa: E402
from src.common.database import db # 使用正确的导入语法 from src.common.logger import get_module_logger # noqa: E402
from src.plugins.schedule.offline_llm import LLMModel from src.plugins.schedule.offline_llm import LLMModel # noqa: E402
from src.common.logger import get_module_logger from src.plugins.chat.config import global_config # noqa: E402
logger = get_module_logger("scheduler") logger = get_module_logger("scheduler")

2
src/plugins/schedule/schedule_generator.py
View File

@@ -1,8 +1,6 @@
import datetime import datetime
import json import json
import re import re
import os
import sys
from typing import Dict, Union from typing import Dict, Union
from nonebot import get_driver from nonebot import get_driver