Spaces:
Sleeping
Sleeping
File size: 1,984 Bytes
64d7fdf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | from langchain_community.chat_message_histories import RedisChatMessageHistory
from langchain_core.runnables.history import RunnableWithMessageHistory
from app.config import config
from app.utils.logger import logger
class ConversationMemory:
def __init__(self):
self.max_messages = config["rag"]["memory"]["max_messages"]
self.redis_url = config["database"]["redis"]["url"]
def get_message_history(self, session_id: str) -> RedisChatMessageHistory:
return RedisChatMessageHistory(
session_id=session_id,
url=self.redis_url,
ttl=86400
)
def create_history_runnable(self, runnable):
return RunnableWithMessageHistory(
runnable,
self.get_message_history,
input_messages_key="input",
history_messages_key="chat_history"
)
def get_messages(self, session_id: str):
history = self.get_message_history(session_id)
messages = history.messages
return messages[-self.max_messages:] if len(messages) > self.max_messages else messages
def add_message(self, session_id: str, role: str, content: str):
"""Add a message to the conversation history."""
try:
history = self.get_message_history(session_id)
if role == "user":
history.add_user_message(content)
elif role == "assistant":
history.add_ai_message(content)
logger.debug(f"Added {role} message to session {session_id}")
except Exception as e:
logger.error(f"Error adding message: {e}")
def clear(self, session_id: str):
try:
history = self.get_message_history(session_id)
history.clear()
logger.info(f"Cleared memory for session {session_id}")
except Exception as e:
logger.error(f"Memory clear error: {e}")
conversation_memory = ConversationMemory()
|