Files
dsProject/dsLightRag/Config/Config.py
2025-08-22 09:44:05 +08:00

85 lines
2.9 KiB
Python

# Elasticsearch配置
ES_CONFIG = {
'hosts': ['https://localhost:9200'],
'basic_auth': ('elastic', 'jv9h8uwRrRxmDi1dq6u8'),
'verify_certs': False,
'index_name': 'ds_db', # 默认索引名称
'student_info_index': 'student_info' # 添加student_info索引名称配置
}
# 阿里云的配置信息【绘智科技】
ALY_AK = 'LTAI5tE4tgpGcKWhbZg6C4bh'
ALY_SK = 'oizcTOZ8izbGUouboC00RcmGE8vBQ1'
ALY_ENDPOINT = "https://oss-cn-beijing.aliyuncs.com"
ALY_REGION = "cn-beijing"
ALY_BUCKET = "hzkc"
ALY_OSS_PREFIX = "HuangHai"
# DeepSeek大模型 【DeepSeek深度求索官方】训练时用这个
LLM_API_KEY = "sk-44ae895eeb614aa1a9c6460579e322f1"
LLM_BASE_URL = "https://api.deepseek.com"
LLM_MODEL_NAME = "deepseek-chat"
# LLM_MODEL_NAME = "deepseek-reasoner"
# 嵌入向量模型
EMBED_MODEL_NAME = "BAAI/bge-m3"
EMBED_API_KEY = "sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl"
EMBED_BASE_URL = "https://api.siliconflow.cn/v1"
EMBED_DIM = 1024
EMBED_MAX_TOKEN_SIZE = 8192
# 重排模型
RERANK_MODEL = 'BAAI/bge-reranker-v2-m3'
RERANK_BASE_URL = 'https://api.siliconflow.cn/v1/rerank'
RERANK_BINDING_API_KEY = 'sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl'
# 图数据库
NEO4J_URI = "bolt://localhost:7687"
NEO4J_USERNAME = "neo4j"
NEO4J_PASSWORD = "DsideaL147258369"
NEO4J_AUTH = (NEO4J_USERNAME, NEO4J_PASSWORD)
# POSTGRESQL配置信息
POSTGRES_HOST = "10.10.14.208"
POSTGRES_PORT = 5432
POSTGRES_USER = "postgres"
POSTGRES_PASSWORD = "postgres"
POSTGRES_DATABASE = "rag"
# GLM
GLM_BASE_URL = "https://api.siliconflow.cn/v1/chat/completions"
GLM_API_KEY = "sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl"
GLM_MODEL_NAME = "THUDM/GLM-4.1V-9B-Thinking"
# 阿里云API信息【YLT】
ALY_LLM_API_KEY = "sk-f6da0c787eff4b0389e4ad03a35a911f"
ALY_LLM_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
#ALY_LLM_MODEL_NAME = "deepseek-r1"
ALY_LLM_MODEL_NAME = "qwen-plus"
# ALY_LLM_MODEL_NAME = "deepseek-v3"
# 华为云云存储
OBS_AK = "WAFBGJACKDOQZDH1MKZ1"
OBS_SK = "dlWTUbqgCICaYJG3n0Rot4jXaen2HnfFtMVxiPEo"
OBS_SERVER = "obs.cn-north-1.myhuaweicloud.com"
OBS_BUCKET = "dsideal"
OBS_PREFIX = "HuangHai"
# 智谱的API KEY【吴缤申请个人版免费】
ZHIPU_API_KEY = "78dc1dfe37e04f29bd4ca9a49858a969.gn7TIZTfzpY35nx9"
# GPTNB的API KEY
GPTNB_API_KEY = "sk-amQHwiEzPIZIB2KuF5A10dC23a0e4b02B48a7a2b6aFa0662"
# JWT配置信息
JWT_SECRET_KEY = "ZXZnZWVr5b+r5LmQ5L2g55qE5Ye66KGM"
ALGORITHM = "HS256"
ACCESS_TOKEN_EXPIRE_MINUTES = 300000 # 访问令牌过期时间(分钟)
# 火山引擎相关信息
# 语音识别 https://console.volcengine.com/speech/service/10012?AppID=1513532196
HS_APP_ID= "1513532196"
HS_ACCESS_TOKEN= "ti2NW-kbjIbkzc6BK55actnI_wTR8xkH"
HS_CLUSTER_ID="volcano_tts"
HS_VOICE_TYPE = "zh_female_wanwanxiaohe_moon_bigtts" # 中文女声 湾湾小何