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

---
 metagpt/rag/schema.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/metagpt/rag/schema.py b/metagpt/rag/schema.py
index a60f294b7..e7b2e5ce9 100644
--- a/metagpt/rag/schema.py
+++ b/metagpt/rag/schema.py
@@ -124,7 +124,7 @@ class CohereRerankConfig(BaseRankerConfig):
     api_key: str = Field(default="YOUR_COHERE_API")
 
 
-class FlagEmbeddingConfig(BaseRankerConfig):
+class BGERerankConfig(BaseRankerConfig):
     model: str = Field(default="BAAI/bge-reranker-large", description="BAAI Reranker model name.")
     use_fp16: bool = Field(default=True, description="Whether to use fp16 for inference.")