from common_logging import get_logger

logger = get_logger(__name__)

KB_NAME = '税务知识库'
KB_CODE = 'tax_knowledge'
KB_DESCRIPTION = '中国税务法律法规、部门规章、规范性文件知识库，支持税务 RAG 检索'
CATEGORY_NAMES: dict[int, str] = {1: '法律', 2: '行政法规', 3: '国务院文件', 4: '税务部门规章', 5: '财税文件', 6: '税务规范性文件', 7: '其他文件', 8: '工作通知'}
DOC_TYPE_MAP: dict[int, str] = {1: 'law', 2: 'regulation', 3: 'notice', 4: 'regulation', 5: 'notice', 6: 'notice', 7: 'notice', 8: 'notice'}
BATCH_SIZE = 50
VECTORIZE_BATCH_SIZE = 100
AUTOTAG_BATCH_SIZE = 50
REQUEST_DELAY = 0.2