'commit'

dsLightRag/ElasticSearch/T6_SelectByVector.py

@@ -0,0 +1,29 @@
+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()