embeddings api

Former-commit-id: a360314c8f0f86cc9e059c60f96e91e0c067e6df
pull/160/head
Kye 2 years ago
parent 0db24eb3fa
commit 68f3f74e49

@ -145,8 +145,9 @@ class Worker:
"""
Set up memory for the worker.
"""
openai_api_key = OPENAI_API_KEY
try:
embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key)
embeddings_model = OpenAIEmbeddings(openai_api_key=openai_api_key)
embedding_size = 1536
index = faiss.IndexFlatL2(embedding_size)

Loading…
Cancel
Save