embed all at once

emb_sim.py

@@ -14,14 +14,10 @@ openai_api_key = os.getenv("OPENAI_API_KEY")
 oai_client = OpenAI(api_key=openai_api_key)
 
 
-def get_openai_embedding(word):
-    response = oai_client.embeddings.create(input=word, model="text-embedding-3-small")
-    return response.data[0].embedding
-
-
 def calculate_embeddings(words):
     # Get word embeddings
-    embeddings = [get_openai_embedding(word) for word in words]
+    response = oai_client.embeddings.create(input=words, model="text-embedding-3-small")
+    embeddings = [e.embedding for e in response.data]
     return embeddings