mirror of
https://github.com/kennethreitz/langchain.git
synced 2026-06-05 23:00:18 +00:00
shibuiwilliam
f29a5d4bcc
# What - This is to add test for knn retriever. --------- Co-authored-by: Bagatur <baskaryan@gmail.com>
12 lines
413 B
Python
12 lines
413 B
Python
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
|