fix wechat

This commit is contained in:
yichuan520030910320
2025-08-03 23:02:06 -07:00
parent 58556ef44c
commit 85277ba67a
4 changed files with 6 additions and 3 deletions

View File

@@ -192,7 +192,7 @@ All RAG examples share these common parameters. **Interactive mode** is availabl
# Search Parameters
--top-k N # Number of results to retrieve (default: 20)
--search-complexity N # Search complexity for graph traversal (default: 64)
--search-complexity N # Search complexity for graph traversal (default: 32)
# Chunking Parameters
--chunk-size N # Size of text chunks (default varies by source: 256 for most, 192 for WeChat)

View File

@@ -18,6 +18,7 @@ class EmailRAG(BaseRAGExample):
def __init__(self):
# Set default values BEFORE calling super().__init__
self.max_items_default = -1 # Process all emails by default
self.embedding_model_default = (
"sentence-transformers/all-MiniLM-L6-v2" # Fast 384-dim model
)

View File

@@ -19,7 +19,7 @@ class WeChatRAG(BaseRAGExample):
def __init__(self):
# Set default values BEFORE calling super().__init__
self.max_items_default = 50 # Match original default
self.max_items_default = -1 # Match original default
self.embedding_model_default = (
"sentence-transformers/all-MiniLM-L6-v2" # Fast 384-dim model
)
@@ -143,6 +143,7 @@ class WeChatRAG(BaseRAGExample):
return []
print(f"\nTotal loaded {len(all_documents)} chat documents from {len(export_dirs)} exports")
print("now starting to split into text chunks ... take some time")
# Convert to text chunks with contact information
all_texts = []

View File

@@ -463,8 +463,9 @@ class LeannSearcher:
print(
f"Leann metadata file not found at {self.meta_path_str}, and you may need to rm -rf {parent_dir}"
)
# highlight in red the filenotfound error
raise FileNotFoundError(
f"Leann metadata file not found at {self.meta_path_str}, you may need to rm -rf {parent_dir}"
f"Leann metadata file not found at {self.meta_path_str}, \033[91m you may need to rm -rf {parent_dir}\033[0m"
)
with open(self.meta_path_str, encoding="utf-8") as f:
self.meta_data = json.load(f)