Skip to content

Commit 30796c6

Browse files
authored
Merge pull request #1262 from xinnan-tech/web-headerbar-modify
update:修复init_memorybug
2 parents 696405a + 86b5eb1 commit 30796c6

File tree

4 files changed

+13
-9
lines changed

4 files changed

+13
-9
lines changed

main/xiaozhi-server/core/connection.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,9 @@ def save_memory_task():
231231
# 创建新事件循环(避免与主循环冲突)
232232
loop = asyncio.new_event_loop()
233233
asyncio.set_event_loop(loop)
234-
loop.run_until_complete(self.memory.save_memory(self.dialogue.dialogue))
234+
loop.run_until_complete(
235+
self.memory.save_memory(self.dialogue.dialogue)
236+
)
235237
except Exception as e:
236238
self.logger.bind(tag=TAG).error(f"保存记忆失败: {e}")
237239
finally:
@@ -441,10 +443,10 @@ def _initialize_private_config(self):
441443
def _initialize_memory(self):
442444
"""初始化记忆模块"""
443445
self.memory.init_memory(
444-
self.device_id,
445-
self.llm,
446-
self.config["summaryMemory"],
447-
not self.read_config_from_api,
446+
role_id=self.device_id,
447+
llm=self.llm,
448+
summary_memory=self.config.get("summaryMemory", None),
449+
save_to_file=not self.read_config_from_api,
448450
)
449451

450452
def _initialize_intent(self):

main/xiaozhi-server/core/providers/intent/intent_llm/intent_llm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ async def detect_intent(self, conn, dialogue_history: List[Dict], text: str) ->
215215

216216
# 记录识别到的function call
217217
logger.bind(tag=TAG).info(
218-
f"识别到function call: {function_name}, 参数: {function_args}"
218+
f"llm 识别到意图: {function_name}, 参数: {function_args}"
219219
)
220220

221221
# 添加到缓存

main/xiaozhi-server/core/providers/memory/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,6 @@ async def query_memory(self, query: str) -> str:
2121
"""Query memories for specific role based on similarity"""
2222
return "please implement query method"
2323

24-
def init_memory(self, role_id, llm, summary_memory=None):
24+
def init_memory(self, role_id, llm, **kwargs):
2525
self.role_id = role_id
2626
self.llm = llm

main/xiaozhi-server/core/providers/memory/mem_local_short/mem_local_short.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,8 +112,10 @@ def __init__(self, config, summary_memory):
112112
self.memory_path = get_project_dir() + "data/.memory.yaml"
113113
self.load_memory(summary_memory)
114114

115-
def init_memory(self, role_id, llm, summary_memory=None, save_to_file=True):
116-
super().init_memory(role_id, llm)
115+
def init_memory(
116+
self, role_id, llm, summary_memory=None, save_to_file=True, **kwargs
117+
):
118+
super().init_memory(role_id, llm, **kwargs)
117119
self.save_to_file = save_to_file
118120
self.load_memory(summary_memory)
119121

0 commit comments

Comments
 (0)