From 3e5ad8f652a82795d17bd244a6b676902516fe9d Mon Sep 17 00:00:00 2001
From: "jinli.yl" <jinli.yl@alibaba-inc.com>
Date: Thu, 27 Jun 2024 14:12:46 +0800
Subject: [PATCH] [dev] rename Reranker to ranker

---
 ...{llama_index_rerank_model.py => llama_index_rank_model.py} | 2 +-
 tests/models/test_models_lli_rerank.py                        | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)
 rename memory_scope/models/{llama_index_rerank_model.py => llama_index_rank_model.py} (97%)

diff --git a/memory_scope/models/llama_index_rerank_model.py b/memory_scope/models/llama_index_rank_model.py
similarity index 97%
rename from memory_scope/models/llama_index_rerank_model.py
rename to memory_scope/models/llama_index_rank_model.py
index c76cbd10..095fda5e 100644
--- a/memory_scope/models/llama_index_rerank_model.py
+++ b/memory_scope/models/llama_index_rank_model.py
@@ -9,7 +9,7 @@ from memory_scope.models.base_model import BaseModel, MODEL_REGISTRY
 from memory_scope.models.model_response import ModelResponse
 
 
-class LlamaIndexRerankModel(BaseModel):
+class LlamaIndexRankModel(BaseModel):
     m_type: ModelEnum = ModelEnum.RANK_MODEL
 
     MODEL_REGISTRY.register("dashscope_rank", DashScopeRerank)
diff --git a/tests/models/test_models_lli_rerank.py b/tests/models/test_models_lli_rerank.py
index c7bdf2f6..1238f7f8 100644
--- a/tests/models/test_models_lli_rerank.py
+++ b/tests/models/test_models_lli_rerank.py
@@ -1,7 +1,7 @@
 import json
 import unittest
 
-from memory_scope.models.llama_index_rerank_model import LlamaIndexRerankModel
+from memory_scope.models.llama_index_rerank_model import LlamaIndexRankModel
 
 
 class TestLLIReRank(unittest.TestCase):
@@ -13,7 +13,7 @@ class TestLLIReRank(unittest.TestCase):
             "model_name": "gte-rerank",
             "clazz": "models.llama_index_rerank_model"
         }
-        self.reranker = LlamaIndexRerankModel(**config)
+        self.reranker = LlamaIndexRankModel(**config)
 
     def test_rerank(self):
         query = "吃啥？"