From 74f7ededcd0d3a653eac5d9788569bcd5dd1f052 Mon Sep 17 00:00:00 2001 From: mtayfur Date: Thu, 9 Oct 2025 11:41:20 +0300 Subject: [PATCH] Refactor memory retrieval logging to include retrieval method, enhancing clarity in LLM and semantic decision processes --- memory_system.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/memory_system.py b/memory_system.py index 5d30273..3828f2d 100644 --- a/memory_system.py +++ b/memory_system.py @@ -784,7 +784,7 @@ CANDIDATE MEMORIES: should_use_llm, decision_reason = self._should_use_llm_reranking(candidate_memories) - analysis_info = {"llm_decision": should_use_llm, "decision_reason": decision_reason, "candidate_count": len(candidate_memories)} + analysis_info = {"llm_decision": should_use_llm, "decision_reason": decision_reason, "candidate_count": len(candidate_memories)} if should_use_llm: extended_count = int(self.memory_system.valves.max_memories_returned * Constants.EXTENDED_MAX_MEMORY_MULTIPLIER) @@ -803,11 +803,12 @@ CANDIDATE MEMORIES: else: logger.info(f"Skipping LLM reranking: {decision_reason}") selected_memories = candidate_memories[:max_injection] - + duration = time.time() - start_time duration_text = f" in {duration:.2f}s" if duration >= 0.01 else "" - await self.memory_system._emit_status(emitter, f"🎯 Memory Retrieval Complete{duration_text}", done=True) - logger.info(f"🎯 Memory Retrieval Complete{duration_text}") + retrieval_method = "LLM" if should_use_llm else "Semantic" + await self.memory_system._emit_status(emitter, f"🎯 {retrieval_method} Memory Retrieval Complete{duration_text}", done=True) + logger.info(f"🎯 {retrieval_method} Memory Retrieval Complete{duration_text}") return selected_memories, analysis_info