test_similar.py 303 B

12345678
  1. from agent.core.es import bm25_vector_search
  2. from agent.core.vector import get_embeddings
  3. phrase = "职业:海鲜批发;投向:海鲜批发;用途:海鲜批发"
  4. phrase = "海鲜批发"
  5. embeddings = get_embeddings([phrase])[0]
  6. print(bm25_vector_search(phrase, embeddings,rrf_score_threshold=0.015))