47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
|
import warnings
|
|||
|
|
|
|||
|
|
from elasticsearch import Elasticsearch
|
|||
|
|
|
|||
|
|
from Config import Config
|
|||
|
|
|
|||
|
|
# 抑制HTTPS相关警告
|
|||
|
|
warnings.filterwarnings('ignore', message='Connecting to .* using TLS with verify_certs=False is insecure')
|
|||
|
|
warnings.filterwarnings('ignore', message='Unverified HTTPS request is being made to host')
|
|||
|
|
|
|||
|
|
# 初始化ES连接
|
|||
|
|
es = Elasticsearch(
|
|||
|
|
hosts=Config.ES_CONFIG['hosts'],
|
|||
|
|
basic_auth=Config.ES_CONFIG['basic_auth'],
|
|||
|
|
verify_certs=False
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 定义mapping结构
|
|||
|
|
mapping = {
|
|||
|
|
"mappings": {
|
|||
|
|
"properties": {
|
|||
|
|
"embedding": {
|
|||
|
|
"type": "dense_vector",
|
|||
|
|
"dims": 200, # embedding维度为200
|
|||
|
|
"index": True,
|
|||
|
|
"similarity": "l2_norm" # 使用L2距离
|
|||
|
|
},
|
|||
|
|
"user_input": {"type": "text"},
|
|||
|
|
"tags": {
|
|||
|
|
"type": "object",
|
|||
|
|
"properties": {
|
|||
|
|
"tags": {"type": "keyword"},
|
|||
|
|
"full_content": {"type": "text"}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 创建索引
|
|||
|
|
index_name = Config.ES_CONFIG['index_name']
|
|||
|
|
if es.indices.exists(index=index_name):
|
|||
|
|
es.indices.delete(index=index_name)
|
|||
|
|
print(f"删除已存在的索引 '{index_name}'")
|
|||
|
|
|
|||
|
|
es.indices.create(index=index_name, body=mapping)
|
|||
|
|
print(f"索引 '{index_name}' 创建成功，mapping结构已设置。")
|