refactor: add BaseReranker

Author: sigridjineth

retriever-simple-benchmark/module/reranker/flag_reranker.py

@@ -1,8 +1,9 @@
 import numpy as np
 from FlagEmbedding import FlagReranker
+from .base import BaseReranker
 
 
-class FlagRerankerType:
+class FlagRerankerType(BaseReranker):
     """
     A multi-GPU capable Reranker based on the FlagEmbedding library.
     Uses FlagAutoReranker.from_finetuned(...) under the hood.

retriever-simple-benchmark/module/reranker/huggingface_reranker.py

@@ -1,9 +1,10 @@
 import torch
 import numpy as np
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from .base import BaseReranker
 
 
-class HuggingFaceReranker:
+class HuggingFaceReranker(BaseReranker):
     def __init__(self, model_path, use_fp16=True, max_length=512):
         self.tokenizer = AutoTokenizer.from_pretrained(
             model_path, trust_remote_code=True