From 20c9e68ac8f1f14e5ba39139ac3f67cd1de69c81 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Mon, 16 Dec 2024 00:05:06 +0100 Subject: [PATCH] updated --- lollms/databases/skills_database.py | 2 +- lollms/personality.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) 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(