kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-06-02 00:57:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(knowledge): exclude root-level files from knowledge stats to preserve empty state

Browse Source
This commit is contained in:
zhayujie
2026-04-16 22:55:46 +08:00
parent 2384bd0e10
commit 426fb88ce7
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
agent/knowledge/service.py
View File

@@ -68,7 +68,7 @@ class KnowledgeService:
return {"tree": [], "stats": {"pages": 0, "size": 0}, "enabled": conf().get("knowledge", True)}
stats = {"pages": 0, "size": 0}
root_files, tree = self._scan_dir(self.knowledge_dir, stats)
root_files, tree = self._scan_dir(self.knowledge_dir, stats, is_root=True)
return {
"root_files": root_files,
@@ -77,7 +77,7 @@ class KnowledgeService:
"enabled": conf().get("knowledge", True),
}
def _scan_dir(self, dir_path: str, stats: dict) -> tuple:
def _scan_dir(self, dir_path: str, stats: dict, is_root: bool = False) -> tuple:
"""
Recursively scan a directory.
@@ -95,6 +95,7 @@ class KnowledgeService:
children.append({"dir": name, "files": sub_files, "children": sub_children})
elif name.endswith(".md"):
size = os.path.getsize(full)
if not is_root:
stats["pages"] += 1
stats["size"] += size
title = name.replace(".md", "")