Spaces:

MrA7A1
/

AiCoderClean

Sleeping

AiCoderClean / brain_server /rag /retriever.py

Initial modernized KAPO runtime upload

06ce7ac verified 7 months ago

395 Bytes

"""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 []

	"""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 []