brain_server/rag/retriever.py

"""RAG Retriever: ??????? FAISS."""
import logging
from memory.knowledge_vector import KnowledgeVectorStore

logger = logging.getLogger("kapo.rag.retriever")


def retrieve(query: str, top_k: int = 3):
    try:
        store = KnowledgeVectorStore()
        return store.query(query, top_k=top_k)
    except Exception:
        logger.exception("Retrieve failed")
        return []