github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 07:09:31 +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

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

@ -38,6 +38,9 @@ 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."""
if "client" in values:
return values
else:
try: try:
from flashrank import Ranker from flashrank import Ranker
except ImportError: except ImportError: