#!/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])