add get_available_rerankers function

Author: Zhe Yu

src/vectorcode/subcommands/query/reranker/__init__.py

@@ -43,6 +43,10 @@ class CustomReranker(RerankerBase):
         raise TypeError(error_message)
 
 
+def get_available_rerankers():
+    return list(__supported_rerankers.values())
+
+
 def get_reranker(configs: Config) -> RerankerBase:
     if configs.reranker:
         if hasattr(sys.modules[__name__], configs.reranker):
@@ -56,7 +60,7 @@ def get_reranker(configs: Config) -> RerankerBase:
 
     # TODO: replace the following with an Exception before the release of 0.6.0.
     logger.warning(
-        f""""reranker" option should be set to one of the following: {list(i for i in __all__ if i != "RerankerBase")}.
+        f""""reranker" option should be set to one of the following: {list(i.__name__ for i in get_available_rerankers())}.
 To choose a CrossEncoderReranker model, you can set the "model_name_or_path" key in the "reranker_params" option to the name/path of the model.
 To use NaiveReranker, set the "reranker" option to "NaiveReranker".
 The old configuration syntax will be DEPRECATED in v0.6.0

tests/subcommands/query/test_reranker.py

@@ -7,6 +7,9 @@
     CrossEncoderReranker,
     NaiveReranker,
     RerankerBase,
+    __supported_rerankers,
+    add_reranker,
+    get_available_rerankers,
     get_reranker,
 )
 
@@ -248,41 +251,32 @@ def test_get_reranker():
 
 def test_supported_rerankers_initialization():
     """Test that __supported_rerankers contains the expected default rerankers"""
-    from vectorcode.subcommands.query.reranker import __supported_rerankers
 
-    assert "CrossEncoderReranker" in __supported_rerankers
-    assert "NaiveReranker" in __supported_rerankers
-    assert len(__supported_rerankers) == 2
+    assert isinstance(
+        get_reranker(Config(reranker="CrossEncoderReranker")), CrossEncoderReranker
+    )
+    assert isinstance(get_reranker(Config(reranker="NaiveReranker")), NaiveReranker)
+    assert len(get_available_rerankers()) == 2
 
 
 def test_add_reranker_success():
     """Test successful registration of a new reranker"""
-    from vectorcode.subcommands.query.reranker import (
-        RerankerBase,
-        __supported_rerankers,
-        add_reranker,
-    )
 
-    original_count = len(__supported_rerankers)
+    original_count = len(get_available_rerankers())
 
     @add_reranker
     class TestReranker(RerankerBase):
         def rerank(self, results, query_chunks):
             return []
 
-    assert len(__supported_rerankers) == original_count + 1
+    assert len(get_available_rerankers()) == original_count + 1
     assert "TestReranker" in __supported_rerankers
     assert isinstance(get_reranker(Config(reranker="TestReranker")), TestReranker)
     __supported_rerankers.pop("TestReranker")
 
 
 def test_add_reranker_duplicate():
     """Test duplicate reranker registration raises error"""
-    from vectorcode.subcommands.query.reranker import (
-        RerankerBase,
-        __supported_rerankers,
-        add_reranker,
-    )
 
     # First registration should succeed
     @add_reranker
@@ -298,9 +292,6 @@ def rerank(self, results, query_chunks):
 
 def test_add_reranker_invalid_baseclass():
     """Test that non-RerankerBase classes can't be registered"""
-    from vectorcode.subcommands.query.reranker import (
-        add_reranker,
-    )
 
     with pytest.raises(TypeError):