27 lines
967 B
Python
27 lines
967 B
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索引名称配置
|
|
}
|
|
|
|
# 嵌入向量模型
|
|
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" |