From 2e90230097b6d7476637f21625bd5679d5bfb7d6 Mon Sep 17 00:00:00 2001 From: catlog22 Date: Mon, 5 Jan 2026 23:13:52 +0800 Subject: [PATCH] feat: Update import path for TextCrossEncoder to support fastembed versioning and add fallback for older versions --- codex-lens/src/codexlens/cli/model_manager.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/codex-lens/src/codexlens/cli/model_manager.py b/codex-lens/src/codexlens/cli/model_manager.py index 30b53395..36e6802a 100644 --- a/codex-lens/src/codexlens/cli/model_manager.py +++ b/codex-lens/src/codexlens/cli/model_manager.py @@ -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