29 lines
790 B
Python
29 lines
790 B
Python
|
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()
|