Spaces:
Sleeping
Sleeping
File size: 295 Bytes
e490ab5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
# embeddings.py
import os
from openai import OpenAI
client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
def get_embedding(text: str, model="text-embedding-3-small"):
response = client.embeddings.create(
model=model,
input=text
)
return response.data[0].embedding
|