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,10 +13,16 @@ except ImportError:
FASTEMBED_AVAILABLE = False
try:
from fastembed import TextCrossEncoder
# fastembed >= 0.4.0 moved TextCrossEncoder to rerank.cross_encoder
from fastembed.rerank.cross_encoder import TextCrossEncoder
RERANKER_AVAILABLE = True
except ImportError:
RERANKER_AVAILABLE = False
try:
# Fallback for older versions
from fastembed import TextCrossEncoder
RERANKER_AVAILABLE = True
except ImportError:
RERANKER_AVAILABLE = False
# Reranker model profiles with metadata