Ubuntu
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
添加新知识到知识库 - 使用方法：
|
|
python3 add_knowledge.py "问题" "答案"
|
|
"""
|
|
import sys
|
|
from pathlib import Path
|
|
from datetime import datetime
|
|
|
|
KNOWLEDGE_FILE = Path(__file__).parent / "knowledge_base" / "custom_faq.txt"
|
|
|
|
def add_qa(question: str, answer: str):
|
|
"""添加Q&A到知识库"""
|
|
|
|
# 确保文件存在
|
|
if not KNOWLEDGE_FILE.exists():
|
|
KNOWLEDGE_FILE.write_text("# 自定义知识库\n\n", encoding="utf-8")
|
|
|
|
# 格式化内容
|
|
content = f"\nQ: {question}\nA: {answer}\n"
|
|
|
|
# 追加到文件
|
|
with open(KNOWLEDGE_FILE, "a", encoding="utf-8") as f:
|
|
f.write(content)
|
|
|
|
print(f"✅ 已添加到知识库: {KNOWLEDGE_FILE}")
|
|
print(f" Q: {question}")
|
|
print(f" A: {answer[:50]}...")
|
|
print("\n⚠️ 请重启服务使新知识生效！")
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) != 3:
|
|
print("用法: python3 add_knowledge.py '问题' '答案'")
|
|
print("示例: python3 add_knowledge.py '如何退货' '7天内可申请退货...'")
|
|
sys.exit(1)
|
|
|
|
add_qa(sys.argv[1], sys.argv[2])
|