优化日志

This commit is contained in:
Windpicker-owo
2025-11-26 21:16:16 +08:00
parent e0157256b1
commit 0908fb50a0
100 changed files with 493 additions and 574 deletions

View File

@@ -542,7 +542,7 @@ class ExpressionLearner:
# 获取 StyleLearner 实例
learner = style_learner_manager.get_learner(chat_id)
logger.info(f"开始训练 StyleLearner: chat_id={chat_id}, 样本数={len(expr_list)}")
logger.debug(f"开始训练 StyleLearner: chat_id={chat_id}, 样本数={len(expr_list)}")
# 为每个学习到的表达方式训练模型
# 使用 situation 作为输入style 作为目标
@@ -566,12 +566,12 @@ class ExpressionLearner:
# 保存模型
if learner.save(style_learner_manager.model_save_path):
logger.info(f"StyleLearner 模型保存成功: {chat_id}")
logger.debug(f"StyleLearner 模型保存成功: {chat_id}")
else:
logger.error(f"StyleLearner 模型保存失败: {chat_id}")
except Exception as e:
logger.error(f"训练 StyleLearner 失败: {e}", exc_info=True)
logger.error(f"训练 StyleLearner 失败: {e}")
return learnt_expressions
return None

View File

@@ -38,7 +38,7 @@ class ExpressorModel:
self._candidates: dict[str, str] = {} # cid -> text (style)
self._situations: dict[str, str] = {} # cid -> situation (不参与计算)
logger.info(
logger.debug(
f"ExpressorModel初始化完成 (alpha={alpha}, beta={beta}, gamma={gamma}, vocab_size={vocab_size}, use_jieba={use_jieba})"
)
@@ -199,8 +199,6 @@ class ExpressorModel:
with open(path, "wb") as f:
pickle.dump(data, f)
logger.info(f"模型已保存到 {path}")
def load(self, path: str):
"""
从文件加载模型
@@ -230,7 +228,7 @@ class ExpressorModel:
for cid, tc in data["nb_token_counts"].items():
self.nb.token_counts[cid] = defaultdict(float, tc)
logger.info(f"模型已从 {path} 加载")
logger.debug(f"模型已从 {path} 加载")
def get_stats(self) -> dict:
"""获取模型统计信息"""

View File

@@ -24,7 +24,7 @@ class Tokenizer:
import rjieba # noqa: F401
# rjieba 会自动初始化,无需手动调用
logger.info("RJieba分词器初始化成功")
logger.debug("RJieba分词器初始化成功")
except ImportError:
logger.warning("RJieba未安装将使用字符级分词")
self.use_jieba = False

View File

@@ -51,8 +51,6 @@ class StyleLearner:
"last_update": time.time(),
}
logger.info(f"StyleLearner初始化成功: chat_id={chat_id}")
def add_style(self, style: str, situation: str | None = None) -> bool:
"""
动态添加一个新的风格
@@ -178,7 +176,7 @@ class StyleLearner:
logger.debug(f"被删除的风格样例(前5): {deleted_styles[:5]}")
except Exception as e:
logger.error(f"清理风格失败: {e}", exc_info=True)
logger.error(f"清理风格失败: {e}")
def learn_mapping(self, up_content: str, style: str) -> bool:
"""
@@ -271,7 +269,7 @@ class StyleLearner:
return best_style, style_scores
except Exception as e:
logger.error(f"预测style失败: {e}", exc_info=True)
logger.error(f"预测style失败: {e}")
return None, {}
def get_style_info(self, style: str) -> tuple[str | None, str | None]:
@@ -373,7 +371,6 @@ class StyleLearner:
with open(meta_path, "wb") as f:
pickle.dump(meta_data, f)
logger.info(f"StyleLearner保存成功: {save_dir}")
return True
except Exception as e:
@@ -421,7 +418,6 @@ class StyleLearner:
if "style_last_used" not in self.learning_stats:
self.learning_stats["style_last_used"] = {}
logger.info(f"StyleLearner加载成功: {save_dir}")
return True
except Exception as e:
@@ -454,7 +450,7 @@ class StyleLearnerManager:
# 确保保存目录存在
os.makedirs(model_save_path, exist_ok=True)
logger.info(f"StyleLearnerManager初始化成功, 模型保存路径: {model_save_path}")
logger.debug(f"StyleLearnerManager初始化成功, 模型保存路径: {model_save_path}")
def get_learner(self, chat_id: str, model_config: dict | None = None) -> StyleLearner:
"""
@@ -520,7 +516,7 @@ class StyleLearnerManager:
if not learner.save(self.model_save_path):
success = False
logger.info(f"保存所有StyleLearner {'成功' if success else '部分失败'}")
logger.debug(f"保存所有StyleLearner {'成功' if success else '部分失败'}")
return success
def cleanup_all_old_styles(self, ratio: float | None = None) -> dict[str, int]:
@@ -540,7 +536,7 @@ class StyleLearnerManager:
cleanup_results[chat_id] = cleaned
total_cleaned = sum(cleanup_results.values())
logger.info(f"清理所有StyleLearner完成: 总共清理了 {total_cleaned} 个风格")
logger.debug(f"清理所有StyleLearner完成: 总共清理了 {total_cleaned} 个风格")
return cleanup_results
def apply_decay_all(self, factor: float | None = None):
@@ -553,7 +549,7 @@ class StyleLearnerManager:
for learner in self.learners.values():
learner.apply_decay(factor)
logger.info("对所有StyleLearner应用知识衰减")
logger.debug("对所有StyleLearner应用知识衰减")
def get_all_stats(self) -> dict[str, dict]:
"""