From aa68e90fb81f4c9e9e4ee6ae6cb8f229d0b6a732 Mon Sep 17 00:00:00 2001
From: YangQianli92 <108046369+YangQianli92@users.noreply.github.com>
Date: Thu, 18 Apr 2024 11:47:59 +0800
Subject: [PATCH] Delete metagpt/rag/rankers/ranker.py

---
 metagpt/rag/rankers/ranker.py | 77 -----------------------------------
 1 file changed, 77 deletions(-)
 delete mode 100644 metagpt/rag/rankers/ranker.py

diff --git a/metagpt/rag/rankers/ranker.py b/metagpt/rag/rankers/ranker.py
deleted file mode 100644
index b75745a1f..000000000
--- a/metagpt/rag/rankers/ranker.py
+++ /dev/null
@@ -1,77 +0,0 @@
-"""RAG Ranker Factory."""
-
-from llama_index.core.llms import LLM
-from llama_index.core.postprocessor import LLMRerank
-from llama_index.core.postprocessor.types import BaseNodePostprocessor
-
-from metagpt.rag.factories.base import ConfigBasedFactory
-from metagpt.rag.rankers.object_ranker import ObjectSortPostprocessor
-from metagpt.rag.schema import (
-    BaseRankerConfig,
-    ColbertRerankConfig,
-    LLMRankerConfig,
-    ObjectRankerConfig,
-    CohereRerankConfig,
-    BGERerankConfig
-)
-
-
-class RankerFactory(ConfigBasedFactory):
-    """Modify creators for dynamically instance implementation."""
-
-    def __init__(self):
-        creators = {
-            LLMRankerConfig: self._create_llm_ranker,
-            ColbertRerankConfig: self._create_colbert_ranker,
-            ObjectRankerConfig: self._create_object_ranker,
-            CohereRerankConfig: self._create_cohere_rerank,
-            BGERerankConfig: self._create_bge_rerank,
-        }
-        super().__init__(creators)
-
-    def get_rankers(self, configs: list[BaseRankerConfig] = None, **kwargs) -> list[BaseNodePostprocessor]:
-        """Creates and returns a retriever instance based on the provided configurations."""
-        if not configs:
-            return []
-
-        return super().get_instances(configs, **kwargs)
-
-    def _create_llm_ranker(self, config: LLMRankerConfig, **kwargs) -> LLMRerank:
-        config.llm = self._extract_llm(config, **kwargs)
-        return LLMRerank(**config.model_dump())
-
-    def _create_colbert_ranker(self, config: ColbertRerankConfig, **kwargs) -> LLMRerank:
-        try:
-            from llama_index.postprocessor.colbert_rerank import ColbertRerank
-        except ImportError:
-            raise ImportError(
-                "`llama-index-postprocessor-colbert-rerank` package not found, please run `pip install llama-index-postprocessor-colbert-rerank`"
-            )
-        return ColbertRerank(**config.model_dump())
-
-    def _create_cohere_rerank(self, config: CohereRerankConfig, **kwargs) -> LLMRerank:
-        try:
-            from llama_index.postprocessor.cohere_rerank import CohereRerank
-        except ImportError:
-            raise ImportError(
-                "`llama-index-postprocessor-cohere-rerank` package not found, please run `pip install llama-index-postprocessor-cohere-rerank`"
-            )
-        return CohereRerank(**config.model_dump())
-
-    def _create_bge_rerank(self, config: BGERerankConfig, **kwargs) -> LLMRerank:
-        try:
-            from llama_index.postprocessor.flag_embedding_reranker import FlagEmbeddingReranker
-        except ImportError:
-            raise ImportError(
-                "`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)
-
-
-get_rankers = RankerFactory().get_rankers