diff --git a/tests/unit_tests/retrievers/test_knn.py b/tests/unit_tests/retrievers/test_knn.py new file mode 100644 index 00000000000..e0486d7efa7 --- /dev/null +++ b/tests/unit_tests/retrievers/test_knn.py @@ -0,0 +1,11 @@ +from langchain.embeddings import FakeEmbeddings +from langchain.retrievers.knn import KNNRetriever + + +class TestKNNRetriever: + def test_from_texts(self) -> None: + input_texts = ["I have a pen.", "Do you have a pen?", "I have a bag."] + knn_retriever = KNNRetriever.from_texts( + texts=input_texts, embeddings=FakeEmbeddings(size=100) + ) + assert len(knn_retriever.texts) == 3