apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-15 11:02:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
YangQianli92 2024-04-18 11:48:25 +08:00 • committed by GitHub
parent aa68e90fb8
commit 60dd3b0c8c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/rag/factories/ranker.py
View file

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