From 1c9a6dc0b3d29f7116851dbf40c1c70848db753a Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Sun, 14 Apr 2024 12:02:13 +0200
Subject: [PATCH] fixed export and import of discussions

---
 lollms/app.py                                | 2 +-
 lollms/databases/discussions_database.py     | 6 +++---
 lollms/server/endpoints/lollms_discussion.py | 3 ++-
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/lollms/app.py b/lollms/app.py
index 899d098..96cc512 100644
--- a/lollms/app.py
+++ b/lollms/app.py
@@ -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")
diff --git a/lollms/databases/discussions_database.py b/lollms/databases/discussions_database.py
index 45c1f7c..4d64fb6 100644
--- a/lollms/databases/discussions_database.py
+++ b/lollms/databases/discussions_database.py
@@ -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
diff --git a/lollms/server/endpoints/lollms_discussion.py b/lollms/server/endpoints/lollms_discussion.py
index e70ae36..4fbcca4 100644
--- a/lollms/server/endpoints/lollms_discussion.py
+++ b/lollms/server/endpoints/lollms_discussion.py
@@ -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