add default folder for the cache

VinciGit00 · VinciGit00 · commit 543b48764a29 · 2024-06-07T09:47:21.000+02:00
diff --git a/scrapegraphai/nodes/rag_node.py b/scrapegraphai/nodes/rag_node.py
@@ -99,7 +99,7 @@ def execute(self, state: dict) -> dict:
         )
         embeddings = self.embedder_model
 
-        folder_name = "cache"
+        folder_name = self.node_config.get("cache", "cache")
 
         if self.node_config.get("cache", False) and not os.path.exists(folder_name):
             index = FAISS.from_documents(chunked_docs, embeddings)

Original file line number	Diff line number	Diff line change
`@@ -99,7 +99,7 @@ def execute(self, state: dict) -> dict:`
`99`	`99`	`)`
`100`	`100`	`embeddings = self.embedder_model`
`101`	`101`
`102`		`- folder_name = "cache"`
	`102`	`+ folder_name = self.node_config.get("cache", "cache")`
`103`	`103`
`104`	`104`	`if self.node_config.get("cache", False) and not os.path.exists(folder_name):`
`105`	`105`	`index = FAISS.from_documents(chunked_docs, embeddings)`