import time

from app.embeddings.pgvector_store import get_vector_store

start_time = time.perf_counter()
programs_vector_store = get_vector_store("programs")
blogs_vector_store = get_vector_store("blogs")

k = 20

blogs_docs = blogs_vector_store.similarity_search(
    query="What is the best uni for nepali students?",
    k=k,
)

end_time = time.perf_counter()

print(f"Retrieving {k} docs took: {end_time - start_time:.2}s")