scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-10-09 03:20:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[dev] rename Reranker to ranker

Browse source
This commit is contained in:
jinli.yl 2024-06-27 14:12:46 +08:00
parent d7b6e20777
commit 3e5ad8f652
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
memory_scope/models/llama_index_rerank_model.py → memory_scope/models/llama_index_rank_model.py
View file

@ -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)

4
tests/models/test_models_lli_rerank.py
View file

@ -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 = "吃啥？"