from elasticsearch import Elasticsearch
import warnings
from Config import Config
from Config.Config import ES_CONFIG
import urllib3

# 禁用SSL警告
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

# 1. 初始化Elasticsearch连接
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": [
                {
                    "terms": {
                        "tags.tags": query_tag
                    }
                }
            ]
        }
    },
    "size": 1000
}

# 4. 执行查询并处理结果
try:
    response = es.search(index="knowledge_base", body=query)
    hits = response['hits']['hits']
    
    if not hits:
        print(f"未找到标签为 '{query_tag}' 的数据。")
    else:
        print(f"找到 {len(hits)} 条标签为 '{query_tag}' 的数据：")
        for i, hit in enumerate(hits, 1):
            print(f"{i}. ID: {hit['_id']}")
            print(f"   内容: {hit['_source'].get('user_input', '')}")
            print(f"   标签: {hit['_source'].get('tags', '')}")
            print("-" * 50)
except Exception as e:
    print(f"查询出错: {str(e)}")

# 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}")