test_as_retriever_sparse_mode() — langchain Function Reference

Source Code

libs/partners/qdrant/tests/integration_tests/qdrant_vector_store/test_search.py lines 350–382

def test_as_retriever_sparse_mode(location: str) -> None:
    """Test that as_retriever() works in SPARSE mode."""
    # Use from_texts to create the vectorstore, which handles collection creation
    texts = ["test document"]
    vectorstore = QdrantVectorStore.from_texts(
        texts,
        embedding=None,  # No dense embedding for SPARSE mode
        location=location,
        retrieval_mode=RetrievalMode.SPARSE,
        sparse_embedding=ConsistentFakeSparseEmbeddings(),
        sparse_vector_name="sparse",
    )

    # Add test documents
    docs = [
        Document(page_content="Python programming", metadata={"topic": "programming"}),
        Document(page_content="Machine learning", metadata={"topic": "AI"}),
        Document(page_content="Data analysis", metadata={"topic": "data"}),
    ]
    vectorstore.add_documents(docs)

    # Test basic as_retriever() functionality
    retriever = vectorstore.as_retriever()
    results = retriever.invoke("programming")

    # Should return documents
    assert len(results) > 0
    assert all(isinstance(doc, Document) for doc in results)

    # Test that retriever has tags
    assert hasattr(retriever, "tags")
    assert isinstance(retriever.tags, list)
    assert "QdrantVectorStore" in retriever.tags