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

Update schema.py

Browse source
This commit is contained in:
YangQianli92 2024-04-16 11:37:10 +08:00 • committed by GitHub
parent 211ba3dce1
commit 2e3a73b2f9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 0 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
metagpt/rag/schema.py
View file

@ -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.")