Update src/llamaindex_backend.py

src/llamaindex_backend.py

@@ -63,18 +63,17 @@ class GLlamaIndex():
             result = await self.index.retrieve_context(query)
             return result["result"]
             
-        extended_query = f"<chat history>]\n{history[-1]}\n</chat history><new query>\n{query}\n</new query>"
-        print(f"history[-1]: {history[-1]}")
-        print(f"history[:-1]: {history[:-1]}")
+        extended_query = f"{format_history(history[-2:])}\n{query}"
+
         results = await self.index.aretrieve_context_multi(
             [query, extended_query]
         )
-        print("retrieval results", results)
+        self.logger.info(f"retrieval results: {results}")
         eval_results = await self.aevaluate_context_multi(
             [query, extended_query],
             [r["result"] for r in results]
         )
-        print("eval results", eval_results)
+        self.logger.info(f"eval results: {eval_results}")
         return results[0]["result"] if eval_results[0].score > eval_results[1].score \
             else results[1]["result"]