From dfbc6f1942a81565676516ab44977f20d15f9f41 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 4 Oct 2023 21:40:57 -0400
Subject: [PATCH] clean up

---
 swarms/workers/worker.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/swarms/workers/worker.py b/swarms/workers/worker.py
index d3ab487e..be729253 100644
--- a/swarms/workers/worker.py
+++ b/swarms/workers/worker.py
@@ -131,7 +131,9 @@ class Worker:
                 temperature=0.5)
         ```
         """
-        query_website_tool = WebpageQATool(qa_chain=load_qa_with_sources_chain(self.llm))
+        query_website_tool = WebpageQATool(
+            qa_chain=load_qa_with_sources_chain(self.llm)
+        )
 
         self.tools = [
             WriteFileTool(root_dir=ROOT_DIR),
@@ -155,7 +157,12 @@ class Worker:
             embedding_size = 1536
             index = faiss.IndexFlatL2(embedding_size)
 
-            self.vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
+            self.vectorstore = FAISS(
+                embeddings_model.embed_query, 
+                index, 
+                InMemoryDocstore({}), {}
+            )
+            
         except Exception as error:
             raise RuntimeError(f"Error setting up memory perhaps try try tuning the embedding size: {error}")