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