File size: 464 Bytes
71d239c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | """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]
|