langchain/libs/community/tests/unit_tests/retrievers/test_imports.py

from langchain_community.retrievers import __all__, _module_lookup

EXPECTED_ALL = [
    "AmazonKendraRetriever",
    "AmazonKnowledgeBasesRetriever",
    "ArceeRetriever",
    "ArxivRetriever",
    "AzureAISearchRetriever",
    "AzureCognitiveSearchRetriever",
    "BreebsRetriever",
    "ChatGPTPluginRetriever",
    "ChaindeskRetriever",
    "CohereRagRetriever",
    "DriaRetriever",
    "ElasticSearchBM25Retriever",
    "EmbedchainRetriever",
    "GoogleDocumentAIWarehouseRetriever",
    "GoogleCloudEnterpriseSearchRetriever",
    "GoogleVertexAIMultiTurnSearchRetriever",
    "GoogleVertexAISearchRetriever",
    "KayAiRetriever",
    "KNNRetriever",
    "LlamaIndexGraphRetriever",
    "LlamaIndexRetriever",
    "MetalRetriever",
    "MilvusRetriever",
    "OutlineRetriever",
    "PineconeHybridSearchRetriever",
    "PubMedRetriever",
    "QdrantSparseVectorRetriever",
    "RemoteLangChainRetriever",
    "SVMRetriever",
    "TavilySearchAPIRetriever",
    "TFIDFRetriever",
    "BM25Retriever",
    "VespaRetriever",
    "WeaviateHybridSearchRetriever",
    "WikipediaRetriever",
    "YouRetriever",
    "ZepRetriever",
    "ZillizRetriever",
    "DocArrayRetriever",
    "NeuralDBRetriever",
]


def test_all_imports() -> None:
    assert set(__all__) == set(EXPECTED_ALL)
    assert set(__all__) == set(_module_lookup.keys())
community[patch]: import flattening fix (#20110) This PR should make it easier for linters to do type checking and for IDEs to jump to definition of code. See #20050 as a template for this PR. - As a byproduct: Added 3 missed `test_imports`. - Added missed `SolarChat` in to __init___.py Added it into test_import ut. - Added `# type: ignore` to fix linting. It is not clear, why linting errors appear after ^ changes. --------- Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com> 5 months ago			`from langchain_community.retrievers import __all__, _module_lookup`
add import tests to all modules (#12806) 11 months ago
			`EXPECTED_ALL = [`
			`"AmazonKendraRetriever",`
langchain[minor]: Add retriever for Knowledge Bases for Amazon Bedrock (#13980) - Description: Adds a retriever implementation for [Knowledge Bases for Amazon Bedrock](https://aws.amazon.com/bedrock/knowledge-bases/), a new service announced at AWS re:Invent, shortly before this PR was opened. This depends on the `bedrock-agent-runtime` service, which will be included in a future version of `boto3` and of `botocore`. We will open a follow-up PR documenting the minimum required versions of `boto3` and `botocore` after that information is available. - Issue: N/A - Dependencies: `boto3>=1.33.2, botocore>=1.33.2` - Tag maintainer: @baskaryan - Twitter handles: `@pjain7` `@dead_letter_q` This PR includes a documentation notebook under `docs/docs/integrations/retrievers`, which I (@dlqqq) have verified independently. EDIT: `bedrock-agent-runtime` service is now included in `boto3>=1.33.2`: https://github.com/boto/boto3/commit/5cf793f49369607f2557fd11741c357df866e4a5 --------- Co-authored-by: Piyush Jain <piyushjain@duck.com> Co-authored-by: Erick Friis <erick@langchain.dev> Co-authored-by: Bagatur <baskaryan@gmail.com> 10 months ago			`"AmazonKnowledgeBasesRetriever",`
add import tests to all modules (#12806) 11 months ago			`"ArceeRetriever",`
			`"ArxivRetriever",`
Community: Updating Azure Retriever and Docs to be Azure AI Search instead of Azure Cognitive Search (#19925) Last year Microsoft [changed the name](https://learn.microsoft.com/en-us/azure/search/search-what-is-azure-search) of Azure Cognitive Search to Azure AI Search. This PR updates the Langchain Azure Retriever API and it's associated docs to reflect this change. It may be confusing for users to see the name Cognitive here and AI in the Microsoft documentation which is why this is needed. I've also added a more detailed example to the Azure retriever doc page. There are more places that need a similar update but I'm breaking it up so the PRs are not too big 😄 Fixing my errors from the previous PR. Twitter: @marlene_zw Two new tests added to test backward compatibility in `libs/community/tests/integration_tests/retrievers/test_azure_cognitive_search.py` --------- Co-authored-by: Chester Curme <chester.curme@gmail.com> 5 months ago			`"AzureAISearchRetriever",`
add import tests to all modules (#12806) 11 months ago			`"AzureCognitiveSearchRetriever",`
community[minor]: Breebs docs retriever (#16578) - Description: Implementation of breeb retriever with integration tests -> libs/community/tests/integration_tests/retrievers/test_breebs.py and documentation (notebook) -> docs/docs/integrations/retrievers/breebs.ipynb. - Dependencies: None 7 months ago			`"BreebsRetriever",`
add import tests to all modules (#12806) 11 months ago			`"ChatGPTPluginRetriever",`
			`"ChaindeskRetriever",`
			`"CohereRagRetriever",`
community[minor]: Add Dria retriever (#17098) [Dria](https://dria.co/) is a hub of public RAG models for developers to both contribute and utilize a shared embedding lake. This PR adds a retriever that can retrieve documents from Dria. 6 months ago			`"DriaRetriever",`
add import tests to all modules (#12806) 11 months ago			`"ElasticSearchBM25Retriever",`
Add embedchain retriever (#13553) Description: This commit adds embedchain retriever along with tests and docs. Embedchain is a RAG framework to create data pipelines. Twitter handle: - [Taranjeet's twitter](https://twitter.com/taranjeetio) and [Embedchain's twitter](https://twitter.com/embedchain) Reviewer @hwchase17 --------- Co-authored-by: Bagatur <baskaryan@gmail.com> 10 months ago			`"EmbedchainRetriever",`
add import tests to all modules (#12806) 11 months ago			`"GoogleDocumentAIWarehouseRetriever",`
			`"GoogleCloudEnterpriseSearchRetriever",`
			`"GoogleVertexAIMultiTurnSearchRetriever",`
			`"GoogleVertexAISearchRetriever",`
			`"KayAiRetriever",`
			`"KNNRetriever",`
			`"LlamaIndexGraphRetriever",`
			`"LlamaIndexRetriever",`
			`"MetalRetriever",`
			`"MilvusRetriever",`
FEATURE: Add retriever for Outline (#13889) - Description: Added a retriever for the Outline API to ask questions on knowledge base - Issue: resolves #11814 - Dependencies: None - Tag maintainer: @baskaryan 10 months ago			`"OutlineRetriever",`
add import tests to all modules (#12806) 11 months ago			`"PineconeHybridSearchRetriever",`
			`"PubMedRetriever",`
community[minor]: Qdrant sparse vector retriever (#14814) ## Description This PR intends to add support for Qdrant's new [sparse vector retrieval](https://qdrant.tech/articles/sparse-vectors/) by introducing a new retriever class, `QdrantSparseVectorRetriever`. Necessary usage docs and integration tests have been added for the retriever. --------- Co-authored-by: Bagatur <baskaryan@gmail.com> 9 months ago			`"QdrantSparseVectorRetriever",`
add import tests to all modules (#12806) 11 months ago			`"RemoteLangChainRetriever",`
			`"SVMRetriever",`
			`"TavilySearchAPIRetriever",`
			`"TFIDFRetriever",`
			`"BM25Retriever",`
			`"VespaRetriever",`
			`"WeaviateHybridSearchRetriever",`
			`"WikipediaRetriever",`
community: Add you.com utility, update you retriever integration docs (#17014) <!-- Thank you for contributing to LangChain! Please title your PR "<package>: <description>", where <package> is whichever of langchain, community, core, experimental, etc. is being modified. Replace this entire comment with: - Description: a description of the change, - Issue: the issue # it fixes if applicable, - Dependencies: any dependencies required for this change, - Twitter handle: we announce bigger features on Twitter. If your PR gets announced, and you'd like a mention, we'll gladly shout you out! Please make sure your PR is passing linting and testing before submitting. Run `make format`, `make lint` and `make test` from the root of the package you've modified to check this locally. See contribution guidelines for more information on how to write/run tests, lint, etc: https://python.langchain.com/docs/contributing/ If you're adding a new integration, please include: 1. a test for the integration, preferably unit tests that do not rely on network access, 2. an example notebook showing its use. It lives in `docs/docs/integrations` directory. If no one reviews your PR within a few days, please @-mention one of @baskaryan, @eyurtsev, @hwchase17. --> - Description: changes to you.com files - general cleanup - adds community/utilities/you.py, moving bulk of code from retriever -> utility - removes `snippet` as endpoint - adds `news` as endpoint - adds more tests <s>Description: update community MAKE file - adds `integration_tests` - adds `coverage`</s> - Issue: the issue # it fixes if applicable, - [For New Contributors: Update Integration Documentation](https://github.com/langchain-ai/langchain/issues/15664#issuecomment-1920099868) - Dependencies: n/a - Twitter handle: @scottnath - Mastodon handle: scottnath@mastodon.social --------- Co-authored-by: Bagatur <baskaryan@gmail.com> 7 months ago			`"YouRetriever",`
add import tests to all modules (#12806) 11 months ago			`"ZepRetriever",`
			`"ZillizRetriever",`
			`"DocArrayRetriever",`
community[minor]: Added integrations for ThirdAI's NeuralDB as a Retriever (#17334) Description: Adds ThirdAI NeuralDB retriever integration. NeuralDB is a CPU-friendly and fine-tunable text retrieval engine. We previously added a vector store integration but we think that it will be easier for our customers if they can also find us under under langchain-community/retrievers. --------- Co-authored-by: kartikTAI <129414343+kartikTAI@users.noreply.github.com> Co-authored-by: Kartik Sarangmath <kartik@thirdai.com> 5 months ago			`"NeuralDBRetriever",`
add import tests to all modules (#12806) 11 months ago			`]`


			`def test_all_imports() -> None:`
			`assert set(__all__) == set(EXPECTED_ALL)`
community[patch]: import flattening fix (#20110) This PR should make it easier for linters to do type checking and for IDEs to jump to definition of code. See #20050 as a template for this PR. - As a byproduct: Added 3 missed `test_imports`. - Added missed `SolarChat` in to __init___.py Added it into test_import ut. - Added `# type: ignore` to fix linting. It is not clear, why linting errors appear after ^ changes. --------- Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com> 5 months ago			`assert set(__all__) == set(_module_lookup.keys())`