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

fixbug: circular import

Browse source
This commit is contained in:
莘权 马 2024-09-02 16:24:20 +08:00
parent 0a1d0e720d
commit 1976142c28
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
metagpt/tools/libs/index_repo.py
View file

@ -11,7 +11,6 @@ from pydantic import BaseModel, Field, model_validator
from metagpt.config2 import Config
from metagpt.logs import logger
from metagpt.rag.engines import SimpleEngine
from metagpt.rag.factories.embedding import RAGEmbeddingFactory
from metagpt.rag.schema import FAISSIndexConfig, FAISSRetrieverConfig, LLMRankerConfig
from metagpt.utils.common import aread, awrite, generate_fingerprint, list_files
@ -100,6 +99,8 @@ class IndexRepo(BaseModel):
await self._add_batch(filenames=filter_filenames, delete_filenames=delete_filenames)
async def _add_batch(self, filenames: List[Union[str, Path]], delete_filenames: List[Union[str, Path]]):
from metagpt.rag.engines import SimpleEngine
if not filenames:
return
engine = None
@ -160,6 +161,8 @@ class IndexRepo(BaseModel):
return pathnames, excludes
async def _search(self, query: str, filters: Set[str]) -> List[NodeWithScore]:
from metagpt.rag.engines import SimpleEngine
if not Path(self.filename).exists():
return []
engine = SimpleEngine.from_index(