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

rename func

Browse source
This commit is contained in:
seehi 2024-03-13 11:17:49 +08:00
parent 0576ab2ed1
commit 33c6f1e17c
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/index.py
View file

@ -23,11 +23,11 @@ class RAGIndexFactory(ConfigFactory):
"""Key is PersistType."""
return super().get_instance(config, **kwargs)
def extract_embed_model(self, config, **kwargs) -> BaseEmbedding:
def _extract_embed_model(self, config, **kwargs) -> BaseEmbedding:
return self._val_from_config_or_kwargs("embed_model", config, **kwargs)
def _create_faiss(self, config: FAISSIndexConfig, **kwargs) -> VectorStoreIndex:
embed_model = self.extract_embed_model(config, **kwargs)
embed_model = self._extract_embed_model(config, **kwargs)
vector_store = FaissVectorStore.from_persist_dir(str(config.persist_path))
storage_context = StorageContext.from_defaults(vector_store=vector_store, persist_dir=config.persist_path)
@ -35,7 +35,7 @@ class RAGIndexFactory(ConfigFactory):
return index
def _create_chroma(self, config: ChromaIndexConfig, **kwargs) -> VectorStoreIndex:
embed_model = self.extract_embed_model(config, **kwargs)
embed_model = self._extract_embed_model(config, **kwargs)
db = chromadb.PersistentClient(str(config.persist_path))
chroma_collection = db.get_or_create_collection(config.collection_name)