|
|
@ -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)
|
|
|
|
|
|
|
|
|
|
|
|