feat: Update import path for TextCrossEncoder to support fastembed versioning and add fallback for older versions

This commit is contained in:
catlog22
2026-01-05 23:13:52 +08:00
parent f90c6b9fab
commit 2e90230097

View File

@@ -13,6 +13,12 @@ except ImportError:
FASTEMBED_AVAILABLE = False FASTEMBED_AVAILABLE = False
try: try:
# fastembed >= 0.4.0 moved TextCrossEncoder to rerank.cross_encoder
from fastembed.rerank.cross_encoder import TextCrossEncoder
RERANKER_AVAILABLE = True
except ImportError:
try:
# Fallback for older versions
from fastembed import TextCrossEncoder from fastembed import TextCrossEncoder
RERANKER_AVAILABLE = True RERANKER_AVAILABLE = True
except ImportError: except ImportError: