mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-06 08:16:00 +00:00
26 lines
719 B
Python
26 lines
719 B
Python
import json
|
|
import unittest
|
|
|
|
from memory_scope.models.llama_index_rerank_model import LlamaIndexRerankModel
|
|
|
|
|
|
class TestLLIReRank(unittest.TestCase):
|
|
"""Tests for LlamaIndexRerankModel"""
|
|
|
|
def setUp(self):
|
|
config = {
|
|
"method_type": "DashScopeRerank",
|
|
"model_name": "gte-rerank",
|
|
"clazz": "models.llama_index_rerank_model"
|
|
}
|
|
self.reranker = LlamaIndexRerankModel(**config)
|
|
|
|
def test_rerank(self):
|
|
query = "吃啥？"
|
|
documents = ["您吃了吗？",
|
|
"吃了吗您？"]
|
|
result = self.reranker.call(
|
|
stream=False,
|
|
documents=documents,
|
|
query=query)
|
|
print(result)
|