mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-09 06:53:59 +00:00
Add Qdrant named arguments (#1386)
This PR: - Increases `qdrant-client` version to 1.0.4 - Introduces custom content and metadata keys (as requested in #1087) - Moves all the `QdrantClient` parameters into the method parameters to simplify code completion
This commit is contained in:
@@ -1,18 +1,46 @@
|
||||
"""Test Qdrant functionality."""
|
||||
import pytest
|
||||
|
||||
from langchain.docstore.document import Document
|
||||
from langchain.vectorstores import Qdrant
|
||||
from tests.integration_tests.vectorstores.fake_embeddings import FakeEmbeddings
|
||||
|
||||
|
||||
def test_qdrant() -> None:
|
||||
@pytest.mark.parametrize(
|
||||
["content_payload_key", "metadata_payload_key"],
|
||||
[
|
||||
(Qdrant.CONTENT_KEY, Qdrant.METADATA_KEY),
|
||||
("foo", "bar"),
|
||||
(Qdrant.CONTENT_KEY, "bar"),
|
||||
("foo", Qdrant.METADATA_KEY),
|
||||
],
|
||||
)
|
||||
def test_qdrant(content_payload_key: str, metadata_payload_key: str) -> None:
|
||||
"""Test end to end construction and search."""
|
||||
texts = ["foo", "bar", "baz"]
|
||||
docsearch = Qdrant.from_texts(texts, FakeEmbeddings(), host="localhost")
|
||||
docsearch = Qdrant.from_texts(
|
||||
texts,
|
||||
FakeEmbeddings(),
|
||||
host="localhost",
|
||||
content_payload_key=content_payload_key,
|
||||
metadata_payload_key=metadata_payload_key,
|
||||
)
|
||||
output = docsearch.similarity_search("foo", k=1)
|
||||
assert output == [Document(page_content="foo")]
|
||||
|
||||
|
||||
def test_qdrant_with_metadatas() -> None:
|
||||
@pytest.mark.parametrize(
|
||||
["content_payload_key", "metadata_payload_key"],
|
||||
[
|
||||
(Qdrant.CONTENT_KEY, Qdrant.METADATA_KEY),
|
||||
("test_content", "test_payload"),
|
||||
(Qdrant.CONTENT_KEY, "payload_test"),
|
||||
("content_test", Qdrant.METADATA_KEY),
|
||||
],
|
||||
)
|
||||
def test_qdrant_with_metadatas(
|
||||
content_payload_key: str, metadata_payload_key: str
|
||||
) -> None:
|
||||
"""Test end to end construction and search."""
|
||||
texts = ["foo", "bar", "baz"]
|
||||
metadatas = [{"page": i} for i in range(len(texts))]
|
||||
@@ -21,12 +49,25 @@ def test_qdrant_with_metadatas() -> None:
|
||||
FakeEmbeddings(),
|
||||
metadatas=metadatas,
|
||||
host="localhost",
|
||||
content_payload_key=content_payload_key,
|
||||
metadata_payload_key=metadata_payload_key,
|
||||
)
|
||||
output = docsearch.similarity_search("foo", k=1)
|
||||
assert output == [Document(page_content="foo", metadata={"page": 0})]
|
||||
|
||||
|
||||
def test_qdrant_max_marginal_relevance_search() -> None:
|
||||
@pytest.mark.parametrize(
|
||||
["content_payload_key", "metadata_payload_key"],
|
||||
[
|
||||
(Qdrant.CONTENT_KEY, Qdrant.METADATA_KEY),
|
||||
("test_content", "test_payload"),
|
||||
(Qdrant.CONTENT_KEY, "payload_test"),
|
||||
("content_test", Qdrant.METADATA_KEY),
|
||||
],
|
||||
)
|
||||
def test_qdrant_max_marginal_relevance_search(
|
||||
content_payload_key: str, metadata_payload_key: str
|
||||
) -> None:
|
||||
"""Test end to end construction and MRR search."""
|
||||
texts = ["foo", "bar", "baz"]
|
||||
metadatas = [{"page": i} for i in range(len(texts))]
|
||||
@@ -35,6 +76,8 @@ def test_qdrant_max_marginal_relevance_search() -> None:
|
||||
FakeEmbeddings(),
|
||||
metadatas=metadatas,
|
||||
host="localhost",
|
||||
content_payload_key=content_payload_key,
|
||||
metadata_payload_key=metadata_payload_key,
|
||||
)
|
||||
output = docsearch.max_marginal_relevance_search("foo", k=2, fetch_k=3)
|
||||
assert output == [
|
||||
|
Reference in New Issue
Block a user