diff --git a/lollms/databases/skills_database.py b/lollms/databases/skills_database.py index cd141c4..7cab0e9 100644 --- a/lollms/databases/skills_database.py +++ b/lollms/databases/skills_database.py @@ -28,7 +28,7 @@ class SkillsLibrary: else: from lollmsvectordb.lollms_vectorizers.semantic_vectorizer import SemanticVectorizer - v = SemanticVectorizer("BAAI/bge-m3") + v = SemanticVectorizer(self.config.rag_vectorizer_model) self.vectorizer = VectorDatabase("", v, TikTokenTokenizer(),chunk_size, overlap, n_neighbors) ASCIIColors.green("Vecorizer ready") diff --git a/lollms/personality.py b/lollms/personality.py index e5d3541..eef269e 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -869,7 +869,10 @@ Don't forget encapsulate the code inside a html code tag. This is mandatory. else: return code else: - return None + if return_full_generated_code: + return None, None + else: + return None def generate_text(