Update __init__.py

This commit is contained in:
Saifeddine ALOUI 2023-11-30 16:32:01 +01:00 committed by GitHub
parent a6720dd56e
commit 96e5cc297c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -98,10 +98,10 @@ def prepare_query(discussion, prompt, n_tokens: int = 0) -> Tuple[str, str, List
# Check if there is discussion history to add to the prompt
history = ""
if lollms_app.config.use_discussions_history and lollms_app.discussions_store is not None:
if lollms_app.config.use_discussions_history and lollms_app.long_term_memory is not None:
if history=="":
documentation="!@>History:\n"
docs, sorted_similarities = lollms_app.discussions_store.recover_text(current_message.content, top_k=lollms_app.config.data_vectorization_nb_chunks)
docs, sorted_similarities = lollms_app.long_term_memory.recover_text(current_message.content, top_k=lollms_app.config.data_vectorization_nb_chunks)
for doc, infos in zip(docs, sorted_similarities):
history += f"discussion chunk:\ndiscussion title: {infos[0]}\nchunk content:{doc}"