# -*- coding: utf-8 -*-
"""
智能荐股系统 - 全局配置
所有环境相关配置集中在此，便于迁移与扩展（后期接入真实行情/新闻只需替换数据层）
"""
import os

BASE_DIR = os.path.dirname(os.path.abspath(__file__))
DATA_DIR = os.path.join(BASE_DIR, "data")
LOG_DIR = os.path.join(BASE_DIR, "logs")
STATIC_DIR = os.path.join(BASE_DIR, "static")
TEMPLATE_DIR = os.path.join(BASE_DIR, "templates")
DB_PATH = os.path.join(DATA_DIR, "stock_advisor.db")

os.makedirs(DATA_DIR, exist_ok=True)
os.makedirs(LOG_DIR, exist_ok=True)

# ---------------- 大模型（DeepSeek） ----------------
LLM_BASE_URL = "https://api.deepseek.com"
LLM_API_KEY = "sk-edb9df58ff574f8c98df1cd6a425e97c"
LLM_MODEL = "deepseek-v4-flash"
LLM_TIMEOUT = 150          # 推理模型生成深度报告较慢，放宽超时
LLM_MAX_TOKENS = 4096
LLM_TEMPERATURE = 0.5

# ---------------- Embedding 服务（本地 16011，OpenAI 兼容） ----------------
EMBEDDING_API_URL = "http://121.40.164.32:16011/v1/embeddings"
EMBEDDING_MODEL = "bge-large-zh-v1.5"   # 1024 维
EMBEDDING_DIM = 1024

# ---------------- Rerank 服务（本地 16011，Cohere 兼容，可选增强） ----------------
RERANK_API_URL = "http://121.40.164.32:16011/v1/rerank"
RERANK_MODEL = "bge-reranker-v2-m3"
USE_RERANK = False          # 荐股场景先关闭，纯检索足够

# ---------------- Chroma 向量库（本地 16010） ----------------
CHROMA_HOST = "121.40.164.32"
CHROMA_PORT = 16010
CHROMA_NEWS_COLLECTION = "stock_news_v1"       # 财经新闻语义索引
CHROMA_PROFILE_COLLECTION = "stock_profiles_v1"  # 公司概况语义索引

# ---------------- 服务 ----------------
SERVICE_PORT = 16095
SERVICE_HOST = "0.0.0.0"
SERVICE_NAME = "智能荐股系统"
IS_MOCK = True             # 当前数据为模拟数据（后期接入真实数据后改为 False）

# ---------------- 邮件通知（默认值，可在设置区修改） ----------------
MAIL_DEFAULTS = {
    "smtp_host": "mail.tphai.com",
    "smtp_port": 587,
    "smtp_user": "hz4th_coder@tphai.com",
    "smtp_pass": "hz4th_coder@!",
    "smtp_mode": "plain",           # plain(无加密) / starttls / ssl
    "email_to": "wlq@tphai.com",
    "sender_name": "智能荐股系统",
}

# ---------------- 舆情监控自动化（默认值，可在设置区修改） ----------------
MONITOR_DEFAULTS = {
    "monitor_enabled": "1",
    "monitor_interval": "30",         # 分钟
    "monitor_categories": "",          # 空=全部；逗号分隔如：公司,业绩,机构观点
    "monitor_sentiment": "0.20",       # 情感绝对值权重阈值
    "monitor_importance": "45",        # 重要度阈值(0-100)，>= 则通知
    "monitor_keywords": "回购,中标,减持,问询,停牌,上调,下调,超预期,不及预期,预警,重组,增持,定增,业绩,退市",
}

# ---------------- 持仓跟踪智能体（默认值，可在设置区修改） ----------------
TRACKING_DEFAULTS = {
    "tracking_enabled": "1",
    "tracking_interval": "60",          # 分钟
    "tracking_notify": "1",             # 重大变化时邮件通知
    "tracking_impact_threshold": "65",  # 影响度评分 >= 此值判定为重大变化
}