Update chatbot_tab.py

tabs/chatbot_tab.py

@@ -30,7 +30,7 @@ model = ChatMistralAI(model="mistral-large-latest")
 dataPath = st.session_state.DataPath
 
 trimmer = trim_messages(
-    max_tokens=1000,
+    max_tokens=10000,
     strategy="last",
     token_counter=model,
     include_system=True,
@@ -80,14 +80,9 @@ def run():
     st.write("")
     st.title(tr(title))
     
-    context = """Je souhaites que tu simule une conversation entre un commercial, Marc, de mon entreprise et le le Directeur commercial d'un de mes prospect. 
-Chez mon prospect, tu es donc le Directeur commercial.  
-Répond à mes questions en tant que Directeur commercial.  
+    context = """Tu es un Directeur Commercial d'une entreprise qui commercialise une solution technologique B2B"""
+    human_message1 = """Je souhaites que tu simule une conversation entre un commercial, Marc, de mon entreprise et toi, je prospecte.
 Mon entreprise propose une solution logicielle pour gérer la proposition de valeur d’entreprises B2B qui commercialises des solutions technologiques.   
-  
-L'entreprise que je prospecte à l'un des profils suivants:  
-- Les scale-up françaises qui commercialisent une solution technologique B2B  
-- Les sociétés françaises qui commercialisent une solution technologique B2B   
     
 Les problématiques adressées par ma solution sont:  
 - Il est difficile pour les startups de concevoir et formaliser une proposition de valeur unique et pertinente
@@ -112,14 +107,17 @@ Les cas d’usages adressés par ma solution sont  :
 - Partager les retours terrain
 - Mettre en oeuvre des meilleures pratiques commerciales
 - Identifier et reproduire les messages de vente gagnants
+
+Je suis Marc, le vendeur.
+Répond à mes questions en tant que Directeur commercial.  
  
- Tu es le Directeur commercial, chez mon prospect.
- Je suis Marc, le vendeur.
- Attention: Ce n'est pas toi qui m'aide, c'est moi qui t'aide avec ma solution.
+Attention: Ce n'est pas toi qui m'aide, c'est moi qui t'aide avec ma solution.
 """
   
     messages = [
-        SystemMessage(content=context)
+        SystemMessage(content=context),
+        HumanMessage(content=human_message1),
+        AIMessage(content="J'ai bien compris, je suis un Directeur Commercial prospecté et je réponds à tes questions"),
         ]
     
     trimmer.invoke(messages)