frdel commited on
Commit
6e2f878
·
1 Parent(s): 9fd3191

minor fixes

Browse files
Files changed (3) hide show
  1. initialize.py +2 -1
  2. memory/.gitkeep +0 -0
  3. python/helpers/vector_db.py +1 -1
initialize.py CHANGED
@@ -26,8 +26,9 @@ def initialize():
26
  chat_model = chat_llm,
27
  utility_model = utility_llm,
28
  embeddings_model = embedding_llm,
29
- prompts_subdir = "custom",
30
  # memory_subdir = "",
 
31
  auto_memory_count = 0,
32
  # auto_memory_skip = 2,
33
  # rate_limit_seconds = 60,
 
26
  chat_model = chat_llm,
27
  utility_model = utility_llm,
28
  embeddings_model = embedding_llm,
29
+ # prompts_subdir = "",
30
  # memory_subdir = "",
31
+ # knowledge_subdir: str = ""
32
  auto_memory_count = 0,
33
  # auto_memory_skip = 2,
34
  # rate_limit_seconds = 60,
memory/.gitkeep ADDED
File without changes
python/helpers/vector_db.py CHANGED
@@ -42,7 +42,7 @@ class VectorDB:
42
 
43
 
44
  # if db folder exists and is not empty:
45
- if os.path.exists(self.db_dir) and len(os.listdir(self.db_dir)) > 0:
46
  self.db = FAISS.load_local(
47
  folder_path=self.db_dir,
48
  embeddings=self.embedder,
 
42
 
43
 
44
  # if db folder exists and is not empty:
45
+ if os.path.exists(self.db_dir) and files.exists(self.db_dir,"index.faiss"):
46
  self.db = FAISS.load_local(
47
  folder_path=self.db_dir,
48
  embeddings=self.embedder,