fix(service): update

InternLM · Jan 11, 2024 · 4d57258 · 4d57258
1 parent 57c8ff8
commit 4d57258
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/service/feature_store.py b/service/feature_store.py
@@ -326,10 +326,9 @@ def query_source(self, question: str):
     def preprocess(self, repo_dir: str, work_dir: str):
         markdown_dir = os.path.join(work_dir, 'preprocess')
         if os.path.exists(markdown_dir):
-            logger.warning(f'{markdown_dir} already exists, skip')
-            return markdown_dir
-        else:
-            os.makedirs(markdown_dir)
+            logger.warning(f'{markdown_dir} already exists, remove and regenerate.')
+            shutil.rmtree(markdown_dir)
+        os.makedirs(markdown_dir)
 
         # find all .md files except those containing mdb
         mds = []