From 87e15cc191f43e04aad5309d52a65b26469fa1c5 Mon Sep 17 00:00:00 2001
From: Ana Pacheco <it28pachecoanne@gmail.com>
Date: Thu, 3 Oct 2024 11:52:05 -0600
Subject: [PATCH] Improve prompt

---
 backend/managers/RagManager.py         | 9 ++++++---
 backend/managers/ResourcesManager.py   | 1 -
 backend/managers/VoicesFacesManager.py | 1 -
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/backend/managers/RagManager.py b/backend/managers/RagManager.py
index 74f8318b..1b96170f 100644
--- a/backend/managers/RagManager.py
+++ b/backend/managers/RagManager.py
@@ -165,9 +165,12 @@ async def retrieve_and_generate(self, collection_name, query, llm) -> str:
         persona = await personas_m.retrieve_persona(persona_id)
         personality_prompt = persona.description
         # Combine the system prompt and context        
-        system_prompt = (os.environ.get('SYSTEM_PROMPT') + "\n\n{context}" +
-                        "\n\nHere is some information about the assistant expertise to help you answer your questions: " +
-                        personality_prompt)
+        system_prompt = (os.environ.get('SYSTEM_PROMPT') + 
+                         "\n\n{context}" +
+                        "\n\nHere is some information about the assistant expertise to help you answer your questions: " + personality_prompt + 
+                        ".\n\nIf the user asks you a question about the assistant information, example: 'What can you tell me about the assistant?', 'What is the name of the assistant?', 'Who is the assistant?', etc." +
+                        "\n\nPlease answer the question using the following information:\n\n" + personality_prompt
+                        )
         
         prompt = ChatPromptTemplate.from_messages([
             ("system", system_prompt),
diff --git a/backend/managers/ResourcesManager.py b/backend/managers/ResourcesManager.py
index 6e9ca8b0..85d3fac5 100644
--- a/backend/managers/ResourcesManager.py
+++ b/backend/managers/ResourcesManager.py
@@ -225,7 +225,6 @@ async def get_ollama_installed_models(self) ->List:
                 if response.status_code == 200:
                     data = response.json()
                     models = data.get("models", [])
-                    print(models)                                      
                     return models
                 else:
                     return []
\ No newline at end of file
diff --git a/backend/managers/VoicesFacesManager.py b/backend/managers/VoicesFacesManager.py
index 4eeb5c93..e220f3e6 100644
--- a/backend/managers/VoicesFacesManager.py
+++ b/backend/managers/VoicesFacesManager.py
@@ -126,7 +126,6 @@ async def text_to_speech(self, voice_id: str, body: str, assistant_id: str, msg_
         directory.mkdir(parents=True, exist_ok=True)
         file_path = directory / f'{msg_id}.mp3'
         audio_msg_path = str(file_path)
-        print('audio_msg_path: ', audio_msg_path)
         tts_url = f"https://api.elevenlabs.io/v1/text-to-speech/{xi_id}/stream"
 
         headers = {