better:优化style总结

pull/1450/head
SengokuCola 2025-12-19 22:52:27 +08:00
parent a4f0bef79e
commit a3af590d09
2 changed files with 51 additions and 29 deletions

View File

@ -554,7 +554,7 @@ class ExpressionLearner:
async def _find_exact_style_match(self, style: str) -> Optional[Expression]: async def _find_exact_style_match(self, style: str) -> Optional[Expression]:
""" """
查找具有完全匹配 style Expression 记录 查找具有完全匹配 style Expression 记录
检查 style 字段和 style_list 中的每一项 只检查 style_list 中的每一项不检查 style 字段因为 style 可能是总结后的概括性描述
Args: Args:
style: 要查找的 style style: 要查找的 style
@ -566,11 +566,7 @@ class ExpressionLearner:
all_expressions = Expression.select().where(Expression.chat_id == self.chat_id) all_expressions = Expression.select().where(Expression.chat_id == self.chat_id)
for expr in all_expressions: for expr in all_expressions:
# 检查 style 字段 # 只检查 style_list 中的每一项
if expr.style == style:
return expr
# 检查 style_list 中的每一项
style_list = self._parse_style_list(expr.style_list) style_list = self._parse_style_list(expr.style_list)
if style in style_list: if style in style_list:
return expr return expr
@ -580,7 +576,7 @@ class ExpressionLearner:
async def _find_similar_style_expression(self, style: str, similarity_threshold: float = 0.75) -> Optional[Expression]: async def _find_similar_style_expression(self, style: str, similarity_threshold: float = 0.75) -> Optional[Expression]:
""" """
查找具有相似 style Expression 记录 查找具有相似 style Expression 记录
检查 style 字段和 style_list 中的每一项 只检查 style_list 中的每一项不检查 style 字段因为 style 可能是总结后的概括性描述
Args: Args:
style: 要查找的 style style: 要查找的 style
@ -596,13 +592,7 @@ class ExpressionLearner:
best_similarity = 0.0 best_similarity = 0.0
for expr in all_expressions: for expr in all_expressions:
# 检查 style 字段 # 只检查 style_list 中的每一项
similarity = calculate_style_similarity(style, expr.style)
if similarity >= similarity_threshold and similarity > best_similarity:
best_similarity = similarity
best_match = expr
# 检查 style_list 中的每一项
style_list = self._parse_style_list(expr.style_list) style_list = self._parse_style_list(expr.style_list)
for existing_style in style_list: for existing_style in style_list:
similarity = calculate_style_similarity(style, existing_style) similarity = calculate_style_similarity(style, existing_style)
@ -640,19 +630,48 @@ class ExpressionLearner:
if not styles or len(styles) <= 1: if not styles or len(styles) <= 1:
return None return None
prompt = ( # 计算输入列表中最长项目的长度
"请阅读以下多个语言风格/表达方式,并将它们概括成一句简短的话," max_input_length = max(len(s) for s in styles) if styles else 0
"长度不超过20个字保留共同特点\n" max_summary_length = max_input_length * 2
f"{chr(10).join(f'- {s}' for s in styles[-10:])}\n只输出概括内容。"
) # 最多重试3次
max_retries = 3
retry_count = 0
while retry_count < max_retries:
# 如果是重试,在 prompt 中强调要更简洁
length_hint = f"长度不超过{max_summary_length}个字符," if retry_count > 0 else "长度不超过20个字"
prompt = (
"请阅读以下多个语言风格/表达方式,对其进行总结。"
"不要对其进行语义概括,而是尽可能找出其中不变的部分或共同表达,尽量使用原文"
f"{length_hint}保留共同特点:\n"
f"{chr(10).join(f'- {s}' for s in styles[-10:])}\n只输出概括内容。不要输出其他内容"
)
try: try:
summary, _ = await self.summary_model.generate_response_async(prompt, temperature=0.2) summary, _ = await self.summary_model.generate_response_async(prompt, temperature=0.2)
summary = summary.strip() summary = summary.strip()
if summary: if summary:
return summary # 检查总结长度是否超过限制
except Exception as e: if len(summary) <= max_summary_length:
logger.error(f"概括表达风格失败: {e}") return summary
else:
retry_count += 1
logger.debug(
f"总结长度 {len(summary)} 超过限制 {max_summary_length} "
f"(输入最长项长度: {max_input_length}),重试第 {retry_count}"
)
continue
except Exception as e:
logger.error(f"概括表达风格失败: {e}")
return None
# 如果重试多次后仍然超过长度,返回 None不进行总结
logger.warning(
f"总结多次后仍超过长度限制,放弃总结。"
f"输入最长项长度: {max_input_length}, 最大允许长度: {max_summary_length}"
)
return None return None
async def _summarize_situations(self, situations: List[str]) -> Optional[str]: async def _summarize_situations(self, situations: List[str]) -> Optional[str]:

View File

@ -1,5 +1,5 @@
[inner] [inner]
version = "7.2.3" version = "7.2.4"
#----以下是给开发人员阅读的,如果你只是部署了麦麦,不需要阅读---- #----以下是给开发人员阅读的,如果你只是部署了麦麦,不需要阅读----
# 如果你想要修改配置文件请递增version的值 # 如果你想要修改配置文件请递增version的值
@ -29,8 +29,11 @@ reply_style = "请不要刻意突出自身学科背景。可以参考贴吧,
# 多种回复风格列表,可选配置:当 non-empty 且 multiple_probability>0 时,会按概率随机从中选择一个替换 reply_style # 多种回复风格列表,可选配置:当 non-empty 且 multiple_probability>0 时,会按概率随机从中选择一个替换 reply_style
multiple_reply_style = [ multiple_reply_style = [
"你的风格平淡但不失讽刺,不过分兴奋,很简短。可以参考贴吧,知乎和微博的回复风格。", # "你的风格平淡但不失讽刺,很简短,很白话。可以参考贴吧,微博的回复风格。",
"说话偏口语化偶尔带一点情绪但不要太长篇像B站评论区一样自然。", # "用1-2个字进行回复",
# "用1-2个符号进行回复",
# "言辭凝練古雅,穿插《論語》經句卻不晦澀,以文言短句為基,輔以淺白語意,持長者溫和風範,全用繁體字表達,具先秦儒者談吐韻致。",
# "带点翻译腔,但不要太长",
] ]
# 替换概率:每次构建回复时,以该概率从 multiple_reply_style 中随机选择一个替换 reply_style0.0-1.0 # 替换概率:每次构建回复时,以该概率从 multiple_reply_style 中随机选择一个替换 reply_style0.0-1.0