Dockerfile model load and import fix

Dockerfile

@@ -11,8 +11,15 @@ COPY requirements-prod.txt .
 RUN pip install --no-cache-dir -r requirements-prod.txt
 
 RUN python -c "\
-from fastembed import TextEmbedding; \
+from fastembed import TextEmbedding, SparseTextEmbedding; \
 TextEmbedding(model_name='intfloat/multilingual-e5-large', cache_dir='./model_cache'); \
+SparseTextEmbedding(model_name='prithivida/Splade_PP_en_v1', cache_dir='./model_cache'); \
+"
+
+RUN python -c "\
+try: from fastembed.rerank.cross_encoder import TextCrossEncoder; \
+except ImportError: from fastembed import TextCrossEncoder; \
+TextCrossEncoder(model_name='BAAI/bge-reranker-v2-m3', cache_dir='./model_cache'); \
 "
 
 COPY . .

src/clients/__init__.py

@@ -3,7 +3,6 @@ from src.clients.embedding_client import EmbeddingClient
 from src.clients.sparse_embedding_client import SparseEmbeddingClient
 from src.clients.qdrant import QdrantClientManager
 from src.clients.redis import RedisClient
-from src.clients.reranker_client import RerankerClient
 
 __all__ = [
     "LLMClient",
@@ -11,5 +10,4 @@ __all__ = [
     "SparseEmbeddingClient",
     "QdrantClientManager",
     "RedisClient",
-    "RerankerClient",
 ]

src/clients/reranker_client.py

@@ -1,9 +1,14 @@
-from fastembed import TextCrossEncoder
 from typing import Optional
 import logging
 
 logger = logging.getLogger(__name__)
 
+# fastembed changed import paths across versions
+try:
+    from fastembed.rerank.cross_encoder import TextCrossEncoder
+except ImportError:
+    from fastembed import TextCrossEncoder
+
 
 class RerankerClient:
     DEFAULT_MODEL = "BAAI/bge-reranker-v2-m3"