智能荐股系统 v1.0.0:股票池/行情/新闻RAG/机构持仓/多因子评分/AI深度研报
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
# -*- 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)
|
||||
Reference in New Issue
Block a user