github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 14:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: FlashrankRerank support loading customer client (#23350)

Browse Source 
Description: FlashrankRerank Document compressor support loading
customer client
Issue: #23338

Co-authored-by: gongwn1 <gongwn1@lenovo.com>
This commit is contained in:
wenngong 2024-06-25 05:50:08 +08:00 committed by GitHub
parent f58c40b4e3
commit b33d2346db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
libs/community/langchain_community/document_compressors/flashrank_rerank.py
View File

@ -38,17 +38,20 @@ class FlashrankRerank(BaseDocumentCompressor):
@root_validator(pre=True) @root_validator(pre=True)
def validate_environment(cls, values: Dict) -> Dict: def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key and python package exists in environment.""" """Validate that api key and python package exists in environment."""
try: if "client" in values:
from flashrank import Ranker return values
except ImportError: else:
raise ImportError( try:
"Could not import flashrank python package. " from flashrank import Ranker
"Please install it with `pip install flashrank`." except ImportError:
) raise ImportError(
"Could not import flashrank python package. "
"Please install it with `pip install flashrank`."
)
values["model"] = values.get("model", DEFAULT_MODEL_NAME) values["model"] = values.get("model", DEFAULT_MODEL_NAME)
values["client"] = Ranker(model_name=values["model"]) values["client"] = Ranker(model_name=values["model"])
return values return values
def compress_documents( def compress_documents(
self, self,