test_bm25.py 367 B

12345678910
  1. from agent.api_outter import vector_similarity_search
  2. import agent.core.dao as dao
  3. phrase = "职业:民宿服务; 投向:民宿服务; 用途:经营民宿"
  4. r = vector_similarity_search(phrase)
  5. print(r)
  6. tags = dao.query_dict(""" select id,tag_nm as tag_name,tag_code, tag_path,category_id,tag_prompt from aitag_tag_info where id in %s """, (tuple(r),))
  7. print(tags)