From cd8bd2821668c3949ab737c182e927e2e6b1d841 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= Date: Wed, 4 Sep 2024 11:03:43 +0800 Subject: [PATCH] feat: model: Optional[str] = None --- metagpt/tools/libs/index_repo.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/metagpt/tools/libs/index_repo.py b/metagpt/tools/libs/index_repo.py index a3efa6d27..1501b8e28 100644 --- a/metagpt/tools/libs/index_repo.py +++ b/metagpt/tools/libs/index_repo.py @@ -29,7 +29,7 @@ class IndexRepo(BaseModel): filename: str root_path: str fingerprint_filename: str = "fingerprint.json" - model: str = "text-embedding-ada-002" + model: Optional[str] = None min_token_count: int = 5000 max_token_count: int = 100000 recall_count: int = 5 @@ -95,7 +95,8 @@ class IndexRepo(BaseModel): """ if not self.embedding: config = Config.default() - config.embedding.model = self.model + if self.model: + config.embedding.model = self.model factory = RAGEmbeddingFactory(config) self.embedding = factory.get_rag_embedding()