Harrison/tair (#3770)

Co-authored-by: Seth Huang <848849+seth-hg@users.noreply.github.com>
This commit is contained in:
Harrison Chase
2023-04-28 21:25:33 -07:00
committed by GitHub
parent 502ba6a0be
commit 0c0f14407c
6 changed files with 455 additions and 0 deletions

View File

@@ -0,0 +1,15 @@
"""Test tair functionality."""
from langchain.docstore.document import Document
from langchain.vectorstores.tair import Tair
from tests.integration_tests.vectorstores.fake_embeddings import FakeEmbeddings
def test_tair() -> None:
"""Test end to end construction and search."""
texts = ["foo", "bar", "baz"]
docsearch = Tair.from_texts(
texts, FakeEmbeddings(), tair_url="redis://localhost:6379"
)
output = docsearch.similarity_search("foo", k=1)
assert output == [Document(page_content="foo")]