BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update：修正记忆提取llm统计

pull/1354/head
SengokuCola 2025-11-10 00:46:48 +08:00
parent 53bbda0c07
commit 70cffcc387
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/chat/utils/statistic.py
View File

@ -165,11 +165,13 @@ class StatisticOutputTask(AsyncTask):
self.stat_period: List[Tuple[str, timedelta, str]] = [ self.stat_period: List[Tuple[str, timedelta, str]] = [
("all_time", now - deploy_time, "自部署以来"), # 必须保留"all_time" ("all_time", now - deploy_time, "自部署以来"), # 必须保留"all_time"
("last_7_days", timedelta(days=7), "最近7天"), ("last_30_days", timedelta(days=30), "近30天"),
("last_3_days", timedelta(days=3), "最近3天"), ("last_7_days", timedelta(days=7), "近7天"),
("last_24_hours", timedelta(days=1), "最近24小时"), ("last_3_days", timedelta(days=3), "近3天"),
("last_3_hours", timedelta(hours=3), "最近3小时"), ("last_24_hours", timedelta(days=1), "近1天"),
("last_hour", timedelta(hours=1), "最近1小时"), ("last_3_hours", timedelta(hours=3), "近3小时"),
("last_hour", timedelta(hours=1), "近1小时"),
("last_15_minutes", timedelta(minutes=15), "近15分钟"),
] ]
""" """
统计时间段 [(统计名称, 统计时间段, 统计描述), ...] 统计时间段 [(统计名称, 统计时间段, 统计描述), ...]

2
src/memory_system/memory_retrieval.py
View File

@ -178,6 +178,7 @@ async def _react_agent_solve_question(
success, response, reasoning_content, model_name = await llm_api.generate_with_model( success, response, reasoning_content, model_name = await llm_api.generate_with_model(
prompt, prompt,
model_config=model_config.model_task_config.tool_use, model_config=model_config.model_task_config.tool_use,
request_type="memory.react",
) )
logger.info(f"ReAct Agent 第 {iteration + 1} 次迭代 LLM响应: {response}") logger.info(f"ReAct Agent 第 {iteration + 1} 次迭代 LLM响应: {response}")
@ -474,6 +475,7 @@ async def build_memory_retrieval_prompt(
success, response, reasoning_content, model_name = await llm_api.generate_with_model( success, response, reasoning_content, model_name = await llm_api.generate_with_model(
question_prompt, question_prompt,
model_config=model_config.model_task_config.tool_use, model_config=model_config.model_task_config.tool_use,
request_type="memory.question",
) )
logger.info(f"记忆检索问题生成提示词: {question_prompt}") logger.info(f"记忆检索问题生成提示词: {question_prompt}")