LawranceLiao
722c690f4f
* fix(memory): isolate queued memory updates by agent * fix(memory): include user in queue identity * Potential fix for pull request finding Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> * Fix the lint error --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
"""Hooks fired before summarization removes messages from state."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from deerflow.agents.memory.message_processing import detect_correction, detect_reinforcement, filter_messages_for_memory
|
|
from deerflow.agents.memory.queue import get_memory_queue
|
|
from deerflow.agents.middlewares.summarization_middleware import SummarizationEvent
|
|
from deerflow.config.memory_config import get_memory_config
|
|
from deerflow.runtime.user_context import resolve_runtime_user_id
|
|
|
|
|
|
def memory_flush_hook(event: SummarizationEvent) -> None:
|
|
"""Flush messages about to be summarized into the memory queue."""
|
|
if not get_memory_config().enabled or not event.thread_id:
|
|
return
|
|
|
|
filtered_messages = filter_messages_for_memory(list(event.messages_to_summarize))
|
|
user_messages = [message for message in filtered_messages if getattr(message, "type", None) == "human"]
|
|
assistant_messages = [message for message in filtered_messages if getattr(message, "type", None) == "ai"]
|
|
if not user_messages or not assistant_messages:
|
|
return
|
|
|
|
correction_detected = detect_correction(filtered_messages)
|
|
reinforcement_detected = not correction_detected and detect_reinforcement(filtered_messages)
|
|
user_id = resolve_runtime_user_id(event.runtime)
|
|
queue = get_memory_queue()
|
|
queue.add_nowait(
|
|
thread_id=event.thread_id,
|
|
messages=filtered_messages,
|
|
agent_name=event.agent_name,
|
|
user_id=user_id,
|
|
correction_detected=correction_detected,
|
|
reinforcement_detected=reinforcement_detected,
|
|
)
|