agent.py

from memory import MemoryVectorStore

memory = MemoryVectorStore()

def run_agent(data):
    user_input = data.get("input", "")
    memory.add(user_input)
    context = memory.retrieve(user_input)
    return {
        "echo": user_input,
        "context": context
    }