| """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] | |