AI-RAG-Interface-to-Hub

Running

AFischer1985 commited on May 27, 2024

Commit

05ce570

verified ·

1 Parent(s): 56c713d

Update run.py

Files changed (1) hide show

run.py CHANGED Viewed

@@ -198,8 +198,9 @@ def response(
         seed=42,
     )
     addon=""
-    if(len(history)==0)first_message=message
-    first_message=history[0]["user_message"]
     results=collection.query(
       query_texts=[first_message],
       n_results=2,

         seed=42,
     )
     addon=""
+    first_message=message
+    if(len(history)>0):
+      first_message=history[0]["user_message"]
     results=collection.query(
       query_texts=[first_message],
       n_results=2,