improvement in the past context management with chatbot

src/control/control.py

@@ -15,15 +15,16 @@ class Controller:
         self.llm = llm
 
     def get_response(self, query_fr: str, histo_fr: [(str, str)]) -> (str, [Block]):
-        query = self.llm.translate(text=query_fr) if self.plan_language == 'en' else query_fr
-        block_sources = self.retriever.similarity_search(query=query)
+        histo_conversation, histo_queries = self._get_histo(histo_fr)
+        queries = self.llm.translate(text=histo_queries) if self.plan_language == 'en' else histo_queries
+        block_sources = self.retriever.similarity_search(query=queries)
         block_sources = self._select_best_sources(block_sources)
         for block in block_sources:
-            self._expand_block_with_specials(block, query_fr)
+            self._expand_block_with_specials(block,  histo_queries)
         sources_contents = [s.content for s in block_sources]
         context = '\n'.join(sources_contents)
-        histo = self._get_histo_str(histo_fr)
-        answer = self.llm.generate_paragraph(query=query, histo=histo, context=context, language=self.content_language)
+        answer = self.llm.generate_paragraph(query=queries, histo=histo_conversation, context=context,
+                                             language=self.content_language)
         sources_contents_fr = [s.content_fr for s in block_sources[:2]]
         context_fr = '\n'.join(sources_contents_fr)
         if self.content_language == 'en':
@@ -32,11 +33,15 @@ class Controller:
         return answer, block_sources
 
     @staticmethod
-    def _get_histo_str(histo: [(str, str)]) -> str:
-        histo_str = ""
-        for (query, answer) in histo[1:5]:
-            histo_str += f'user: {query} \n botagent: {answer}\n'
-        return histo_str
+    def _get_histo(histo: [(str, str)]) -> str:
+        histo_conversation = ""
+        histo_queries = ""
+
+        for (query, answer) in histo[-5:]:
+            histo_conversation += f'user: {query} \n bot: {answer}\n'
+            histo_queries += query + '\n'
+        return histo_conversation[:-1], histo_queries
+
 
     @staticmethod
     def _select_best_sources(sources: [Block], delta_1_2=0.15, delta_1_n=0.3, absolute=1.2, alpha=0.9) -> [Block]:

src/tools/llm.py

@@ -5,14 +5,22 @@ class LlmAgent:
 
     def generate_paragraph(self, query: str, context: {}, histo: [(str, str)], language='fr') -> str:
         """generates the  answer"""
-        template = (f"You are a botagent designed to answer to the {query} from users based on the context "
-                    f"delimited by triple backticks: ``` {context}``` and on the previous part of the conversation in"
-                    f"french and delimited by triple backticks ``` {histo} ```"
-                    f" The response shall be in {language} and shall be concise and based on the context provided. "
+        template = (f"You are a bot designed to answer to the query from users delimited by triple backticks: \\n"
+                    f"``` {query} ```\\n"
+                    f"Your answer is based on the context delimited by triple backticks: \\n"
+                    f"``` {context} ```\\n"
+                    f"Your answer takes into account previous part of the conversation in french delimited by triple "
+                    f"backticks and shall avoid redundancies in the answers\\n "
+                    f"``` {histo} ```\\n"
+                    f"Your response shall be in {language} and shall be concise/"
                     f"In case the provided context is not relevant to answer to the question, just return that you "
                     f"don't know the answer ")
 
         p = self.llm(template)
+        print("****************")
+        print(template)
+        print("----")
+        print(p)
         return p
 
 

src/view/view.py

@@ -23,14 +23,12 @@ def run(ctrl: Controller, config: {}):
                     interactive=True,
                     placeholder="Posez votre question ici",
                 )
-                gr.Examples(
-                    list(config['examples'].values()),
-                    input_text_comp,
-                    None,
-                    lambda: None,
-                )
-
                 clear_btn = gr.Button("Clear")
+                input_example_comp = gr.Radio(
+                    label="Examples",
+                    choices=list(config['examples'].values()),
+                    value="",
+                )
                 source_text_comp = []
                 for i in range(4):
                     source_text_comp.append(gr.Textbox(
@@ -47,6 +45,7 @@ def run(ctrl: Controller, config: {}):
             histo_text_.append((input_text_, None))
             update_ = {
                 histo_text_comp: gr.update(visible=True, value=histo_text_),
+                input_example_comp: gr.update(visible=False, interactive=False),
             }
             for i in range(4):
                 update_[source_text_comp[i]] = gr.update(visible=False)
@@ -66,10 +65,22 @@ def run(ctrl: Controller, config: {}):
                 update_[source_text_comp[i]] = gr.update(visible=True, value=source_text, label=source_label)
             return update_
 
+        def input_example_fn(input_example_, histo_text_):
+            histo_text_.append((input_example_, None))
+            update_ = {
+                input_text_comp: gr.update(value=input_example_),
+                histo_text_comp: gr.update(visible=True, value=histo_text_),
+                input_example_comp: gr.update(visible=False, interactive=False, value=''),
+            }
+            for i in range(4):
+                update_[source_text_comp[i]] = gr.update(visible=False)
+            return update_
+
         def clear_fn():
             update_ = {
                 input_text_comp: gr.update(value=''),
                 histo_text_comp: gr.update(value='', visible=False),
+                input_example_comp: gr.update(value='', visible=True, interactive=True),
             }
             for i in range(4):
                 update_[source_text_comp[i]] = gr.update(visible=False, value='hello')
@@ -78,15 +89,24 @@ def run(ctrl: Controller, config: {}):
         input_text_comp \
             .submit(input_text_fn1,
                     inputs=[input_text_comp, histo_text_comp],
-                    outputs=[histo_text_comp,
+                    outputs=[histo_text_comp, input_example_comp,
                              source_text_comp[0], source_text_comp[1], source_text_comp[2], source_text_comp[3]])\
             .then(input_text_fn2,
                   inputs=[input_text_comp, histo_text_comp],
                   outputs=[input_text_comp, histo_text_comp,
                            source_text_comp[0], source_text_comp[1], source_text_comp[2], source_text_comp[3]])
+        input_example_comp \
+            .input(input_example_fn,
+                   inputs=[input_example_comp, histo_text_comp],
+                   outputs=[input_text_comp, histo_text_comp, input_example_comp,
+                            source_text_comp[0], source_text_comp[1], source_text_comp[2], source_text_comp[3]])\
+            .then(input_text_fn2,
+                  inputs=[input_text_comp, histo_text_comp],
+                  outputs=[input_text_comp, histo_text_comp,
+                           source_text_comp[0], source_text_comp[1], source_text_comp[2], source_text_comp[3]])
         clear_btn.click(clear_fn,
                         inputs=None,
-                        outputs=[input_text_comp, histo_text_comp,
+                        outputs=[input_text_comp, histo_text_comp, input_example_comp,
                                  source_text_comp[0], source_text_comp[1], source_text_comp[2], source_text_comp[3]])
 
     return qna