OSError / FileNotFoundError

import os
from FlagEmbedding import FlagReranker

# BROKEN: Model name typo + no error handling
reranker = FlagReranker('BAAI/bge-reranker-v2-m3')  # ← This model doesn't exist (missing -en suffix)
pairs = [["query", "passage1"], ["query", "passage2"]]
scores = reranker.compute_score(pairs)
print(scores)

import os
from FlagEmbedding import FlagReranker
from huggingface_hub import login

# Set HuggingFace token if repo is private (optional)
if hf_token := os.environ.get('HF_TOKEN'):
    login(token=hf_token)

# FIXED: Verify model name, add error handling, use fallback
model_names = [
    'BAAI/bge-reranker-v2-m3',  # Primary model (if this doesn't exist, check HF Hub)
    'BAAI/bge-reranker-v2-m3-en',  # Corrected name for English variant
    'BAAI/bge-reranker-large'  # Fallback model
]

reranker = None
for model_name in model_names:
    try:
        print(f"Loading reranker: {model_name}")
        reranker = FlagReranker(
            model_name_or_path=model_name,
            use_fp16=True,
            cache_dir=os.environ.get('HF_HOME', os.path.expanduser('~/.cache/huggingface'))
        )
        print(f"✓ Successfully loaded: {model_name}")
        break
    except OSError as e:
        print(f"✗ Model not found: {model_name}\nError: {e}")
        continue

if reranker is None:
    raise RuntimeError("Could not load any reranker model. Check HF_TOKEN and model names.")

pairs = [["query", "passage1"], ["query", "passage2"]]
scores = reranker.compute_score(pairs)
print(f"Reranker scores: {scores}")