# pip install openai; export BAY_RUN_API_KEY=bay_live_...
import os
from openai import OpenAI

texts = ["Reset my password", "I forgot my password", "Invoice me", "Send a billing invoice"]
client = OpenAI(api_key=os.environ["BAY_RUN_API_KEY"], base_url="https://run.huggingbay.xyz/v1")
response = client.embeddings.create(model="BAAI/bge-small-en-v1.5", input=texts)
vectors = [item.embedding for item in response.data]

def cosine(left, right):
    return sum(a * b for a, b in zip(left, right, strict=True))

clusters = []
for text, vector in zip(texts, vectors, strict=True):
    for cluster in clusters:
        if cosine(vector, cluster["vector"]) >= 0.80:
            cluster["texts"].append(text)
            break
    else:
        clusters.append({"vector": vector, "texts": [text]})

for number, cluster in enumerate(clusters, 1):
    print(f"Cluster {number}: {cluster['texts']}")