2024-04-10 17:01:19 +00:00
|
|
|
from langchain_community.retrievers import __all__, _module_lookup
|
2023-11-02 22:32:55 +00:00
|
|
|
|
|
|
|
EXPECTED_ALL = [
|
|
|
|
"AmazonKendraRetriever",
|
2023-11-28 22:10:23 +00:00
|
|
|
"AmazonKnowledgeBasesRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"ArceeRetriever",
|
|
|
|
"ArxivRetriever",
|
2024-04-08 15:12:41 +00:00
|
|
|
"AzureAISearchRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"AzureCognitiveSearchRetriever",
|
2024-02-05 23:51:08 +00:00
|
|
|
"BreebsRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"ChatGPTPluginRetriever",
|
|
|
|
"ChaindeskRetriever",
|
|
|
|
"CohereRagRetriever",
|
2024-04-01 19:04:19 +00:00
|
|
|
"DriaRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"ElasticSearchBM25Retriever",
|
2023-11-20 01:35:03 +00:00
|
|
|
"EmbedchainRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"GoogleDocumentAIWarehouseRetriever",
|
|
|
|
"GoogleCloudEnterpriseSearchRetriever",
|
|
|
|
"GoogleVertexAIMultiTurnSearchRetriever",
|
|
|
|
"GoogleVertexAISearchRetriever",
|
|
|
|
"KayAiRetriever",
|
|
|
|
"KNNRetriever",
|
|
|
|
"LlamaIndexGraphRetriever",
|
|
|
|
"LlamaIndexRetriever",
|
|
|
|
"MetalRetriever",
|
|
|
|
"MilvusRetriever",
|
2023-11-27 02:56:12 +00:00
|
|
|
"OutlineRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"PineconeHybridSearchRetriever",
|
|
|
|
"PubMedRetriever",
|
2023-12-20 07:22:19 +00:00
|
|
|
"QdrantSparseVectorRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"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>
2024-02-08 21:47:50 +00:00
|
|
|
"YouRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"ZepRetriever",
|
|
|
|
"ZillizRetriever",
|
|
|
|
"DocArrayRetriever",
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
def test_all_imports() -> None:
|
|
|
|
assert set(__all__) == set(EXPECTED_ALL)
|
2024-04-10 17:01:19 +00:00
|
|
|
assert set(__all__) == set(_module_lookup.keys())
|