From 00e4cd9a8076d725c269a0b30e36622dea5e587d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= Date: Fri, 20 Sep 2024 16:52:42 +0800 Subject: [PATCH] feat: +report --- metagpt/tools/libs/index_repo.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/metagpt/tools/libs/index_repo.py b/metagpt/tools/libs/index_repo.py index f72f7ef3e..4c4e6c59b 100644 --- a/metagpt/tools/libs/index_repo.py +++ b/metagpt/tools/libs/index_repo.py @@ -19,6 +19,7 @@ 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 from metagpt.utils.file import File +from metagpt.utils.report import EditorReporter UPLOADS_INDEX_ROOT = "/data/.index/uploads" DEFAULT_INDEX_ROOT = UPLOADS_INDEX_ROOT @@ -89,8 +90,9 @@ class IndexRepo(BaseModel): filenames, excludes = await self._filter(filenames) if not filenames: raise ValueError(f"Unsupported file types: {[str(i) for i in excludes]}") + resource = EditorReporter() for i in filenames: - self.resource.report(str(i), "path") + await resource.async_report(str(i), "path") filter_filenames = set() meta = await self._read_meta() new_files = {}