'commit'

dsSchoolBuddy/ElasticSearch/T2_Vector.py

@@ -1,5 +1,5 @@
 # pip install pydantic requests
-from ElasticSearch.Utils.VectorUtil import text_to_vector_db, query_vector_db
+from ElasticSearch.Utils.VectorDBUtil import VectorDBUtil
 
 
 def main():
@@ -16,12 +16,15 @@ def main():
 
 随着建筑技术的发展，高性能混凝土、自密实混凝土、再生骨料混凝土等新型混凝土不断涌现，为土木工程领域提供了更多的选择。"""
 
+    # 创建工具实例
+    vector_util = VectorDBUtil()
+
     # 调用文本入库功能
-    vector_store, doc_count, split_count = text_to_vector_db(long_text)
+    vector_util.text_to_vector_db(long_text)
 
     # 调用文本查询功能
     query = "混凝土"
-    reranked_results = query_vector_db(vector_store, query, k=4)
+    reranked_results = vector_util.query_vector_db(query, k=4)
 
     # 打印所有查询结果及其可信度
     print("最终查询结果：")