# Elasticsearch配置
ES_CONFIG = {
    "hosts": "https://127.0.0.1:9200",
    "basic_auth": ("elastic", "jv9h8uwRrRxmDi1dq6u8"),
    "verify_certs": False,
    "ssl_show_warn": False,
    "index_name": "ds_kb"
}

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

# 阿里云API信息【HZKJ】
ALY_LLM_API_KEY = "sk-01d13a39e09844038322108ecdbd1bbc"
ALY_LLM_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
ALY_LLM_MODEL_NAME = "qwen-plus"
#ALY_LLM_MODEL_NAME = "deepseek-r1"
# ALY_LLM_MODEL_NAME = "deepseek-v3"