Spaces:

gourisankar85
/

realtime-rag-pipeline

Running

realtime-rag-pipeline / retriever /retrieve_documents.py

Gourisankar Padihary

Initial commit

026aeba 4 months ago

117 Bytes

	def retrieve_top_k_documents(vector_store, query, top_k=5):
	return vector_store.similarity_search(query, k=top_k)