Ubuntu
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
"""
|
|
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. 保持礼貌和耐心
|
|
"""
|