from agent.core.es import bm25_vector_search
from agent.core.vector import get_embeddings

phrase = "渔业产品批发,海带边周转"
phrase_vector = get_embeddings([phrase])[0]
r = bm25_vector_search(phrase,phrase_vector)
print(r)
# from openpyxl import load_workbook
# workbook = load_workbook('data/样本数据标注后-完整版.xlsx', data_only=True)
# sheet = workbook['核对结果'] 
# for row in sheet.iter_rows(min_row=2,min_col=2, max_col=3, values_only=True):
#     touxiang = row[0] # B列的值
#     yongtu = row[1] # C列的值
#     phrase = f"投向:{touxiang},用途:{yongtu}"
#     phrase_vector = get_embeddings([phrase])[0]
#     r = bm25_vector_search(phrase,phrase_vector)
#     print(f"输入:{phrase}")
#     print(f"输出:{r}")
# workbook.close()