diff --git a/yt_fts/embeddings.py b/yt_fts/embeddings.py index b2d1505..cbeb4a1 100644 --- a/yt_fts/embeddings.py +++ b/yt_fts/embeddings.py @@ -39,7 +39,9 @@ def add_embeddings_to_chroma(subs, openai_client): -def get_embedding(text, model="text-embedding-ada-002", client=OpenAI()): +def get_embedding(text, model="text-embedding-ada-002", client=None): + if client is None: + client = OpenAI() text = text.replace("\n", " ") return client.embeddings.create(input = [text], model=model).data[0].embedding