Update run.py

run.py

@@ -187,9 +187,10 @@ def response(
     message, history, temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,
 ):
     addon=""
-    first_message = history[0][0] if history else message #[-1][0]
+    first_message = history[0][0] if history else message 
+    last_message = history[-1][0] if history else message # RAG based on last user-message
     results=collection.query(
-      query_texts=[first_message],
+      query_texts=[last_message],
       n_results=2,
       #where={"source": "google-docs"}
       #where_document={"$contains":"search_string"}
@@ -205,14 +206,14 @@ def response(
     system="Du bist ein deutschsprachiges KI-basiertes Assistenzsystem, das zu jedem Anliegen möglichst geeignete KI-Tools empfiehlt." #+addon #+"\n\nUser-Anliegen:"   
     #body={"prompt":system+"### Instruktion:\n"+message+"\n\n### Antwort:","max_tokens":500, "echo":"False","stream":"True"} #e.g. SauerkrautLM
     #formatted_prompt = extend_prompt(system+"\n"+prompt,  None) #history)
-    removeHTML=False
+    removeHTML=True
     prompt=extend_prompt(
       message,                  # current message of the user
       history,                  # complete history 
       system,                   # system prompt
       addon,                    # RAG-component added to the system prompt
       None,                     # fictive first words of the AI (neither displayed nor stored)
-      historylimit=0,           # number of past messages to consider for response to current message
+      historylimit=2,           # number of past messages to consider for response to current message
       removeHTML=removeHTML     # remove HTML-components from History (to prevent bugs with Markdown)
     )
   
@@ -245,7 +246,7 @@ def response(
           response += part
           if removeHTML==True: response = re.sub("<(.*?)>","\n", response) # remove HTML-components in general (may cause bugs with markdown-rendering)
           yield response
-      if(len(history)==0):
+      if(True): #len(history)==0):
         response=response+"\n\n<details open><summary><strong>Sources</strong></summary><br><ul>"+ "".join(["<li>" + s + "</li>" for s in combination])+"</ul></details>"
       yield response
     
@@ -283,7 +284,7 @@ def response(
             pass
         if removeHTML==True: response = re.sub("<(.*?)>","\n", response) # remove HTML-components in general (may cause bugs with markdown-rendering)
         yield response
-      if(len(history)==0): 
+      if(True): #len(history)==0): 
         response=response+"\n\n<details open><summary><strong>Sources</strong></summary><br><ul>"+ "".join(["<li>" + s + "</li>" for s in combination])+"</ul></details>"
       yield response
       #history.append((message, response)) # add current dialog to history