config.py 1.2 KB

1234567891011121314151617181920212223242526272829
  1. """
  2. 配置文件
  3. """
  4. import os
  5. from dotenv import load_dotenv
  6. load_dotenv()
  7. # OpenAI API配置
  8. OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "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")
  9. OPENAI_BASE_URL = os.getenv("OPENAI_BASE_URL", "https://api.openai.com/v1")
  10. OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-5.1")
  11. # 知识点匹配配置
  12. CONFIDENCE_THRESHOLD = float(os.getenv("CONFIDENCE_THRESHOLD", "0.7")) # 置信度阈值,低于此值允许生成参考知识点
  13. # 向量数据库配置
  14. # 向量数据库文件路径(可以是本地路径或网络共享路径)
  15. # 例如:本地路径 "knowledge_points.index" 或网络路径 "\\\\server\\share\\knowledge_points.index"
  16. FAISS_INDEX_PATH = os.getenv("FAISS_INDEX_PATH", "knowledge_points.index")
  17. FAISS_METADATA_PATH = os.getenv("FAISS_METADATA_PATH", "knowledge_points_metadata.pkl")
  18. # 数据库配置
  19. DB_HOST = os.getenv("DB_HOST", "rm-f8ze60yirdj8786u2wo.mysql.rds.aliyuncs.com")
  20. DB_PORT = int(os.getenv("DB_PORT", "3306"))
  21. DB_DATABASE = os.getenv("DB_DATABASE", "math-conten-online2")
  22. DB_USERNAME = os.getenv("DB_USERNAME", "root")
  23. DB_PASSWORD = os.getenv("DB_PASSWORD", "csqz@20255")