Spaces:
Runtime error
Runtime error
File size: 691 Bytes
0f35f83 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
from openai import OpenAI
class OpenAIAda002():
def name(self):
return "OpenAI/text-embedding-ada-002"
def embed(self, text):
client = OpenAI()
resp = client.embeddings.create(
input=[text],
model="text-embedding-ada-002"
)
embedding = resp.data[0].embedding
return embedding
class OpenAI3Large():
def name(self):
return "OpenAI/text-embedding-3-large"
def embed(self, text):
client = OpenAI()
resp = client.embeddings.create(
input=[text],
model="text-embedding-3-large"
)
embedding = resp.data[0].embedding
return embedding |