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

@@ -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 = []