Update mapper.py

mapper.py

@@ -4,12 +4,12 @@ class Mapper:
   def __init__(self, repo: str, model: str):
     self.__model = SentenceTransformer(f"{repo}/{model}")
 
-  def __call__(self, query: str, *data):
+  def __call__(self, query: str, data: list[str]):
     query_emb = self.__model.encode(query)
     data_emb = self.__model.encode(data)
 
-    scores = util.dot_score(query_emb, data_emb)[0].cpu().tolist()
-    data_score_pairs = list(zip(data, scores))
+    scores: list[float] = util.dot_score(query_emb, data_emb)[0].cpu().tolist()
+    data_score_pairs = list(zip(range(0, len(data)), scores))
 
     return sorted(
       data_score_pairs,