menchuangfallback.py 331 B

1234567
  1. def menchuangfallback(A, B, aiclient, qwclient, menchuang_collection, model):
  2. sentence=["特征描述:" + B['mc'] + "\n" + B['tz']]
  3. embeddings = model.encode(sentence)
  4. result = menchuang_collection.query(query_embeddings=embeddings, n_results=25)
  5. print(result['documents'][0])
  6. print(result['distances'][0])