# 嵌入向量模型
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'

# 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"
# 阿里云API信息【绘智科技】
ALY_LLM_API_KEY = "sk-01d13a39e09844038322108ecdbd1bbc"
ALY_LLM_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
# 通义千问
# ALY_LLM_MODEL_NAME = "qwen-plus"
# Kimi K2大模型
ALY_LLM_MODEL_NAME = "Moonshot-Kimi-K2-Instruct"

# 智谱的API KEY【吴缤申请个人版免费】
ZHIPU_API_KEY = "78dc1dfe37e04f29bd4ca9a49858a969.gn7TIZTfzpY35nx9"

# GPTNB的API KEY
GPTNB_API_KEY = "sk-amQHwiEzPIZIB2KuF5A10dC23a0e4b02B48a7a2b6aFa0662"
GPTNB_BASE_URL = "https://goapi.gptnb.ai"