Improve prompt

pAI-OS · Oct 3, 2024 · 87e15cc · 87e15cc
1 parent ce9b559
commit 87e15cc
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git 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
@@ -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 []
diff --git 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 = {