27 lines
889 B
Python
27 lines
889 B
Python
# 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" |