worker node clean up name atribute error

pull/48/head
Kye 2 years ago
parent d7ef20031b
commit fc83185ee4

@ -173,7 +173,7 @@ class WorkerNode:
try: try:
embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key) embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key)
embedding_size = self.embedding_size embedding_size = self.embedding_size
index = faiss.IndexFlatL2(embedding_size=embedding_size) index = faiss.IndexFlatL2(embedding_size)
return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {}) return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
except Exception as e: except Exception as e:
@ -247,7 +247,6 @@ class WorkerNode:
human_in_the_loop=human_in_the_loop, human_in_the_loop=human_in_the_loop,
search_kwargs=search_kwargs, search_kwargs=search_kwargs,
) )
worker_node.name = worker_name # Setting the name here
return worker_node return worker_node
except Exception as e: except Exception as e:
logging.error(f"Failed to create worker node: {e}") logging.error(f"Failed to create worker node: {e}")

Loading…
Cancel
Save