From 2e3a73b2f9897d2bed729a9ed8891f5bca4654f6 Mon Sep 17 00:00:00 2001 From: YangQianli92 <108046369+YangQianli92@users.noreply.github.com> Date: Tue, 16 Apr 2024 11:37:10 +0800 Subject: [PATCH] Update schema.py --- metagpt/rag/schema.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/metagpt/rag/schema.py b/metagpt/rag/schema.py index 1036c9dad..a60f294b7 100644 --- a/metagpt/rag/schema.py +++ b/metagpt/rag/schema.py @@ -121,13 +121,11 @@ class ColbertRerankConfig(BaseRankerConfig): class CohereRerankConfig(BaseRankerConfig): model: str = Field(default="rerank-english-v3.0") - top_n: int = Field(default=2) api_key: str = Field(default="YOUR_COHERE_API") class FlagEmbeddingConfig(BaseRankerConfig): model: str = Field(default="BAAI/bge-reranker-large", description="BAAI Reranker model name.") - top_n: int = Field(default=2, description="Number of nodes to return sorted by score.") use_fp16: bool = Field(default=True, description="Whether to use fp16 for inference.")