txtai-simple.py

# pip install txtai

from txtai.embeddings import Embeddings
import requests

# Embeddings with sentence-transformers backend
embeddings = Embeddings({"method": "transformers", "path": "sentence-transformers/paraphrase-mpnet-base-v2"})

# Query HN
data = [x["title"] for x in requests.get("https://hn.algolia.com/api/v1/search?tags=front_page").json()["hits"]]

# Get best matches for topic
print([data[uid] for uid, score in embeddings.similarity("programming", data) if score >= 0.3])