From 1c4bc70a45c1767d1e7acaf27ede00822a3fe52e Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Wed, 10 Apr 2024 20:59:37 +0200
Subject: [PATCH] added new extensions

---
 lollms/personality.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lollms/personality.py b/lollms/personality.py
index 368e542..75609e5 100644
--- a/lollms/personality.py
+++ b/lollms/personality.py
@@ -880,7 +880,7 @@ class AIPersonality:
                             database_dict=None)
                 ASCIIColors.green("Ok")
             else:
-                files = [f for f in self.data_path.iterdir() if f.suffix.lower() in [".txt", ".pdf", ".docx", ".pptx", ".md", ".py", ".c", ".cpp"] ]
+                files = [f for f in self.data_path.iterdir() if f.suffix.lower() in ['.sh', '.json', '.sym', '.log', '.snippet', '.se', '.yml', '.snippets', '.lua', '.pdf', '.md', '.docx', '.yaml', '.inc', '.txt', '.ini', '.pas', '.pptx', '.map', '.php', '.xlsx', '.rtf', '.hpp', '.h', '.asm', '.xml', '.hh', '.sql', '.java', '.c', '.html', '.inf', '.rb', '.py', '.cs', '.js', '.bat', '.css', '.s', '.cpp', '.csv'] ]
                 if len(files)>0:
                     dl = GenericDataLoader()
                     self.persona_data_vectorizer = TextVectorizer(