You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
461 B
24 lines
461 B
from openai import OpenAI
|
|
|
|
client = OpenAI()
|
|
|
|
|
|
|
|
def get_ada_embeddings(text: str, model: str = "text-embedding-ada-002"):
|
|
"""
|
|
Simple function to get embeddings from ada
|
|
|
|
Usage:
|
|
>>> get_ada_embeddings("Hello World")
|
|
>>> get_ada_embeddings("Hello World", model="text-embedding-ada-001")
|
|
|
|
"""
|
|
|
|
|
|
text = text.replace("\n", " ")
|
|
|
|
return client.embeddings.create(input=[text],
|
|
model=model)["data"][
|
|
0
|
|
]["embedding"]
|