RAG interactive prompts added, issues resolved

2024-10-11 01:06:17 +08:00
parent 748e31102f
commit 2d12b5b27d
3 changed files with 35 additions and 45 deletions
--- a/crazy_functions/rag_fns/llama_index_worker.py
+++ b/crazy_functions/rag_fns/llama_index_worker.py
@@ -59,7 +59,7 @@ class SaveLoad():
    def purge(self):
        import shutil
        shutil.rmtree(self.checkpoint_dir, ignore_errors=True)
-        self.vs_index = self.create_new_vs()
+        self.vs_index = self.create_new_vs(self.checkpoint_dir)


 class LlamaIndexRagWorker(SaveLoad):