Update run.py

run.py

@@ -112,9 +112,9 @@ def response(
         do_sample=True,
         seed=42,
     )
-    addon=""
+    search_prompt = format_prompt(prompt,history)
     results=collection.query(
-      query_texts=[prompt],
+      query_texts=[search_prompt],
       n_results=60,
       #where={"source": "google-docs"}
       #where_document={"$contains":"search_string"}
@@ -133,7 +133,7 @@ def response(
     combination = [' '.join(triplets) for triplets in combination]
     #print(str(prompt)+"\n\n"+str(combination))
     if(len(results)>1):
-      addon=" Bitte berücksichtige bei deiner Antwort ausschießlich folgende Auszüge aus unserer Datenbank, sofern sie für die Antwort erforderlich sind. Beantworte die Frage knapp und präzise. Ignoriere unpassende Datenbank-Auszüge OHNE sie zu kommentieren, zu erwähnen oder aufzulisten:\n"+"\n".join(results)
+      addon="Bitte berücksichtige bei deiner Antwort ausschießlich folgende Auszüge aus unserer Datenbank, sofern sie für die Antwort erforderlich sind. Beantworte die Frage knapp und präzise. Ignoriere unpassende Datenbank-Auszüge OHNE sie zu kommentieren, zu erwähnen oder aufzulisten:\n"+"\n".join(results)
     system="Du bist ein deutschsprachiges KI-basiertes Studienberater Assistenzsystem, das zu jedem Anliegen möglichst geeignete Studieninformationen empfiehlt."+addon+"\n\nUser-Anliegen:"   
     formatted_prompt = format_prompt(system+"\n"+prompt,history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)