From 60dd3b0c8c691953ea6d19de192c2bfbc889c1aa Mon Sep 17 00:00:00 2001
From: YangQianli92 <108046369+YangQianli92@users.noreply.github.com>
Date: Thu, 18 Apr 2024 11:48:25 +0800
Subject: [PATCH] Add files via upload

---
 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 222930591..b75745a1f 100644
--- a/metagpt/rag/factories/ranker.py
+++ b/metagpt/rag/factories/ranker.py
@@ -12,7 +12,7 @@ from metagpt.rag.schema import (
     LLMRankerConfig,
     ObjectRankerConfig,
     CohereRerankConfig,
-    FlagEmbeddingConfig
+    BGERerankConfig
 )
 
 
@@ -25,7 +25,7 @@ class RankerFactory(ConfigBasedFactory):
             ColbertRerankConfig: self._create_colbert_ranker,
             ObjectRankerConfig: self._create_object_ranker,
             CohereRerankConfig: self._create_cohere_rerank,
-            FlagEmbeddingConfig: self._create_flag_rerank,
+            BGERerankConfig: self._create_bge_rerank,
         }
         super().__init__(creators)
 
@@ -58,7 +58,7 @@ class RankerFactory(ConfigBasedFactory):
             )
         return CohereRerank(**config.model_dump())
 
-    def _create_flag_rerank(self, config: FlagEmbeddingConfig, **kwargs) -> LLMRerank:
+    def _create_bge_rerank(self, config: BGERerankConfig, **kwargs) -> LLMRerank:
         try:
             from llama_index.postprocessor.flag_embedding_reranker import FlagEmbeddingReranker
         except ImportError: