from Config import Config
from ElasticSearch.Utils.EsSearchUtil import EsSearchUtil


def main():
    # 初始化搜索工具
    search_util = EsSearchUtil(Config.ES_CONFIG)

    # 输入查询文本
    query = "混凝土"
    print(f"查询文本: {query}")

    # 获取查询向量
    query_embedding = search_util.get_query_embedding(query)
    print(f"查询向量维度: {len(query_embedding)}")

    # 向量搜索
    search_results = search_util.search_by_vector(query_embedding, k=10)
    print(f"向量搜索结果数量: {len(search_results)}")

    # 结果重排
    reranked_results = search_util.rerank_results(query, search_results)

    # 显示结果
    search_util.display_results(reranked_results)


if __name__ == "__main__":
    main()