diff --git a/scripts/make_rag_database.py b/scripts/make_rag_database.py index 61bffc7..8b7fad3 100644 --- a/scripts/make_rag_database.py +++ b/scripts/make_rag_database.py @@ -59,6 +59,8 @@ def main(save_path = "assets/xiaozhan_emb"): "k":3 }) + u = loaded_vectorstore.similarity_search("灯与尘", k=2) + res = retriever.invoke("灯与尘") for doc in res: