api/agent/memory.py

from langchain.memory import ConversationSummaryBufferMemory

shared_memory = None

def get_memory(llm):
    global shared_memory
    if shared_memory is None:
        shared_memory = ConversationSummaryBufferMemory(
            llm=llm,
            max_token_limit=1000, # Merangkum percakapan jika melebihi 1000 token
            memory_key="chat_history",
            return_messages=True
        )
    return shared_memory