main
HuangHai 4 weeks ago
parent 8a42084208
commit 058f59b71c

@ -8,7 +8,7 @@ logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)
# 初始化EsSearchUtil
es_search_util = EsSearchUtil(ES_CONFIG)
esClient = EsSearchUtil(ES_CONFIG)
def main():
@ -19,12 +19,12 @@ def main():
print(f"原始查询文本: {query}")
# 执行混合搜索
es_conn = es_search_util.es_pool.get_connection()
es_conn = esClient.es_pool.get_connection()
try:
# 向量搜索
print("\n=== 向量搜索阶段 ===")
print("1. 文本分词和向量化处理中...")
query_embedding = es_search_util.text_to_embedding(query)
query_embedding = esClient.text_to_embedding(query)
print(f"2. 生成的查询向量维度: {len(query_embedding)}")
print(f"3. 前3维向量值: {query_embedding[:3]}")
@ -99,7 +99,7 @@ def main():
# print(f" 详细: {hit['_source']['tags']['full_content']}")
finally:
es_search_util.es_pool.release_connection(es_conn)
esClient.es_pool.release_connection(es_conn)
if __name__ == "__main__":
main()

Loading…
Cancel
Save