"""
NER 服务配置
"""
import os
from pydantic_settings import BaseSettings
from typing import Optional, List


class Settings(BaseSettings):
    """应用配置"""
    
    # 服务配置
    app_name: str = "NER Service"
    app_version: str = "1.0.0"
    debug: bool = False
    host: str = "0.0.0.0"
    port: int = 8001
    
    # NER 模型配置
    ner_model: str = "rule"  # rule / spacy / transformers / api
    ner_model_name: Optional[str] = None  # 具体模型名称
    use_gpu: bool = False
    max_text_length: int = 50000
    
    # API 配置（用于 API 模式的后备方案）
    api_base_url: Optional[str] = None
    api_key: Optional[str] = None
    api_model: str = "qwen-plus"
    
    # 实体类型配置
    entity_types: List[str] = [
        "PERSON", "ORG", "LOC", "DATE", "NUMBER", "DEVICE", "TERM", "PROJECT", "COMPANY"
    ]
    
    # 日志配置
    log_level: str = "INFO"
    
    class Config:
        env_file = ".env"
        env_file_encoding = "utf-8"


settings = Settings()