change the query code from
results = collection.query(query_embeddings=[query_embedding], n_results=1)
to
results = collection.query(query_embeddings=[query_embedding], n_results=1, include=["embeddings"])
would fix the problem