I before E (#6015)

searx_updates
Zander Chase 11 months ago committed by GitHub
parent a7227ee01b
commit 614cff89bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1,4 +1,5 @@
"""Milvus Retriever"""
import warnings
from typing import Any, Dict, List, Optional
from langchain.embeddings.base import Embeddings
@ -8,7 +9,7 @@ from langchain.vectorstores.milvus import Milvus
# TODO: Update to MilvusClient + Hybrid Search when available
class MilvusRetreiver(BaseRetriever):
class MilvusRetriever(BaseRetriever):
def __init__(
self,
embedding_function: Embeddings,
@ -41,3 +42,12 @@ class MilvusRetreiver(BaseRetriever):
async def aget_relevant_documents(self, query: str) -> List[Document]:
raise NotImplementedError
def MilvusRetreiver(*args: Any, **kwargs: Any) -> MilvusRetriever:
warnings.warn(
"MilvusRetreiver will be deprecated in the future. "
"Please use MilvusRetriever ('i' before 'e') instead.",
DeprecationWarning,
)
return MilvusRetriever(*args, **kwargs)

@ -1,4 +1,5 @@
"""Zilliz Retriever"""
import warnings
from typing import Any, Dict, List, Optional
from langchain.embeddings.base import Embeddings
@ -8,7 +9,7 @@ from langchain.vectorstores.zilliz import Zilliz
# TODO: Update to ZillizClient + Hybrid Search when available
class ZillizRetreiver(BaseRetriever):
class ZillizRetriever(BaseRetriever):
def __init__(
self,
embedding_function: Embeddings,
@ -41,3 +42,12 @@ class ZillizRetreiver(BaseRetriever):
async def aget_relevant_documents(self, query: str) -> List[Document]:
raise NotImplementedError
def ZillizRetreiver(*args: Any, **kwargs: Any) -> ZillizRetriever:
warnings.warn(
"ZillizRetreiver will be deprecated in the future. "
"Please use ZillizRetriever ('i' before 'e') instead.",
DeprecationWarning,
)
return ZillizRetriever(*args, **kwargs)

Loading…
Cancel
Save