Apppublic

DialogueRobust/RobustDialogueDemo

sourceHugging Faceccupdated 11mo agoView on Hugging Face

0likes

embeddings.py13 linesDownload Raw Back to root

1# embeddings.py2import os3from openai import OpenAI4 5client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])6 7def get_embedding(text: str, model="text-embedding-3-small"):8    response = client.embeddings.create(9        model=model,10        input=text11    )12    return response.data[0].embedding13