You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.4 KiB
49 lines
1.4 KiB
1 month ago
|
from elasticsearch import Elasticsearch
|
||
|
|
import warnings
|
||
|
|
from Config import Config
|
||
|
|
from Config.Config import ES_CONFIG
|
||
|
|
|
||
|
|
# 1. 初始化Elasticsearch连接
|
||
|
|
warnings.filterwarnings('ignore', category=DeprecationWarning)
|
||
|
|
es = Elasticsearch(
|
||
|
|
hosts=Config.ES_CONFIG['hosts'],
|
||
|
|
basic_auth=Config.ES_CONFIG['basic_auth'],
|
||
|
|
verify_certs=False
|
||
|
|
)
|
||
|
|
|
||
|
|
# 2. 直接在代码中指定要查询的标签
|
||
|
|
query_tag = "MATH_DATA_1" # 可以修改为其他需要的标签
|
||
|
|
|
||
|
|
# 3. 构建查询条件
|
||
|
|
query = {
|
||
|
|
"query": {
|
||
|
|
"bool": {
|
||
|
|
"must": [
|
||
|
|
{
|
||
|
|
"term": {
|
||
|
|
"tags": query_tag
|
||
|
|
}
|
||
|
|
}
|
||
|
|
]
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"size": 1000
|
||
|
|
}
|
||
|
|
|
||
|
|
# 4. 执行查询
|
||
|
|
try:
|
||
|
|
results = es.search(index=ES_CONFIG['index_name'], body=query)
|
||
|
|
print(f"查询标签 '{query_tag}' 结果：")
|
||
|
|
if results['hits']['hits']:
|
||
|
|
for hit in results['hits']['hits']:
|
||
|
|
doc = hit['_source']
|
||
|
|
print(f"ID: {hit['_id']}")
|
||
|
|
print(f"标签: {doc['tags']['tags']}")
|
||
|
|
print(f"用户问题: {doc['user_input']}")
|
||
|
|
print(f"时间: {doc['timestamp']}")
|
||
|
|
print(f"向量: {doc['embedding'][:5]}...")
|
||
|
|
print("-" * 40)
|
||
|
|
else:
|
||
|
|
print(f"未找到标签为 '{query_tag}' 的数据。")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"查询失败: {e}")
|