From e60468f0d1defc949749037d557be4bd45cb671f Mon Sep 17 00:00:00 2001 From: YangQianli92 <108046369+YangQianli92@users.noreply.github.com> Date: Wed, 17 Apr 2024 10:52:06 +0800 Subject: [PATCH] Update ranker.py --- metagpt/rag/factories/ranker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/metagpt/rag/factories/ranker.py b/metagpt/rag/factories/ranker.py index 61b81ccdc..222930591 100644 --- a/metagpt/rag/factories/ranker.py +++ b/metagpt/rag/factories/ranker.py @@ -49,9 +49,6 @@ class RankerFactory(ConfigBasedFactory): ) return ColbertRerank(**config.model_dump()) - def _create_object_ranker(self, config: ObjectRankerConfig, **kwargs) -> LLMRerank: - return ObjectSortPostprocessor(**config.model_dump()) - def _create_cohere_rerank(self, config: CohereRerankConfig, **kwargs) -> LLMRerank: try: from llama_index.postprocessor.cohere_rerank import CohereRerank @@ -69,6 +66,9 @@ class RankerFactory(ConfigBasedFactory): "`llama-index-postprocessor-flag-embedding-reranker` package not found, please run `pip install llama-index-postprocessor-flag-embedding-reranker`" ) return FlagEmbeddingReranker(**config.model_dump()) + + def _create_object_ranker(self, config: ObjectRankerConfig, **kwargs) -> LLMRerank: + return ObjectSortPostprocessor(**config.model_dump()) def _extract_llm(self, config: BaseRankerConfig = None, **kwargs) -> LLM: return self._val_from_config_or_kwargs("llm", config, **kwargs)