"""Turn text into embeddings (numbers a computer can compare by meaning)."""
from openai import OpenAI

import config

client = OpenAI(api_key=config.OPENAI_API_KEY)


def embed_text(text):
    """Return the embedding vector for a single piece of text."""
    resp = client.embeddings.create(model=config.EMBED_MODEL, input=text)
    return resp.data[0].embedding


def embed_many(texts):
    """Embed a list of texts."""
    return [embed_text(t) for t in texts]