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

9
agent/knowledge/service.py
View File

@@ -68,7 +68,7 @@ class KnowledgeService:
return {"tree": [], "stats": {"pages": 0, "size": 0}, "enabled": conf().get("knowledge", True)} return {"tree": [], "stats": {"pages": 0, "size": 0}, "enabled": conf().get("knowledge", True)}
stats = {"pages": 0, "size": 0} 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 { return {
"root_files": root_files, "root_files": root_files,
@@ -77,7 +77,7 @@ class KnowledgeService:
"enabled": conf().get("knowledge", True), "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. Recursively scan a directory.
@@ -95,8 +95,9 @@ class KnowledgeService:
children.append({"dir": name, "files": sub_files, "children": sub_children}) children.append({"dir": name, "files": sub_files, "children": sub_children})
elif name.endswith(".md"): elif name.endswith(".md"):
size = os.path.getsize(full) size = os.path.getsize(full)
stats["pages"] += 1 if not is_root:
stats["size"] += size stats["pages"] += 1
stats["size"] += size
title = name.replace(".md", "") title = name.replace(".md", "")
try: try:
with open(full, "r", encoding="utf-8") as f: with open(full, "r", encoding="utf-8") as f: