Add prompt system

config.yaml

@@ -30,6 +30,8 @@ Paramètre 3 : {param3}
 Documents partagées : {commonContext}
 Document de référence : {documentContext}
 
+{prompt_system}
+
 Voici l'historique des messages : {messages}
 Les attentes de l'utilisateur sont : {query}
 "

pages/chatbot.py

@@ -21,8 +21,10 @@ def process_input():
     if "user_input" in st.session_state and st.session_state["user_input"] and len(st.session_state["user_input"].strip()) > 0:
         user_text = st.session_state["user_input"].strip()
 
+        prompt_sys = st.session_state.prompt_system if 'prompt_system' in st.session_state and st.session_state.prompt_system != '' else ""
+    
         with st.session_state["thinking_spinner"], st.spinner(f"Je réfléchis"):
-            agent_text = st.session_state["assistant"].ask(user_text, st.session_state["messages"] if "messages" in st.session_state else [], variables=st.session_state["data_dict"])
+            agent_text = st.session_state["assistant"].ask(user_text, prompt_sys, st.session_state["messages"] if "messages" in st.session_state else [], variables=st.session_state["data_dict"])
 
         st.session_state["messages"].append((user_text, True))
         st.session_state["messages"].append((agent_text, False))
@@ -57,9 +59,6 @@ def page():
 
     selector.ModelSelector()
 
-    # Get prompt system
-    # prompt_sys = st.session_state.prompt_system if 'prompt_system' in st.session_state and st.session_state.prompt_system != '' else "Renseignez votre prompt system"
-    
     display_messages()
 
     st.text_input("Message", key="user_input", on_change=process_input)

prompt_template.py

@@ -1,12 +0,0 @@
-base_template = '''
-
-Paramètre 1 : {param_1}
-Paramètre 2 : {param_2}
-Paramètre 3 : {param_3}
-
-Documents partagées : {commonContext}
-Document de référence : {documentContext}
-
-Voici l'historique des messages : {messages}
-Les attentes de l'utilisateur sont : {query}
-'''

rag.py

@@ -72,7 +72,7 @@ class Rag:
             },
         )
 
-    def ask(self, query: str, messages: list, variables: list = None):
+    def ask(self, query: str, prompt_system: str, messages: list, variables: list = None):
         self.chain = self.prompt | self.model | StrOutputParser()
         
         # Retrieve the context document
@@ -89,6 +89,7 @@ class Rag:
             "query": query,
             "documentContext": documentContext,
             "commonContext": contextCommon,
+            "prompt_system": prompt_system,
             "messages": messages
         }