Update personality.py

This commit is contained in:
Saifeddine ALOUI 2024-09-02 09:25:58 +02:00 committed by GitHub
parent 43a0ab68b7
commit ac1bb16dd7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -907,7 +907,7 @@ class AIPersonality:
v = TFIDFVectorizer()
elif self.config.rag_vectorizer == "openai":
from lollmsvectordb.lollms_vectorizers.openai_vectorizer import OpenAIVectorizer
v = OpenAIVectorizer()
v = OpenAIVectorizer(api_key=self.config.rag_vectorizer_openai_key)
self.persona_data_vectorizer = VectorDatabase(self.database_path, v, TikTokenTokenizer(), self.config.rag_chunk_size, self.config.rag_overlap)