fixed export and import of discussions

This commit is contained in:
Saifeddine ALOUI 2024-04-14 12:02:13 +02:00
parent ec9736cd6b
commit 1c9a6dc0b3
3 changed files with 6 additions and 5 deletions

View File

@ -656,7 +656,7 @@ class LollmsApplication(LoLLMsCom):
try:
docs, sorted_similarities, document_ids = self.personality.vectorizer.recover_text(query, top_k=self.config.data_vectorization_nb_chunks)
for doc, infos in zip(docs, sorted_similarities):
documentation += f"document chunk:\nchunk path: {infos[0]}\nchunk content:{doc}"
documentation += f"document chunk:\nchunk path: {infos[0]}\nchunk content:\n{doc}"
documentation += "\n!@>important information: Use the documentation data to answer the user questions. If the data is not present in the documentation, please tell the user that the information he is asking for does not exist in the documentation section. It is strictly forbidden to give the user an answer without having actual proof from the documentation."
except:
self.warning("Couldn't add documentation to the context. Please verify the vector database")

View File

@ -388,9 +388,9 @@ class DiscussionsDB:
discussions = []
data = json_data
for discussion_data in data:
discussion_id = discussion_data.get("id")
discussion_title = discussion_data.get("title")
messages_data = discussion_data.get("messages", [])
discussion_id = discussion_data.id
discussion_title = discussion_data.title
messages_data = discussion_data.messages
discussion = {"id": discussion_id, "title": discussion_title, "messages": []}
# Insert discussion into the database

View File

@ -190,7 +190,8 @@ async def export_multiple_discussions(discussion_export: DiscussionExport):
class DiscussionInfo(BaseModel):
id: int
content: str
title: str
messages: List[dict]
class DiscussionImport(BaseModel):
client_id: str