mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-21 05:14:45 +08:00
45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
|
"""Memory module for DeerFlow.
|
||
|
|
|
||
|
|
This module provides a global memory mechanism that:
|
||
|
|
- Stores user context and conversation history in memory.json
|
||
|
|
- Uses LLM to summarize and extract facts from conversations
|
||
|
|
- Injects relevant memory into system prompts for personalized responses
|
||
|
|
"""
|
||
|
|
|
||
|
|
from src.agents.memory.prompt import (
|
||
|
|
FACT_EXTRACTION_PROMPT,
|
||
|
|
MEMORY_UPDATE_PROMPT,
|
||
|
|
format_conversation_for_update,
|
||
|
|
format_memory_for_injection,
|
||
|
|
)
|
||
|
|
from src.agents.memory.queue import (
|
||
|
|
ConversationContext,
|
||
|
|
MemoryUpdateQueue,
|
||
|
|
get_memory_queue,
|
||
|
|
reset_memory_queue,
|
||
|
|
)
|
||
|
|
from src.agents.memory.updater import (
|
||
|
|
MemoryUpdater,
|
||
|
|
get_memory_data,
|
||
|
|
reload_memory_data,
|
||
|
|
update_memory_from_conversation,
|
||
|
|
)
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
# Prompt utilities
|
||
|
|
"MEMORY_UPDATE_PROMPT",
|
||
|
|
"FACT_EXTRACTION_PROMPT",
|
||
|
|
"format_memory_for_injection",
|
||
|
|
"format_conversation_for_update",
|
||
|
|
# Queue
|
||
|
|
"ConversationContext",
|
||
|
|
"MemoryUpdateQueue",
|
||
|
|
"get_memory_queue",
|
||
|
|
"reset_memory_queue",
|
||
|
|
# Updater
|
||
|
|
"MemoryUpdater",
|
||
|
|
"get_memory_data",
|
||
|
|
"reload_memory_data",
|
||
|
|
"update_memory_from_conversation",
|
||
|
|
]
|