"""
AI 客服系统配置
"""
import os
from pathlib import Path

# 项目根目录
BASE_DIR = Path(__file__).parent.parent

# API 配置
# 使用 DeepSeek API（推荐，便宜好用）
LLM_API_KEY = os.getenv("DEEPSEEK_API_KEY", "sk-8cb88f614b5b4def882d0ac46025cdcc")
LLM_BASE_URL = os.getenv("LLM_BASE_URL", "https://api.deepseek.com")
LLM_MODEL = os.getenv("LLM_MODEL", "deepseek-chat")

# 也可以使用其他兼容 OpenAI 的 API
# LLM_BASE_URL = "https://api.openai.com"
# LLM_MODEL = "gpt-3.5-turbo"

# 服务配置
HOST = "0.0.0.0"
PORT = 8890

# 知识库配置
KNOWLEDGE_DIR = BASE_DIR / "knowledge_base"
CHROMA_DB_DIR = BASE_DIR / "data" / "chroma_db"

# 向量模型（使用 HuggingFace 免费模型，本地 CPU 运行）
# 使用 BAAI/bge-small-zh-v1.5 轻量级中文模型，无需 API Key

# 检索配置
RETRIEVAL_TOP_K = 3  # 检索 top-k 个相关文档
CHUNK_SIZE = 500     # 文档分块大小
CHUNK_OVERLAP = 50   # 分块重叠大小

# 客服人设
SYSTEM_PROMPT = """你是「小助手」AI客服，专门为用户提供友好、专业的咨询服务。

【角色设定】
- 你是某公司的智能客服助手
- 你的回答应该亲切、简洁、有帮助
- 如果不确定答案，请诚实地告诉用户你会转接人工客服

【回答原则】
1. 基于提供的知识库内容回答
2. 如果知识库中没有相关信息，请说："抱歉，这个问题我暂时无法回答，为您转接人工客服..."
3. 不要编造答案
4. 保持礼貌和耐心
"""