import os
from dotenv import load_dotenv

load_dotenv()

class Config:
    # API 配置
    OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "xxx")
    OPENAI_BASE_URL = os.getenv("OPENAI_BASE_URL", "https://matrixllm.alipay.com/v1")
    MODEL_NAME = os.getenv("MODEL_NAME", "gpt-5.1")
    EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL", "text-embedding-ada-002")
    
    # 文本处理配置
    MAX_CHUNK_SIZE = 2000  # 每个文本块的最大字符数
    CHUNK_OVERLAP = 200    # 文本块之间的重叠
    
    # 角色提取配置
    MIN_CHARACTER_MENTIONS = 10  # 最小出场次数
    TOP_N_CHARACTERS = 20        # 提取前N个主要角色
    
    # 角色分析配置
    MAX_ANALYSIS_CHUNKS = 10     # 每个角色最多分析的文本块数量
    ANALYSIS_TEMPERATURE = 0.3
    
    # 对话配置
    MAX_HISTORY = 10
    CONVERSATION_TEMPERATURE = 0.8
    MAX_MEMORY_RETRIEVAL = 5     # 从记忆中检索的最大片段数
    
    # 缓存配置
    CACHE_DIR = "cache"
    ENABLE_CACHE = True
    
    # 向量数据库配置
    VECTOR_DB_PATH = "cache/vectordb"