embeddings api

pull/64/head
Kye 2 years ago
parent 080ade7cc4
commit 95f569c9b2

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

Loading…
Cancel
Save