Function main

examples/mem_feedback/example_feedback.py:153–244

Main program flow: 1. Initialize components. 2. Simulate a conversation scenario and existing (possibly incorrect) memory. 3. Receive user feedback (correct memory). 4. Process feedback and update memory store. 5. Display processing results.

()

def main():
    """
    Main program flow:
    1. Initialize components.
    2. Simulate a conversation scenario and existing (possibly incorrect) memory.
    3. Receive user feedback (correct memory).
    4. Process feedback and update memory store.
    5. Display processing results.
    """
    # Load dotenv in main to avoid affecting module import order
    from dotenv import load_dotenv

    load_dotenv()

    # Lazy import to avoid E402
    from memos.mem_feedback.utils import make_mem_item

    feedback_server, memory_manager, embedder = init_components()
    print("-" * 50)
    print("Initialization Done. Processing Feedback...")
    print("-" * 50)

    # 1. Simulate Chat History
    # Simulate a conversation between user and assistant, where the assistant's response contains a statement about user preferences.
    history = [
        {"role": "user", "content": "我喜欢什么水果,不喜欢什么水果"},
        {"role": "assistant", "content": "你喜欢苹果,不喜欢香蕉"},
    ]

    # 2. Simulate Initial Memory
    # We manually add a memory to the database, representing what the system currently believes to be a "fact".
    # This memory content is "你喜欢苹果,不喜欢香蕉", which we will later correct via feedback.
    mem_text = "你喜欢苹果,不喜欢香蕉"
    memory_manager.add(
        [
            make_mem_item(
                mem_text,
                user_id="user_id_001",
                user_name="cube_id_001_0115",
                session_id="session_id",
                tags=["fact"],
                key="food_preference",
                sources=[{"type": "chat"}],
                background="init from chat history",
                embedding=embedder.embed([mem_text])[
                    0
                ],  # Generate embedding for subsequent retrieval
                info={
                    "user_id": "user_id_001",
                    "user_name": "cube_id_001_0115",
                    "session_id": "session_id",
                },
            )
        ],
        user_name="cube_id_001_0115",
        mode="sync",
    )