feat: add bot general

core/chat/bot_service.py

@@ -10,14 +10,24 @@ from llama_index.core.llms import MessageRole
 
 from core.chat.engine import Engine
 from core.chat.chatstore import ChatStore
-from core.parser import clean_text, update_response, sort_and_renumber_sources
+from core.parser import (
+    update_response,
+    sort_and_renumber_sources,
+    redesign_structure_message,
+)
 
 from service.dto import ChatMessage
 from pymongo.mongo_client import MongoClient
 
 
 class ChatCompletionService:
-    def __init__(self, session_id: str, user_request: str, titles: List = None, type_bot: str = "general"):
+    def __init__(
+        self,
+        session_id: str,
+        user_request: str,
+        titles: List = None,
+        type_bot: str = "general",
+    ):
         self.session_id = session_id
         self.user_request = user_request
         self.titles = titles
@@ -29,7 +39,9 @@ class ChatCompletionService:
 
     def generate_completion(self):
         if not self._ping_mongo():
-            return JSONResponse(status_code=500, content="Database Error: Unable to connect to MongoDB")
+            return JSONResponse(
+                status_code=500, content="Database Error: Unable to connect to MongoDB"
+            )
 
         try:
             # Load and retrieve chat engine with appropriate index
@@ -38,19 +50,26 @@ class ChatCompletionService:
 
             # Generate chat response
             response = chat_engine.chat(self.user_request)
+
             sources = response.sources
             number_reference_sorted = self._extract_sorted_references(response)
-            print("number reference sorted : ",number_reference_sorted)
 
-            contents, metadata_collection, scores = self._process_sources(sources, number_reference_sorted)
+            contents, metadata_collection, scores = self._process_sources(
+                sources, number_reference_sorted
+            )
 
             # Update response and renumber sources
             response = update_response(str(response))
             contents = sort_and_renumber_sources(contents)
-            print(contents)
 
             # Add contents to metadata
-            metadata_collection = self._attach_contents_to_metadata(contents, metadata_collection)
+            metadata_collection = self._attach_contents_to_metadata(
+                contents, metadata_collection
+            )
+
+            if self.type_bot == "general":
+                response = redesign_structure_message(response, metadata_collection)
+                print(response)
 
             # Save the message to chat store
             self._store_message_in_chatstore(response, metadata_collection)
@@ -58,13 +77,12 @@ class ChatCompletionService:
         except Exception as e:
             logging.error(f"An error occurred in generate text: {e}")
             return JSONResponse(
-                status_code=500,
-                content=f"An internal server error occurred: {e}"
+                status_code=500, content=f"An internal server error occurred: {e}"
             )
 
         try:
             if self.type_bot == "specific":
-                self._save_chat_history_to_db(str(response), metadata_collection)
+                self._save_chat_history_to_db(response, metadata_collection)
 
             return str(response), metadata_collection, scores
 
@@ -72,7 +90,7 @@ class ChatCompletionService:
             logging.error(f"An error occurred while saving chat history: {e}")
             return JSONResponse(
                 status_code=500,
-                content=f"An internal server error occurred while saving chat history: {e}"
+                content=f"An internal server error occurred while saving chat history: {e}",
             )
 
     def _ping_mongo(self):
@@ -87,7 +105,9 @@ class ChatCompletionService:
     def _get_chat_engine(self, index):
         if self.type_bot == "general":
             return self.engine.get_chat_engine(self.session_id, index)
-        return self.engine.get_chat_engine(self.session_id, index, self.titles, self.type_bot)
+        return self.engine.get_chat_engine(
+            self.session_id, index, self.titles, self.type_bot
+        )
 
     def _extract_sorted_references(self, response):
         number_reference = list(set(re.findall(r"\[(\d+)\]", str(response))))
@@ -124,9 +144,7 @@ class ChatCompletionService:
 
     def _store_message_in_chatstore(self, response, metadata_collection):
         message = ChatMessage(
-            role=MessageRole.ASSISTANT,
-            content=response,
-            metadata=metadata_collection
+            role=MessageRole.ASSISTANT, content=response, metadata=metadata_collection
         )
         self.chatstore.delete_last_message(self.session_id)
         self.chatstore.add_message(self.session_id, message)

core/parser.py

@@ -123,3 +123,24 @@ def join_list(items):
         return f"{items[0]} and {items[1]}"
     else:
         return ", ".join(items[:-1]) + " and " + items[-1]
+
+def redesign_structure_message(message, metadata):
+    """ 
+    This function replaces occurrences of '[n]' in the message 
+    with the title of the book found in metadata[n-1]["title"].
+    """
+    if not metadata or metadata == []:
+        return message  # Return the original message if metadata is not valid
+    
+    # Create a function to replace each citation with the corresponding book title
+    def replace_citation(match):
+        citation_number = int(match.group(1))  # Extract the citation number
+        # Check if the citation number corresponds to a title in metadata
+        if 1 <= citation_number <= len(metadata):
+            return f"[*{metadata[citation_number - 1]['title']}*]"  # Return the title in italics
+        return match.group(0)  # Return the original citation if out of bounds
+
+    # Use regex to find all citations in the format '[n]'
+    redesigned_message = re.sub(r'\[(\d+)\]', replace_citation, message)
+    
+    return redesigned_message

core/prompt.py

@@ -18,6 +18,37 @@ ADDITIONAL_INFORMATIONS = """
 Kemudian, kamu menjawab pertanyan user dari buku {titles}, jadi jika user bertaya kamu pastikan akan mengacu buku tersebut yang didapatkan dari tools dari yang kamu punya. 
 """
 
+SYSTEM_BOT_GENERAL_TEMPLATE = """
+Kamu adalah Medbot yang selalu menggunakan tools untuk menjawab pertanyaan medis. Jika pengguna bertanya tentang topik non-medis, arahkan mereka untuk bertanya di bidang medis. Tugasmu adalah memberikan jawaban yang penjelasan singkat tentang medis berdasarkan tools yang tersedia sebagai gambaran dari isi buku. Pastikan kamu hanya memberikan informasi dari buku yang telah disediakan, jangan sampai menjawab pertanyaan yang tidak terdapat dalam buku atau tools yang kamu gunakan. Jika di konteks tidak terdapat citation atau kutipan seperti [1], [2] dan lainnya kamu cukup bilang mohon maaf, saya tidak memiliki referensi atau buku yang kamu minati.
+
+Kamu juga harus memperhatikan instruksi : 
+
+**Referensi dan Kutipan**: 
+- Jangan menghapus sumber kutipan dari teks yang diberikan. Contohnya, jika teksnya adalah "Ilmu kedokteran sangat dibutuhkan [2]", pastikan untuk menyertakan kutipan sumbernya yaitu [2] dalam jawabanmu.
+
+**Gaya Jawaban:**
+* Referensi Buku: Gunakan format referensi buku saat menjelaskan. Misalnya:
+ - "Kanker adalah penyakit yang ditandai oleh pertumbuhan sel yang tidak terkontrol dalam tubuh. Penyakit ini sering kali terjadi seiring bertambahnya usia, dengan dua pertiga dari semua kasus kanker terjadi pada individu yang berusia lebih dari 65 tahun [1]. Ubah menjadi --> Jika Anda tertarik untuk belajar tentang kanker, kami sarankan Anda membaca buku [1] karena menjelaskan secara mendalam tentang penyakit ini."
+ - "termasuk mutasi somatik yang mempengaruhi sel-sel, dan bagaimana beberapa peristiwa mutasi kumulatif dapat mengubah sel normal menjadi sel kanker [2] ubah menjadi --> Jika Anda ingin mempelajari lebih lanjut tentang mutasi, kami merekomendasikan untuk membaca buku [2]."
+ - Setiap ada penjelasan: penjelasan [n] diubah menjadi --> buku [n] menjelaskan (gambaran besar tentang penjelasan tersebut)
+ - Jangan digabungkan, seperti : 
+   - Jika anda tertarik maka kami merekomendasikan buku [1] dan [2] melainkan : 
+     * Jika anda tertarik belajar (sebutkan alasannya) kami rekomendasikan buku [1], 
+       Kemudian, jika tertarik belajar (sebutkan alasannya) buku [2] juga menjelaskan materi tersebut. 
+     * Lakukan hal yang sama jika terdapat lebih dari 2 citation. 
+ 
+**Gambaran Umum:**
+* Isi Buku: Jika ada citation atau [1], [2] atau lainnya maka berikan gambaran umum yang jelas dan menarik tentang topik-topik medis yang dibahas, tanpa masuk ke rincian yang berlebihan. Jika tidak ada, maka anda dapat mengatakan kami tidak memiliki referensi apa yang anda minta.
+
+**Gaya Komunikasi:**
+* Persuasif dan Ramah: Gunakan gaya komunikasi yang persuasif, ramah, dan profesional. Pastikan penjelasanmu menarik dan mudah dipahami oleh semua kalangan.
+
+**Penutup:**
+* Mengundang Minat: Akhiri dengan kalimat yang mengundang minat, seperti:
+ - "Jangan lewatkan kesempatan untuk memperdalam pengetahuan medis Anda dengan buku ini ✨😇"
+ - "Dapatkan buku ini sekarang dan tingkatkan pemahaman Anda tentang kesehatan 😊"
+"""
+
 SYSTEM_TOPIC_TEMPLATE = """
 You are tasked with analyzing a table of contents from a book. Your goal is to identify and extract the main topics and subtopics. Please provide a clear and organized list of these topics and subtopics. The list should reflect the structure and hierarchy presented in the table of contents.
 """