Update RAG

run.py

@@ -2,7 +2,7 @@
 # Title:  Gradio Interface to LLM-chatbot with dynamic RAG-funcionality and ChromaDB
 # Author: Andreas Fischer
 # Date:   October 10th, 2024
-# Last update: October 22th, 2024
+# Last update: October 24th, 2024
 ##########################################################################################
 
 import os
@@ -294,8 +294,9 @@ def multimodal_response(message, history, dropdown, hfToken, request: gr.Request
     collection=add_doc(message["text"], session)
   client = chromadb.PersistentClient(path=dbPath)
   print(str(client.list_collections()))
-  x=collection.get(include=[])["ids"]  
-  context=collection.query(query_texts=[query], n_results=1)
+  x=collection.get(include=[])["ids"] 
+  ragQuery=[format_prompt(query, history) if len(history)>0 else query]
+  context=collection.query(query_texts=ragQuery, n_results=1)
   context=["<Kontext "+str(i)+"> "+str(c)+"</Kontext "+str(i)+">" for i,c in enumerate(context["documents"][0])]
   gr.Info("Kontext:\n"+str(context))    
   generate_kwargs = dict(