logger for reranker.

Author: Zhe Yu

src/vectorcode/subcommands/query/reranker.py

@@ -1,4 +1,5 @@
 import heapq
+import logging
 from abc import abstractmethod
 from collections import defaultdict
 from typing import Any, DefaultDict
@@ -8,6 +9,8 @@
 
 from vectorcode.cli_utils import Config, QueryInclude
 
+logger = logging.getLogger(name=__name__)
+
 
 class RerankerBase:
     def __init__(self, configs: Config, **kwargs: Any):
@@ -42,7 +45,7 @@ def rerank(self, results: QueryResult) -> list[str]:
                     # so that vectorcode doesn't break on old collections.
                     continue
                 documents[identifier].append(distance)
-
+        logger.debug("Document scores: %s", documents)
         top_k = int(numpy.mean(tuple(len(i) for i in documents.values())))
         for key in documents.keys():
             documents[key] = heapq.nsmallest(top_k, documents[key])
@@ -80,7 +83,7 @@ def rerank(self, results: QueryResult) -> list[str]:
                     documents[chunk_metas[rank["corpus_id"]]["path"]].append(
                         float(rank["score"])
                     )
-
+        logger.debug("Document scores: %s", documents)
         top_k = int(numpy.mean(tuple(len(i) for i in documents.values())))
         for key in documents.keys():
             documents[key] = heapq.nlargest(top_k, documents[key])