Update run.py

run.py

@@ -198,8 +198,9 @@ def response(
         seed=42,
     )
     addon=""
+    first_message=history[1]["user_message"]
     results=collection.query(
-      query_texts=[message],
+      query_texts=[first_message],
       n_results=2,
       #where={"source": "google-docs"}
       #where_document={"$contains":"search_string"}
@@ -224,7 +225,7 @@ def response(
       historylimit=4,           # number of past messages to consider for response to current message
       removeHTML=True           # remove HTML-components from History (to prevent bugs with Markdown)
     )
-    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
+    stream = client.text_generation(prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
     for response in stream:
         output += response.token.text