# 阿里云的配置信息【绘智科技】
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"